Update app.py

app.py

@@ -1,4 +1,5 @@
 from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from duckduckgo_search import DDGS
 import datetime
 import pytz
 import yaml
@@ -7,8 +8,11 @@ import yfinance as yf
 from Gradio_UI import GradioUI
 import json
 
+#---------------------------------------------------------------------------
+#---------------------------------------------------------------------------tools
+#---------------------------------------------------------------------------
 
-# ---- Nouvel outil custom ----
+#stock
 @tool
 def get_stock_price(ticker: str) -> str:
     """Un outil qui retourne le prix actuel d’un actif coté en bourse au format JSON.
@@ -27,8 +31,22 @@ def get_stock_price(ticker: str) -> str:
     except Exception as e:
         return json.dumps({"error": f"Impossible de récupérer le cours pour {ticker}: {str(e)}"})
 
+#web search
+@tool
+def web_search(query: str, max_results: int = 5) -> str:
+    """Recherche des informations sur internet via DuckDuckGo et retourne une liste de résultats JSON.
+    Args:
+        query: La requête de recherche.
+        max_results: Nombre maximum de résultats à retourner.
+    """
+    try:
+        with DDGS() as ddgs:
+            results = list(ddgs.text(query, max_results=max_results))
+        return json.dumps(results, ensure_ascii=False, indent=2)
+    except Exception as e:
+        return json.dumps({"error": str(e)})
 
-# ---- Exemple déjà présent : l’heure dans un fuseau horaire ----
+#timezone
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """Un outil qui retourne l’heure locale actuelle dans un fuseau horaire donné.
@@ -42,13 +60,12 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Erreur pour le fuseau horaire '{timezone}': {str(e)}"
 
+#---------------------------------------------------------------------------
+#---------------------------------------------------------------------------init
+#---------------------------------------------------------------------------
 
-# ---- Initialisation ----
 final_answer = FinalAnswerTool()
 
-# Ajout de l'outil de recherche internet (DuckDuckGo)
-web_search = load_tool("duckduckgo_search", trust_remote_code=True)
-
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -67,7 +84,7 @@ agent = CodeAgent(
         final_answer,
         get_stock_price,
         get_current_time_in_timezone,
-        web_search   # 👈 Ajouté ici
+        web_search 
     ],
     max_steps=6,
     verbosity_level=1,
@@ -78,5 +95,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# ---- Lancement UI ----
+#---------------------------------------------------------------------------main
 GradioUI(agent).launch()