Update app.py

app.py

@@ -1,20 +1,23 @@
-import requests
-import datetime
 import http.server
-import websockets
-import websocket
 import asyncio
-import sqlite3
 import json
-import gradio as gr
-from bs4 import BeautifulSoup
-from gradio_client import Client
+import random
+import datetime
+import sqlite3
 import time
+import websockets
+import gradio as gr
 
-client_messages = []
-server_responses = []
-messages = []
-used_ports = []
+client_messages = []  # List to store client messages
+server_responses = []  # List to store server responses
+messages = []  # List to store all messages
+used_ports = []  # List to store used WebSocket ports
+
+# Set up the SQLite database
+db = sqlite3.connect('chat-hub.db')
+cursor = db.cursor()
+cursor.execute('CREATE TABLE IF NOT EXISTS messages (id INTEGER PRIMARY KEY AUTOINCREMENT, sender TEXT, message TEXT, timestamp TEXT)')
+db.commit()
 
 websocket_server = None
 stop = asyncio.Future()
@@ -58,39 +61,23 @@ 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 askQuestion(question):
-    try:
-        response = requests.post(
-            "https://flowiseai-flowise.hf.space/api/v1/prediction/8fd03bee-7752-484f-844f-bdae1e4cb4fe",
-            headers={"Content-Type": "application/json"},
-            json={"question": question},
-        )
-        response_content = response.content.decode('utf-8')
-        client_messages.append(response_content)
-        return response_content
-    except Exception as e:
-        print(e)
-        
-# 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,
-                  "topK":2
-        }
-        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
-    except Exception as e:
-        print(e)    
+
+async def handle_message(client_responses, user_input, user_message, history):
+    client_responses.append(user_input)
+    # Provide a response to the user
+    return "You typed: " + user_message, history + [[user_message, None]]
+
+async def ask_question(agent_responses, user_input, user_message):
+    agent_responses.append(user_input)
+    bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+    return bot_message
+    
+# Define the WebSocket server handler
+async def websocket_server(ws, path):
+    async for message in ws:
+        # Handle incoming WebSocket messages using the handle_message function
+        response, _ = await handle_message(client_messages, message, message, [])
+        await ws.send(response)
 
 async def handleWebSocket(ws):
     print('New connection')
@@ -163,27 +150,52 @@ async def start_client():
             return server_message
             await asyncio.sleep(0.1)
 
+# Start the WebSocket server
+start_server = websockets.serve(websocket_server, "localhost", 8765)
+
 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_msg = gr.Textbox(lines=15, max_lines=130, label="Client inputs")     
-                    # Use the server_responses list to update the serverMessageTextbox
-                    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")            
-                    stopWebsockets = gr.Button("Stop WebSocket Server")               
-                with gr.Row():    
-                    Bot1 = gr.Button("Bot 1")    
-                with gr.Row():
-                    gui = gr.Button("connect interface")
-                with gr.Row():   
-                    port = gr.Textbox()  
-                    startWebsockets.click(start_websockets, inputs=websocketPort, outputs=port)
-                    Bot1.click(askQuestion, inputs=client_msg, outputs=server_msg)                  
- 
-demo.queue()    
-demo.launch()
+    with gr.Column(scale=1, min_width=600):
+        with gr.Row():
+            input_msg1 = gr.Textbox(lines=15, max_lines=130, label="inputs", interactive=False)
+            response_msg1 = gr.Textbox(lines=15, max_lines=130, label="Client responses", interactive=False)
+        with gr.Row():
+            user_input1 = gr.Textbox(label="User Input")
+        with gr.Row():
+            bot1 = gr.Button("Ask Agent1")
+            bot2 = gr.Button("Ask Agent2")
+        with gr.Row():
+            websocket_port = gr.Slider(minimum=1000, maximum=9999, label="Websocket server port", interactive=True, randomize=False)
+            start_client = gr.Button("Start WebSocket client")
+            stop_client = gr.Button("Stop WebSocket client")
+        with gr.Row():
+            port_in_use = gr.Textbox()
+        with gr.Row():
+            chat = gr.ChatInterface(handle_message, inputs=input_msg1, outputs=[response_msg1])
+            start_client.click(start_client, inputs=websocket_port, outputs=[port_in_use]).then(handle_message, inputs=None, outputs=chat)
+            bot1.click(ask_question, inputs=user_input1, outputs=response_msg1)
+    
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.Button("Clear")
+
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+        history[-1][1] = ""
+        for character in bot_message:
+            history[-1][1] += character
+            time.sleep(0.05)
+            yield history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    clear.click(lambda: None, None, chatbot, queue=False)
+    
+    server = await websockets.serve(websocket_server, "localhost", 8765)  # Set your desired WebSocket server parameters
+    await server.wait_closed()
+    
+demo.queue()
+demo.launch(share=True, server_name="0.0.0.0", server_port=8765)