Update app.py

app.py

@@ -10,7 +10,7 @@ from Gradio_UI import GradioUI
 # 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: str) -> str:
+def get_audiobook_recommendation(category=None):
     """
     Tool function that recommends audiobooks based on specified category.
     
@@ -122,36 +122,48 @@ def get_audiobook_recommendation(category: str) -> str:
         ]
     }
     
-    # Normalize category input if provided
-    if 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]
+    try:
+        # Normalize category input if provided
+        if category is not None:
+            if not isinstance(category, str):
+                category = str(category)
             
-            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)
-                return {
-                    "result": recommendation,
-                    "message": f"No exact match for '{category}'. Here's a random recommendation instead."
-                }
-    else:
-        # If no category specified, choose random category
-        category = random.choice(list(audiobook_catalog.keys()))
-    
-    # Select a random book from the category
-    recommendation = random.choice(audiobook_catalog[category])
-    
-    return {
-        "result": recommendation,
-        "message": f"Here's a {category} audiobook recommendation for you!"
-    }
+            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]
+                
+                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)
+                    return {
+                        "result": recommendation,
+                        "message": f"No exact match for '{category}'. Here's a random recommendation instead."
+                    }
+        else:
+            # If no category specified, choose random category
+            category = random.choice(list(audiobook_catalog.keys()))
+        
+        # Select a random book from the category
+        recommendation = random.choice(audiobook_catalog[category])
+        
+        return {
+            "result": recommendation,
+            "message": f"Here's a {category} audiobook recommendation for you!"
+        }
+    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)
+        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: