"""Add text overlay fields to campaigns table"""
import sys
import os
sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))

from app.extensions import db

def upgrade():
    """Add text overlay fields to campaigns table"""
    # Check if columns exist
    inspector = db.inspect(db.engine)
    columns = [col['name'] for col in inspector.get_columns('campaigns')]

    if 'text_overlays' not in columns:
        print("Adding text_overlays column...")
        db.session.execute(db.text("ALTER TABLE campaigns ADD COLUMN text_overlays JSONB"))

    if 'overlay_instructions' not in columns:
        print("Adding overlay_instructions column...")
        db.session.execute(db.text("ALTER TABLE campaigns ADD COLUMN overlay_instructions JSONB"))

    if 'video_with_text_url' not in columns:
        print("Adding video_with_text_url column...")
        db.session.execute(db.text("ALTER TABLE campaigns ADD COLUMN video_with_text_url TEXT"))

    db.session.commit()
    print("✅ Text overlay fields added successfully!")

if __name__ == "__main__":
    from app import create_app
    app = create_app()
    with app.app_context():
        upgrade()