Update app.py

app.py

@@ -17,38 +17,48 @@ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnl
 treasury_usdc = 0.0
 treasury_btc = 0.0
 
-def check_bill(url, premium=False, api_key=None):
+def check_bill(url, premium=False):
     global treasury_usdc, treasury_btc
 
     if premium:
-        if not api_key:
-            main_out = "Enter API key for premium access."
+        if not stripe.api_key:
+            main_out = "**Error: Stripe key missing.** Add `STRIPE_SECRET_KEY` in HF Secrets."
             treasury_out = f"**Treasury:** ${treasury_usdc:,.0f} USDC | {treasury_btc:.4f} BTC"
             return main_out, treasury_out
+
         try:
-            session = stripe.checkout.sessions.create(
+            session = stripe.checkout.Session.create(  # Note: 'create' not 'checkout.sessions.create'
                 payment_method_types=['card'],
                 line_items=[{
                     'price_data': {
                         'currency': 'usd',
                         'product_data': {'name': 'BillionDollarOracle Premium'},
-                        'unit_amount': 100,
+                        'unit_amount': 100,  # $1.00
                         'recurring': {'interval': 'month'},
                     },
                     'quantity': 1,
                 }],
                 mode='subscription',
-                success_url=f"{SITE_URL}/success",
+                success_url=f"{SITE_URL}/success?session_id={{CHECKOUT_SESSION_ID}}",
                 cancel_url=f"{SITE_URL}/cancel",
                 metadata={'dao_vote': 'true'}
             )
-            main_out = f"[Pay $1/month → Unlimited + Vote]({session.url})\n\n90% → BTC treasury | 10% → `{CREATOR_WALLET}`"
-            treasury_out = f"**Treasury:** ${treasury_usdc:,.0f} USDC | {treasury_btc:.4f} BTC"
-            return main_out, treasury_out
+            # Force redirect in Gradio (popup/redirect)
+            main_out = f"""
+            **Premium Unlock:**  
+            Redirecting to Stripe... (if no popup, click below)  
+            [Pay $1/month → Unlimited + Vote]({session.url})  
+            90% → BTC treasury | 10% → `{CREATOR_WALLET}`
+            """
+        except stripe.error.InvalidRequestError as e:
+            main_out = f"**Stripe Error:** {str(e)} (Check key/mode in dashboard)."
         except Exception as e:
-            main_out = f"Payment failed: {e}"
-            treasury_out = f"**Treasury:** ${treasury_usdc:,.0f} USDC | {treasury_btc:.4f} BTC"
-            return main_out, treasury_out
+            main_out = f"**Unexpected Error:** {str(e)}. Use free mode."
+        else:
+            main_out = main_out  # From try block
+
+        treasury_out = f"**Treasury:** ${treasury_usdc:,.0f} USDC | {treasury_btc:.4f} BTC"
+        return main_out, treasury_out                
 
     try:
         text = requests.get(url, timeout=10).text.lower()