from typing import List, Optional

from pydantic import BaseModel


class PlanResponse(BaseModel):
    id: str
    name: str
    display_name: str
    price: float
    currency: str
    features: List[str]
    max_conversations: Optional[int] = None
    max_chapters: Optional[int] = None
    max_words: Optional[int] = None
    is_popular: bool = False


class CurrentPlanResponse(BaseModel):
    plan_id: str
    plan_name: str
    subscription_type: str
    expires_at: Optional[str] = None
    features: List[str]
    usage: dict