models.py

from pydantic import BaseModel
from typing import List, Optional

class WardrobeItem(BaseModel):
    id: Optional[int] = None
    category: str
    style: str
    color: Optional[str] = None
    brand: Optional[str] = None
    name: Optional[str] = None

class ChatRequest(BaseModel):
    message: str
    session_id: Optional[str] = "default"
    wardrobe: Optional[List[WardrobeItem]] = None
    wardrobe_description: Optional[str] = None
    images: Optional[List[str]] = None

class ChatResponse(BaseModel):
    response: str
    session_id: str

class TextRequest(BaseModel):
    message: str
    session_id: Optional[str] = "default"