Update app.py

app.py

@@ -94,7 +94,7 @@ def extract_json_from_answer(answer, image_analysis):
             end_idx = answer.rfind('}')
             if start_idx != -1 and end_idx != 0:
                 json_str = answer[start_idx:end_idx]
-                json_str += f",image_analysis: ['prediction': {image_analysis['prediction']}, 'confidence': {image_analysis['confidence']}%.]"
+                json_str += f",'image_analysis': ['prediction': {image_analysis['prediction']}, 'confidence': {image_analysis['confidence']}%.]"
                 return json.loads(json_str)
         except (json.JSONDecodeError, ValueError):
             logger.error("Failed to parse JSON from response")
@@ -208,10 +208,10 @@ def gradio_interface(patient_info, image):
             raise ValueError("Invalid response structure from LLM")
         
         # Extract and clean JSON from the response
-        json_data = extract_json_from_answer(llm_response['data']['answer'], image_analysis)
         logger.info(f"llm_response: {llm_response}")
         logger.info(f"llm_response[data]: {llm_response['data']}")
         logger.info(f"llm_response[data][answer]: {llm_response['data']['answer']}")
+        json_data = extract_json_from_answer(llm_response['data']['answer'], image_analysis)
         return json.dumps(json_data, indent=2)
         
     except Exception as e: