Update app.py

app.py

@@ -51,10 +51,9 @@ def generate_image(prompt: str) -> AgentImage:
     Args:
         prompt: The text description of the image to generate.
     """
-    img = _image_generator(prompt=prompt)  # returns a PIL image in your tests
-    out_path = f"/tmp/{uuid.uuid4().hex}.png"
-    img.save(out_path)
-    return AgentImage(out_path)
+    pil_img = _image_generator(prompt=prompt)  # PIL image
+    return AgentImage(pil_img)  # ✅ wrap the PIL object, not a filepath
+
 
 
 final_answer = FinalAnswerTool()