Update app.py

app.py

@@ -81,15 +81,26 @@ def roboflow(image, confidence, overlap, stroke_width=1, labels=False):
     print("JSON Response Headers:", json_request.headers)
     print("Image Response Headers:", image_request.headers)
 
+    json_response = {}
+    image_response = None
+
     if json_request.status_code == 200:
-        json_response = json_request.json()
+        try:
+            json_response = json_request.json()
+        except json.JSONDecodeError:
+            json_response = {"error": "Invalid JSON response"}
     else:
-        json_response = {"error": "Failed to get JSON response"}
+        json_response = {"error": f"Failed to get JSON response, status code: {json_request.status_code}"}
 
     if image_request.status_code == 200 and 'image' in image_request.headers.get('Content-Type', ''):
-        image_response = Image.open(io.BytesIO(image_request.content))
+        try:
+            image_response = Image.open(io.BytesIO(image_request.content))
+        except Exception as e:
+            image_response = None
+            print(f"Failed to open image: {e}")
     else:
-        image_response = None
+        print(f"Failed to retrieve image, status code: {image_request.status_code}")
+        print("Image Response Content:", image_request.content)
 
     return {
         "json": json_response,