Update app.py

app.py

@@ -11,25 +11,43 @@ WORKSPACE = "naveen-kumar-hnmil"
 WORKFLOW = "detect-count-and-visualize-5"
 
 def detect_image(image):
-    # Convert PIL to bytes
-    buffered = BytesIO()
-    image.save(buffered, format="JPEG")
-    image_bytes = buffered.getvalue()
+    try:
+        # Convert PIL to bytes
+        buffered = BytesIO()
+        image.save(buffered, format="JPEG")
+        image_bytes = buffered.getvalue()
 
-    # Build full API endpoint
-    endpoint = f"{API_URL}/{WORKSPACE}/{WORKFLOW}?api_key={API_KEY}"
+        # Build full API endpoint
+        endpoint = f"{API_URL}/{WORKSPACE}/{WORKFLOW}?api_key={API_KEY}"
 
-    # Make the POST request to Roboflow
-    response = requests.post(endpoint, files={"file": image_bytes}, data={"confidence": "0.5", "overlap": "0.4"})
+        # Make POST request to Roboflow
+        response = requests.post(
+            endpoint,
+            files={"file": image_bytes},
+            data={"confidence": "0.5", "overlap": "0.4"}
+        )
 
-    if response.status_code != 200:
-        return f"Error from Roboflow: {response.text}"
+        # Log raw response
+        print("Roboflow status:", response.status_code)
+        print("Roboflow response:", response.text)
 
-    # Roboflow returns image as bytes
-    result = requests.get(response.json()["image"]["url"])
-    result_img = Image.open(BytesIO(result.content))
+        if response.status_code != 200:
+            return f"Roboflow error: {response.text}"
+
+        json_response = response.json()
+        if "image" not in json_response or "url" not in json_response["image"]:
+            return f"Unexpected response format: {json_response}"
+
+        # Download result image
+        image_url = json_response["image"]["url"]
+        result = requests.get(image_url)
+        result_img = Image.open(BytesIO(result.content))
+
+        return result_img
+
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-    return result_img
 
 # Gradio UI
 gr.Interface(