#!/usr/bin/env python3
"""
Script para adicionar o campo welcome_message ao banco de dados
"""

import sys
import os
from dotenv import load_dotenv

load_dotenv()

sys.path.append(os.path.dirname(os.path.abspath(__file__)))

def add_welcome_message_column():
    from src import create_app
    app = create_app()
    with app.app_context():
        from src.extensions import db
        from sqlalchemy import text
        
        try:
            result = db.session.execute(text("SHOW COLUMNS FROM agente LIKE 'welcome_message'"))
            if result.fetchone():
                print("✅ Coluna welcome_message já existe")
                return True
            
            db.session.execute(text("ALTER TABLE agente ADD COLUMN welcome_message TEXT NULL"))
            db.session.commit()
            print("✅ Coluna welcome_message adicionada com sucesso!")
            return True
            
        except Exception as e:
            print(f"❌ Erro ao adicionar coluna welcome_message: {str(e)}")
            db.session.rollback()
            return False

if __name__ == '__main__':
    add_welcome_message_column()