Update app.py

app.py

@@ -35,11 +35,23 @@ def get_completion(image, parameters=None, endpoint_url=endpoint_url):
 
     response = requests.post(endpoint_url, headers=headers, data=json.dumps(data))
 
-    # Check for errors
     if response.status_code != 200:
         return {"error": response.text}
 
-    return json.loads(response.content.decode("utf-8"))
+    try:
+        # Try parsing the response as JSON
+        response_data = json.loads(response.content.decode("utf-8"))
+
+        # Check if it's a list and extract the first item
+        if isinstance(response_data, list) and len(response_data) > 0:
+            return response_data[0]
+        elif isinstance(response_data, dict):
+            return response_data
+        else:
+            return {"error": "Unexpected response format"}
+    except json.JSONDecodeError:
+        return {"error": "Failed to decode API response"}
+
 
 # Helper function to download and process the image from a URL
 def caption_image(image_url):