from __future__ import annotations

import uuid

from sqlalchemy import JSON, ForeignKey, String
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import Mapped, mapped_column

from cpv3.db.base import Base, BaseModelMixin


class Transcription(Base, BaseModelMixin):
    __tablename__ = "transcriptions"

    project_id: Mapped[uuid.UUID | None] = mapped_column(
        UUID(as_uuid=True),
        ForeignKey("projects.id", ondelete="RESTRICT"),
        nullable=True,
        index=True,
    )
    source_file_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), ForeignKey("files.id", ondelete="RESTRICT"), index=True
    )
    artifact_id: Mapped[uuid.UUID | None] = mapped_column(
        UUID(as_uuid=True),
        ForeignKey("artifact_media_files.id", ondelete="RESTRICT"),
        nullable=True,
        index=True,
    )

    engine: Mapped[str] = mapped_column(String(32), default="LOCAL_WHISPER")
    language: Mapped[str | None] = mapped_column(String(3), nullable=True)

    document: Mapped[dict] = mapped_column(JSON)
    transcribe_options: Mapped[dict | None] = mapped_column(JSON, nullable=True)