Update app.py

app.py

@@ -4,70 +4,24 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
-from bs4 import BeautifulSoup
 
 @tool
-def recherche_google(query: str) -> dict:
-    """Recherche sur Google un mot-clé donné.
+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.
     """
-    url_search = "https://www.google.com/search"
-    headers = {
-        "User-Agent": (
-            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
-            "AppleWebKit/537.36 (KHTML, like Gecko) "
-            "Chrome/98.0.4758.102 Safari/537.36"
-        )
-    }
-    params = {"q": query, "hl": "fr"}
-    
-    response = requests.get(url_search, params=params, headers=headers)
-    if response.status_code != 200:
-        raise Exception("Erreur lors de la requête vers Google")
-    
-    soup = BeautifulSoup(response.text, "html.parser")
-    # Recherche des blocs de résultats ; la structure de Google pouvant varier,
-    # nous itérons sur les résultats pour trouver un lien valide.
-    results = soup.find_all("div", class_="g")
-    for result in results:
-        a_tag = result.find("a")
-        if a_tag:
-            url_result = a_tag.get("href")
-            titre = a_tag.get_text(strip=True)
-            if url_result and titre:
-                return {"url": url_result, "titre": titre}
+    ddg_tool = DuckDuckGoSearchTool()
+    result = ddg_tool.search(query)
+    if result is not None and isinstance(result, dict):
+        return {"url": result.get("url"), "titre": result.get("title")}
     return {"url": None, "titre": None}
 
-@tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
 final_answer = FinalAnswerTool()
 model = HfApiModel(
     max_tokens=2096,
@@ -76,7 +30,7 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Import tool from Hub (exemple d'outil de génération d'image)
+# 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:
@@ -84,7 +38,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, recherche_google, my_custom_tool, get_current_time_in_timezone],
+    tools=[final_answer, duckduckgo_search],
     max_steps=6,
     verbosity_level=1,
     grammar=None,