Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+from PIL import Image
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -57,7 +58,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 @tool
-def generate_travel_brochure_image(description: str) -> str:
+def generate_travel_brochure_image(description: str) -> Image.Image:
     """A tool that generates a travel brochure image based on a description.
     Args:
         description: A string representing text to be converted into a brochure.
@@ -85,10 +86,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
+        generate_travel_brochure_image,
+        final_answer
     ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,