from brave_search_mcp import BraveSearchMCP
import json

def main():
    # Remplacez "mcp-config.json" par le chemin vers votre fichier de configuration
    brave_mcp = BraveSearchMCP("mcp-config.json")
    
    # Effectuer une recherche
    search_results = brave_mcp.search("Python programming", {
        "count": 5,
        "search_lang": "fr"
    })
    
    # Afficher les résultats
    print("Résultats de recherche:")
    for i, result in enumerate(search_results.get("web", {}).get("results", []), 1):
        print(f"{i}. {result['title']}")
        print(f"   URL: {result['url']}")
        print(f"   Description: {result['description'][:100]}...")
        print()
    
    # Utiliser l'autocomplétion
    suggestions = brave_mcp.autocomplete("machine lear")
    
    # Afficher les suggestions
    print("\nSuggestions d'autocomplétion:")
    for suggestion in suggestions.get("suggestions", []):
        print(f"- {suggestion['query']}")
    
    # N'oubliez pas que le serveur sera automatiquement arrêté grâce à atexit

if __name__ == "__main__":
    main()