Update app.py

app.py

@@ -46,7 +46,10 @@ 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")
+    if not params.get("normalfileID",[]):
+        file_ids = [None]*len(image_urls)
+    else:
+        file_ids = params.get("normalfileID",[])
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
 
@@ -62,7 +65,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, file_id)  # Create solutions with image URLs and bounding boxes
+    solutions = create_solutions(image_urls, names, file_ids)  # Create solutions with image URLs and bounding boxes
     print("prediction done")
 
     # result_url = f"{api}/{job_id}"