Update app.py

app.py

@@ -23,11 +23,11 @@ def detect_objects(images):
         names.append(arr)
     return names, probss
 
-def create_solutions(image_urls, names, probss):
+def create_solutions(image_urls, names, probss, file_ids):
     solutions = []   #list to store all the objects
 
-    for image_url, class_name, prob in zip(image_urls, names, probss):      
-        obj = {"image": image_url, "answer": [class_name], "qcUser" : None}   
+    for image_url, class_name, prob, file_id in zip(image_urls, names, probss, file_ids):      
+        obj = {"image": image_url, "answer": [class_name], "qcUser" : None, "normalfileID": file_id}   
         solutions.append(obj)
         print("image done")
     return solutions
@@ -50,7 +50,7 @@ def process_images(params):
     
     image_urls = params.get("urls", [])
     # normalFileId = 0
-    # normalFileId = params.get("normalfileID",[])
+    file_id = params.get("normalfileID",[])
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
 
@@ -66,7 +66,7 @@ def process_images(params):
         return {"error": f"Error loading images: {str(e)}"}
         
     names, probss = detect_objects(images)  # Perform object detection
-    solutions = create_solutions(image_urls, names, probss)  # Create solutions with image URLs and bounding boxes
+    solutions = create_solutions(image_urls, names, probss, file_id)  # Create solutions with image URLs and bounding boxes
 
     # result_url = f"{api}/{job_id}"
     # send_results_to_api(solutions, result_url)