Update app.py

app.py

@@ -15,8 +15,8 @@ from huggingface_hub import HfApi
 # -------------------------
 DATASET_REPO_ID = os.environ.get("DATASET_REPO_ID") # Space secret
 HF_TOKEN = os.environ["HF_TOKEN"]          # Space secret
+OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]  # Space secret
 
-KEYS_PATH = "keys.csv"
 SENT_PATH = "sent.csv"
 
 # Maximum keys per order = quantity * KEYS_PER_QUANTITY
@@ -29,7 +29,6 @@ _lock = threading.Lock()
 # Simple cache to reduce Hub reads
 _cache = {
     "ts": 0.0,
-    "keys": None,  # type: ignore
     "sent": None,  # type: ignore
 }
 CACHE_TTL_SEC = 10.0
@@ -61,29 +60,6 @@ def _upload_csv(repo_id: str, path_in_repo: str, content: str, commit_message: s
         token=HF_TOKEN,
     )
 
-def _parse_keys(csv_text: str) -> list:
-    keys = []
-    reader = csv.DictReader(io.StringIO(csv_text))
-    for r in reader:
-        if r["key"].strip():
-            keys.append({
-                "key": r["key"].strip(),
-                "sent": (r.get("sent") or "").strip(),
-            })
-    return keys
-
-def _serialize_keys(keys: list) -> str:
-    out = io.StringIO()
-    fieldnames = ["key", "sent"]
-    w = csv.DictWriter(out, fieldnames=fieldnames)
-    w.writeheader()
-    for k in keys:
-        w.writerow({
-            "key": k["key"],
-            "sent": k.get("sent", ""),
-        })
-    return out.getvalue()
-
 def _parse_sent(csv_text: str) -> dict:
     """Parse sent.csv and return a dict keyed by order_number"""
     sent = {}
@@ -120,19 +96,15 @@ def _serialize_sent(sent_dict: dict) -> 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["sent"]
+    if (not force) and _cache["sent"] is not None and (now - _cache["ts"] < CACHE_TTL_SEC):
+        return _cache["sent"]
 
-    keys_csv = _download_csv(DATASET_REPO_ID, KEYS_PATH)
     sent_csv = _download_csv(DATASET_REPO_ID, SENT_PATH)
-
-    keys = _parse_keys(keys_csv)
     sent = _parse_sent(sent_csv)
 
     _cache["ts"] = now
-    _cache["keys"] = keys
     _cache["sent"] = sent
-    return keys, sent
+    return sent
 
 # -------------------------
 # Core API
@@ -154,7 +126,7 @@ def claim_c_key(
         return "", "Please provide an order number."
 
     with _lock:
-        keys, sent = load_state(force=True)
+        sent = load_state(force=True)
 
         # Check if order exists in sent.csv
         if order_number not in sent:
@@ -172,21 +144,12 @@ def claim_c_key(
         if keys_already_sent >= max_keys:
             return "", f"Key limit reached for order {order_number} ({keys_already_sent}/{max_keys} keys sent)."
 
-        # Find available keys
-        available_keys = [k for k in keys if not k["sent"]]
-
-        if len(available_keys) < 1:
-            return "", "No available keys remaining."
-
-        # Send the next available key
-        key_to_send = available_keys[0]
-
         # Create ephemeral OpenAI key using the Realtime API (expires in 1 hour)
         try:
             response = requests.post(
                 "https://api.openai.com/v1/realtime/client_secrets",
                 headers={
-                    "Authorization": f"Bearer {key_to_send['key']}",
+                    "Authorization": f"Bearer {OPENAI_API_KEY}",
                     "Content-Type": "application/json"
                 },
                 json={
@@ -209,9 +172,6 @@ def claim_c_key(
         except Exception as e:
             return "", f"Failed to create ephemeral key: {str(e)}"
 
-        # Mark key as sent
-        key_to_send["sent"] = _utc_now_iso()
-
         # Update sent.csv - store the ephemeral key
         existing_keys = sent[order_number]["keys_sent"]
         if existing_keys:
@@ -220,17 +180,9 @@ def claim_c_key(
             sent[order_number]["keys_sent"] = ephemeral_key
         sent[order_number]["quantity_keys_sent"] += 1
 
-        # Save changes to both files
-        updated_keys_csv = _serialize_keys(keys)
+        # Save changes
         updated_sent_csv = _serialize_sent(sent)
 
-        _upload_csv(
-            DATASET_REPO_ID,
-            KEYS_PATH,
-            updated_keys_csv,
-            commit_message=f"Key sent for order {order_number} at {_utc_now_iso()}",
-        )
-
         _upload_csv(
             DATASET_REPO_ID,
             SENT_PATH,
@@ -264,4 +216,3 @@ with gr.Blocks(title="API") as demo:
 
 demo.queue()
 demo.launch()
-