.

main.py

@@ -403,6 +403,15 @@ async def websocket_endpoint(websocket: WebSocket):
                     # Close connection at 1 minute
                     if time_since_activity >= CONNECTION_TIMEOUT:
                         print(f"[{current_time:.3f}] Closing connection {client_id} due to timeout")
+                        
+                        # Clear the input queue before closing
+                        while not input_queue.empty():
+                            try:
+                                input_queue.get_nowait()
+                                input_queue.task_done()
+                            except asyncio.QueueEmpty:
+                                break
+                        
                         await websocket.close(code=1000, reason="User inactivity timeout")
                         return
                     

static/index.html

@@ -293,7 +293,7 @@
             autoInputInterval = setInterval(() => {
                 const currentTime = Date.now();
                 
-                if (!autoInputEnabled || !lastSentPosition || !isConnected) {
+                if (!autoInputEnabled || !lastSentPosition || !isConnected || socket.readyState !== WebSocket.OPEN) {
                     return;
                 }
                 
@@ -330,6 +330,8 @@
                         lastAutoInputTime = currentTime;
                     } catch (error) {
                         console.error("Error sending auto-input:", error);
+                        // Stop auto-input if there's an error (connection likely closed)
+                        stopAutoInput();
                     }
                 }
             }, 100); // Check every 100ms
@@ -410,14 +412,18 @@
         function resetTimeout() {
             // Send a heartbeat to reset the server's timeout
             if (socket && socket.readyState === WebSocket.OPEN) {
-                socket.send(JSON.stringify({ type: "heartbeat" }));
+                try {
+                    socket.send(JSON.stringify({ type: "heartbeat" }));
+                } catch (error) {
+                    console.error("Error sending heartbeat:", error);
+                }
             }
             stopTimeoutCountdown();
         }
 
         function sendInputState(x, y, isLeftClick = false, isRightClick = false) {
             const currentTime = Date.now();
-            if (isConnected && (isLeftClick || isRightClick || !lastSentPosition || currentTime - lastSentTime >= SEND_INTERVAL)) {
+            if (isConnected && socket.readyState === WebSocket.OPEN && (isLeftClick || isRightClick || !lastSentPosition || currentTime - lastSentTime >= SEND_INTERVAL)) {
                 try {
                     socket.send(JSON.stringify({
                         "x": x,
@@ -497,7 +503,7 @@
         });
 
         document.addEventListener("keyup", function (event) {
-            if (!isConnected) return;
+            if (!isConnected || socket.readyState !== WebSocket.OPEN) return;
             
             // Remove the key from our set of pressed keys
             pressedKeys.delete(event.key);