Upload restore.sh with huggingface_hub

restore.sh

@@ -0,0 +1,39 @@
+#!/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).
+
+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
+"