Update stripe_checkout.py

stripe_checkout.py

@@ -3,30 +3,19 @@
 import os
 import stripe
 
-# Load your Stripe secret key
 stripe.api_key = os.getenv("STRIPE_API_KEY")
 
-# Load the redirect URLs from environment (must be set in HF Secrets)
-SUCCESS_URL = os.getenv("SUCCESS_URL")
-CANCEL_URL  = os.getenv("CANCEL_URL")
-
-if not SUCCESS_URL:
-    raise RuntimeError("❌ SUCCESS_URL is not set. Please add it to your Hugging Face Space Secrets.")
-if not CANCEL_URL:
-    raise RuntimeError("❌ CANCEL_URL is not set. Please add it to your Hugging Face Space Secrets.")
+PRICE_ID     = os.getenv("PRICE_ID")
+SUCCESS_URL  = os.getenv("SUCCESS_URL")
+CANCEL_URL   = os.getenv("CANCEL_URL")
+if not all([PRICE_ID, SUCCESS_URL, CANCEL_URL]):
+    raise RuntimeError("❌ PRICE_ID, SUCCESS_URL or CANCEL_URL missing in secrets")
 
 def create_stripe_session():
-    """
-    Creates a Stripe Checkout Session in subscription mode.
-    """
     session = stripe.checkout.Session.create(
         payment_method_types=["card"],
         line_items=[{
-            "price_data": {
-                "currency": "usd",
-                "product_data": {"name": "AutoExec AI Pro Subscription"},
-                "unit_amount": 4900,  # $49.00
-            },
+            "price": PRICE_ID,   # ← Use your recurring Price ID
             "quantity": 1,
         }],
         mode="subscription",