neural-os

Runtime error

App Files Files Community

da03 commited on May 10

Commit

274b099

1 Parent(s): 2b99d5a

.

Browse files

Files changed (2) hide show

main.py +30 -0
static/index.html +18 -0

main.py CHANGED Viewed

@@ -235,6 +235,30 @@ async def websocket_endpoint(websocket: WebSocket):
         input_queue = asyncio.Queue()
         is_processing = False
         async def process_input(data):
             nonlocal previous_frame, hidden_states, keys_down, frame_num, frame_count, is_processing
@@ -377,6 +401,12 @@ async def websocket_endpoint(websocket: WebSocket):
                     await websocket.send_json({"type": "heartbeat_response"})
                     continue
                 # Add the input to our queue
                 await input_queue.put(data)
                 print(f"[{receive_time:.3f}] Received input. Queue size now: {input_queue.qsize()}")

         input_queue = asyncio.Queue()
         is_processing = False
+        # Add a function to reset the simulation
+        async def reset_simulation():
+            nonlocal previous_frame, hidden_states, keys_down, frame_num, is_processing, input_queue
+            # Clear the input queue
+            while not input_queue.empty():
+                try:
+                    input_queue.get_nowait()
+                    input_queue.task_done()
+                except asyncio.QueueEmpty:
+                    break
+            # Reset all state variables
+            previous_frame = padding_image
+            hidden_states = None
+            keys_down = set()
+            frame_num = -1
+            is_processing = False
+            print(f"[{time.perf_counter():.3f}] Simulation reset to initial state")
+            # Send confirmation to client
+            await websocket.send_json({"type": "reset_confirmed"})
         async def process_input(data):
             nonlocal previous_frame, hidden_states, keys_down, frame_num, frame_count, is_processing
                     await websocket.send_json({"type": "heartbeat_response"})
                     continue
+                # Handle reset command
+                if data.get("type") == "reset":
+                    print(f"[{receive_time:.3f}] Received reset command")
+                    await reset_simulation()
+                    continue
                 # Add the input to our queue
                 await input_queue.put(data)
                 print(f"[{receive_time:.3f}] Received input. Queue size now: {input_queue.qsize()}")

static/index.html CHANGED Viewed

@@ -14,6 +14,9 @@
 <body>
     <canvas id="displayCanvas" width="512" height="384"></canvas>
     <script>
         const canvas = document.getElementById('displayCanvas');
         const ctx = canvas.getContext('2d');
@@ -61,6 +64,9 @@
                         //isProcessing = false; // Reset the processing flag when we get a response
                     };
                     img.src = 'data:image/png;base64,' + data.image;
                 }
             };
         }
@@ -212,6 +218,18 @@
                 }
             }
         });
     </script>
 </body>
 </html>

 <body>
     <canvas id="displayCanvas" width="512" height="384"></canvas>
+    <!-- Add this button somewhere in your UI, perhaps near other controls -->
+    <button id="resetButton" class="control-button">Reset Simulation</button>
     <script>
         const canvas = document.getElementById('displayCanvas');
         const ctx = canvas.getContext('2d');
                         //isProcessing = false; // Reset the processing flag when we get a response
                     };
                     img.src = 'data:image/png;base64,' + data.image;
+                } else if (data.type === "reset_confirmed") {
+                    console.log("Simulation reset confirmed by server");
+                    // Optionally update UI to reflect reset state
                 }
             };
         }
                 }
             }
         });
+        // Add event listener for the reset button
+        document.getElementById('resetButton').addEventListener('click', function() {
+            if (socket && socket.readyState === WebSocket.OPEN) {
+                console.log("Sending reset command to server");
+                socket.send(JSON.stringify({
+                    type: "reset"
+                }));
+            } else {
+                console.error("WebSocket not connected, cannot reset");
+            }
+        });
     </script>
 </body>
 </html>