tools approach

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 
 import re
 import requests
-from duckduckgo_search import DDGS
+
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -14,41 +14,51 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-
-
-class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
-        self.searcher = DDGS()
-
-    def web_search(self, query: str, max_results: int = 3) -> str:
-        """Perform a DuckDuckGo search and return snippets."""
+class MathTool:
+    def run(self, query):
         try:
-            results = list(self.searcher.text(query, max_results=max_results))
-            if not results:
-                return "No search results found."
-            # Concatenate top snippets
-            snippets = [r.get("body", "") for r in results if "body" in r]
-            return " ".join(snippets[:3])
-        except Exception as e:
-            return f"Search error: {e}"
-
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+            return str(eval(query))
+        except Exception:
+            return "Math error."
+
+class GoogleSearchTool:
+    def __init__(self, api_key, engine_id):
+        self.api_key = api_key
+        self.engine_id = engine_id
+    
+    def run(self, query):
+        url = "https://www.googleapis.com/customsearch/v1"
+        params = {"key": self.api_key, "cx": self.engine_id, "q": query}
+        r = requests.get(url, params=params).json()
+        items = r.get("items", [])
+        return " ".join([item["snippet"] for item in items[:3]])
+
+class WikipediaTool:
+    def run(self, query):
+        url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}"
+        r = requests.get(url)
+        if r.status_code == 200:
+            return r.json().get("extract", "")
+        return "No wiki entry found."
 
-        q = question.strip()
-
-        # If it looks like a math problem, handle locally
-        if re.search(r"\d+\s*[\+\-\*\/]\s*\d+", q):
-            try:
-                return str(eval(q))
-            except Exception:
-                pass
-
-        # Otherwise: do a web search
-        print("Delegating to web search...")
-        snippets = self.web_search(q, max_results=3)
-        return snippets if snippets else "I could not find an answer."
+class BasicAgent:
+    def __init__(self, tools):
+        self.tools = tools
+    
+    def __call__(self, question: str):
+        print(f"Reasoning on: {question[:50]}...")
+        
+        # Decide which tool to use
+        if re.search(r"\d+\s*[\+\-\*\/]\s*\d+", question):
+            return self.tools["math"].run(question)
+        
+        if "who" in question.lower() or "what" in question.lower():
+            wiki_ans = self.tools["wiki"].run(question)
+            if wiki_ans:
+                return wiki_ans
+        
+        # Default: search
+        return self.tools["google"].run(question)
         
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """