Update app.py

app.py

@@ -403,6 +403,16 @@ CONTROL_INPUT_JS = """
             triggerUpdate();
         }
     }
+    
+    // Expose setCapturing globally so we can trigger it from Start Game button
+    window.worldEngineSetCapturing = setCapturing;
+    window.worldEngineRequestPointerLock = () => {
+        if (!isMobile) {
+            document.body.requestPointerLock();
+        } else {
+            setCapturing(true);
+        }
+    };
 
     // Mobile: show toggle button and enable controls on tap
     if (isMobile) {
@@ -647,7 +657,7 @@ def create_app():
 
         Interactive frame-by-frame world generation on ZeroGPU.
 
-        **Controls:** Click "Start Game" → Click game window to capture inputs → WASD to move • Mouse to look • ESC to release
+        **Controls:** Click "Start Game" → WASD to move • Mouse to look • Press ESC to release controls
         """)
 
         with gr.Row(elem_classes=["main-row"]):
@@ -716,6 +726,9 @@ def create_app():
 
         def on_start(selected_url, uploaded_image, prompt):
             """Start GPU session - generator that shows loading then first frame."""
+            # Show info about controls
+            gr.Info("Controls locked! Press ESC to release mouse/keyboard capture.", duration=5)
+            
             # Show loading state immediately
             loading_img = create_loading_image(text="Generating World ...")
             yield (
@@ -849,6 +862,7 @@ def create_app():
             inputs=[selected_seed_url, seed_image_upload, prompt_input],
             outputs=[session_state, latest_frame, latest_frame_count,
                     video_output, frame_display, start_btn, stop_btn],
+            js="() => { setTimeout(() => { if (window.worldEngineRequestPointerLock) window.worldEngineRequestPointerLock(); }, 500); }",
         )
 
         stop_btn.click(
@@ -875,14 +889,20 @@ def create_app():
             outputs=[latest_frame, latest_frame_count, video_output, frame_display],
         )
 
-        # Pointer lock JS
+        # Pointer lock JS - also allows clicking the game window
         demo.load(fn=None, js="""
         () => {
             const insertButton = () => {
                 const output = document.querySelector('#video-output');
                 if (!output) { setTimeout(insertButton, 100); return; }
                 output.style.cursor = 'pointer';
-                output.onclick = () => document.body.requestPointerLock();
+                output.onclick = () => {
+                    if (window.worldEngineRequestPointerLock) {
+                        window.worldEngineRequestPointerLock();
+                    } else {
+                        document.body.requestPointerLock();
+                    }
+                };
             };
             insertButton();
         }