#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
beeRoot - Exemplo de uso da API
"""

import requests

# URL da sua abelha
BEEROOT_URL = "https://seu-usuario-sua-abelha.hf.space"

def search(query, top_k=10):
    """Busca documentos"""
    response = requests.post(
        f"{BEEROOT_URL}/search/embedding",
        json={"query": query, "top_k": top_k}
    )

    return response.json()

# Exemplo 1: Busca simples
print("🔍 Busca: FGTS")
results = search("FGTS", top_k=5)

for doc in results['results']:
    print(f"\nID: {doc['id']} | Score: {doc['score']:.4f}")
    print(f"Ementa: {doc['ementa'][:200]}...")

# Exemplo 2: Múltiplas queries
queries = [
    "FGTS correção monetária",
    "aposentadoria servidor público",
    "execução fiscal prescrição"
]

for query in queries:
    print(f"\n{'='*60}")
    print(f"Query: {query}")
    print('='*60)

    results = search(query, top_k=3)

    for doc in results['results']:
        print(f"  • ID {doc['id']}: {doc['score']:.2f}")