Update app.py

app.py

@@ -6,12 +6,13 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 import PIL
 from PIL import Image
+from typing import Any, Optional
 
 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(arg1:str, img: bool)-> str: #it's import to specify the return type
+def my_custom_tool(arg1:str, img: bool)-> str|Any: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that uses 'DuckDuckGoSearchTool' tool to answer any query.
     Args:
@@ -20,7 +21,7 @@ def my_custom_tool(arg1:str, img: bool)-> str: #it's import to specify the retur
     """
     try:
         if img:
-            return Image.open(image_generation_tool(arg1)).convert('RGB')
+            return image_generation_tool
         else:
             search_obj = DuckDuckGoSearchTool()
             answer = search_obj(arg1)