from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI

@tool
def duckduckgo_search(query: str) -> dict:
    """Un outil qui effectue une recherche sur DuckDuckGo pour un mot-clé donné.
    
    Args:
        query: Le mot-clé à rechercher.
    
    Returns:
        dict: Dictionnaire contenant l'URL et le titre du premier résultat.
    """
    ddg_tool = DuckDuckGoSearchTool()
    # On utilise la méthode run() pour exécuter la recherche
    result = ddg_tool.run(query)
    if result is not None and isinstance(result, dict):
        return {"url": result.get("url"), "titre": result.get("title")}
    return {"url": None, "titre": None}

final_answer = FinalAnswerTool()
model = HfApiModel(
    max_tokens=2096,
    temperature=0.5,
    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
    custom_role_conversions=None,
)

# Import d'un outil depuis le Hub (exemple d'outil de génération d'image)
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[final_answer, duckduckgo_search],
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)

GradioUI(agent).launch()