Update app.py

app.py

@@ -9,7 +9,7 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_pokemon_data(pokemon_name:str)-> str:
+def get_pokemon_type(pokemon_name:str)-> str:
     """A tool that retrieves data about a pokemon 
     Args:
         pokemon_name: the pokemon name
@@ -17,7 +17,7 @@ def get_pokemon_data(pokemon_name:str)-> str:
     try:
         response = requests.get(f"https://pokeapi.co/api/v2/pokemon/{pokemon_name}")
         pokemon_data = response.json()
-        return pokemon_data
+        return pokemon_data['types'][0]['type']['name']
     except Exception as e:
         return f"Error fetching pokemon data for {pokemon_name}"