Update app.py

app.py

@@ -16,7 +16,7 @@ DATASET_REPO_ID = os.environ.get("DATASET_REPO_ID") # Space secret
 HF_TOKEN = os.environ["HF_TOKEN"]          # Space secret
 
 KEYS_PATH = "keys.csv"
-OPENAI_KEYS_PATH = "openai_keys.csv"
+B_KEYS_PATH = "b_keys.csv"
 
 
 api = HfApi(token=HF_TOKEN)
@@ -26,7 +26,7 @@ _lock = threading.Lock()
 _cache = {
     "ts": 0.0,
     "keys": None,  # type: ignore
-    "openai_keys": None,  # type: ignore
+    "b_keys": None,  # type: ignore
 }
 CACHE_TTL_SEC = 10.0
 
@@ -68,7 +68,7 @@ def _parse_keys(csv_text: str) -> list:
             })
     return keys
 
-def _parse_openai_keys(csv_text: str) -> list:
+def _parse_b_keys(csv_text: str) -> list:
     keys = []
     reader = csv.DictReader(io.StringIO(csv_text))
     for r in reader:
@@ -106,18 +106,18 @@ def _serialize_b_keys(keys: list) -> str:
 def load_state(force: bool = False):
     now = time.time()
     if (not force) and _cache["keys"] is not None and (now - _cache["ts"] < CACHE_TTL_SEC):
-        return _cache["keys"], _cache["openai_keys"]
+        return _cache["keys"], _cache["b_keys"]
 
     keys_csv = _download_csv(DATASET_REPO_ID, KEYS_PATH)
-    openai_keys_csv = _download_csv(DATASET_REPO_ID, OPENAI_KEYS_PATH)
+    b_keys_csv = _download_csv(DATASET_REPO_ID, B_KEYS_PATH)
 
     keys = _parse_keys(keys_csv)
-    openai_keys = _parse_openai_keys(openai_keys_csv)
+    b_keys = _parse_b_keys(b_keys_csv)
 
     _cache["ts"] = now
     _cache["keys"] = keys
-    _cache["openai_keys"] = openai_keys
-    return keys, openai_keys
+    _cache["b_keys"] = b_keys
+    return keys, b_keys
 
 # -------------------------
 # Core API
@@ -132,7 +132,7 @@ def claim_key(
     _ = request
 
     with _lock:
-        keys, openai_keys = load_state(force=True)
+        keys, _ = load_state(force=True)
 
         # Find available keys
         available_keys = [k for k in keys if not k["sent"]]
@@ -165,8 +165,8 @@ def claim_b_key(
     request: Optional[gr.Request] = None,
 ) -> Tuple[str, str]:
     """
-    Returns the OpenAI API key and increments the usage count.
-    Returns (openai_key, status_message)
+    Returns the  key and increments the usage count.
+    Returns (b_key, status_message)
     """
     _ = request
 
@@ -187,7 +187,7 @@ def claim_b_key(
 
         _upload_csv(
             DATASET_REPO_ID,
-            OPENAI_KEYS_PATH,
+            B_KEYS_PATH,
             updated_b_keys_csv,
             commit_message=f"Key requested at {_utc_now_iso()}",
         )