#!/usr/bin/env python3
"""
Script para verificar as mensagens de boas-vindas dos 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 src.models.chat import Chat
from src.models.message import Message
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 check_welcome_messages():
    app = create_app()
    
    with app.app_context():
        print("=== Verificando Mensagens de Boas-vindas ===\n")
        
        agentes = Agente.query.all()
        print(f"Total de agentes encontrados: {len(agentes)}\n")
        
        for agente in agentes:
            print(f"Agente ID: {agente.id}")
            print(f"Nome: {agente.nome}")
            print(f"Mensagem de boas-vindas: {repr(agente.welcome_message)}")
            print("-" * 50)
        
        print("\n=== Verificando Chats Recentes ===\n")
        recent_chats = Chat.query.order_by(Chat.created_at.desc()).limit(5).all()
        
        for chat in recent_chats:
            print(f"Chat ID: {chat.id} | Agent ID: {chat.agent_id} | Criado em: {chat.created_at}")
            messages = Message.query.filter_by(chat_id=chat.id).order_by(Message.created_at).all()
            print(f"Total de mensagens: {len(messages)}")
            
            for i, msg in enumerate(messages):
                user_type = "Usuário" if msg.is_user else "Agente"
                print(f"  {i+1}. {user_type}: {repr(msg.message[:100])}...")
            
            print("-" * 50)

if __name__ == "__main__":
    check_welcome_messages()