from sqlalchemy import Float, ForeignKey, Integer, String
from sqlalchemy.orm import Mapped, mapped_column

from app.models.base import Base, TimestampMixin


class CreditRatio(Base, TimestampMixin):
    __tablename__ = "credit_ratios"

    id: Mapped[str] = mapped_column(String(32), primary_key=True)
    model_config_id: Mapped[str] = mapped_column(
        String(32), ForeignKey("model_configs.id")
    )
    gen_type: Mapped[str] = mapped_column(String(16), default="video")
    resolution: Mapped[str] = mapped_column(String(16), nullable=False)
    ratio: Mapped[float] = mapped_column(Float, nullable=False)
    base_credits: Mapped[float] = mapped_column(Float, default=80.0)
    per_second_credits: Mapped[float] = mapped_column(Float, default=2.0)