Update app.py

app.py

@@ -25,7 +25,7 @@ def create_solutions(image_urls, all_bboxes, all_bboxes2):
     img_id =1
     box_id =1
     cat_id =1
-    for image_url, bbox, bbox2 in zip(image_urls, all_bboxes, all_bboxes2):       # Loop through each image ID and its corresponding prediction
+    for image_url, bbox, bbox2, file_id in zip(image_urls, all_bboxes, all_bboxes2, file_ids):       # Loop through each image ID and its corresponding prediction
 
         for subbox, subbox2 in zip(bbox, bbox2):
 
@@ -36,9 +36,11 @@ def create_solutions(image_urls, all_bboxes, all_bboxes2):
             ans = {"segmentation":seg, "area":area, 'iscrowd':0, "image_id":img_id, "bbox": subbox, "category_id":cat_id, "id":box_id }
             ansx=[]
             ansx.append(ans)
-            obj = {"url": image_url, 'answer':ansx}                 
+                            
             box_id +=1
-            solutions.append(obj)                    
+        } 
+            
+        solutions.append({"url": image_url,'answer':ansx, "qcUser" : None, "normalfileID": file_id })
         img_id +=1
     return solutions
 
@@ -59,6 +61,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_ids = params.get("normalfileID", [])
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
     
@@ -72,7 +75,7 @@ def process_images(params):
         return {"error": f"Error loading images: {str(e)}"}
         
     all_bboxes, all_bboxes2 = detect_objects(images)  # Perform object detection
-    solutions = create_solutions(image_urls, all_bboxes, all_bboxes2)  # Create solutions with image URLs and bounding boxes
+    solutions = create_solutions(image_urls, all_bboxes, all_bboxes2, file_ids)  # Create solutions with image URLs and bounding boxes
 
     # result_url = f"{api}/{job_id}"
     # send_results_to_api(solutions, result_url)