Update app.py

app.py

@@ -19,54 +19,38 @@ class AudiobookRecommendation(TypedDict):
 # 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: Optional[str] = None) -> Dict[str, AudiobookRecommendation | str]:
+def get_audiobook_recommendation(category: Optional[str] = None) -> AudiobookRecommendation:
     """
-    Tool function that recommends audiobooks based on specified category.
-    
-    Args:
-        category (Optional[str]): Genre or topic of interest. If None, returns a random recommendation.
-        
-    Returns:
-        Dict[str, AudiobookRecommendation | str]: Dictionary with 'result' (audiobook details) and 'message' (string).
+    Returns a single audiobook recommendation as a flat dictionary.
     """
     audiobook_catalog = {
         "science fiction": [
             {"title": "Dune", "author": "Frank Herbert", "narrator": "Simon Vance", "length": "21h 8m", "description": "An epic science fiction novel..."},
             # ... other books
         ],
-        # ... other categories (omitted for brevity)
+        # ... other categories
     }
 
     try:
         if category is not None:
             category = category.lower().strip()
             if category in audiobook_catalog:
-                recommendation = random.choice(audiobook_catalog[category])
-                return {
-                    "result": recommendation,
-                    "message": f"Here's a {category} audiobook recommendation for you!"
-                }
-            else:
-                all_books = [book for books in audiobook_catalog.values() for book in books]
-                recommendation = random.choice(all_books)
-                return {
-                    "result": recommendation,
-                    "message": f"Category '{category}' not found. Here's a random recommendation."
-                }
-        else:
-            random_category = random.choice(list(audiobook_catalog.keys()))
-            recommendation = random.choice(audiobook_catalog[random_category])
-            return {
-                "result": recommendation,
-                "message": f"No category specified. Here's a {random_category} recommendation!"
-            }
-    except Exception as e:
+                return random.choice(audiobook_catalog[category])
+        random_category = random.choice(list(audiobook_catalog.keys()))
+        return random.choice(audiobook_catalog[random_category])
+    except Exception:
         all_books = [book for books in audiobook_catalog.values() for book in books]
-        recommendation = random.choice(all_books)
-        return {
-            "result": recommendation,
-            "message": f"Error occurred: {str(e)}. Here's a random recommendation."
-        }
+        return random.choice(all_books)
+
+# Test with Gradio
+import gradio as gr
+
+interface = gr.Interface(
+    fn=get_audiobook_recommendation,
+    inputs=gr.Textbox(label="Category (optional)"),
+    outputs=gr.JSON(label="Recommendation"),
+    title="Audiobook Recommender"
+)
         
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: