custom_types.py

from __future__ import annotations

from typing import Literal

from pydantic import BaseModel, Field, model_validator


class PlateBBox(BaseModel):
    x1: float = Field(..., description="Left coordinate.")
    y1: float = Field(..., description="Top coordinate.")
    x2: float = Field(..., description="Right coordinate.")
    y2: float = Field(..., description="Bottom coordinate.")

    @model_validator(mode="after")
    def validate_bounds(self):
        if min(self.x1, self.y1, self.x2, self.y2) < 0:
            raise ValueError("Bounding box coordinates must be non-negative")
        if self.x2 <= self.x1 or self.y2 <= self.y1:
            raise ValueError("Bounding box coordinates are invalid")
        return self


class PlateResult(BaseModel):
    status: Literal["success", "not_found"]
    plate_text: str
    confidence: float = Field(ge=0.0, le=1.0)
    bbox: PlateBBox


__all__ = ["PlateBBox", "PlateResult"]