"""
Pydantic response models
"""
from pydantic import BaseModel
from typing import Optional, List, Dict, Any


class GeneratedImage(BaseModel):
    """Single generated image"""
    data: str  # base64
    index: Optional[int] = None
    prompt: str
    aspectRatio: str


class GenerateResponse(BaseModel):
    """Response from image generation"""
    images: List[GeneratedImage]


class VideoResponse(BaseModel):
    """Response from video generation"""
    success: bool
    id: Optional[str] = None
    url: Optional[str] = None
    status: Optional[str] = None


class ReferenceUploadResponse(BaseModel):
    """Response from reference upload"""
    success: bool
    id: str


class MetaImageResult(BaseModel):
    """Meta AI generated image"""
    data: Optional[str] = None
    url: Optional[str] = None
    prompt: str
    model: str
    aspectRatio: str = "1:1"


class MetaGenerateResponse(BaseModel):
    """Response from Meta AI generation"""
    success: bool
    images: List[MetaImageResult]


class MetaVideoResult(BaseModel):
    """Meta AI video result"""
    url: str
    prompt: str


class MetaVideoResponse(BaseModel):
    """Response from Meta AI video generation"""
    success: bool
    videos: List[MetaVideoResult]
    conversation_id: Optional[str] = None


class Prompt(BaseModel):
    """Prompt library item"""
    id: int
    title: str
    description: str
    prompt: str
    category: str
    source: str
    source_url: str
    images: Optional[List[str]] = None
    useCount: int = 0
    lastUsedAt: Optional[int] = None
    createdAt: Optional[int] = None


class PromptCache(BaseModel):
    """Prompt library cache"""
    prompts: List[Prompt]
    last_updated: Optional[str] = None
    lastSync: Optional[int] = None
    categories: Dict[str, List[str]] = {}
    total_count: int = 0
    sources: List[str] = []


class SyncResponse(BaseModel):
    """Response from prompt sync"""
    success: bool
    count: int
    added: int


class HistoryItem(BaseModel):
    """Upload history item"""
    id: str
    url: str
    originalName: str
    category: str
    mediaId: Optional[str] = None
    createdAt: Optional[int] = None


class HistoryResponse(BaseModel):
    """Response from history endpoint"""
    history: List[HistoryItem]


class ErrorResponse(BaseModel):
    """Standard error response"""
    error: str
    details: Optional[str] = None