#!/usr/bin/env python3
"""
Script para adicionar mensagem de boas-vindas a um agente específico
"""
import sys
import os

sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

from src.config.database import db
from src.models.agente import Agente
from flask import Flask

def create_app():
    app = Flask(__name__)
    app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://fernando.barreto:fT3&r$qId&LA3v$9@dns.auditto.com.br:33306/ollama'
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
    db.init_app(app)
    return app

def update_agent_welcome_message():
    app = create_app()
    
    with app.app_context():
        print("=== Atualizando Mensagem de Boas-vindas ===\n")
        
        agente = Agente.query.first()
        
        if not agente:
            print("Nenhum agente encontrado!")
            return
        
        print(f"Agente selecionado: {agente.nome} (ID: {agente.id})")
        print(f"Mensagem atual: {repr(agente.welcome_message)}")
        
        nova_mensagem = "🤖 Olá! Sou seu assistente inteligente. Como posso ajudá-lo hoje? Estou aqui para responder suas perguntas e auxiliá-lo em suas tarefas!"
        agente.welcome_message = nova_mensagem
        
        db.session.commit()
        
        print(f"Nova mensagem: {repr(agente.welcome_message)}")
        print("✅ Mensagem de boas-vindas atualizada com sucesso!")

if __name__ == "__main__":
    update_agent_welcome_message()