deploy app, storage, readme

app.py

@@ -13,7 +13,8 @@ from storage import VoteStorage
 
 LOCAL_DATA_DIR = 'data'
 DEBUG_MODE = os.getenv("DEBUG", "0").lower() in ("1", "true", "yes", "on")
-VOTE_STORAGE = VoteStorage(mode="local" if DEBUG_MODE else "hf", local_dir=LOCAL_DATA_DIR)
+RATINGS_APP_TOKEN = os.getenv("RATINGS_APP_TOKEN")
+VOTE_STORAGE = VoteStorage(mode="local" if DEBUG_MODE else "hf", local_dir=LOCAL_DATA_DIR, token=RATINGS_APP_TOKEN)
 
 # -- Pool dataset -----------------------------------------------------------
 if DEBUG_MODE:
@@ -24,6 +25,7 @@ else:
         repo_id="taigasan/e6-visual-ratings",
         filename="pool.parquet",
         repo_type="dataset",
+        token=RATINGS_APP_TOKEN,
     )
 _pool_df = pd.read_parquet(_pool_path)
 _pool_group_dfs = {g: gdf for g, gdf in _pool_df.groupby("group")}

storage.py

@@ -28,10 +28,11 @@ VOTE_COLUMNS = [
 
 
 class VoteStorage:
-    def __init__(self, mode: str, local_dir: str = "ratings_data"):
+    def __init__(self, mode: str, local_dir: str = "ratings_data", token: str | None = None):
         assert mode in ("hf", "local"), f"Unsupported storage mode: {mode}"
         self.mode = mode
         self.local_dir = local_dir
+        self._token = token
         is_debug_mode = self.mode == "local"
         self._flush_every = 3 if is_debug_mode else 50
         self._flush_interval_sec = 15.0 if is_debug_mode else 300.0
@@ -50,7 +51,7 @@ class VoteStorage:
         return log_dir
 
     def _hf_token(self) -> str | None:
-        return os.getenv("RATINGS_APP_TOKEN")
+        return self._token
 
     def _empty_votes_df(self) -> pd.DataFrame:
         return pd.DataFrame(columns=VOTE_COLUMNS)