"""MLSettings — single-row table holding ML pipeline tunables."""

from datetime import datetime

from sqlalchemy import CheckConstraint, DateTime, Float, Integer, String, func
from sqlalchemy.orm import Mapped, mapped_column

from .base import Base


class MLSettings(Base):
    __tablename__ = "ml_settings"
    # Bare name — Base.metadata's naming convention prepends ck_<table>_,
    # producing the final ck_ml_settings_singleton (matches migration 0003).
    __table_args__ = (CheckConstraint("id = 1", name="singleton"),)

    id: Mapped[int] = mapped_column(Integer, primary_key=True)
    suggestion_threshold_artist: Mapped[float] = mapped_column(
        Float, nullable=False, default=0.30
    )
    suggestion_threshold_character: Mapped[float] = mapped_column(
        Float, nullable=False, default=0.50
    )
    suggestion_threshold_copyright: Mapped[float] = mapped_column(
        Float, nullable=False, default=0.50
    )
    suggestion_threshold_general: Mapped[float] = mapped_column(
        Float, nullable=False, default=0.95
    )
    centroid_similarity_threshold: Mapped[float] = mapped_column(
        Float, nullable=False, default=0.55
    )
    min_reference_images: Mapped[int] = mapped_column(
        Integer, nullable=False, default=5
    )
    tagger_model_version: Mapped[str] = mapped_column(
        String(128), nullable=False, default="camie-tagger-v2"
    )
    embedder_model_version: Mapped[str] = mapped_column(
        String(128), nullable=False, default="siglip-so400m-patch14-384"
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, server_default=func.now()
    )