import os
import psycopg2
from dotenv import load_dotenv

load_dotenv()

# Usar DATABASE_URL del archivo .env
DATABASE_URL = os.getenv("DATABASE_URL")

# Si no está en .env, usar el fallback directo (con el puerto correcto para transacciones/directo 5432 o 6543)
if not DATABASE_URL:
    DATABASE_URL = "postgresql://postgres.egkydqberhebzkqtltic:Isaac%23105%2Amay%40@aws-1-us-east-1.pooler.supabase.com:6543/postgres"

def migrate():
    print(f"Conectando a la base de datos...")
    conn = psycopg2.connect(DATABASE_URL)
    cur = conn.cursor()
    
    print("Agregando columna 'bpm' a la tabla 'lyrics'...")
    try:
        cur.execute("ALTER TABLE lyrics ADD COLUMN bpm INTEGER NULL;")
        conn.commit()
        print("Columna 'bpm' agregada exitosamente.")
    except Exception as e:
        print("Error al agregar columna (es posible que ya exista):", e)
        conn.rollback()
        
    cur.close()
    conn.close()

if __name__ == "__main__":
    migrate()