Update app.py

app.py

@@ -186,7 +186,7 @@ def create_chat_session():
         
         logger.info("Creating chat session...")
         response = requests.post(create_session_url, headers=create_session_headers, json=create_session_body)
-        response.raise_for_status()  # Raise an exception for bad status codes
+        response.raise_for_status()
         
         response_data = response.json()
         logger.info(f"Session created successfully: {json.dumps(response_data, indent=2)}")
@@ -249,16 +249,24 @@ def gradio_interface(patient_info, query_type):
         # Navigate the response structure with detailed logging
         logger.info(f"Processing LLM response: {json.dumps(llm_response, indent=2)}")
         
-        if 'data' not in llm_response:
-            logger.error("Response missing 'data' field")
+        # Check for answer in the correct field
+        if 'data' not in llm_response or 'answer' not in llm_response['data']:
+            logger.error("Response missing required fields")
             return f"Error: Unexpected response structure\nFull response: {json.dumps(llm_response, indent=2)}"
         
-        message = llm_response.get('data', {}).get('message')
-        if not message:
-            logger.error("No message found in response data")
-            return f"Error: No message in response\nFull response: {json.dumps(llm_response, indent=2)}"
+        answer = llm_response['data']['answer']
+        if not answer:
+            logger.error("No answer found in response data")
+            return f"Error: No answer in response\nFull response: {json.dumps(llm_response, indent=2)}"
+        
+        # Format the response
+        response = f"Patient Info: {patient_info}\n\nQuery Type: {query_type}\n\nAnalysis:\n{answer}"
+        
+        # Add completion metrics if available
+        if 'metrics' in llm_response['data']:
+            metrics = llm_response['data']['metrics']
+            response += f"\n\nProcessing Time: {metrics.get('totalTimeSec', 'N/A')} seconds"
         
-        response = f"Patient Info: {patient_info}\nQuery Type: {query_type}\n\nLLM Response:\n{message}"
         return response
     
     except Exception as e: