from app import create_app, db
import sqlalchemy

app = create_app()
with app.app_context():
    inspector = sqlalchemy.inspect(db.engine)
    columns = [col['name'] for col in inspector.get_columns('company_info')]
    if 'intro_banner_url' not in columns:
        with db.engine.begin() as conn:
            conn.execute(sqlalchemy.text('ALTER TABLE company_info ADD COLUMN intro_banner_url VARCHAR(256)'))
        print("Added intro_banner_url column")
    else:
        print("Column already exists")