from pydantic import BaseModel, Field


class OcrRequest(BaseModel):
    pdf_url: str = Field(..., alias="pdfUrl")
    script_type: str = Field("UNKNOWN", alias="scriptType")
    language: str = "de"


class OcrBlock(BaseModel):
    page_number: int = Field(..., alias="pageNumber")
    x: float
    y: float
    width: float
    height: float
    polygon: list[list[float]] | None = None
    text: str

    class Config:
        populate_by_name = True