Update app.py

added display_image

app.py

@@ -5,7 +5,7 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 import cv2
-
+from PIL import Image
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -35,6 +35,21 @@ def gaussian_blur(image_path: str, output_path: str, kernel_size: int, sigma: in
     
     return output_path
 
+@tool
+def display_image(image_path: str) -> None:
+    """
+    Opens and displays an image using Pillow.
+
+    Args:
+        image_path: The file path of the image to be displayed.
+    """
+    try:
+        image = Image.open(image_path)
+        image.show()
+    except Exception as e:
+        raise ValueError(f"Error opening image: {e}")
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -72,7 +87,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[image_generation_tool, get_current_time_in_timezone, gaussian_blur, DuckDuckGoSearchTool(), final_answer], ## add your tools here (don't remove final answer)
+    tools=[image_generation_tool, get_current_time_in_timezone, gaussian_blur, display_image, DuckDuckGoSearchTool(), final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,