Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 from ultralytics import YOLO
 import requests
 import json
+import logging
 
 model = YOLO("BP_Multiple_Objects_Complicated_v1.pt")
 
@@ -55,37 +56,51 @@ def create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments):
         img_id += 1
     return solutions
 
-def send_results_to_api(data, result_url):
-    headers = {"Content-Type": "application/json"}
-    response = requests.post(result_url, json=data, headers=headers)
-    if response.status_code == 200:
-        return response.json()
-    else:
-        return {"error": f"Failed to send results to API: {response.status_code}"}
 
-def process_images(params):
-    params = json.loads(params)
-    
-    image_urls = params.get("image_urls", [])
-    api = params.get("api", "")
-    job_id = params.get("job_id", "")
+# def send_results_to_api(data, result_url):
+#     headers = {"Content-Type": "application/json"}
+#     response = requests.post(result_url, json=data, headers=headers)
+#     if response.status_code == 200:
+#         return response.json()
+#     else:
+#         return {"error": f"Failed to send results to API: {response.status_code}"}
 
-    images = [Image.open(requests.get(url, stream=True).raw) for url in image_urls]
 
+def process_images(params):
+    try:
+        params = json.loads(params)
+    except json.JSONDecodeError as e:
+        logging.error(f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}")
+        return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
+    
+    image_urls = params.get("urls", [])
+    # api = params.get("api", "")
+    # job_id = params.get("job_id", "")
+
+    if not image_urls:
+        logging.error("Missing required parameters: 'urls'")
+        return {"error": "Missing required parameters: 'urls'"}
+
+    try:
+        images = [Image.open(requests.get(url, stream=True).raw) for url in image_urls]
+    except Exception as e:
+        logging.error(f"Error loading images: {e}")
+        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)
 
-    result_url = f"{api}/{job_id}"
-    send_results_to_api(solutions, result_url)
+    # result_url = f"{api}/{job_id}"
+    # send_results_to_api(solutions, result_url)
 
-    return json.dumps({"solutions": solutions}, indent=4)
+    return json.dumps({"solutions": solutions})
 
 
 inputt = gr.Textbox(label="Parameters (JSON format)")
 outputs = gr.JSON()
 
 application = gr.Interface(fn=process_images, inputs=inputt, outputs=outputs, title="Multiple Object Segmentation with API Integration")
-application.launch(share=True)
+application.launch()