Update app.py

app.py

@@ -20,12 +20,12 @@ def detect_objects(images):
         all_bboxes2.append(boxes2)
     return all_bboxes, all_bboxes2
 
-def create_solutions(image_urls, all_bboxes, all_bboxes2):
+def create_solutions(image_urls, all_bboxes, all_bboxes2, file_ids):
     solutions = []
     img_id = 1
     box_id = 1
     category_id = 1
-    for image_url, bboxes, bboxes2 in zip(image_urls, all_bboxes, all_bboxes2):
+    for image_url, bboxes, bboxes2, file_id in zip(image_urls, all_bboxes, all_bboxes2, file_ids):
         for box, box2 in zip(bboxes, bboxes2):
             if isinstance(box2[0], list):
                 w = box2[0][2]
@@ -38,7 +38,7 @@ def create_solutions(image_urls, all_bboxes, all_bboxes2):
             seg = [[]]
 
             ans = {"segmentation": seg,"area": area,"iscrowd": 0,"image_id": img_id,"bbox": box,"category_id": category_id,"id": box_id}
-            obj = {"url": image_url, "answer": [ans], "qcUser" : None }
+            obj = {"url": image_url, "answer": [ans], "qcUser" : None, "normalfileID": file_id}
             solutions.append(obj)
             box_id += 1
         img_id += 1
@@ -61,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_id = params.get("normalfileID",[])
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
     print(image_urls)
@@ -76,7 +77,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_id)  # Create solutions with image URLs and bounding boxes
 
     # result_url = f"{api}/{job_id}"
     # send_results_to_api(solutions, result_url)