Update app.py

app.py

@@ -15,24 +15,25 @@ import time
 
 
 
-# @tool
-# def show_image(image : smolagents.agent_types.AgentImage ): #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 shows image generated by image_generation_tool  
-#     Args:
-#         image: the input image with the original type
-#     """
-#     # Assuming 'agent_image' is your smolagents.agent_types.AgentImage object
-#     if isinstance(image, Image.Image):
-#         pil_image = image
-#     elif hasattr(image, 'to_pil'):
-#         pil_image = agent_image.to_pil()
-#     elif hasattr(image, 'image') and isinstance(agent_image.image, Image.Image):
-#         pil_image = agent_image.image
-#     else:
-#         # If none of the above, you'll need to investigate further
-#         print("Could not directly convert. Check the object's attributes and methods.")
-#         # For example, print(dir(agent_image)) to see what's available
+@tool
+def show_image(image : PIL.JpegImagePlugin.JpegImageFile ): #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 shows image generated by image_generation_tool  
+    Args:
+        image: the input image with the original type
+    """
+    # Assuming 'agent_image' is your smolagents.agent_types.AgentImage object
+    if isinstance(image, Image.Image):
+        pil_image = image
+        image.show()
+    elif hasattr(image, 'to_pil'):
+        pil_image = agent_image.to_pil()
+    elif hasattr(image, 'image') and isinstance(agent_image.image, Image.Image):
+        pil_image = agent_image.image
+    else:
+        # If none of the above, you'll need to investigate further
+        print("Could not directly convert. Check the object's attributes and methods.")
+        # For example, print(dir(agent_image)) to see what's available
 
 
 @tool
@@ -209,7 +210,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=gemini_model,
-    tools=[final_answer,get_current_time_in_timezone, search_tool, web_visit, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer,get_current_time_in_timezone, search_tool, web_visit, image_generation_tool, show_image], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,