Update app.py

app.py

@@ -108,9 +108,20 @@ def process_image(image):
 def gradio_interface(patient_info, query_type, image):
     if image is not None:
         image_response = process_image(image)
+
+        # Call LLM with patient info and query
         session_id = create_chat_session()
-        llm_response = submit_query(session_id, f"Patient Info: {patient_info}. Query Type: {query_type}.")
-        return f"Patient Info: {patient_info}\nQuery Type: {query_type}\n\n{image_response}\n\nLLM Response:\n{llm_response['data']['message']}"
+        query = f"Patient Info: {patient_info}\nQuery Type: {query_type}"
+        llm_response = submit_query(session_id, query)
+        
+        # Debug: Print the full response to inspect it
+        print("LLM Response:", llm_response)
+
+        # Handle missing 'message' field safely
+        message = llm_response.get('data', {}).get('message', 'No message returned from LLM')
+
+        response = f"Patient Info: {patient_info}\nQuery Type: {query_type}\n\n{image_response}\n\nLLM Response:\n{message}"
+        return response
     else:
         return "Please upload an image."