Update app.py

app.py

@@ -7,6 +7,7 @@ import asyncio
 import sqlite3
 import json
 import gradio as gr
+from bs4 import BeautifulSoup
 from gradio_client import Client
 import time
 
@@ -22,6 +23,17 @@ stop = asyncio.Future()
 messageTextbox = None
 serverMessageTextbox = None
 
+def slow_echo(message, history):
+    for i in range(len(message)):
+        time.sleep(0.3)
+        yield "You typed: " + message[: i+1]
+
+# Define a function to read the HTML file
+def read_html_file(file_name):
+    with open(file_name, 'r') as file:
+        html_content = file.read()
+    return html_content        
+
 # Set up the HTTP server
 class SimpleHTTPRequestHandler(http.server.SimpleHTTPRequestHandler):
     def do_GET(self):
@@ -46,21 +58,34 @@ def sendErrorMessage(ws, errorMessage):
     errorResponse = {'error': errorMessage}
     ws.send(json.dumps(errorResponse))
 
-# Function to send a question to the chatbot and get the response
+# 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]
         response = requests.post(
-            "https://flowiseai-flowise.hf.space/api/v1/prediction/522afa32-484c-471e-9ba5-4d6d2edfb89b",
+            "https://flowiseai-flowise.hf.space/api/v1/prediction/8f9d1727-7f27-4445-a5c6-9efd72dd0320",
             headers={"Content-Type": "application/json"},
             json={"question": message},
         )
         response_content = response.content.decode('utf-8')
-     
+        
         return response_content
     except Exception as e:
         print(e)
 
-
 async def listen_for_messages():
     while True:
         if len(client_messages) > 0:
@@ -88,7 +113,6 @@ async def handleWebSocket(ws):
         message_copy = message
         client_messages.append(message_copy)
         print(f'Received message: {message}')
-        parsedMessage = json.loads(message)        
         messageText = message
         messages.append(message)
         timestamp = datetime.datetime.now().isoformat()
@@ -128,6 +152,25 @@ 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
@@ -139,24 +182,27 @@ async def start_websockets(websocketPort):
     return "Used ports:\n" + '\n'.join(map(str, used_ports))
 
 with gr.Blocks() as demo:
-    
-    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")     
-        # Use the server_responses list to update the serverMessageTextbox
-        server_message = 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():                  
-        gui = gr.Button("connect interface")
-        with gr.Row():   
-        port = gr.Textbox()  
-        startWebsockets.click(start_websockets, inputs=websocketPort, outputs=port)
-        gui.click(listen_for_messages, inputs=None, outputs={client_message, server_message})      
-        stopWebsockets.click(stop_websockets)
+    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")     
+                    # Use the server_responses list to update the serverMessageTextbox
+                    server_message = 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():                  
+                    gui = gr.Button("connect interface")
+                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])
+  
+        with gr.TabItem("FalconChat", elem_id="falconchat", id=1):
+                gr.load("HuggingFaceH4/starchat-playground", src="spaces")
 
 demo.queue()    
-demo.launch(share=True)
+demo.launch(share=True, server_port=1111)