Update prompts.yaml

prompts.yaml

@@ -13,7 +13,7 @@
   ---
   Task: "Generate an image of the oldest person in this document."
 
-  Thought: I will proceed step by step and use the following tools: `document_qa` to find the oldest person in the document, then `image_generator` to generate an image according to the answer.
+  Thought: I will proceed step by step and use the following tools: `document_qa` to find the oldest person in the document, then `image_generator` to generate an image according to the answer. I must save the image to disk before calling final_answer.
   Code:
   ```py
   answer = document_qa(document=document, question="Who is the oldest person mentioned?")
@@ -21,11 +21,17 @@
   ```<end_code>
   Observation: "The oldest person in the document is John Doe, a 55 year old lumberjack living in Newfoundland."
 
-  Thought: I will now generate an image showcasing the oldest person.
+  Thought: I will now generate an image showcasing the oldest person, save it locally, and return the path.
   Code:
   ```py
-  image = image_generator("A portrait of John Doe, a 55-year-old man living in Canada.")
-  final_answer(image)
+  import os
+  os.makedirs("generated_images", exist_ok=True)
+
+  image = image_generator(prompt="A portrait of John Doe, a 55-year-old man living in Canada.")
+  path = "generated_images/john_doe.png"
+  image.save(path)
+
+  final_answer(path)
   ```<end_code>
 
   ---