Update app.py

app.py

@@ -150,12 +150,17 @@ async def start_websocket_server(host='0.0.0.0', port=8765):
     logger.info(f"WebSocket server started on ws://{host}:{port}")
     return server
 
+# Global variables for event loop and queue
+main_event_loop = None
+message_queue = []
+
 def send_message(message, username, room_id):
     """Function to send a message from the Gradio interface."""
     if not message.strip():
         return None
     
-    loop = asyncio.get_event_loop()
+    global message_queue
+    
     msg_data = {
         "type": "chat",
         "content": message,
@@ -163,9 +168,10 @@ def send_message(message, username, room_id):
         "room_id": room_id
     }
     
-    loop.create_task(broadcast_message(msg_data, room_id))
+    # Add to queue for processing by the main loop
+    message_queue.append(msg_data)
     
-    # Format the message for display
+    # Format the message for display in the UI
     formatted_msg = f"{username}: {message}"
     return formatted_msg
 
@@ -244,14 +250,35 @@ def create_gradio_interface():
     
     return interface
 
+async def process_message_queue():
+    """Process messages in the queue and broadcast them."""
+    global message_queue
+    
+    while True:
+        # Check if there are messages to process
+        if message_queue:
+            # Get the oldest message
+            msg_data = message_queue.pop(0)
+            # Broadcast it
+            await broadcast_message(msg_data, msg_data["room_id"])
+            
+        # Sleep to avoid busy-waiting
+        await asyncio.sleep(0.1)
+
 async def main():
     """Main function to start the application."""
-    global NODE_NAME
+    global NODE_NAME, main_event_loop
     NODE_NAME, port = get_node_name()
     
+    # Store the main event loop for later use
+    main_event_loop = asyncio.get_running_loop()
+    
     # Start WebSocket server
     server = await start_websocket_server()
     
+    # Start message queue processor
+    asyncio.create_task(process_message_queue())
+    
     # Create and launch Gradio interface
     interface = create_gradio_interface()
     
@@ -283,10 +310,11 @@ async def main():
     
     logger.info(f"Starting Gradio interface on http://0.0.0.0:{port} with node name '{NODE_NAME}'")
     
-    await server.serve()
+    # Start message processor
+    logger.info("Starting message queue processor")
     
-    # Keep the WebSocket server running
-    await asyncio.Future()
+    # Run the server and keep it running
+    await server.serve()
 
 if __name__ == "__main__":
     asyncio.run(main())