Update app.py

app.py

@@ -41,20 +41,21 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
         
-# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 @tool 
-def text_to_image(text: str) -> image:
-    """A tool that takes string and creates image with textual information.
+def text_to_image(text: str) -> str:
+    """
+    A tool that takes a string and creates an image with textual information.
     Args: 
-        text: A string representing the textual information about the image to be created.
+        text (str): A string representing the textual information about the image to be created.
+    Returns:
+        str: Image URL or path
     """
-    try :
+    try:
         return image_generation_tool(text)
-
     except Exception as e:
-        return "Error creating the image"    
+        return f"Error creating the image: {str(e)}"
 
 final_answer = FinalAnswerTool()