Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 import pytz
 import yaml
 from PIL import Image
+import os
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -55,10 +56,11 @@ def generate_travel_brochure_image(description: str) -> AgentImage:
     Args:
         description: A string representing text to be converted into a brochure.
     """
-    image = image_generation_tool(description)
-    if isinstance(image, Image.Image):
+    image_path = image_generation_tool(description)
+    if isinstance(image_path, str) and os.path.exists(image_path):
+        image = Image.open(image_path)
         return AgentImage(image)
-    raise ValueError("Image generation failed or returned invalid data.")
+    raise ValueError("Image generation failed or returned invalid path.")
 
 
 
@@ -81,10 +83,10 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     tools=[
+        final_answer,
         get_current_time_in_timezone,
         travel_recommendations_for_city,
-        generate_travel_brochure_image,
-        final_answer
+        generate_travel_brochure_image
     ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,