Update app.py

app.py

@@ -20,6 +20,37 @@ 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
+
+# Step 2: Submit a query to the API
+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()
+
 def load_model():
     try:
         model_path = 'model_epoch_01.h5.keras'