Update app.py

app.py

@@ -45,6 +45,24 @@ def search_audiobooks(topic:str, limit:int=3)-> str: #it's import to specify the
             })
     return audiobooks
 
+@tool
+# Function to estimate if audiobook fits time constraint
+def filter_by_time(audiobooks, free_time):
+    """The tool is designed to match the time in terms of duration of the audiobook which is called estimated_minutes and the free time that the person have 
+    Args:
+        audiobooks: the list of the audiobooks
+        free_time: The free time the person inform
+    """
+    try:
+        if "hour" in free_time.lower():
+            hours = float(free_time.split()[0])
+            max_minutes = hours * 60
+        else:
+            max_minutes = float(free_time.split()[0])
+    except:
+        max_minutes = 180  # Default to 3 hours
+    return [book for book in audiobooks if book["estimated_minutes"] <= max_minutes]
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -82,7 +100,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,search_audiobooks,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer,search_audiobooks,get_current_time_in_timezone,filter_by_time], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,