fund-tracer/backend/app/schemas/image.py

from datetime import datetime
from uuid import UUID

from app.models.evidence_image import SourceApp, PageType, OcrStatus
from app.schemas.base import CamelModel


class ImageOut(CamelModel):
    id: UUID
    case_id: UUID
    url: str = ""
    thumb_url: str = ""
    source_app: SourceApp
    page_type: PageType
    ocr_status: OcrStatus
    file_hash: str
    uploaded_at: datetime


class OcrBlockOut(CamelModel):
    id: UUID
    content: str
    bbox: dict
    seq_order: int
    confidence: float


class ImageDetailOut(ImageOut):
    ocr_blocks: list[OcrBlockOut] = []


class OcrFieldCorrection(CamelModel):
    field_name: str
    old_value: str
    new_value: str


class OcrBlockUpdateIn(CamelModel):
    content: str
    confidence: float = 0.5


class CaseOcrStartIn(CamelModel):
    include_done: bool = False
    image_ids: list[UUID] = []


class CaseImagesDeleteIn(CamelModel):
    image_ids: list[UUID] = []