Update app.py

app.py

@@ -42,17 +42,21 @@ def analyze_image(image, prompt):
     full_response = processor.decode(output[0])
     print("Full response:", full_response)  # Debug print
 
-    return full_response
-    # try:
-    #     json_match = re.search(r'\{.*?\}', full_response, re.DOTALL)
-    #     if json_match:
-    #         json_str = json_match.group(0)
-    #         try:
-    #             return json.loads(json_str)
-    #         except json.JSONDecodeError as e:
-    #             return full_response
-    # except Exception as e:
-    #     return full_response
+    # return full_response
+    try:
+        json_match = re.search(r'\{.*?\}', full_response, re.DOTALL)
+        if json_match:
+            json_str = json_match.group(0)
+            try:
+                return json.loads(json_str)
+            except json.JSONDecodeError as e:
+                print(f"JSON decode error: {e}")
+                return {"error": "Invalid JSON in model output", "full_response": full_response}
+        else:
+            return {"error": "No JSON found in model output", "full_response": full_response}
+    except Exception as e:
+        print(f"Error in analyze_image: {e}")
+        return {"Full Response": str(e), "full_response": full_response}
 
 
 default_prompt = """Analyze this image and determine if it contains a data logger.