Update routes/subscription.py

routes/subscription.py

@@ -833,10 +833,8 @@ def create_checkout_session(
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
-        # 🔹 1. Validate client token and get user_id
         user_id = verify_token(user_token)
 
-        # 🔹 2. Get stylist data from Supabase
         response_stylist = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.id}",
             headers=SUPABASE_HEADERS
@@ -853,7 +851,6 @@ def create_checkout_session(
         if not stylist_stripe_id or not price_id:
             raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
 
-        # 🔹 3. Get client data from Supabase
         response_user = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
             headers=SUPABASE_HEADERS
@@ -868,12 +865,10 @@ def create_checkout_session(
         if not user_stripe_id:
             raise HTTPException(status_code=400, detail="Client does not have a Stripe Customer ID")
 
-        # 🔹 4. Get price details from Stripe
         price_data = stripe.Price.retrieve(price_id)
-        price_amount = price_data.unit_amount / 100  # Convert cents to dollars
+        price_amount = price_data.unit_amount / 100
         currency = price_data.currency.upper()
 
-        # 🔹 5. Create a subscription
         subscription = stripe.Subscription.create(
             customer=user_stripe_id,
             items=[{"price": price_id}],
@@ -884,11 +879,11 @@ def create_checkout_session(
                 "stylist_stripe_id": stylist_stripe_id,
                 "user_id": user_id,
                 "user_stripe_id": user_stripe_id,
-                "price_id": price_id
+                "price_id": price_id,
+                "consultations": data.consultations
             }
         )
         
-        # Also add metadata to the payment intent
         payment_intent = subscription.latest_invoice.payment_intent
         stripe.PaymentIntent.modify(
             payment_intent.id,
@@ -897,7 +892,8 @@ def create_checkout_session(
                 "stylist_stripe_id": stylist_stripe_id,
                 "user_id": user_id,
                 "user_stripe_id": user_stripe_id,
-                "price_id": price_id
+                "price_id": price_id,
+                "consultations": data.consultations
             }
         )