"""Pydantic models for request/response schemas."""

from pydantic import BaseModel, HttpUrl


class EmbedRequest(BaseModel):
    image_url: HttpUrl


class BBox(BaseModel):
    x: int
    y: int
    w: int
    h: int


class FaceEmbedding(BaseModel):
    bbox: BBox
    score: float
    embedding: list[float]


class EmbedAvatarResponse(BaseModel):
    embedding: list[float]
    bbox: BBox
    score: float
    processed_width: int | None = None
    processed_height: int | None = None


class EmbedImageResponse(BaseModel):
    faces: list[FaceEmbedding]
    processed_width: int | None = None
    processed_height: int | None = None


class MatchResult(BaseModel):
    tag: str
    found: bool
    score: float | None = None
    bbox: BBox | None = None