feat: update subscription response to track AI credits instead of tokens

app/api/v1/subscription.py

@@ -7,6 +7,7 @@ from app.api.deps import get_current_user
 from app.models.user import User
 from app.models.wardrobe import WardrobeItem
 from app.models.chat import ChatMessage
+from app.models.tryon import TryOnTask
 from app.schemas.subscription import SubscriptionResponse
 
 router = APIRouter()
@@ -14,9 +15,10 @@ router = APIRouter()
 # Free-tier caps advertised on the landing page's pricing table.
 FREE_WARDROBE_ITEM_LIMIT = 30
 FREE_AI_RECOMMENDATIONS_PER_WEEK = 5
-# AI Stylist usage is metered like an LLM API quota: total tokens consumed
-# by "ai" responses over a rolling 30-day window.
-FREE_AI_TOKEN_LIMIT_PER_30D = 50_000
+# AI Stylist usage is metered like an LLM API quota, expressed in "credits"
+# (1 credit = 1,000 tokens) over a rolling 30-day window.
+TOKENS_PER_CREDIT = 1_000
+FREE_AI_CREDIT_LIMIT_PER_30D = 500
 
 
 def _build_response(db: Session, user: User) -> SubscriptionResponse:
@@ -26,13 +28,13 @@ def _build_response(db: Session, user: User) -> SubscriptionResponse:
         db.query(WardrobeItem).filter(WardrobeItem.user_id == user.id).count()
     )
 
+    # "AI outfits" = virtual try-on generations (each one produces one AI-styled outfit).
     week_ago = datetime.now(timezone.utc) - timedelta(days=7)
     ai_used = (
-        db.query(ChatMessage)
+        db.query(TryOnTask)
         .filter(
-            ChatMessage.user_id == user.id,
-            ChatMessage.role == "ai",
-            ChatMessage.created_at >= week_ago,
+            TryOnTask.user_id == user.id,
+            TryOnTask.created_at >= week_ago,
         )
         .count()
     )
@@ -47,6 +49,8 @@ def _build_response(db: Session, user: User) -> SubscriptionResponse:
         )
         .scalar()
     )
+    # Round up: any partial usage still consumes a whole credit, like real LLM billing.
+    credits_used = -(-int(tokens_used or 0) // TOKENS_PER_CREDIT)
 
     return SubscriptionResponse(
         plan=user.plan,
@@ -55,8 +59,8 @@ def _build_response(db: Session, user: User) -> SubscriptionResponse:
         wardrobe_items_limit=None if is_premium else FREE_WARDROBE_ITEM_LIMIT,
         ai_recommendations_used=ai_used,
         ai_recommendations_limit=None if is_premium else FREE_AI_RECOMMENDATIONS_PER_WEEK,
-        ai_tokens_used=int(tokens_used or 0),
-        ai_tokens_limit=None if is_premium else FREE_AI_TOKEN_LIMIT_PER_30D,
+        ai_credits_used=credits_used,
+        ai_credits_limit=None if is_premium else FREE_AI_CREDIT_LIMIT_PER_30D,
     )
 
 

app/schemas/subscription.py

@@ -9,8 +9,8 @@ class SubscriptionResponse(BaseModel):
     wardrobe_items_limit: Optional[int]  # None = unlimited
     ai_recommendations_used: int
     ai_recommendations_limit: Optional[int]  # None = unlimited
-    ai_tokens_used: int
-    ai_tokens_limit: Optional[int]  # None = unlimited
+    ai_credits_used: int  # 1 credit = 1,000 LLM tokens
+    ai_credits_limit: Optional[int]  # None = unlimited
 
     class Config:
         from_attributes = True