Restart Crowd Counter hourly

AGENTS.md

@@ -7,6 +7,7 @@ Type: code
 - Structure: `app.py` (UI + ZeroGPU handler), `count.py` (model build/inference/tiling/overlay), `PET/` (patched model code), `weights/` (git-lfs).
 - Rules: this Space is Gradio 6 — `theme`/`css`/`js`/`head` go on `demo.launch()`, NOT `gr.Blocks()` (Gradio 6 moved them). UI must mirror the sibling Spaces' design language (interior-ai-designer is the reference): Markdown hero "# 👥 ... by [SkyeBrowse]", sponsor banner, inter-app nav cluster, forceLight JS, link-rewrite JS, JSON-LD + canonical head, SEO footer.
 - `@spaces.GPU(duration=7)` — Bobby raised the cap from 5s (2026-07-03) so the two-grid ensemble completes on large drone stills; ZeroGPU quota charges actual seconds used, not the reservation, so typical 2-3s runs cost the same.
+- The Space self-restarts after one hour of container uptime, matching 3D AI and Anime AI; the external five-minute watchdog still handles `RUNTIME_ERROR` when the container cannot run its own restart thread.
 - Always LIGHT mode on every host (Bobby 2026-07-02): enforced by the `_force_light_theme` FastAPI middleware (307 `/` → `/?__theme=light` for Accept: text/html; relative Location; probes get 200). Gradio 6's CSR frontend does NOT execute `launch(js=...)`/`head=` scripts, so the sibling-style client-side redirect never runs — do not rely on custom_js for anything load-bearing in this repo.
 - Embedding is restricted through the `Content-Security-Policy: frame-ancestors` response header in `_force_light_theme`: `https://skyebrowse.com`, `https://*.skyebrowse.com`, plus `http://localhost:8080` for local webapp development.
 - Successful user uploads save the original and first counted overlay to the configured Google Drive folder through `drive_storage.py`; threshold recounts and bundled examples do not create duplicate Drive files. Drive saving is asynchronous and best-effort so storage failures never break inference.

app.py

@@ -1,5 +1,7 @@
 import json
 import os
+import threading
+import time
 
 os.environ.setdefault("PYTORCH_ENABLE_MPS_FALLBACK", "1")
 
@@ -10,6 +12,7 @@ import gradio as gr
 import gradio.route_utils
 import spaces
 import torch
+from huggingface_hub import HfApi
 from PIL import Image
 from pillow_heif import register_heif_opener
 from starlette.middleware import Middleware
@@ -69,6 +72,25 @@ async def _force_light_theme(request, call_next):
 
 force_light_middleware = Middleware(BaseHTTPMiddleware, dispatch=_force_light_theme)
 
+RESTART_INTERVAL_SECONDS = 3600
+
+
+def _restart_space_after_interval(space_id):
+    time.sleep(RESTART_INTERVAL_SECONDS)
+    try:
+        HfApi().restart_space(space_id)
+    except Exception as exc:
+        print(f"Auto-restart failed: {exc}", flush=True)
+
+
+space_id = os.getenv("SPACE_ID")
+if space_id:
+    threading.Thread(
+        target=_restart_space_after_interval,
+        args=(space_id,),
+        daemon=True,
+    ).start()
+
 TILE = 768
 OVERLAP = 192
 UPSCALE = 1.0