Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from smolagents import Tool, CodeAgent
 import requests
+import yaml
 
 # ---------------------------
 # TOOLS
@@ -8,45 +9,38 @@ import requests
 
 @Tool
 def web_search(query: str, max_results: int = 5) -> str:
-    """
-    Perform a web search.
-
-    Args:
-        query (str): The search query.
-        max_results (int): Maximum number of results to return.
-
-    Returns:
-        str: A string with the search results.
-    """
-    # Dummy search implementation (replace with real API call if needed)
+    """Perform a web search (dummy implementation)."""
     return f"Search results for '{query}' (top {max_results} results)."
 
 @Tool
 def visit_webpage(url: str) -> str:
-    """
-    Visit a webpage and return its HTML content.
-
-    Args:
-        url (str): The webpage URL.
-
-    Returns:
-        str: The HTML text of the page.
-    """
+    """Visit a webpage and return its HTML content."""
     try:
         response = requests.get(url, timeout=5)
-        return response.text[:2000]  # limit output size
+        return response.text[:2000]
     except Exception as e:
         return f"Error visiting {url}: {e}"
 
+# ---------------------------
+# LOAD CONFIG.YAML
+# ---------------------------
+
+with open("config.yaml", "r") as f:
+    config = yaml.safe_load(f)
+
+system_prompt = config.get("system_prompt", "")
+prompt_templates = config.get("prompt_templates", {})
+
 # ---------------------------
 # AGENT SETUP
 # ---------------------------
 
 agent = CodeAgent(
-    model="google/flan-t5-small",  # modèle gratuit
+    model="google/flan-t5-small",
     tools=[web_search, visit_webpage],
     max_steps=6,
-    prompt_config_path="config.yaml",  # le yaml corrigé
+    system_prompt=system_prompt,
+    prompt_templates=prompt_templates,
 )
 
 # ---------------------------