Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, HfApiModel, tool
 import requests
 import base64
 import datetime
@@ -16,7 +16,10 @@ def generate_image(prompt: str) -> str:
     Generate an image from a text prompt.
 
     Args:
-        prompt: description of the image to generate
+        prompt: A description of the image to generate.
+    
+    Returns:
+        A base64-encoded string of the generated image.
     """
     API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
 
@@ -44,7 +47,15 @@ def generate_image(prompt: str) -> str:
 # -------------------------
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetches the current local time in a specified timezone."""
+    """
+    Fetches the current local time in a specified timezone.
+
+    Args:
+        timezone: A string representing a valid timezone, e.g., 'America/New_York'
+    
+    Returns:
+        A string with the current time in the specified timezone.
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -82,7 +93,7 @@ with open("prompts.yaml", 'r') as stream:
 # -------------------------
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, generate_image],
+    tools=[final_answer, get_current_time_in_timezone, generate_image],  # keep final_answer first
     max_steps=6,
     verbosity_level=1,
     name="FirstAgent",