Update app.py

app.py

@@ -6,24 +6,25 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+from typing import Dict, Union, List, Optional, Any
+import random
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 # The next step should be connect directly to a spotify API for example or Audible API, but let's keep it simple for now.
 @tool
-def get_audiobook_recommendation(category=None):
+def get_audiobook_recommendation(category: Optional[str] = None) -> Dict[str, Any]:
     """
     Tool function that recommends audiobooks based on specified category.
     
     Args:
-        category (str, optional): Genre or topic of interest. If None, returns a random recommendation.
+        category (Optional[str]): Genre or topic of interest. If None, returns a random recommendation.
         
     Returns:
-        dict: Audiobook recommendation with title, author, and description
+        Dict[str, Any]: Audiobook recommendation with title, author, and description
     """
-    import random
-    
     # Audiobook catalog organized by category
-    audiobook_catalog = {
+    audiobook_catalog: Dict[str, List[Dict[str, str]]] = {
         "science fiction": [
             {
                 "title": "Dune",
@@ -125,22 +126,25 @@ def get_audiobook_recommendation(category=None):
     try:
         # Normalize category input if provided
         if category is not None:
-            if not isinstance(category, str):
-                category = str(category)
-            
             category = category.lower().strip()
             
             # Check if category exists in catalog
             if category not in audiobook_catalog:
                 # Try partial matching
-                matching_categories = [cat for cat in audiobook_catalog.keys() if category in cat]
+                matching_categories: List[str] = [
+                    cat for cat in audiobook_catalog.keys() 
+                    if category in cat
+                ]
                 
                 if matching_categories:
                     category = matching_categories[0]
                 else:
                     # Return message if no matching category
-                    all_books = [book for booklist in audiobook_catalog.values() for book in booklist]
-                    recommendation = random.choice(all_books)
+                    all_books: List[Dict[str, str]] = [
+                        book for booklist in audiobook_catalog.values() 
+                        for book in booklist
+                    ]
+                    recommendation: Dict[str, str] = random.choice(all_books)
                     return {
                         "result": recommendation,
                         "message": f"No exact match for '{category}'. Here's a random recommendation instead."
@@ -150,7 +154,7 @@ def get_audiobook_recommendation(category=None):
             category = random.choice(list(audiobook_catalog.keys()))
         
         # Select a random book from the category
-        recommendation = random.choice(audiobook_catalog[category])
+        recommendation: Dict[str, str] = random.choice(audiobook_catalog[category])
         
         return {
             "result": recommendation,
@@ -158,13 +162,16 @@ def get_audiobook_recommendation(category=None):
         }
     except Exception as e:
         # Provide a fallback recommendation if anything goes wrong
-        all_books = [book for booklist in audiobook_catalog.values() for book in booklist]
-        recommendation = random.choice(all_books)
+        all_books: List[Dict[str, str]] = [
+            book for booklist in audiobook_catalog.values() 
+            for book in booklist
+        ]
+        recommendation: Dict[str, str] = random.choice(all_books)
         return {
             "result": recommendation,
             "message": f"Encountered an error: {str(e)}. Here's a random recommendation instead."
         }
-
+        
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.