Update routes/subscription.py

routes/subscription.py

@@ -87,13 +87,24 @@ async def generate_dashboard_link(data: UserIDRequest):
                 logger.error(f"Error creating login link for stylist account: {str(e)}")
                 raise HTTPException(status_code=500, detail="Error creating login link for stylist account.")
         
-        # Caso seja um cliente, não podemos criar um login link
+        # Caso seja um cliente, criamos uma sessão do portal de cobrança
         elif user_id.startswith("cus_"):
-            return {
-                "status": "success",
-                "dashboard_link": f"https://billing.stripe.com/p/{user_id}"
-            }
+            try:
+                # Cria uma sessão do portal de cobrança para o cliente
+                session = stripe.billing_portal.Session.create(
+                    customer=user_id,
+                    return_url="https://your-website.com/account"  # URL para onde o cliente será redirecionado após sair do portal
+                )
+                
+                return {
+                    "status": "success",
+                    "dashboard_link": session.url  # Retorna o link para o painel do cliente
+                }
 
+            except stripe.error.StripeError as e:
+                logger.error(f"Error creating billing portal session for customer: {str(e)}")
+                raise HTTPException(status_code=500, detail="Error creating billing portal session for customer.")
+        
         else:
             raise HTTPException(status_code=400, detail="Invalid user ID format. Must start with 'acct_' for stylist or 'cus_' for customer.")