Fix server startup and backup script

backup_to_hub.py

@@ -3,24 +3,30 @@ Automatic backup of Argilla annotations to HF Dataset.
 Runs as background process, syncs every 10 minutes.
 """
 import os
+import sys
 import json
 import time
 from pathlib import Path
 from datetime import datetime
-from huggingface_hub import CommitScheduler, login
-import argilla as rg
 
 # Configuration
 EXPORT_DIR = Path("/exports")
-EXPORT_DIR.mkdir(exist_ok=True)
 BACKUP_REPO = os.environ.get("BACKUP_DATASET_REPO", "argus-systems/vuln-preferences-data")
 SYNC_INTERVAL = int(os.environ.get("SYNC_INTERVAL_MINUTES", "10"))
 ARGILLA_API_URL = "http://localhost:6900"
 
-# Login to HF (uses HF_TOKEN secret)
+# Check for HF_TOKEN before importing heavy dependencies
 hf_token = os.environ.get("HF_TOKEN")
-if hf_token:
-    login(token=hf_token)
+if not hf_token:
+    print("ERROR: HF_TOKEN not set, backup script exiting")
+    sys.exit(1)
+
+from huggingface_hub import CommitScheduler, login
+import argilla as rg
+
+# Setup
+EXPORT_DIR.mkdir(exist_ok=True)
+login(token=hf_token)
 
 # Setup CommitScheduler
 scheduler = CommitScheduler(

start.sh

@@ -1,7 +1,11 @@
 #!/bin/bash
 
-# Start backup script in background
-python /app/backup_to_hub.py &
+# Start backup script in background (only if HF_TOKEN is set)
+if [ -n "$HF_TOKEN" ]; then
+    python /app/backup_to_hub.py &
+else
+    echo "WARNING: HF_TOKEN not set, backup disabled"
+fi
 
-# Start Argilla server (original command)
-python -m argilla server start --host 0.0.0.0 --port 6900
+# Start Argilla server using the CLI from argilla-server package
+exec argilla server start --host 0.0.0.0 --port 6900