Update app.py

app.py

@@ -30,12 +30,12 @@ def detect_objects(images):
 
     return all_bboxes, all_bboxes2, all_segments
 
-def create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments):
+def create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments, file_ids):
     solutions = []   
     img_id = 1
     box_id = 1
     cat_id = 1
-    for image_url, bbox, bbox2, segmnt in zip(image_urls, all_bboxes, all_bboxes2, all_segments):       
+    for image_url, bbox, bbox2, segmnt, file_id in zip(image_urls, all_bboxes, all_bboxes2, all_segments, file_ids):       
         temp=[]
         for subbox, subbox2, subsegmnt in zip(bbox, bbox2, segmnt):
             w = subbox2[2]
@@ -48,7 +48,7 @@ def create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments):
             box_id += 1
             temp.append(ans)
         img_id += 1
-        obj ={"url": image_url, "answer":temp}
+        obj ={"url": image_url, "answer":temp, "qcUser" : None, "normalfileID" : file_id}
         solutions.append(obj)
     return solutions
 
@@ -70,6 +70,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", "")
 
@@ -84,7 +85,7 @@ def process_images(params):
         return {"error": f"Error loading images: {str(e)}"}
     
     all_bboxes, all_bboxes2, all_segments = detect_objects(images)
-    solutions = create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments)
+    solutions = create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments, file_ids)
 
     # result_url = f"{api}/{job_id}"
     # send_results_to_api(solutions, result_url)