use hf api

src/database.py

@@ -9,6 +9,8 @@ from contextlib import contextmanager
 from datetime import datetime, timezone
 from typing import Optional
 
+from huggingface_hub import CommitOperationAdd, HfApi
+
 REQUIRED_TABLES = ('tasks', 'assignments', 'task_config', 'answers', 'allocation_history')
 
 
@@ -134,7 +136,8 @@ class DB:
     # ========== Auto Commit (For HF Spaces) ==========
     def commit_and_push_db(self) -> None:
         """Commit and push the database to the repository."""
-        if not os.getenv('SPACE_ID'):
+        space_id = os.getenv('SPACE_ID', None)
+        if not space_id:
             return
 
         try:
@@ -148,9 +151,12 @@ class DB:
             ts = _get_time_now()
             db_rel = os.path.relpath(self.db_path, repo_root)
             subprocess.run(['git', 'add', db_rel], check=True, cwd=repo_root)
-            diff = subprocess.run(['git', 'diff', '--cached', '--quiet'], check=False, cwd=repo_root)
-            if diff.returncode != 0:
-                subprocess.run(['git', 'commit', '-m', f'update db {ts}'], check=True, cwd=repo_root)
-                subprocess.run(['git', 'push', 'origin', 'main'], check=True, cwd=repo_root)
+            api = HfApi(token=os.getenv('HF_TOKEN', None))
+            api.create_commit(
+                repo_id=space_id,
+                repo_type='space',
+                operations=[CommitOperationAdd(path_in_repo=db_rel, path_or_fileobj=self.db_path)],
+                commit_message=f'update db {ts}',
+            )
         except Exception as e:
             raise RuntimeError(f'Failed to commit and push the database: {e}')