Update app.py

app.py

@@ -8,35 +8,48 @@ from Gradio_UI import GradioUI
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetch current time in a timezone"""
+    """
+    Get the current local time in a specified timezone.
+
+    Args:
+        timezone (str): A valid timezone string such as 'Asia/Kolkata' or 'America/New_York'.
+
+    Returns:
+        str: The current local time in the specified timezone.
+    """
     try:
         tz = pytz.timezone(timezone)
         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:
-        return str(e)
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+# Final answer tool (mandatory for agent)
 final_answer = FinalAnswerTool()
 
 
+# LLM model for agent reasoning
 model = HfApiModel(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     max_tokens=2096,
-    temperature=0.5
+    temperature=0.5,
 )
 
 
+# Image generation tool from Hugging Face Agents course
 image_generation_tool = load_tool(
     "agents-course/text-to-image",
     trust_remote_code=True
 )
 
 
+# Load prompt templates
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
 
+# Create agent
 agent = CodeAgent(
     model=model,
     tools=[
@@ -47,8 +60,13 @@ agent = CodeAgent(
     ],
     max_steps=6,
     verbosity_level=2,
+    grammar=None,
+    planning_interval=None,
+    name="MultiTool Agent",
+    description="Agent that can search the web, generate images, and fetch timezone information.",
     prompt_templates=prompt_templates
 )
 
 
+# Launch Gradio UI
 GradioUI(agent).launch()