Update main.py

main.py

@@ -180,9 +180,12 @@ async def analyze_image(file: UploadFile = File(...)):
     # Generate content using the AI model
     response = model.generate_content([prompt, blob])
 
-    # Remove extra characters and escape sequences
-    cleaned_response = re.sub(r'\\n|json', '', response.text).strip()
+    # Extract JSON string from Markdown-formatted JSON string
+    json_string = re.search(r'```json(.*?)```', response.text, re.DOTALL).group(1)
     
+    # Clean JSON string
+    cleaned_response = json_string.strip()
+
     # Parse the cleaned string as JSON
     data = json.loads(cleaned_response)
     fd = json.dumps(data, indent=4)