Update app.py

app.py

@@ -149,53 +149,55 @@ async def start_client():
             await ws.send(client_response)
             return server_message
             await asyncio.sleep(0.1)
+async def start_websocket_server():
+    server = await websockets.serve(websocket_server, "localhost", 8765)  # Set your desired WebSocket server parameters
+    await server.wait_closed()
 
-# Start the WebSocket server
-start_server = websockets.serve(websocket_server, "localhost", 8765)
-
-with gr.Blocks() as demo:
-    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)
+if __name__ == "__main__":
+    demo = gr.Interface.load()
     
-    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)
+    with gr.Blocks() as demo:
+        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)
     
-demo.queue()
-demo.launch(share=True, server_name="0.0.0.0", server_port=8765)
+    asyncio.run(start_websocket_server())