import uuid
from sqlalchemy import Column, String, SmallInteger, Float, Text, TIMESTAMP, ForeignKey, Index
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.sql import func
from app.db.engine import Base


class ConversationUtterance(Base):
    __tablename__ = "conversation_utterance"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    session_id = Column(UUID(as_uuid=True), ForeignKey("intake_session.id", ondelete="CASCADE"), nullable=False)
    sequence_number = Column(SmallInteger, nullable=False)
    speaker_role = Column(String(20), nullable=False, default="UNKNOWN")
    raw_transcript = Column(Text, nullable=False)
    language = Column(String(20), nullable=False, default="te-IN")
    mapped_question_id = Column(String(255), nullable=True)
    mapping_confidence = Column(Float, nullable=True)
    mapping_status = Column(String(20), nullable=False, default="UNMAPPED")
    created_at = Column(TIMESTAMP(timezone=True), server_default=func.now(), nullable=False)

    __table_args__ = (
        Index("conversation_utterance_session_idx", "session_id", "sequence_number"),
    )