Update app.py

app.py

@@ -49,9 +49,6 @@ custom_role_conversions=None,
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-print("Image tool name:", image_generation_tool.name)
-print("Image tool:", image_generation_tool)
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
@@ -67,12 +64,17 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-print("Agent tools (raw):", agent.tools)
-print("Agent tools (types):", [type(t) for t in agent.tools])
+img = image_generation_tool(prompt="A cute cat with prominent moustaches, photorealistic, high-resolution")
+
+print("Returned type:", type(img))
+print("Returned value:", img)
 
-for i, t in enumerate(agent.tools):
-    name = getattr(t, "name", None)
-    print(f"Tool {i}: type={type(t)} name={name} value={t}")
+# If it's a PIL image, this will work:
+try:
+    img.save("/tmp/test.png")
+    print("Saved to /tmp/test.png")
+except Exception as e:
+    print("Could not save as PIL:", e)
 
 
 GradioUI(agent).launch()