Upload 20 files

requirements.txt

@@ -4,4 +4,5 @@ websockets==14.1
 python-multipart==0.0.18
 aiofiles==24.1.0
 httpx==0.28.1
+huggingface_hub>=0.25.0
 

routers/backup.py

@@ -122,20 +122,18 @@ async def list_backups(request: Request):
 
 
 def _upload_to_hf(creds: dict, zone_name: str, archive_path: Path):
-    """Upload archive directly to HuggingFace Dataset API."""
+    """Upload archive directly to HuggingFace Dataset via huggingface_hub."""
+    from huggingface_hub import HfApi
+
     file_path = f"{creds['path_prefix']}/{zone_name}.tar.gz"
-    hf_token = creds["hf_token"]
-    repo = creds["repo"]
-
-    with httpx.Client(timeout=300) as client:
-        with open(archive_path, "rb") as f:
-            resp = client.post(
-                f"{HF_API}/datasets/{repo}/upload/main/{file_path}",
-                headers={"Authorization": f"Bearer {hf_token}"},
-                content=f.read(),
-            )
-        if resp.status_code not in (200, 201):
-            raise ValueError(f"HF upload error: {resp.status_code} {resp.text}")
+    api = HfApi(token=creds["hf_token"])
+    api.upload_file(
+        path_or_fileobj=str(archive_path),
+        path_in_repo=file_path,
+        repo_id=creds["repo"],
+        repo_type="dataset",
+        commit_message=f"Backup zone: {zone_name}",
+    )
 
 
 @router.post("/zone/{zone_name}")
@@ -236,11 +234,10 @@ async def backup_all(request: Request, background_tasks: BackgroundTasks):
 def _download_from_hf(creds: dict, zone_name: str) -> bytes:
     """Download archive directly from HuggingFace Dataset."""
     file_path = f"{creds['path_prefix']}/{zone_name}.tar.gz"
-    with httpx.Client(timeout=300) as client:
+    with httpx.Client(timeout=300, follow_redirects=True) as client:
         resp = client.get(
             f"https://huggingface.co/datasets/{creds['repo']}/resolve/main/{file_path}",
             headers={"Authorization": f"Bearer {creds['hf_token']}"},
-            follow_redirects=True,
         )
         if resp.status_code == 404:
             raise FileNotFoundError(f"Backup zone '{zone_name}' khong ton tai")