Update app.py

app.py

@@ -214,6 +214,21 @@ def get_task_status(api_base_url: str, api_key: str, task_id: str) -> Dict[str,
     return response.json()
 
 
+def send_heartbeat(api_base_url: str, api_key: str, task_id: str) -> None:
+    """Notify the ECS server that the frontend is still alive.
+
+    Swallows all errors so a transient network blip never kills the UI loop.
+    """
+    try:
+        get_http_session().post(
+            f"{normalize_base_url(api_base_url)}/tasks/{task_id}/heartbeat",
+            headers=build_headers(api_key),
+            timeout=5,
+        )
+    except Exception:
+        pass
+
+
 def cancel_remote_task(api_base_url: str, api_key: str, task_state: Dict[str, Any]) -> Tuple[str, Dict[str, Any]]:
     task_id = (task_state or {}).get("task_id")
     if not task_id:
@@ -266,6 +281,10 @@ def run_remote_task(
         except Exception as exc:
             raise gr.Error(f"Failed to query task status: {exc}") from exc
 
+        # Keep the ECS heartbeat watchdog satisfied so it doesn't auto-cancel
+        # this task if it thinks the frontend has disconnected.
+        send_heartbeat(api_base_url, api_key, task_id)
+
         image_url = task.get("display_image_url") or task.get("latest_annotated_screenshot_url") or task.get("latest_screenshot_url")
         if image_url and image_url != last_image_url:
             try: