Upload restore_runner.py

restore_runner.py

@@ -1,31 +1,19 @@
-import os
-import sys
-import subprocess
-from huggingface_hub import snapshot_download
+"""Emergency restore runner - reverts main.py to last known good state."""
+import os, sys
 
-REVISION = os.environ.get("VNEWS_RESTORE_REVISION", "bcaa2dc")
-REPO_ID = os.environ.get("VNEWS_REPO_ID", "bep40/vnews")
+def restore():
+    """Copy backup files back to main."""
+    backup_dir = os.path.join(os.path.dirname(__file__), 'backup')
+    if not os.path.isdir(backup_dir):
+        print("[restore] No backup dir found")
+        return
+    import shutil
+    for f in os.listdir(backup_dir):
+        src = os.path.join(backup_dir, f)
+        dst = os.path.join(os.path.dirname(__file__), f)
+        if os.path.isfile(src):
+            shutil.copy2(src, dst)
+            print(f"[restore] Restored {f}")
 
-# Download exact Space snapshot from Hugging Face Hub.
-# This avoids manually copying huge files from an old commit.
-snapshot_dir = snapshot_download(
-    repo_id=REPO_ID,
-    repo_type="space",
-    revision=REVISION,
-    local_dir="/tmp/vnews_restore",
-    local_dir_use_symlinks=False,
-)
-
-os.chdir(snapshot_dir)
-sys.path.insert(0, snapshot_dir)
-
-# Commit bcaa2dc Dockerfile ran ai_patch:app.
-cmd = [
-    "uvicorn",
-    "ai_patch:app",
-    "--host",
-    "0.0.0.0",
-    "--port",
-    "7860",
-]
-os.execvp(cmd[0], cmd)
+if __name__ == '__main__':
+    restore()