Spaces:

tsalkar
/

pdf-field-extractor-backend

Sleeping

pdf-field-extractor-backend / commonforms /utils.py

Tanishq Salkar

code updated

9de8051 9 months ago

665 Bytes

from __future__ import annotations
from typing import Literal
from pydantic import BaseModel
from dataclasses import dataclass
from PIL import Image


class BoundingBox(BaseModel):
    x0: float
    y0: float
    x1: float
    y1: float

    @classmethod
    def from_yolo(cls, cx: float, cy: float, w: float, h: float) -> BoundingBox:
        return cls(x0=cx - w / 2, y0=cy - h / 2, x1=cx + w / 2, y1=cy + h / 2)


class Widget(BaseModel):
    widget_type: Literal[
        "TextBox",
        "ChoiceButton",
        "Signature",
    ]
    bounding_box: BoundingBox
    page: int


@dataclass
class Page:
    image: Image.Image
    width: float
    height: float

	from __future__ import annotations
	from typing import Literal
	from pydantic import BaseModel
	from dataclasses import dataclass
	from PIL import Image


	class BoundingBox(BaseModel):
	x0: float
	y0: float
	x1: float
	y1: float

	@classmethod
	def from_yolo(cls, cx: float, cy: float, w: float, h: float) -> BoundingBox:
	return cls(x0=cx - w / 2, y0=cy - h / 2, x1=cx + w / 2, y1=cy + h / 2)


	class Widget(BaseModel):
	widget_type: Literal[
	"TextBox",
	"ChoiceButton",
	"Signature",
	]
	bounding_box: BoundingBox
	page: int


	@dataclass
	class Page:
	image: Image.Image
	width: float
	height: float