Update app.py

app.py

@@ -4,8 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
+import uuid
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -23,19 +23,18 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 @tool
 def generate_image(prompt: str) -> str:
-    """
-    Generates an image based on a text prompt.
-
-    Args:
-        prompt: A detailed description of the image to generate.
-    """
     try:
         result = image_generation_tool(prompt)
-        return f"Image successfully generated for prompt: '{prompt}'.\nResult: {result}"
+        image = result["image"]
+
+        filename = f"{uuid.uuid4()}.png"
+        image.save(filename)
+
+        return filename
+
     except Exception as e:
         return f"Error generating image: {str(e)}"
 
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.