Update app.py

app.py

@@ -59,12 +59,12 @@ def encode_image(image):
     image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
-
 def roboflow(image, confidence, overlap, stroke_width=1, labels=False):
     '''
         Send the image to Roboflow API for inference.
         Returns JSON and image with bounding boxes drawn on to it.
     '''
+
     json_url = f"https://detect.roboflow.com/sku-110k/2?api_key=gHiUgOSq9GqTnRy5mErk&confidence={confidence}&overlap={overlap}&format=json"
     image_url = f"https://detect.roboflow.com/sku-110k/2?api_key=gHiUgOSq9GqTnRy5mErk&confidence={confidence}&overlap={overlap}&format=image&labels={str(labels).lower()}&stroke={stroke_width}"
     
@@ -74,12 +74,15 @@ def roboflow(image, confidence, overlap, stroke_width=1, labels=False):
     json_request = requests.post(json_url, data=encoded_image, headers=headers)
     image_request = requests.post(image_url, data=encoded_image, headers=headers)
 
+    print("JSON Response Headers:", json_request.headers)
+    print("Image Response Headers:", image_request.headers)
+
     if json_request.status_code == 200:
         json_response = json_request.json()
     else:
         json_response = {"error": "Failed to get JSON response"}
 
-    if image_request.status_code == 200:
+    if image_request.status_code == 200 and 'image' in image_request.headers.get('Content-Type', ''):
         image_response = Image.open(io.BytesIO(image_request.content))
     else:
         image_response = None
@@ -88,6 +91,8 @@ def roboflow(image, confidence, overlap, stroke_width=1, labels=False):
         "json": json_response,
         "image": image_response
     }
+
+
 # Image Splitting and Merging Functionality
 def split_image(image, tile_size=640, overlap=160):
     img_width, img_height = image.size