auto-restart HF Space every hour to keep Zero GPU healthy

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

app.py

@@ -5,6 +5,7 @@ import importlib.util
 import json
 import os
 import sys
+import threading
 import time
 from datetime import datetime
 from typing import Any, Dict, Generator, List, Optional, Tuple
@@ -1319,6 +1320,32 @@ with gr.Blocks(theme="Taithrah/Minimal", css=css, js=custom_js, head=head_html,
     )
 
 
+RESTART_INTERVAL_SECONDS = 3600  # 1 hour
+
+def _auto_restart_space():
+    """Background thread that restarts this HF Space every hour to keep Zero GPU healthy."""
+    space_id = os.getenv("SPACE_ID")
+    if not space_id:
+        return
+    try:
+        from huggingface_hub import HfApi
+        api = HfApi()
+    except Exception:
+        logger.warning("huggingface_hub not available; auto-restart disabled")
+        return
+    logger.info("Auto-restart thread started — will restart %s every %ds", space_id, RESTART_INTERVAL_SECONDS)
+    time.sleep(RESTART_INTERVAL_SECONDS)
+    logger.info("Auto-restart: restarting space %s now", space_id)
+    try:
+        api.restart_space(space_id)
+    except Exception as e:
+        logger.error("Auto-restart failed: %s", e)
+
+if RUNNING_ON_SPACES:
+    _restart_thread = threading.Thread(target=_auto_restart_space, daemon=True)
+    _restart_thread.start()
+
+
 if __name__ == "__main__":
     if pipeline is None:
         logger.critical("Pipeline failed to initialize. Exiting.")