Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, tool
+
 
 
 # (Keep Constants as is)
@@ -22,6 +23,43 @@ class BasicAgent:
         return fixed_answer
 
         
+@tool
+# this tool allow web search on a local SearXNG instance
+def search_web(query: str, num_results: int = 5) -> list:
+    """
+    Perform a web search using local SearXNG instance.
+    Args:
+        query (str): The search query.
+        num_results (int): The number of results to return.
+    Returns:
+        list: A list of search results sorted by score with {url, title, content, score} for each result.
+    """
+    # local metaserach engine searxng, run on localhost:8888
+    searxng_url = "http://localhost:8888/search"
+    params = {"q": query, "format": 'json'}
+    response = requests.get(searxng_url, params=params)
+    if response.status_code == 200:
+        ret = response.json()
+        # keep only the response'results' array
+        results = ret.get("results", [])
+        # keep only the first num_results   
+        results = results[:num_results]
+        # for each result keep only the url, title and content ans score
+        results = [
+            {
+                "url": result.get("url"),
+                "title": result.get("title"),
+                "content": result.get("content"),
+                "score": result.get("score"),
+            }
+            for result in results
+        ]
+        
+        return results
+        
+    else:
+        print(f"Error: {response.status_code}")
+        return []
 
 
 
@@ -49,8 +87,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     try:
         # agent = BasicAgent()
         model = InferenceClientModel()
-        search_tool = DuckDuckGoSearchTool()
-        my_small_agent = CodeAgent(tools=[search_tool], model=model)
+        my_small_agent = CodeAgent(tools=[search_web], model=model)
         
     except Exception as e:
         print(f"Error instantiating agent: {e}")