Update app.py

app.py

@@ -5,19 +5,17 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Define the tool correctly using @tool decorator
+# Define the tool correctly with a properly formatted docstring
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
     Fetches the current local time in a specified timezone.
-
-    Parameters:
-        timezone (str): The name of the timezone in "Continent/City" format. 
-                        Example: "Asia/Kolkata", "America/New_York".
-
+    
+    Args:
+        timezone (str): The timezone string (e.g., 'America/New_York', 'Asia/Kolkata').
+        
     Returns:
-        str: The current time in the given timezone in 'YYYY-MM-DD HH:MM:SS' format, 
-             or an error message if the timezone is invalid.
+        str: The current local time in the specified timezone.
     """
     try:
         tz = pytz.timezone(timezone)
@@ -47,7 +45,12 @@ with open("prompts.yaml", 'r') as stream:
 # Create the Agent with properly defined tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, DuckDuckGoSearchTool(), get_current_time_in_timezone, image_generation_tool],  
+    tools=[
+        final_answer, 
+        DuckDuckGoSearchTool(), 
+        get_current_time_in_timezone,  # Now properly formatted
+        image_generation_tool
+    ],  
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -58,6 +61,4 @@ agent = CodeAgent(
 )
 
 # Launch UI
-GradioUI(agent).launch()
-
-
+GradioUI(agent).launch()