Update app.py

app.py

@@ -20,35 +20,64 @@ logger = logging.getLogger(__name__)
 api_key = 'KGSjxB1uptfSk8I8A7ciCuNT9Xa3qWC3'
 external_user_id = 'plugin-1717464304'
 
-# Step 1: Create a chat session
 def create_chat_session():
-    create_session_url = 'https://api.on-demand.io/chat/v1/sessions'
-    create_session_headers = {
-        'apikey': api_key
-    }
-    create_session_body = {
-        "pluginIds": [],
-        "externalUserId": external_user_id
-    }
-    # Make the request to create a chat session
-    response = requests.post(create_session_url, headers=create_session_headers, json=create_session_body)
-    response_data = response.json()
-    session_id = response_data['data']['id']
-    return session_id
-
+    try:
+        create_session_url = 'https://api.on-demand.io/chat/v1/sessions'
+        create_session_headers = {
+            'apikey': api_key,
+            'Content-Type': 'application/json'
+        }
+        create_session_body = {
+            "pluginIds": [],
+            "externalUserId": external_user_id
+        }
+        
+        response = requests.post(create_session_url, headers=create_session_headers, json=create_session_body)
+        response.raise_for_status()
+        return response.json()['data']['id']
+    
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Error creating chat session: {str(e)}")
+        raise
+        
 def submit_query(session_id, query):
-    submit_query_url = f'https://api.on-demand.io/chat/v1/sessions/{session_id}/query'
-    submit_query_headers = {
-        'apikey': api_key
-    }
-    submit_query_body = {
-        "endpointId": "predefined-openai-gpt4o",
-        "query": query,
-        "pluginIds": ["plugin-1712327325", "plugin-1713962163"],
-        "responseMode": "sync"
-    }
-    response = requests.post(submit_query_url, headers=submit_query_headers, json=submit_query_body)
-    return response.json()
+    try:
+        submit_query_url = f'https://api.on-demand.io/chat/v1/sessions/{session_id}/query'
+        submit_query_headers = {
+            'apikey': api_key,
+            'Content-Type': 'application/json'
+        }
+        
+        structured_query = f"""
+        Based on the following patient information, provide a detailed medical analysis in JSON format:
+
+        {query}
+
+        Return only valid JSON with these fields:
+        - diagnosis_details
+        - probable_diagnoses (array)
+        - treatment_plans (array)
+        - lifestyle_modifications (array)
+        - medications (array of objects with name and dosage)
+        - additional_tests (array)
+        - precautions (array)
+        - follow_up (string)
+        """
+        
+        submit_query_body = {
+            "endpointId": "predefined-openai-gpt4o",
+            "query": structured_query,
+            "pluginIds": ["plugin-1712327325", "plugin-1713962163"],
+            "responseMode": "sync"
+        }
+        
+        response = requests.post(submit_query_url, headers=submit_query_headers, json=submit_query_body)
+        response.raise_for_status()
+        return response.json()
+    
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Error submitting query: {str(e)}")
+        raise
 
 
 def extract_json_from_answer(answer):