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


class QuestionResponse(Base):
    __tablename__ = "question_response"

    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)
    question_id = Column(String(255), nullable=True)
    question_text = Column(Text, nullable=False)
    transcribed_text = Column(Text, nullable=False)
    input_type = Column(String(10), nullable=False, default="TEXT")
    input_language = Column(String(20), nullable=False, default="ENGLISH")
    is_flagged = Column(Boolean, nullable=False, default=False)
    sequence_number = Column(SmallInteger, nullable=False)
    responded_by_staff_id = Column(UUID(as_uuid=True), ForeignKey("user.id"), nullable=False)
    created_at = Column(TIMESTAMP(timezone=True), server_default=func.now(), nullable=False)

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