Update app.py

app.py

@@ -51,17 +51,21 @@ custom_role_conversions=None,
 # @tool
 # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 @tool
-def get_image_generation_tool() -> str:
-    """Loads the image generation tool for generating images from text prompts.
+def generate_image_from_text(prompt: str) -> str:
+    """Generates an image based on the given text prompt.
+    
+    Args:
+        prompt: A string describing the image to be generated.
     
     Returns:
-        A string confirming the tool has been loaded successfully.
+        A string indicating the success or failure of the image generation process.
     """
     try:
         image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-        return "Image generation tool has been successfully loaded."
+        image_url = image_generation_tool(prompt)
+        return f"Image successfully generated: {image_url}"
     except Exception as e:
-        return f"Error loading image generation tool: {str(e)}"
+        return f"Error generating image: {str(e)}"
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)