src/storage/models.py

from sqlalchemy import Integer, Float, Text
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column

class Base(DeclarativeBase):
    pass

class BoxesData(Base):
    __tablename__= "croped_boxes_metadata"
    
    id: Mapped[int] = mapped_column(Integer, primary_key=True)
    class_name: Mapped[int] = mapped_column(Integer)
  
    x_min: Mapped[float] = mapped_column(Float)
    y_min: Mapped[float] = mapped_column(Float)
    x_max: Mapped[float] = mapped_column(Float)
    y_max: Mapped[float] = mapped_column(Float)
    
    confidence: Mapped[float] = mapped_column(Float)
    saved_img_path: Mapped[str] = mapped_column(Text)