import os
from dotenv import load_dotenv
from sqlalchemy import create_engine, text

# Load environment variables
load_dotenv(".env", override=True)
load_dotenv("../.env", override=False)

DATABASE_URL = os.getenv("DATABASE_URL")

if not DATABASE_URL:
    print("Error: DATABASE_URL not found")
    exit(1)

def add_column():
    engine = create_engine(DATABASE_URL)
    with engine.connect() as conn:
        try:
            # Check if column exists first to avoid error
            check_sql = text("SELECT column_name FROM information_schema.columns WHERE table_name='conversations' AND column_name='summary';")
            result = conn.execute(check_sql)
            if result.fetchone():
                print("Column 'summary' already exists.")
                return

            print("Adding 'summary' column to 'conversations' table...")
            sql = text("ALTER TABLE conversations ADD COLUMN summary TEXT;")
            conn.execute(sql)
            conn.commit()
            print("Successfully added 'summary' column.")
        except Exception as e:
            print(f"Error adding column: {e}")

if __name__ == "__main__":
    add_column()