Update app.py

app.py

@@ -1,5 +1,11 @@
+from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from tools.final_answer import FinalAnswerTool
 from duckduckgo_search import DDGS
-from smolagents import tool
+import datetime
+import pytz
+import yaml
+from Gradio_UI import GradioUI
+
 
 @tool
 def duckduckgo_search(query: str, max_results: int = 5) -> str:
@@ -28,4 +34,34 @@ def duckduckgo_search(query: str, max_results: int = 5) -> str:
         return "\n".join(output) if output else "No results found."
 
     except Exception as e:
-        return f"Search failed due to error: {str(e)}"
+        return f"Search failed due to error: {str(e)}"
+
+@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}"
+
+
+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
+)
+
+GradioUI(agent).launch()