from typing import Optional

from pydantic import BaseModel, Field


class ClassificationResponse(BaseModel):
    is_ai_generated: bool
    ai_probability: float
    human_probability: float


class IconRequest(BaseModel):
    prompt: str = Field(..., min_length=1, max_length=4000)
    color: Optional[str] = None
    shape: Optional[str] = None
    style: Optional[str] = None