"""
Script para busca semântica na base de conhecimento
Descomplicar - Agência de Aceleração Digital
https://www.descomplicar.pt
"""

from processors.embedding_processor import EmbeddingProcessor

def main():
    """Função principal."""
    try:
        # Inicializar processador
        embedding = EmbeddingProcessor()
        
        # Solicitar query do usuário
        query = input("\nDigite sua pergunta: ")
        
        # Buscar documentos similares
        print("\nBuscando documentos relevantes...")
        results = embedding.search_similar(query, limit=5)
        
        # Mostrar resultados
        print("\nResultados encontrados:")
        print("-" * 80)
        
        for i, result in enumerate(results, 1):
            print(f"\n{i}. {result['title']}")
            print(f"Similaridade: {result['doc_similarity']:.1%}")
            print("\nTrechos relevantes:")
            print(result['relevant_chunks'])
            print("-" * 80)

if __name__ == "__main__":
    main()