Update app.py

app.py

@@ -27,33 +27,44 @@ books_service = build("books", "v1", credentials=credentials)
 
 # I want to try connect to a google API to get some recommendation of audiobooks!
 @tool
-def search_audiobooks(topic: str, limit: int = 3)-> list[dict[str, any]]: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """The tool is designed to connect to the google API platform and retrieve some audiobooks recommendations 
+def search_audiobooks(topic: str, limit: int = 3) -> List[Dict[str, Any]]:
+    """
+    Search Google Play Books for audiobooks by topic.
+    
     Args:
-        topic: the topic that I would like to use to retieve some audiobooks
-        limit: it is a constant as I want to recommend only three audiobooks
+        topic (str): Genre or topic to search (e.g., "epic fantasy").
+        limit (int): Maximum number of results to return (default: 3).
+    
+    Returns:
+        List[Dict[str, Any]]: List of audiobook metadata dictionaries.
     """
-    request = books_service.volumes().list(
-        q=f"subject:{topic}+audiobook",
-        maxResults=limit,
-        printType="books",
-        gl="us"
-    )
-    response = request.execute()
-    audiobooks = []
-    for item in response.get("items", []):
-        volume_info = item.get("volumeInfo", {})
-        # Check if audiobook (based on categories or accessInfo)
-        if "Audiobook" in volume_info.get("categories", []) or volume_info.get("accessInfo", {}).get("epub", {}).get("isAvailable", False):
-            audiobooks.append({
-                "title": volume_info.get("title", "Unknown"),
-                "author": ", ".join(volume_info.get("authors", ["Unknown"])),
-                "categories": ", ".join(volume_info.get("categories", ["Unknown"])),
-                "url": volume_info.get("canonicalVolumeLink", "#"),
-                "estimated_minutes": volume_info.get("pageCount", 180)  # Estimate: 1 page ≈ 1 minute
-            })
-    return audiobooks
+    try:
+        request = books_service.volumes().list(
+            q=f"subject:{topic}+audiobook",
+            maxResults=limit,
+            printType="books",
+            country="US"  # Optional: Restrict to US market
+        )
+        response: Dict[str, Any] = request.execute()
+        audiobooks: List[Dict[str, Any]] = []
+        for item in response.get("items", []):
+            volume_info: Dict[str, Any] = item.get("volumeInfo", {})
+            # Check if audiobook (based on categories or accessInfo)
+            if "Audiobook" in volume_info.get("categories", []) or volume_info.get("accessInfo", {}).get("epub", {}).get("isAvailable", False):
+                audiobooks.append({
+                    "title": volume_info.get("title", "Unknown"),
+                    "author": ", ".join(volume_info.get("authors", ["Unknown"])),
+                    "categories": ", ".join(volume_info.get("categories", ["Unknown"])),
+                    "url": volume_info.get("canonicalVolumeLink", "#"),
+                    "estimated_minutes": volume_info.get("pageCount", 180)  # Estimate: 1 page ≈ 1 minute
+                })
+        return audiobooks
+    except HttpError as e:
+        print(f"Google Books API error: {e}")
+        return []
+    except Exception as e:
+        print(f"Unexpected error in search_audiobooks: {e}")
+        return []
 
 @tool
 # Function to estimate if audiobook fits time constraint