Upload app.py

app.py

@@ -1,15 +1,12 @@
+
 import os
 import gradio as gr
 import requests
 import pandas as pd
-#from smolagents import Agent, tool
 from duckduckgo_search import DDGS
 from transformers import pipeline
 from smolagents import tool
 
-# TOOL 1: Wikipedia-based Search Tool
-from duckduckgo_search import DDGS
-
 @tool
 def web_search(query: str) -> str:
     """
@@ -31,7 +28,6 @@ def web_search(query: str) -> str:
                     return f"{snippet}\n\nSource: [{r['href']}]({r['href']})"
         return "Could not find a direct answer from Wikipedia."
 
-# TOOL 2: Citation Tool
 @tool
 def cite(input: str) -> str:
     """
@@ -49,7 +45,6 @@ def cite(input: str) -> str:
     except:
         return "Could not format citation."
 
-# TOOL 3: Math Eval
 @tool
 def python(code: str) -> str:
     """
@@ -67,7 +62,6 @@ def python(code: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-# TOOL 4: Fallback
 @tool
 def fallback(_: str) -> str:
     """
@@ -81,27 +75,18 @@ def fallback(_: str) -> str:
     """
     return "Sorry, I couldn't confidently answer that. Could you rephrase?"
 
-# MAIN AGENT
 class BasicAgent:
     def __call__(self, question: str) -> str:
         q = question.lower()
 
         try:
-            # Wikipedia-focused factual lookups
-            if any(x in q for x in ["who", "what", "when", "where", "how many", "how much", "did", "which", "name"]):
-                return web_search(question)
-
-            # Inline citation formatting
-            elif "|||" in q:
+            if "|||" in question:
                 return cite(question)
-
-            # Math evaluation
-            elif any(op in q for op in ["+", "-", "*", "/"]):
+            if any(op in q for op in ["+", "-", "*", "/"]) and any(c.isdigit() for c in q):
                 return python(question)
-
-            # Default catch-all
-            return fallback(question)
-
+            if len(q.split()) < 3:
+                return fallback(question)
+            return web_search(question)
         except Exception as e:
             return f"Agent error: {str(e)}"