Update app.py

app.py

@@ -22,6 +22,9 @@ SENT_PATH = "sent.csv"
 # Maximum keys per order = quantity * KEYS_PER_QUANTITY
 KEYS_PER_QUANTITY = 10
 
+# Reuse ephemeral keys if created within this time window (in seconds)
+KEY_REUSE_WINDOW_SECONDS = 40 * 60  # 40 minutes
+
 
 api = HfApi(token=HF_TOKEN)
 _lock = threading.Lock()
@@ -69,19 +72,23 @@ def _parse_sent(csv_text: str) -> dict:
         if order_num:
             keys_sent_str = r.get("keys_sent", "").strip()
             quantity_keys_sent_str = r.get("quantity_keys_sent", "").strip()
+            last_key_sent_str = r.get("last_key_sent", "").strip()
+            last_key_created_at_str = r.get("last_key_created_at", "").strip()
 
             sent[order_num] = {
                 "order_number": order_num,
                 "quantity": int(r.get("quantity", "0")),
                 "keys_sent": keys_sent_str if keys_sent_str else "",
                 "quantity_keys_sent": int(quantity_keys_sent_str) if quantity_keys_sent_str else 0,
+                "last_key_sent": last_key_sent_str if last_key_sent_str else "",
+                "last_key_created_at": last_key_created_at_str if last_key_created_at_str else "",
             }
     return sent
 
 def _serialize_sent(sent_dict: dict) -> str:
     """Serialize sent dict back to CSV format"""
     out = io.StringIO()
-    fieldnames = ["order_number", "quantity", "keys_sent", "quantity_keys_sent"]
+    fieldnames = ["order_number", "quantity", "keys_sent", "quantity_keys_sent", "last_key_sent", "last_key_created_at"]
     w = csv.DictWriter(out, fieldnames=fieldnames)
     w.writeheader()
     for order_num in sorted(sent_dict.keys()):
@@ -91,6 +98,8 @@ def _serialize_sent(sent_dict: dict) -> str:
             "quantity": entry["quantity"],
             "keys_sent": entry.get("keys_sent", ""),
             "quantity_keys_sent": entry.get("quantity_keys_sent", 0),
+            "last_key_sent": entry.get("last_key_sent", ""),
+            "last_key_created_at": entry.get("last_key_created_at", ""),
         })
     return out.getvalue()
 
@@ -144,7 +153,25 @@ 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)."
 
-        # Create ephemeral OpenAI key using the Realtime API (expires in 1 hour)
+        # Check if we can reuse a recently created key
+        last_key_sent = sent[order_number].get("last_key_sent", "")
+        last_key_created_at = sent[order_number].get("last_key_created_at", "")
+
+        if last_key_sent and last_key_created_at:
+            try:
+                # Parse the last creation timestamp
+                last_created_time = datetime.fromisoformat(last_key_created_at)
+                now = datetime.now(timezone.utc)
+                time_since_creation = (now - last_created_time).total_seconds()
+
+                # If the last key was created within the reuse window, reuse it
+                if time_since_creation < KEY_REUSE_WINDOW_SECONDS:
+                    return last_key_sent, f"Reused recent key. ({keys_already_sent}/{max_keys} keys sent for this order)"
+            except (ValueError, TypeError):
+                # If there's an error parsing the timestamp, create a new key
+                pass
+
+        # Create a new ephemeral key
         try:
             response = requests.post(
                 "https://api.openai.com/v1/realtime/client_secrets",
@@ -172,13 +199,15 @@ def claim_c_key(
         except Exception as e:
             return "", f"Failed to create ephemeral key: {str(e)}"
 
-        # Update sent.csv - store the ephemeral key
+        # Update sent.csv - store the ephemeral key and timestamp
         existing_keys = sent[order_number]["keys_sent"]
         if existing_keys:
             sent[order_number]["keys_sent"] = existing_keys + "," + ephemeral_key
         else:
             sent[order_number]["keys_sent"] = ephemeral_key
         sent[order_number]["quantity_keys_sent"] += 1
+        sent[order_number]["last_key_sent"] = ephemeral_key
+        sent[order_number]["last_key_created_at"] = _utc_now_iso()
 
         # Save changes
         updated_sent_csv = _serialize_sent(sent)
@@ -187,14 +216,14 @@ def claim_c_key(
             DATASET_REPO_ID,
             SENT_PATH,
             updated_sent_csv,
-            commit_message=f"Updated sent tracking for order {order_number} at {_utc_now_iso()}",
+            commit_message=f"Updated sent tracking at {_utc_now_iso()}",
         )
 
         # refresh cache immediately
         _cache["ts"] = 0.0
 
         keys_sent_count = sent[order_number]["quantity_keys_sent"]
-        return ephemeral_key, f"Ephemeral key sent successfully. ({keys_sent_count}/{max_keys} keys sent for this order)"
+        return ephemeral_key, f"New ephemeral key sent successfully. ({keys_sent_count}/{max_keys} keys sent for this order)"
 
 # -------------------------
 # UI
@@ -216,3 +245,4 @@ with gr.Blocks(title="API") as demo:
 
 demo.queue()
 demo.launch()
+