.

main.py

@@ -152,9 +152,8 @@ async def websocket_endpoint(websocket: WebSocket):
     await websocket.accept()
     previous_frames = []
     previous_actions = []
-    positions = ['496~61', '815~335', '815~335', '815~335', '787~342', '749~345', '749~345', '703~346', '703~346', '654~347', '604~349', '604~349', '555~353', '509~357', '509~357']
     positions = ['815~335', '787~342', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
-    #positions = positions[:4]
+    
     try:
         while True:
             try:
@@ -199,15 +198,15 @@ async def websocket_endpoint(websocket: WebSocket):
             
             except asyncio.TimeoutError:
                 print("WebSocket connection timed out")
-                #break
+                break  # Exit the loop on timeout
             
             except WebSocketDisconnect:
                 print("WebSocket disconnected")
-                #break
+                break  # Exit the loop on disconnect
 
     except Exception as e:
         print(f"Error in WebSocket connection {client_id}: {e}")
     
     finally:
         print(f"WebSocket connection closed: {client_id}")
-        # Remove the explicit websocket.close() call here
+        await websocket.close()  # Ensure the WebSocket is closed

static/index.html

@@ -66,7 +66,11 @@
         function startHeartbeat() {
             heartbeatInterval = setInterval(() => {
                 if (isConnected) {
-                    socket.send(JSON.stringify({ type: "heartbeat" }));
+                    try {
+                        socket.send(JSON.stringify({ type: "heartbeat" }));
+                    } catch (error) {
+                        console.error("Error sending heartbeat:", error);
+                    }
                 }
             }, 15000); // Send heartbeat every 15 seconds
         }
@@ -81,13 +85,16 @@
         function sendMousePosition(x, y, forceUpdate = false) {
             const currentTime = Date.now();
             if (isConnected && !isProcessing && (forceUpdate || !lastSentPosition || currentTime - lastSentTime >= SEND_INTERVAL)) {
-                //isProcessing = true;
-                socket.send(JSON.stringify({
-                    "action_type": "move",
-                    "mouse_position": [x, y]
-                }));
-                lastSentPosition = { x, y };
-                lastSentTime = currentTime;
+                try {
+                    socket.send(JSON.stringify({
+                        "action_type": "move",
+                        "mouse_position": [x, y]
+                    }));
+                    lastSentPosition = { x, y };
+                    lastSentTime = currentTime;
+                } catch (error) {
+                    console.error("Error sending mouse position:", error);
+                }
             }
         }
 
@@ -114,17 +121,25 @@
             let y = event.clientY - rect.top;
 
             isProcessing = true;
-            socket.send(JSON.stringify({
-                "action_type": "left_click",
-                "mouse_position": [x, y]
-            }));
+            try {
+                socket.send(JSON.stringify({
+                    "action_type": "left_click",
+                    "mouse_position": [x, y]
+                }));
+            } catch (error) {
+                console.error("Error sending click action:", error);
+            }
         });
 
         // Graceful disconnection
         window.addEventListener('beforeunload', function (e) {
             if (isConnected) {
-                socket.send(JSON.stringify({ type: "disconnect" }));
-                socket.close();
+                try {
+                    socket.send(JSON.stringify({ type: "disconnect" }));
+                    socket.close();
+                } catch (error) {
+                    console.error("Error during disconnection:", error);
+                }
             }
         });
     </script>