Spaces:
Sleeping
Sleeping
File size: 3,209 Bytes
689227c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
from flask import Flask, request, jsonify
import os
import importlib.util
from datetime import datetime
from predict_task1 import Predictor
from id_mapping import mapping
#from show_annotation import start_annotation_process
#from multiprocessing import Process, Queue
#import cv2
from show_stitched import *
app = Flask(__name__)
config_dir = os.path.abspath(os.path.dirname(__file__))
config_path = os.path.join(config_dir, 'PC_CONFIG.py')
spec = importlib.util.spec_from_file_location("PC_CONFIG", config_path)
PC_CONFIG = importlib.util.module_from_spec(spec)
spec.loader.exec_module(PC_CONFIG)
HOST = PC_CONFIG.HOST
PORT = PC_CONFIG.IMAGE_REC_PORT
UPLOAD_FOLDER = os.path.join(PC_CONFIG.FILE_DIRECTORY,"image-rec","images")
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
def process_file(file_path, direction, task_type):
predictor = Predictor()
print("File received and saved successfully.")
print(f"Direction received: {direction}")
print(f"Task type received: {task_type}")
startTime = datetime.now()
class_name, results, detection_id = predictor.predict_id(file_path, task_type) # Perform prediction
#show_annotation_queue.put((file_path, results, detection_id))
class_id = str(mapping.get(class_name, -1))
endTime = datetime.now()
totalTime = (endTime - startTime).total_seconds()
print(f"Predicted ID: {class_id}")
print(f"Time taken for Predicting Image = {totalTime} s")
return class_id
@app.route('/status', methods=['GET'])
def server_status():
return jsonify({'status': 'OK'})
@app.route('/upload', methods=['POST'])
def upload_file():
if 'file' not in request.files:
return jsonify({'error': 'No file part'}), 400
file = request.files['file']
direction = request.form['direction']
task_type = request.form['task_type']
if file.filename == '':
return jsonify({'error': 'No selected file'}), 400
if file:
filename = os.path.basename(file.filename)
file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
file.save(file_path)
# image = cv2.imread(file_path)
# cv2.imshow("Uploaded Image", image)
# cv2.waitKey(0) # Wait until a key is pressed
# cv2.destroyAllWindows() # Close the window
# Process the file and predict
class_id = process_file(file_path, direction, task_type)
return jsonify({'message': 'File successfully uploaded', 'predicted_id': class_id}), 200
@app.route('/display_stitched', methods=['POST'])
def display_stitched():
showAnnotatedStitched()
return jsonify({'display_stitched': 'OK'})
if __name__ == '__main__':
# show_annotation_queue = Queue()
# process = Process(target=start_annotation_process, args=(show_annotation_queue,))
# process.start()
print()
print(f"UPLOAD FOLDER: {UPLOAD_FOLDER}")
# Port 5000 if free
'''
try:
app.run(host=HOST, port=PORT, debug=False)
except:
print('Unable to Connect to PC_CONFIG Host and Port. Switching to 0.0.0.0:4000.')
app.run(host='0.0.0.0', port=4000, debug=True)
'''
# Run on Port 4000
app.run(host='0.0.0.0', port=4000, debug=True)
#process.join()
|