Update app.py

app.py

@@ -129,7 +129,8 @@ def gradio_interface(patient_info, image):
                 "error": "Model initialization failed. Please check the logs for details.",
                 "status": "error"
             }, indent=2)
-            
+
+        classes = ["Alzheimer's", "Stroke", "Tumor", "Normal"]
         # Process image if provided
         image_analysis = None
         if image is not None:
@@ -144,15 +145,15 @@ def gradio_interface(patient_info, image):
             logger.info(f"Prediction: {prediction}")
             # Format prediction results
             image_analysis = {
-                "prediction": float(prediction[0][0]),
+                "prediction": classes[int(prediction[0][0])],
                 "confidence": float(prediction[0][0]) * 100
             }
             logger.info(f"Image analysis results: {image_analysis}")
-        
+
+        patient_info += f"Prediction based on MRI images: {image_analysis["prediction"]}, Confidence: {image_analysis["confidence"]}"
         # Create chat session and submit query
         session_id = create_chat_session()
-        llm_response = submit_query(session_id, patient_info, 
-                                  json.dumps(image_analysis) if image_analysis else None)
+        llm_response = submit_query(session_id, patient_info)
         
         if not llm_response or 'data' not in llm_response or 'answer' not in llm_response['data']:
             raise ValueError("Invalid response structure from LLM")