Update app.py

app.py

@@ -144,6 +144,16 @@ async def handle_action(ws: WebSocket, msg: dict):
                 "stderr": result["stderr"],
                 "returncode": result["returncode"],
             }))
+            # Auto-screenshot after terminal command
+            await asyncio.sleep(0.5)
+            frame = capture_screen(scale=0.7, quality=70)
+            if frame:
+                await ws.send_text(json.dumps({
+                    "type": "screenshot",
+                    "data": frame,
+                    "ts": int(time.time() * 1000),
+                    "auto": True
+                }))
 
     # ── MOUSE MOVE ───────────────────────────────────
     elif action == "mouse_move":
@@ -221,6 +231,16 @@ async def handle_action(ws: WebSocket, msg: dict):
                 stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
             )
         await ws.send_text(json.dumps({"type": "ack", "action": "open_app"}))
+        # Auto-screenshot after app opens (wait for it to load)
+        await asyncio.sleep(3)
+        frame = capture_screen(scale=0.7, quality=70)
+        if frame:
+            await ws.send_text(json.dumps({
+                "type": "screenshot",
+                "data": frame,
+                "ts": int(time.time() * 1000),
+                "auto": True
+            }))
 
     # ── START STREAM ─────────────────────────────────
     elif action == "start_stream":