Update app.py

app.py

@@ -45,6 +45,39 @@ def generate_surfer_image(next_tide: str) -> str:
         return f"Error generating image: {str(e)}"
 
 
+@tool
+def display_image_tool(image):
+    """
+    A tool that displays an image correctly in different formats.
+
+    Args:
+        image: Can be a PIL image, a URL (string), or a Base64-encoded image.
+    
+    Returns:
+        None. Displays the image in the notebook or script.
+    """
+    if isinstance(image, Image.Image):
+        # If it's a PIL Image, display it directly
+        display.display(image)
+
+    elif isinstance(image, str):
+        if image.startswith("http"):  
+            # If it's a URL, display it as an HTML image tag
+            display.display(display.HTML(f'<img src="{image}" width="512">'))
+
+        elif image.startswith("data:image"):
+            # If it's a Base64-encoded image, display it directly
+            display.display(display.HTML(f'<img src="{image}" width="512">'))
+
+        else:
+            print("Invalid string format: Expected an image URL or Base64 string.")
+
+    else:
+        print("Unsupported image format. Please provide a PIL image, URL, or Base64 string.")
+
+
+
+
 @tool
 def tide_time(date: str) -> str:
     """ A tool that checks when the time of the next high tide in Falesia beach in Algarve using DuckDuckGo search.
@@ -133,7 +166,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_current_time_in_timezone,get_tide_time,generate_surfer_image,final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_current_time_in_timezone,get_tide_time,generate_surfer_image,display_image_tool,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,