Update app.py

app.py

@@ -28,44 +28,56 @@ 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]]:
+def search_audiobooks(topic: str, limit: int = 3) -> List[Dict[str, Any]]:
     """
     Search Google Play Books for audiobooks by topic.
     
     Args:
-        topic (str): Genre or topic to search (e.g., "epic fantasy").
+        topic (str): Genre or topic to search (e.g., "science").
         limit (int): Maximum number of results to return (default: 3).
     
     Returns:
-        list[dict[str, any]]: List of audiobook metadata dictionaries.
+        List[Dict[str, Any]]: List of audiobook metadata dictionaries.
     """
-    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 []
+    audiobooks: List[Dict[str, Any]] = []
+    queries: List[str] = [f"{topic}+audiobook", f"subject:{topic}+audiobook"]  # Try broader query first
+    
+    for query in queries:
+        try:
+            request = books_service.volumes().list(
+                q=query,
+                maxResults=limit,
+                printType="books",
+                country="US"  # Optional: Restrict to US market
+            )
+            response: Dict[str, Any] = request.execute()
+            print(f"API response for query '{query}': {response.get('items', [])}")  # Debug output
+            
+            for item in response.get("items", []):
+                volume_info: Dict[str, Any] = item.get("volumeInfo", {})
+                categories: List[str] = volume_info.get("categories", [])
+                # Check if audiobook (based on categories or title/description)
+                if any("audiobook" in cat.lower() for cat in categories) or "audiobook" in volume_info.get("title", "").lower():
+                    audiobooks.append({
+                        "title": volume_info.get("title", "Unknown"),
+                        "author": ", ".join(volume_info.get("authors", ["Unknown"])),
+                        "categories": ", ".join(categories or ["Unknown"]),
+                        "url": volume_info.get("canonicalVolumeLink", "#"),
+                        "estimated_minutes": volume_info.get("pageCount", 180)  # Estimate: 1 page ≈ 1 minute
+                    })
+            if audiobooks:
+                break  # Exit if results found
+        except HttpError as e:
+            print(f"Google Books API error for query '{query}': {e}")
+            continue
+        except Exception as e:
+            print(f"Unexpected error in search_audiobooks for query '{query}': {e}")
+            continue
+    
+    if not audiobooks:
+        print(f"No audiobooks found for topic: {topic}")
+    return audiobooks[:limit]  # Ensure no more than limit results
+
 
 @tool
 # Function to estimate if audiobook fits time constraint