Redirect to DuckDuckGo as a tool

app.py

@@ -4,41 +4,51 @@ import requests
 import inspect
 import pandas as pd
 
+import re
+import requests
+from duckduckgo_search import DDGS
+
 # (Keep Constants as is)
 # --- Constants ---
 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."""
+        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:
-        # Agent's reasoning process (thoughts)
-        thoughts = self.reason(question)
-        final_answer = self.extract_answer(thoughts)
-        response = (
-            f"THOUGHTS: {thoughts}\n"
-            f"FINAL ANSWER: {final_answer}"
-        )
-        print(f"Agent response:\n{response[:150]}...")
-        return response
-
-    def reason(self, question: str) -> str:
-        # Put chain-of-thought reasoning here,
-        # e.g., use LLM or heuristics to break down the problem.
-        # Example placeholder for demonstration:
-        return "Analyzing the problem step by step, considering all constraints, and isolating the answer format. Selected the most direct solution."
-
-    def extract_answer(self, thoughts: str) -> str:
-        # Normally, use selected answer from reasoning.
-        # For now, this placeholder returns demo result.
-        # Replace demo logic with answer extraction, format as per rules:
-        # - if a number is needed, just plain digits (no commas or units unless specified)
-        # - if string, no articles or abbreviations, digits written out unless specified
-        # - if list, elements comma separated, each following corresponding rules
-        return "42"
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+
+        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."
         
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """