Update app.py

app.py

@@ -11,8 +11,13 @@ from typing import Optional
 from Gradio_UI import GradioUI
 
 @tool
-def my_custom_tool(animal_type: Optional[str] = None, breed: Optional[str] = None):
-    """Finds animals and breeds from Brampton Animal Shelter website."""
+def get_pets_available_for_adoption(animal_type: Optional[str] = None, breed: Optional[str] = None):
+    """A tool that finds the specified animal and breed from the Brampton Animal Shelter website.
+    
+    Args:
+        animal_type: The type of animal to search for (e.g., "dog", "cat", "bird", "other"). If None, returns all types.
+        breed: The specific breed or species name to filter by. If None, returns all breeds.
+    """
     
     base_url = "https://www.brampton.ca/EN/residents/Animal-Services/Pages/Adoption.aspx"
     
@@ -129,7 +134,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_pets_available_for_adoption, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,