# app/schemas.py
from pydantic import BaseModel, Field
from typing import Union, Optional

class GenerateRequest(BaseModel): # <--- 'GenerateRequest' 정의되어 있음
    user_input: str = Field(..., example="미래 도시 풍경, 하늘을 나는 자동차들, 화려한 네온사인, 다양한 캐릭터들, 영화적인 애니메이션 스타일로 그려줘.")
    # ...

class GenerateResponse(BaseModel): # <--- 'GenerateResponse' 정의되어 있음
    generated_prompt: str
    processing_time_seconds: float
    error_message: Optional[str] = None # <--- 이렇게 변경되었는지 확인!

class ResetMemoryResponse(BaseModel): # <--- 'ResetMemoryResponse' 정의되어 있음
    message: str
    cleared_memory: bool
    cleared_wiki_history: bool