fix 3

startup.sh

@@ -10,9 +10,9 @@ if [ -z "$HF_TOKEN" ] || [ -z "$HF_DATASET" ]; then
     exit 0
 fi
 
-# Login to Hugging Face
+# Login to Hugging Face using Python
 echo "Logging in to Hugging Face..."
-huggingface-cli login --token "$HF_TOKEN" --add-to-git-credential
+python3 -c "from huggingface_hub import login; login(token='$HF_TOKEN', add_to_git_credential=True)"
 
 # Directories to persist
 CONFIG_DIR="/home/coder/.config/code-server"
@@ -27,18 +27,35 @@ mkdir -p "$BACKUP_DIR"
 download_from_hf() {
     echo "Downloading persisted data from Hugging Face dataset: $HF_DATASET"
     
-    # Try to download the backup archive
-    if huggingface-cli download "$HF_DATASET" backup.tar.gz --repo-type dataset --local-dir "$BACKUP_DIR" 2>/dev/null; then
-        echo "Found existing backup, restoring..."
-        
-        # Extract the backup
-        cd /home/coder
-        tar -xzf "$BACKUP_DIR/backup.tar.gz" 2>/dev/null || echo "No previous backup found or extraction failed"
-        
-        echo "Restoration complete!"
-    else
-        echo "No existing backup found. Starting fresh."
-    fi
+    # Try to download the backup archive using Python
+    python3 << 'PYTHON_SCRIPT'
+import os
+from huggingface_hub import hf_hub_download
+import tarfile
+
+try:
+    backup_dir = "/tmp/hf_backup"
+    os.makedirs(backup_dir, exist_ok=True)
+    
+    # Download backup file
+    file_path = hf_hub_download(
+        repo_id=os.environ["HF_DATASET"],
+        filename="backup.tar.gz",
+        repo_type="dataset",
+        local_dir=backup_dir
+    )
+    
+    print(f"Found existing backup at {file_path}, restoring...")
+    
+    # Extract the backup
+    with tarfile.open(file_path, "r:gz") as tar:
+        tar.extractall("/home/coder")
+    
+    print("Restoration complete!")
+except Exception as e:
+    print(f"No existing backup found or restoration failed: {e}")
+    print("Starting fresh.")
+PYTHON_SCRIPT
 }
 
 # Function to upload to HF dataset
@@ -57,9 +74,35 @@ upload_to_hf() {
         .local/share/code-server \
         workspace 2>/dev/null || true
     
-    # Upload to HF dataset
-    huggingface-cli upload "$HF_DATASET" "$BACKUP_DIR/backup.tar.gz" backup.tar.gz --repo-type dataset --create || \
-        echo "Upload failed, continuing anyway..."
+    # Upload to HF dataset using Python
+    python3 << 'PYTHON_SCRIPT'
+import os
+from huggingface_hub import HfApi
+
+try:
+    api = HfApi()
+    backup_file = "/tmp/hf_backup/backup.tar.gz"
+    
+    # Create repo if it doesn't exist and upload
+    api.create_repo(
+        repo_id=os.environ["HF_DATASET"],
+        repo_type="dataset",
+        private=True,
+        exist_ok=True
+    )
+    
+    api.upload_file(
+        path_or_fileobj=backup_file,
+        path_in_repo="backup.tar.gz",
+        repo_id=os.environ["HF_DATASET"],
+        repo_type="dataset"
+    )
+    
+    print("Upload successful!")
+except Exception as e:
+    print(f"Upload failed: {e}")
+    print("Continuing anyway...")
+PYTHON_SCRIPT
 }
 
 # Setup trap to save on exit