Update app.py

app.py

@@ -1,53 +1,31 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
-from tools.final_answer import FinalAnswerTool
 from duckduckgo_search import DDGS
-import datetime
-import pytz
-import yaml
-from Gradio_UI import GradioUI
-
+from smolagents import tool
 
 @tool
 def duckduckgo_search(query: str, max_results: int = 5) -> str:
-    """Search the web using DuckDuckGo."""
-    with DDGS() as ddgs:
-        results = ddgs.text(query, max_results=max_results)
-        output = []
-        for i, r in enumerate(results, 1):
-            output.append(
-                f"{i}. {r.get('title')}\n"
-                f"{r.get('href')}\n"
-                f"{r.get('body')}\n"
-            )
-        return "\n".join(output) if output else "No results found."
-
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """Get current time in a timezone."""
-    tz = pytz.timezone(timezone)
-    local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-    return f"The current time in {timezone} is {local_time}"
-
+    """
+    Perform a web search using DuckDuckGo.
+
+    Args:
+        query (str): The search query string.
+        max_results (int): The maximum number of search results to return.
+
+    Returns:
+        str: A formatted string containing search results.
+    """
+    try:
+        with DDGS() as ddgs:
+            results = ddgs.text(query, max_results=max_results)
+
+            output = []
+            for i, r in enumerate(results, 1):
+                output.append(
+                    f"{i}. {r.get('title')}\n"
+                    f"{r.get('href')}\n"
+                    f"{r.get('body')}\n"
+                )
 
-final_answer = FinalAnswerTool()
-
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-)
-
-with open("prompts.yaml", "r") as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-
-agent = CodeAgent(
-    model=model,
-    tools=[duckduckgo_search, get_current_time_in_timezone, final_answer],
-    max_steps=6,
-    verbosity_level=1,
-    prompt_templates=prompt_templates
-)
+        return "\n".join(output) if output else "No results found."
 
-GradioUI(agent).launch()
+    except Exception as e:
+        return f"Search failed due to error: {str(e)}"