Updated my image generator function tool

app.py

@@ -4,12 +4,13 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from PIL import Image
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(image_description:str)-> None: #it's import to specify the return type
+def generate_image_by_description(image_description:str)-> Image.Image: #it's import to specify the return type
     """A tool that generates an image based on the image_description variable
     Args:
         image_description: the description of the image to be generated
@@ -20,9 +21,9 @@ def my_custom_tool(image_description:str)-> None: #it's import to specify the re
 
     image = image_generation_tool(image_description)
 
-    image.show()
+    # image.show()
 
-    # return "What magic will you build ?"
+    return image
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -58,7 +59,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_image_by_description], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,