Update main.py

main.py

@@ -158,7 +158,7 @@ async def websocket_endpoint(websocket: WebSocket):
         while True:
             try:
                 # Receive user input with a timeout
-                data = await asyncio.wait_for(websocket.receive_json(), timeout=90.0)
+                data = await asyncio.wait_for(websocket.receive_json(), timeout=9000.0)
                 
                 if data.get("type") == "heartbeat":
                     await websocket.send_json({"type": "heartbeat_response"})