#!/usr/bin/env python3
"""
Script para adicionar mensagens de boas-vindas a vários agentes
"""
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_multiple_agents():
    app = create_app()
    
    with app.app_context():
        print("=== Atualizando Mensagens de Boas-vindas em Vários Agentes ===\n")
        
        welcome_messages = {
            'Llama': "🦙 Olá! Sou o Llama, seu assistente de IA local. Como posso ajudá-lo hoje?",
            'Gemini': "💎 Olá! Sou o Gemini, powered by Google. Estou aqui para ajudar com suas perguntas!",
            'ChatGPT 3.5 Turbo': "🚀 Olá! Sou o ChatGPT 3.5 Turbo. Como posso auxiliá-lo hoje?",
            'Auditto': "🏢 Olá! Sou o assistente da Auditto. Como posso ajudá-lo com questões corporativas?",
            'Calculadora de Metragem': "📐 Olá! Sou especializado em cálculos de metragem. Como posso ajudar?",
            'Comunicação': "📢 Olá! Sou especializado em comunicação. Como posso ajudar a melhorar sua comunicação?",
            'Tecnologia': "💻 Olá! Sou especializado em tecnologia. Como posso ajudá-lo com questões técnicas?",
            'default': "👋 Olá! Como posso ajudá-lo hoje? Estou aqui para responder suas perguntas!"
        }
        
        agentes = Agente.query.all()
        agentes_atualizados = 0
        
        for agente in agentes:
            if agente.welcome_message:
                print(f"✓ {agente.nome} (ID: {agente.id}) - já tem mensagem configurada")
                continue
            
            mensagem = welcome_messages.get(agente.nome, welcome_messages['default'])
            
            agente.welcome_message = mensagem
            print(f"📝 {agente.nome} (ID: {agente.id}) - mensagem adicionada: {repr(mensagem[:50])}...")
            agentes_atualizados += 1
        
        db.session.commit()
        
        print(f"\n✅ Processo concluído!")
        print(f"Total de agentes: {len(agentes)}")
        print(f"Agentes atualizados: {agentes_atualizados}")
        print(f"Agentes que já tinham mensagem: {len(agentes) - agentes_atualizados}")

if __name__ == "__main__":
    update_multiple_agents()