Update

app.py

@@ -6,6 +6,8 @@ import datetime
 # import requests
 import pytz
 import yaml
+from Gradio_UI import GradioUI
+from loguru import logger
 from smolagents import (
     CodeAgent,
     # DuckDuckGoSearchTool,
@@ -13,7 +15,6 @@ from smolagents import (
     load_tool,
     tool,
 )
-from Gradio_UI import GradioUI
 from tools.final_answer import FinalAnswerTool
 
 
@@ -41,6 +42,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         timezone: A string representing a valid timezone (e.g., 'America/New_York').
 
     """
+    logger.debug(f" get_current_time_in_timezone {timezone=}")
     try:
         # Create timezone object
         tz = pytz.timezone(timezone)
@@ -48,6 +50,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
+        logger.error(e)
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
@@ -68,7 +71,11 @@ with open("prompts.yaml", "r", encoding="utf8") as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[my_cutom_tool, get_current_time_in_timezone, final_answer],  ## add your tools here (don't remove final answer)
+    tools=[
+        my_cutom_tool,
+        get_current_time_in_timezone,
+        final_answer,
+    ],  ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -78,5 +85,11 @@ agent = CodeAgent(
     prompt_templates=prompt_templates,
 )
 
-
-GradioUI(agent).launch()
+debug = True
+share = True
+ssr = True
+GradioUI(agent).launch(
+    debug=debug,
+    share=share,
+    ssr=ssr,
+)

requirements.txt

@@ -3,3 +3,5 @@ smolagents
 requests
 duckduckgo_search
 pandas
+
+loguru