app/schemas.py

"""HTTP request/response models for the asset extractor."""

from pydantic import BaseModel


class AssetExtractRequest(BaseModel):
    url: str
    include_data_uris: bool = False
    groups: list[str] | None = None  # optional subset: image, video, audio, svg_icon, font


class AssetGroup(BaseModel):
    image: list[str] = []
    video: list[str] = []
    audio: list[str] = []
    svg_icon: list[str] = []
    font: list[str] = []


class AssetExtractResponse(BaseModel):
    url: str
    title: str | None = None
    total: int = 0
    assets: AssetGroup