assets/assets/utils/convert/models/tokens.py

from typing import List, Optional, Tuple

from pydantic import BaseModel, Field

Bbox = Tuple[float, float, float, float]


class Offset(BaseModel):
    begin: int
    end: int


class BadgerdocToken(BaseModel):
    type_: str = Field(default="text", alias="type")
    bbox: Bbox
    text: str
    offset: Offset
    previous: Optional[str] = None
    after: Optional[str] = None


class PageSize(BaseModel):
    width: float = Field(..., example=200.0)
    height: float = Field(..., example=300.0)


class Page(BaseModel):
    """A model for the field with bboxes."""

    page_num: int = Field(..., example=1)
    size: PageSize
    objs: List[BadgerdocToken]