Update app.py

app.py

@@ -7,72 +7,39 @@ from duckduckgo_search import DDGS
 from transformers import pipeline
 
 # --- Tool Definitions ---
-@tool
-class WebSearchTool:
-    name = "web_search"
-    description = "Search the web for up-to-date factual information."
-    def use(self, query: str) -> str:
-        with DDGS() as ddgs:
-            results = ddgs.text(query)
-            output = [f"{r['title']} - {r['href']}" for r in results[:3]]
-            return "\n".join(output) if output else "No relevant results found."
-
-@tool
-class CiteTool:
-    name = "cite"
-    description = "Add citation to a given answer with a valid URL."
-    def use(self, input: str) -> str:
-        try:
-            answer, url = input.split("|||")
-            return f"{answer.strip()}\n\nSource: [{url.strip()}]({url.strip()})"
-        except:
-            return "Could not format citation correctly."
-
-summarizer = pipeline("summarization")
-@tool
-class SummarizerTool:
-    name = "summarize"
-    description = "Summarize a long text into a short paragraph."
-    def use(self, input: str) -> str:
-        if len(input) < 50:
-            return input
-        result = summarizer(input, max_length=100, min_length=25, do_sample=False)
-        return result[0]['summary_text']
-
-@tool
-class PythonTool:
-    name = "python"
-    description = "Execute Python code to solve math problems."
-    def use(self, code: str) -> str:
-        try:
-            result = str(eval(code, {"__builtins__": {}}))
-            return f"Answer: {result}"
-        except Exception as e:
-            return f"Error: {str(e)}"
+from tools.search_tool import WebSearchTool
+from tools.citation_tool import CiteTool
+from tools.summarizer_tool import SummarizerTool
+from tools.math_tool import PythonTool
+from tools.fallback_tool import FallbackTool
 
-@tool
-class FallbackTool:
-    name = "fallback"
-    description = "Handle unanswerable or unclear queries."
-    def use(self, _: str) -> str:
-        return "I'm sorry, I couldn't find the answer to your question. Could you rephrase or try something else?"
-
-# --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self):
-        tools = [WebSearchTool(), CiteTool(), SummarizerTool(), PythonTool(), FallbackTool()]
-        self.agent = Agent(
-            tools=tools,
-            system_prompt="""
-You are Smart Answering Agent v3.
-Answer questions factually, concisely, and cite sources when available.
-Route to the correct tool for factual, math, or summarization queries.
-If you don’t know the answer, respond gracefully using the fallback tool.
-Ensure output format is friendly for the GAIA evaluation.
-"""
-        )
+        self.search = WebSearchTool()
+        self.cite = CiteTool()
+        self.summarizer = SummarizerTool()
+        self.math = PythonTool()
+        self.fallback = FallbackTool()
+
     def __call__(self, question: str) -> str:
-        return self.agent.run(question)
+        question_lower = question.lower()
+
+        try:
+            if any(keyword in question_lower for keyword in ["latest", "news", "who", "what", "where", "when"]):
+                return self.search.use(question)
+            elif any(keyword in question_lower for keyword in ["summarize", "summary", "explain"]):
+                return self.summarizer.use(question)
+            elif any(char.isdigit() for char in question) and any(op in question for op in ["+", "-", "*", "/"]):
+                return self.math.use(question)
+            elif "|||" in question:
+                return self.cite.use(question)
+            else:
+                return self.fallback.use(question)
+        except Exception as e:
+            return f"Error during processing: {str(e)}"
+
+
+
 
 # --- Evaluation Logic ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"