- Hopefully this fixes it

app.py

@@ -6,20 +6,27 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 import numpy as np
 from Gradio_UI import GradioUI
-
+from bs4 import BeautifulSoup
+import random
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_animal_image(animal_name:str)-> str: #it's import to specify the return type
+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
     Args:
         animal_name: the animal name
     """
+    # 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
-    image_url = f"https://www.google.com/search?q={animal_name}&tbm=isch"
-    #download the image
-    response = requests.get(image_url)
-    #return the image as base64 string
+    response =   requests.get(image_url)
+    #return the image as numpy array
     return np.array(response.content)
 
 @tool