35 lines
1013 B
Python
Executable File
35 lines
1013 B
Python
Executable File
"""
|
|
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()
|