Update routes/subscription.py

routes/subscription.py

@@ -15,9 +15,11 @@ stripe.api_version = "2023-10-16"
 ### **DATA MODELS** ###
 class SubscriptionRequest(BaseModel):
     user_id: str
-    stylist_id: str  # This must be the Stripe connected account ID (acct_xxxxx)
+    stylist_id: str  # Stripe Connected Account ID (acct_xxxxx)
     amount: int  # Value in cents (e.g., R$25.00 -> 2500)
     stylist_name: str  # Name of the stylist
+    stylist_avatar: str  # URL da foto do estilista
+    consultations_per_month: int  # Número de consultas inclusas no plano
 
 class CancelSubscriptionRequest(BaseModel):
     subscription_id: str
@@ -27,7 +29,12 @@ class CancelSubscriptionRequest(BaseModel):
 def create_checkout_session(data: SubscriptionRequest):
     try:
         product_name = f"{data.stylist_name}'s Subscription"
-        product_description = f"Monthly plan to access exclusive services from {data.stylist_name}."
+        product_description = (
+            f"💼 Exclusive styling plan by {data.stylist_name}.\n\n"
+            f"👕 {data.consultations_per_month} styling consultations per month.\n"
+            f"📞 Personalized fashion advice and outfit recommendations.\n"
+            f"💳 Secure payments powered by Stripe."
+        )
 
         session = stripe.checkout.Session.create(
             success_url="https://yourdomain.com/success",
@@ -40,7 +47,8 @@ def create_checkout_session(data: SubscriptionRequest):
                         "currency": "brl",
                         "product_data": {
                             "name": product_name,
-                            "description": product_description
+                            "description": product_description,
+                            "images": [data.stylist_avatar],  # Adiciona a foto do estilista
                         },
                         "unit_amount": data.amount,
                         "recurring": {"interval": "month"}
@@ -52,6 +60,7 @@ def create_checkout_session(data: SubscriptionRequest):
                 "metadata": {
                     "user_id": data.user_id,
                     "stylist_id": data.stylist_id,
+                    "consultations_per_month": data.consultations_per_month
                 }
             }
         )