Update app.py

app.py

@@ -1,25 +1,31 @@
+
 import asyncio
 import websockets
-print(0)
-# Define the WebSocket server handler
-async def handler(websocket, path):
-    print(f"New connection from {websocket.remote_address}")
+
+async def handle_connection(websocket, path):
+    """Handle a new WebSocket connection."""
+    print(f"New connection from {path}")
+
     try:
         while True:
-            # Wait for a message from the client
+            # Receive message from client
             message = await websocket.recv()
-            print(f"Received: {message}")
+            print(f"Received message: {message}")
 
-            # Echo the message back to the client
-            await websocket.send(f"Echo: {message}")
+            # Send response back to client
+            await websocket.send(f"Server received: {message}")
     except websockets.ConnectionClosed:
-        print(f"Connection closed from {websocket.remote_address}")
+        print(f"Connection closed from {path}")
+    except websockets.WebSocketException as e:
+        print(f"WebSocket error: {e}")
+    except Exception as e:
+        print(f"Error: {e}")
 
-# Start the WebSocket server
-async def start_server():
-    server = await websockets.serve(handler, "0.0.0.0", 7860)
-    print("WebSocket server started on ws://0.0.0.0:7860")
-    await server.wait_closed()
+async def main():
+    """Start the WebSocket server."""
+    async with websockets.serve(handle_connection, "0.0.0.0", 7860):
+        print("Server started on port 8765")
+        await asyncio.Future()  # Run forever
 
-# Run the server
-asyncio.run(start_server())
+if __name__ == "__main__":
+    asyncio.run(main())