import sqlite3
import sys

DB_PATH = "/app/companies_v3_fixed_2.db"

def migrate():
    try:
        conn = sqlite3.connect(DB_PATH)
        cursor = conn.cursor()
        
        print(f"Checking schema in {DB_PATH}...")
        cursor.execute("PRAGMA table_info(companies)")
        columns = [row[1] for row in cursor.fetchall()]
        
        if "ai_opener" in columns:
            print("Column 'ai_opener' already exists. Skipping.")
        else:
            print("Adding column 'ai_opener' to 'companies' table...")
            cursor.execute("ALTER TABLE companies ADD COLUMN ai_opener TEXT")
            conn.commit()
            print("✅ Migration successful.")
            
    except Exception as e:
        print(f"❌ Migration failed: {e}")
    finally:
        if conn: conn.close()

if __name__ == "__main__":
    migrate()
