Update app.py

app.py

@@ -47,37 +47,29 @@ def generate_surfer_image(next_tide: str) -> str:
 
 
 @tool
-def display_image_tool(image: Union[Image.Image, str]) -> None:
+def display_image_tool(image: str) -> None:
     """
-    Display an image in different formats.
+    Display an image from a URL or a Base64 string.
 
     Args:
-        image: The image to display. It can be a PIL Image object; a URL string (starting with "http"), a Base64-encoded image string (starting with "data:image").
+        image: The image to display. It must be: a URL string (starting with "http") or a Base64-encoded image string (starting with "data:image").
     Returns:
-        None: The function does not return anything. It displays the image in the notebook or script.
-    
+        None: The function does not return anything. It displays the image.
+
     Raises:
         ValueError: If the string format is not a valid image URL or Base64 string.
-        TypeError: If the input is not a PIL Image or a string.
     """
-    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">'))
+    if image.startswith("http"):
+        # Display image from URL
+        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.")
+    elif image.startswith("data:image"):
+        # Display Base64-encoded image
+        display.display(display.HTML(f'<img src="{image}" width="512">'))
 
     else:
-        print("Unsupported image format. Please provide a PIL image, URL, or Base64 string.")
+        raise ValueError("Invalid image format: Expected an image URL or Base64 string.")
+