Update app.py

app.py

@@ -1,47 +1,132 @@
-# tool_docstring_test.py
-from smolagents import tool
-import json, sys
-
-results = []
-
-# Variante A : une ligne par argument, sans "Args:" (format minimal)
-try:
-    @tool
-    def f_a(query: str, max_results: int = 5) -> str:
-        """query: La requête exacte à rechercher sur internet
-max_results: Nombre maximum de résultats à retourner"""
-        return "A"
-    results.append(("A", "OK"))
-except Exception as e:
-    results.append(("A", f"ERR: {type(e).__name__}: {e}"))
-
-# Variante B : style Google (Args: block)
-try:
-    @tool
-    def f_b(query: str, max_results: int = 5) -> str:
-        """
-        Perform a web search.
-
-        Args:
-            query (str): The search query.
-            max_results (int): Maximum results to return.
-        """
-        return "B"
-    results.append(("B", "OK"))
-except Exception as e:
-    results.append(("B", f"ERR: {type(e).__name__}: {e}"))
-
-# Variante C : Sphinx-like param lines (param: desc)
-try:
-    @tool
-    def f_c(query: str, max_results: int = 5) -> str:
-        """
-        query (str): The search query.
-        max_results (int): Maximum results to return.
-        """
-        return "C"
-    results.append(("C", "OK"))
-except Exception as e:
-    results.append(("C", f"ERR: {type(e).__name__}: {e}"))
-
-print(json.dumps(results, ensure_ascii=False, indent=2))
+from smolagents import CodeAgent, HfApiModel, tool
+from duckduckgo_search import DDGS
+import requests
+from bs4 import BeautifulSoup
+import datetime
+import pytz
+import yaml
+from tools.final_answer import FinalAnswerTool
+from Gradio_UI import GradioUI
+import json
+
+#---------------------------------------------------------------------------#
+#--------------------------------- Tools ----------------------------------#
+#---------------------------------------------------------------------------#
+
+@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 (str): La requête de recherche.
+        max_results (int): Nombre maximum de résultats à retourner.
+
+    Returns:
+        str: Liste JSON des résultats de recherche.
+    """
+    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)})
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """
+    Retourne l’heure locale actuelle dans un fuseau horaire donné.
+
+    Args:
+        timezone (str): Un fuseau horaire valide (ex: 'America/New_York').
+
+    Returns:
+        str: Heure locale formatée ou message d’erreur.
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"L’heure locale à {timezone} est {local_time}"
+    except Exception as e:
+        return f"Erreur pour le fuseau horaire '{timezone}': {str(e)}"
+
+@tool
+def visit_webpage(url: str) -> str:
+    """
+    Télécharge le HTML complet d’une page web avec User-Agent.
+
+    Args:
+        url (str): URL de la page web.
+
+    Returns:
+        str: Contenu HTML de la page.
+    """
+    headers = {
+        "User-Agent": (
+            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+            "AppleWebKit/537.36 (KHTML, like Gecko) "
+            "Chrome/122.0.0.0 Safari/537.36"
+        )
+    }
+    resp = requests.get(url, headers=headers, timeout=10)
+    resp.raise_for_status()
+    return resp.text
+
+@tool
+def parse_html(html: str, css_selector: str) -> list:
+    """
+    Extrait du texte d’éléments HTML via un sélecteur CSS.
+
+    Args:
+        html (str): Code HTML complet.
+        css_selector (str): Sélecteur CSS (ex: 'table tr td:last-child').
+
+    Returns:
+        list: Liste de textes extraits des éléments correspondant au sélecteur.
+    """
+    soup = BeautifulSoup(html, "html.parser")
+    elements = soup.select(css_selector)
+    return [el.get_text(strip=True) for el in elements]
+
+#---------------------------------------------------------------------------#
+#------------------------------- Agent Init -------------------------------#
+#---------------------------------------------------------------------------#
+
+final_answer = FinalAnswerTool()
+
+# Utiliser un modèle gratuit
+model = HfApiModel(
+    max_tokens=1024,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-small-Instruct',
+    custom_role_conversions=None,
+)
+
+# Charger les templates de prompt
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
+# Créer l’agent avec les outils
+agent = CodeAgent(
+    model=model,
+    tools=[
+        final_answer,
+        web_search,
+        get_current_time_in_timezone,
+        visit_webpage,
+        parse_html
+    ],
+    max_steps=6,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name="InfoAgent",
+    description="Un agent capable de donner l’heure, rechercher sur internet et analyser des pages web",
+    prompt_templates=prompt_templates
+)
+
+#---------------------------------------------------------------------------#
+#------------------------------- Launch UI --------------------------------#
+#---------------------------------------------------------------------------#
+
+GradioUI(agent).launch()