Upload restore.sh with huggingface_hub

restore.sh

@@ -1,39 +1,6 @@
 #!/bin/sh
-# Restore /data/webui.db from the HF backups dataset repo if it doesn't exist.
-# Requires HF_TOKEN env var to be set (Space secret).
-
+# Restore Open WebUI user data from the HF backups dataset when the DB is
+# missing (fresh volume / factory reset). No-op when the DB is already present.
 set -eu
 
-DB_PATH="/data/webui.db"
-REPO_ID="tkgray/open-webui-backups"
-FILE_IN_REPO="backups/webui.db"
-
-if [ -f "$DB_PATH" ]; then
-    echo "[restore] $DB_PATH already exists — skipping restore."
-    exit 0
-fi
-
-if [ -z "${HF_TOKEN:-}" ]; then
-    echo "[restore] HF_TOKEN not set — skipping restore."
-    exit 0
-fi
-
-echo "[restore] $DB_PATH not found — attempting restore from $REPO_ID:$FILE_IN_REPO ..."
-python3 -c "
-from huggingface_hub import hf_hub_download
-import os, sys
-try:
-    path = hf_hub_download(
-        repo_id='$REPO_ID',
-        filename='$FILE_IN_REPO',
-        repo_type='dataset',
-        token=os.environ['HF_TOKEN']
-    )
-    import shutil
-    os.makedirs('/data', exist_ok=True)
-    shutil.copy2(path, '$DB_PATH')
-    print('[restore] Restore OK')
-except Exception as e:
-    print(f'[restore] Restore FAILED (first boot or no backup yet?): {e}', file=sys.stderr)
-    sys.exit(0)  # don't fail boot if no backup exists yet
-"
+exec python3 /restore.py