Update app.py

app.py

@@ -4,19 +4,157 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
+# to use spotify API
+import spotipy
+from spotipy.oauth2 import SpotifyClientCredentials
+
 
 # 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 my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def get_audiobook_recommendation(category=None):
+    """
+    Tool function that recommends audiobooks based on specified category.
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        category (str, optional): Genre or topic of interest. If None, returns a random recommendation.
+        
+    Returns:
+        dict: Audiobook recommendation with title, author, and description
     """
-    return "What magic will you build ?"
+    import random
+    
+    # Audiobook catalog organized by category
+    audiobook_catalog = {
+        "science fiction": [
+            {
+                "title": "Dune",
+                "author": "Frank Herbert",
+                "narrator": "Simon Vance",
+                "length": "21h 8m",
+                "description": "An epic science fiction novel about politics, religion, and ecology on a desert planet."
+            },
+            {
+                "title": "Project Hail Mary",
+                "author": "Andy Weir",
+                "narrator": "Ray Porter",
+                "length": "16h 10m",
+                "description": "A lone astronaut must save humanity from an extinction-level threat using his scientific ingenuity."
+            }
+        ],
+        "fantasy": [
+            {
+                "title": "The Name of the Wind",
+                "author": "Patrick Rothfuss",
+                "narrator": "Nick Podehl",
+                "length": "27h 55m",
+                "description": "First book in the Kingkiller Chronicle, following the magical journey of Kvothe."
+            },
+            {
+                "title": "The Hobbit",
+                "author": "J.R.R. Tolkien",
+                "narrator": "Andy Serkis",
+                "length": "10h 25m",
+                "description": "Bilbo Baggins' unexpected adventure with dwarves and a dragon that changed Middle-earth."
+            }
+        ],
+        "mystery": [
+            {
+                "title": "The Thursday Murder Club",
+                "author": "Richard Osman",
+                "narrator": "Lesley Manville",
+                "length": "12h 25m",
+                "description": "Four septuagenarians meet weekly to solve cold cases, until a live murder occurs right on their doorstep."
+            },
+            {
+                "title": "The Silent Patient",
+                "author": "Alex Michaelides",
+                "narrator": "Jack Hawkins",
+                "length": "8h 43m",
+                "description": "A psychological thriller about a woman who shoots her husband and then never speaks again."
+            }
+        ],
+        "business": [
+            {
+                "title": "Never Split the Difference",
+                "author": "Chris Voss",
+                "narrator": "Michael Kramer",
+                "length": "8h 7m",
+                "description": "Negotiation tactics from a former FBI hostage negotiator."
+            },
+            {
+                "title": "Atomic Habits",
+                "author": "James Clear",
+                "narrator": "James Clear",
+                "length": "5h 35m",
+                "description": "A practical guide to building good habits and breaking bad ones."
+            }
+        ],
+        "history": [
+            {
+                "title": "Sapiens: A Brief History of Humankind",
+                "author": "Yuval Noah Harari",
+                "narrator": "Derek Perkins",
+                "length": "15h 18m",
+                "description": "An exploration of human history from ancient humans to the present day."
+            },
+            {
+                "title": "The Splendid and the Vile",
+                "author": "Erik Larson",
+                "narrator": "John Lee",
+                "length": "17h 49m",
+                "description": "Chronicles Winston Churchill's first year as Prime Minister during the Blitz."
+            }
+        ],
+        "self-improvement": [
+            {
+                "title": "Thinking, Fast and Slow",
+                "author": "Daniel Kahneman",
+                "narrator": "Patrick Egan",
+                "length": "20h 2m",
+                "description": "Explores the two systems that drive how we think and make decisions."
+            },
+            {
+                "title": "The Power of Now",
+                "author": "Eckhart Tolle",
+                "narrator": "Eckhart Tolle",
+                "length": "7h 37m",
+                "description": "A guide to spiritual enlightenment and living in the present moment."
+            }
+        ]
+    }
+    
+    # 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]
+            
+            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!"
+    }
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: