Update app.py

app.py

@@ -43,6 +43,21 @@ def mutante_unicorn(animal_name:str) -> str: #it's import to specify the return
     except Exception as e:
         return f"Error creating a mutant unicorn for '{animal_name}': {str(e)}"
 
+def describe_animal(animal_name:str) -> str:
+    """Look up for the name of the animal inwikipedia using websearch tool, and then provide a 300 characters description about that animal
+    Args:
+        animal_name: The name of the animal to look up for in wikipedia
+    """
+    #search_tool = DuckDuckGoSearchTool()
+    # Use DuckDuckGoSearchTool to search for the wikipedia url of the person and then visit the url
+    query = f"{animal_name} wikipedia"
+    #wikipedia_url = search_tool(query)
+    # Use the wikipedia url to get the person's description, location, and interesting facts
+    wikipedia_results = DuckDuckGoSearchTool(query)
+    #return f"Detailes description: {wikipedia_results['description']}, Location: {wikipedia_results['location']}, Interesting facts: {wikipedia_results['interesting facts']}"
+    return wikipedia_results
+
+
 #image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 final_answer = FinalAnswerTool()
@@ -62,7 +77,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, mutante_unicorn],  # add your tools here (don't remove final answer)
+    tools=[final_answer, mutante_unicorn, describe_animal],  # add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,