"""add conversation_utterance table

Revision ID: f6a7b8c9d0e1
Revises: e5f6a7b8c9d0
Create Date: 2026-07-25
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

revision = 'f6a7b8c9d0e1'
down_revision = 'e5f6a7b8c9d0'
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        'conversation_utterance',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('session_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('intake_session.id', ondelete='CASCADE'), nullable=False),
        sa.Column('sequence_number', sa.SmallInteger(), nullable=False),
        sa.Column('speaker_role', sa.String(length=20), nullable=False, server_default='UNKNOWN'),
        sa.Column('raw_transcript', sa.Text(), nullable=False),
        sa.Column('language', sa.String(length=20), nullable=False, server_default='te-IN'),
        sa.Column('mapped_question_id', sa.String(length=255), nullable=True),
        sa.Column('mapping_confidence', sa.Float(), nullable=True),
        sa.Column('mapping_status', sa.String(length=20), nullable=False, server_default='UNMAPPED'),
        sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.func.now(), nullable=False),
    )
    op.create_index(
        'conversation_utterance_session_idx',
        'conversation_utterance',
        ['session_id', 'sequence_number'],
    )


def downgrade() -> None:
    op.drop_index('conversation_utterance_session_idx', table_name='conversation_utterance')
    op.drop_table('conversation_utterance')