"""
Migration script to add toxicity_manual_override column to comment_analyses table.
Run this once to update your existing database.
"""
from db import engine, SessionLocal
from sqlalchemy import text

def migrate():
    """Add toxicity_manual_override column if it doesn't exist"""
    db = SessionLocal()
    try:
        # Check if column exists
        result = db.execute(text("PRAGMA table_info(comment_analyses)"))
        columns = [row[1] for row in result.fetchall()]
        
        if "toxicity_manual_override" not in columns:
            print("Adding toxicity_manual_override column...")
            db.execute(text("""
                ALTER TABLE comment_analyses 
                ADD COLUMN toxicity_manual_override BOOLEAN
            """))
            db.commit()
            print("✓ Column added successfully!")
        else:
            print("✓ Column already exists, no migration needed.")
    except Exception as e:
        db.rollback()
        print(f"✗ Error: {e}")
        raise
    finally:
        db.close()

if __name__ == "__main__":
    print("Running migration to add toxicity_manual_override column...")
    migrate()
    print("Migration complete!")