"""
Migration: Add ai_prompt column to questions table
Date: 2025-11-21
Description: Adds ai_prompt TEXT column for custom AI analysis prompts per question
"""

# SQL Migration Script
sql_statements = [
    # Add ai_prompt column
    """
    ALTER TABLE questions 
    ADD COLUMN ai_prompt TEXT;
    """,
]

# To apply this migration, run these SQL statements in your PostgreSQL database:
if __name__ == "__main__":
    print("=" * 80)
    print("MIGRATION: Add ai_prompt to questions table")
    print("=" * 80)
    print("\nExecute the following SQL statements in your PostgreSQL database:\n")
    
    for i, statement in enumerate(sql_statements, 1):
        print(f"-- Statement {i}")
        print(statement.strip())
        print()
    
    print("=" * 80)
    print("\nTo verify the migration:")
    print("SELECT column_name, data_type FROM information_schema.columns")
    print("WHERE table_name = 'questions' AND column_name = 'ai_prompt';")
    print("=" * 80)