Create restore.py

restore.py

@@ -0,0 +1,46 @@
+import os, subprocess
+
+def restore_workspace():
+    repo_id = os.environ.get("BACKUP_REPO")
+    hf_token = os.environ.get("HF_TOKEN")
+    repo_type = "space" if "/" in repo_id else "dataset"  # auto-detect
+
+    if not repo_id or not hf_token:
+        print("❌ BACKUP_REPO or HF_TOKEN missing, skipping restore")
+        return
+
+    env = os.environ.copy()
+    env["HF_HOME"] = "/tmp/hf_cache"
+    env["XDG_CACHE_HOME"] = "/tmp/xdg_cache"
+    env["TMPDIR"] = "/tmp"
+
+    os.makedirs(env["HF_HOME"], exist_ok=True)
+    os.makedirs(env["XDG_CACHE_HOME"], exist_ok=True)
+    os.makedirs(env["TMPDIR"], exist_ok=True)
+
+    local_path = "/workspace"
+    os.makedirs(local_path, exist_ok=True)
+
+    cmd = [
+        "hf", "download",
+        repo_id,
+        "--repo-type", repo_type,
+        "--token", hf_token,
+        "--local-dir", local_path,
+        "--exclude", ".git",
+    ]
+
+    print("🔄 Restoring workspace from backup...")
+    process = subprocess.Popen(
+        cmd,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT,
+        env=env,
+        text=True,
+        cwd="/tmp",
+    )
+
+    for line in iter(process.stdout.readline, ''):
+        print("📥", line.strip())
+
+    print("✅ Restore complete.")