Update app.py

app.py

@@ -19,12 +19,6 @@ ORDERS_PATH = "orders.csv"
 SENT_PATH = "sent.csv"
 KEYS_PATH = "keys.csv"
 
-# # Optional SMTP config (Space secrets)
-# SMTP_HOST = os.getenv("SMTP_HOST")
-# SMTP_PORT = int(os.getenv("SMTP_PORT", "587"))
-# SMTP_USER = os.getenv("SMTP_USER")
-# SMTP_PASS = os.getenv("SMTP_PASS")
-# FROM_EMAIL = os.getenv("FROM_EMAIL", SMTP_USER or "[email protected]")
 
 api = HfApi(token=HF_TOKEN)
 _lock = threading.Lock()
@@ -149,39 +143,10 @@ def load_state(force: bool = False):
     _cache["keys"] = keys
     return orders, sent, keys
 
-def send_key_email(to_email: str, keys: list, contact_person: str) -> None:
-    """
-    Send keys to the customer via email.
-    """
-    # if not (SMTP_HOST and SMTP_USER and SMTP_PASS):
-    #     return
-
-    # import smtplib
-    # from email.message import EmailMessage
-
-    # msg = EmailMessage()
-    # msg["Subject"] = "Your Gradium Keys"
-    # msg["From"] = FROM_EMAIL
-    # msg["To"] = to_email
-    # keys_text = "\n".join([f"- {k}" for k in keys])
-    # msg.set_content(
-    #     f"Hi {contact_person},\n\n"
-    #     f"Here are your Gradium keys:\n\n"
-    #     f"{keys_text}\n\n"
-    #     f"Best regards,\nGradium Team\n"
-    # )
-
-    # with smtplib.SMTP(SMTP_HOST, SMTP_PORT) as s:
-    #     s.starttls()
-    #     s.login(SMTP_USER, SMTP_PASS)
-    #     s.send_message(msg)
-
-    pass
-
 # -------------------------
 # Core API
 # -------------------------
-def send_keys_for_order(
+def claim_key(
     order_number: str,
     request: Optional[gr.Request] = None,
 ) -> Tuple[str, str]:
@@ -238,24 +203,19 @@ def send_keys_for_order(
             DATASET_REPO_ID,
             SENT_PATH,
             updated_sent_csv,
-            commit_message=f"Send keys for order {order_number} at {_utc_now_iso()}",
+            commit_message=f"Send keys for new order at {_utc_now_iso()}",
         )
         _upload_csv(
             DATASET_REPO_ID,
             KEYS_PATH,
             updated_keys_csv,
-            commit_message=f"Mark keys as sent for order {order_number} at {_utc_now_iso()}",
+            commit_message=f"Mark keys as sent for new order at {_utc_now_iso()}",
         )
 
         # refresh cache immediately
         _cache["ts"] = 0.0
 
-        # Send email
-        try:
-            send_key_email(order["email_address"], keys_str.split(","), order["contact_person"])
-            return keys_str, f"Keys sent successfully to {order['email_address']}."
-        except Exception as e:
-            return keys_str, f"Keys allocated: {keys_str}. Email failed: {str(e)}"
+        return keys_str, f"Keys sent successfully."
 
 # -------------------------
 # UI
@@ -263,20 +223,21 @@ def send_keys_for_order(
 with gr.Blocks() as demo:
     gr.Markdown(
         "## Gradium Key Distribution System\n\n"
-        "Enter an **order number** to send keys to the customer.\n\n"
+        "Enter an **order number** to claim your key.\n\n"
     )
 
     order_in = gr.Textbox(label="Order number", placeholder="e.g. 0000-1111")
-    send_btn = gr.Button("Send Keys", variant="primary")
-    keys_out = gr.Textbox(label="Keys sent", interactive=False)
+    claim_btn = gr.Button("Claim Key", variant="primary")
+    keys_out = gr.Textbox(label="Your Key(s)", interactive=False)
     status_out = gr.Textbox(label="Status", interactive=False)
 
-    send_btn.click(
-        fn=send_keys_for_order,
+    claim_btn.click(
+        fn=claim_key,
         inputs=[order_in],
         outputs=[keys_out, status_out],
-        api_name="send_keys",
+        api_name="claim",
     )
 
 demo.queue()
 demo.launch()
+