Update app.py

app.py

@@ -48,8 +48,20 @@ custom_role_conversions=None,
 
 
 # Import tool from Hub
+# @tool
+# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 @tool
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+def get_image_generation_tool() -> str:
+    """Loads the image generation tool for generating images from text prompts.
+    
+    Returns:
+        A string confirming the tool has been loaded successfully.
+    """
+    try:
+        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+        return "Image generation tool has been successfully loaded."
+    except Exception as e:
+        return f"Error loading image generation tool: {str(e)}"
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)