Update app.py

app.py

@@ -19,12 +19,12 @@ def detect_objects(images):
         names.append(classes[probs])
     return names
 
-def create_solutions(image_urls, names):
+def create_solutions(image_urls, names, file_id):
     solutions = []   
-    for image_url, prediction in zip(image_urls, names):       
+    for image_url, prediction, file in zip(image_urls, names, file_id):       
         prediction_list=[]
         prediction_list.append(prediction)
-        obj = {"url": image_url, "answer": prediction_list, "qcUser" : None}      
+        obj = {"url": image_url, "answer": prediction_list, "qcUser" : None, "normalfileID": file}      
         solutions.append(obj)   
         print("object prepared")
     return solutions
@@ -46,6 +46,7 @@ def process_images(params):
         return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
         
     image_urls = params.get("urls", [])
+    file_id = params.get("normalfileID")
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
 
@@ -61,7 +62,7 @@ def process_images(params):
         return {"error": f"Error loading images: {str(e)}"}
         
     names = detect_objects(images)  # Perform object detection
-    solutions = create_solutions(image_urls, names)  # Create solutions with image URLs and bounding boxes
+    solutions = create_solutions(image_urls, names, file_id)  # Create solutions with image URLs and bounding boxes
     print("prediction done")
 
     # result_url = f"{api}/{job_id}"