- A more safe approach

app.py

@@ -8,26 +8,26 @@ import numpy as np
 from Gradio_UI import GradioUI
 from bs4 import BeautifulSoup
 import random
+from PIL import Image
+from io import BytesIO
+from typing import Annotated
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_animal_image(animal_name:str)-> np.ndarray: #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 gets a random animal image from the internet and returns an image as numpy array
+def get_animal_image(animal_name: Annotated[str, "The animal name to search for."]) -> Annotated[np.ndarray, "A numpy array of the image."]:
+    """Returns a random animal image from an online image API as a numpy array.
     Args:
-        animal_name: the animal name
+        animal_name: The animal name to search for.
     """
-    # search for the animal name on the internet in images search
-    search_url = f"https://www.google.com/search?q={animal_name}&tbm=isch"
-    #get the image from the internet
-    response = requests.get(search_url)
-    #parse the html
-    soup = BeautifulSoup(response.text, 'html.parser')
-    # select a random image from the search results
-    image_url = random.choice(soup.find_all('img'))['src']
-    #get the image from the internet
-    response =   requests.get(image_url)
-    #return the image as numpy array
-    return np.array(response.content)
+    # Use a real image API like Unsplash or Wikimedia
+    url = f"https://source.unsplash.com/600x400/?{animal_name}"
+    
+    response = requests.get(url)
+    if response.status_code != 200:
+        raise RuntimeError("Failed to retrieve image.")
+    
+    image = Image.open(BytesIO(response.content)).convert("RGB")
+    return np.array(image)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: