Update app.py

app.py

@@ -57,53 +57,38 @@ db.commit()
 def sendErrorMessage(ws, errorMessage):
     errorResponse = {'error': errorMessage}
     ws.send(json.dumps(errorResponse))
-
-# Define a function to ask a question to the chatbot and display the response
-async def askQuestion2(question):
-    try:
-        message = messages[-1]
-        client = Client("http://localhost:1111/")
-        response = client.predict(
-                message,   
-                fn_index=2
-                )        
-        return response
-    except Exception as e:
-        print(e)
-
+    
 # Define a function to ask a question to the chatbot and display the response
 async def askQuestion(question):
     try:
-        message = messages[-1]
+        message = server_responses[-1]
         response = requests.post(
-            "https://flowiseai-flowise.hf.space/api/v1/prediction/8f9d1727-7f27-4445-a5c6-9efd72dd0320",
+            "https://flowiseai-flowise.hf.space/api/v1/prediction/0da97a1a-963b-4f6b-bcad-928c0630e5c1",
             headers={"Content-Type": "application/json"},
             json={"question": message},
         )
         response_content = response.content.decode('utf-8')
-        
+        client_messages.append(response_content)
         return response_content
     except Exception as e:
         print(e)
 
-async def listen_for_messages():
-    while True:
-        if len(client_messages) > 0:
-            # Get the latest client message
-            client_message = client_messages[-1]
-            try:
-                server_message = server_responses[-1]
-            except IndexError:
-                # Handle the case when there are no server responses yet
-                server_message = "connected successfully"
-            
-            return client_message, server_message
-        else:
-            # Handle the case when there are no client messages yet
-            client_message = "connected successfully"
-            server_message = "connected successfully"
-            
-            return client_message, server_message
+# Define a function to ask a question to the chatbot and display the response
+async def askQuestion2(question):
+    try:
+        message = server_responses[-1]
+        url = "https://api.chaindesk.ai/datastores/query/clka5g9zc000drg6mxl671ekv"
+        payload = {"query": message}
+        headers = {
+            "Authorization": "Bearer 5315cd7b-bb79-49bc-bca2-8bcc7b243504",
+            "Content-Type": "application/json"
+        }
+
+response = requests.request("POST", url, json=payload, headers=headers)
+
+print(response.text)
+
+return response.text
 
 async def handleWebSocket(ws):
     print('New connection')
@@ -122,8 +107,9 @@ async def handleWebSocket(ws):
                    (sender, messageText, timestamp))
         db.commit()        
         try:
-            message = messages[-1]
+            message = client_messages[-1]
             answer = await askQuestion(message)  # Use the message directly
+            messages.append(answer)
             response = {'answer': answer}
             serverMessageText = response.get('answer', '')        
             await ws.send(json.dumps(response))
@@ -152,25 +138,6 @@ def stop_websockets():
     else:
         print("WebSocket server is not running.")    
 
-async def start_client():
-    async with websockets.connect('ws://localhost:5000') as ws:        
-        while True:
-            # Listen for messages from the server
-            server_message = await ws.recv()
-            messages.append(server_message)
-            return server_message
-            client_message = await askQuestion2(server_message)
-            
-            # Send the client's response to the server
-            await ws.send(client_message)
-            
-            # Append the client message and server response to the respective lists
-            client_message_textboxes.append(client_message)
-            server_response_textboxes.append(server_message)
-            return client_message            
-            # Pause for a short duration to allow for smooth streaming
-            await asyncio.sleep(0.1)
-
 # Start the WebSocket server 
 async def start_websockets(websocketPort):
     global messageTextbox, serverMessageTextbox, websocket_server
@@ -181,15 +148,28 @@ async def start_websockets(websocketPort):
     print(f"Starting WebSocket server on port {websocketPort}...")
     return "Used ports:\n" + '\n'.join(map(str, used_ports))
 
+async def start_client():
+    async with websockets.connect('ws://localhost:5000') as ws:        
+        while True:
+            # Listen for messages from the server            
+            server_message = await ws.recv()
+            server_responses.append(server_message)
+            messages.append(server_message)
+            client_response = await askQuestion2(server_message)
+            client_messages.append(client_response)
+            await ws.send(client_response)
+            return server_message
+            await asyncio.sleep(0.1)
+
 with gr.Blocks() as demo:
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("Websocket Server", elem_id="websocket_server", id=0):
             with gr.Column(scale=1, min_width=600):   
                 with gr.Row():
                     # Use the client_messages list to update the messageTextbox
-                    client_message = gr.Textbox(lines=15, max_lines=130, label="Client inputs")     
+                    client_msg = gr.Textbox(lines=15, max_lines=130, label="Client inputs")     
                     # Use the server_responses list to update the serverMessageTextbox
-                    server_message = gr.Textbox(lines=15, max_lines=130, label="Server responses")            
+                    server_msg = gr.Textbox(lines=15, max_lines=130, label="Server responses")            
                 with gr.Row():
                     websocketPort = gr.Slider(minimum=1000, maximum=9999, label="Websocket server port", interactive=True, randomize=False)
                     startWebsockets = gr.Button("Start WebSocket Server")            
@@ -199,10 +179,13 @@ with gr.Blocks() as demo:
                 with gr.Row():   
                     port = gr.Textbox()  
                     startWebsockets.click(start_websockets, inputs=websocketPort, outputs=port)
-                    gui.click(start_client, inputs=None, outputs=[client_message, server_message])
+                    gui.click(start_client, inputs=None, outputs=server_msg)
   
+                    client_msg.change(askQuestion, inputs=client_msg, outputs=server_msg)
+                    server_msg.change(askQuestion2, inputs=server_msg, outputs=client_msg).then(askQuestion, inputs=client_msg, outputs=server_msg)
+
         with gr.TabItem("FalconChat", elem_id="falconchat", id=1):
                 gr.load("HuggingFaceH4/starchat-playground", src="spaces")
 
 demo.queue()    
-demo.launch()
+demo.launch(share=True, server_port=1111)