# stripe_checkout.py

import os
import stripe

stripe.api_key = os.getenv("STRIPE_API_KEY")

def create_stripe_session():
    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,
            },
            "quantity": 1,
        }],
        mode="subscription",
        success_url=os.getenv("SUCCESS_URL", "https://example.com/success"),
        cancel_url=os.getenv("CANCEL_URL", "https://example.com/cancel"),
    )
    return session.url