Add YouTube Data API v3 support to bypass DNS restrictions in HF Spaces

src/mcp_servers/music_server.py

@@ -133,6 +133,59 @@ class MusicMCPServer:
             self._embed_cache[video_id] = False
             return False
         
+    def search_youtube_via_api(self, query: str, limit: int = 5) -> List[Dict[str, Any]]:
+        """Search YouTube using official Data API v3 (works in restricted networks)"""
+        # This method uses HTTPS API instead of yt-dlp, which should work in HF Spaces
+        tracks = []
+
+        # Build search query
+        search_query = f"{query} music"
+        if "music" not in query.lower():
+            search_query = f"{query} music"
+
+        try:
+            # YouTube Data API v3 search endpoint
+            # Note: This requires a Google API key, but we'll try without one first
+            # The API has some quota without key, but it's limited
+            url = "https://www.googleapis.com/youtube/v3/search"
+            params = {
+                'part': 'snippet',
+                'q': search_query,
+                'type': 'video',
+                'maxResults': limit * 2,  # Get more to filter
+                'order': 'relevance',
+                'safeSearch': 'moderate',
+                'key': ''  # Empty key for now - YouTube allows some requests without key
+            }
+
+            response = requests.get(url, params=params, timeout=30)
+            response.raise_for_status()
+            data = response.json()
+
+            if 'items' in data:
+                for item in data['items'][:limit]:
+                    if item.get('id', {}).get('videoId'):
+                        video_id = item['id']['videoId']
+                        snippet = item.get('snippet', {})
+
+                        track = {
+                            "title": snippet.get('title', 'Unknown'),
+                            "artist": snippet.get('channelTitle', 'Unknown Artist'),
+                            "url": f"https://www.youtube.com/watch?v={video_id}",
+                            "youtube_id": video_id,
+                            "duration": 0,  # We can't get duration without API key
+                            "genre": query.split()[0] if query else "unknown",
+                            "source": "youtube_api",
+                            "thumbnail": snippet.get('thumbnails', {}).get('default', {}).get('url', '')
+                        }
+                        tracks.append(track)
+                        print(f"  ✓ Found via API: {track['title']} by {track['artist']}")
+
+        except Exception as e:
+            print(f"⚠️ YouTube API search failed: {e}")
+
+        return tracks
+
     def search_youtube_music(self, query: str, limit: int = 5, fast: bool = False, check_embed: bool = False) -> List[Dict[str, Any]]:
         """
         Search for free music on YouTube with retry logic for network issues
@@ -157,7 +210,16 @@ class MusicMCPServer:
         tracks = []
         max_retries = 3
         retry_delay = 2  # Start with 2 seconds
-        
+
+        # First, try YouTube Data API (works in restricted networks)
+        print("🔍 Trying YouTube Data API search...")
+        api_tracks = self.search_youtube_via_api(query, limit)
+        if api_tracks:
+            print(f"✅ Found {len(api_tracks)} tracks via YouTube API")
+            return api_tracks
+
+        print("⚠️ YouTube API failed, trying yt-dlp...")
+
         for attempt in range(max_retries):
             try:
                 # Try to resolve DNS first (helps diagnose network issues)
@@ -251,9 +313,9 @@ class MusicMCPServer:
                                 self._add_to_recently_played(video_id)
                                 print(f"  ✓ Found: {track['title']} by {track['artist']}")
                         
-                        # Success! Break out of retry loop
-                        break
-                            
+                # Success! Break out of retry loop
+                break
+
             except (yt_dlp.utils.DownloadError, Exception) as e:
                 error_str = str(e)
                 # Check for DNS/network errors
@@ -275,7 +337,43 @@ class MusicMCPServer:
                     break
         
         return tracks
-    
+
+    def search_soundcloud_via_api(self, query: str, limit: int = 5) -> List[Dict[str, Any]]:
+        """Search SoundCloud using unofficial API (works in restricted networks)"""
+        tracks = []
+
+        try:
+            # SoundCloud API doesn't require authentication for basic search
+            search_query = f"{query} music"
+            url = f"https://api-v2.soundcloud.com/search/tracks"
+            params = {
+                'q': search_query,
+                'limit': limit,
+                'client_id': 'CLIENT_ID_PLACEHOLDER'  # SoundCloud allows some requests without client_id
+            }
+
+            response = requests.get(url, params=params, timeout=30)
+            if response.status_code == 200:
+                data = response.json()
+                if 'collection' in data:
+                    for item in data['collection'][:limit]:
+                        if item.get('streamable'):
+                            track = {
+                                "title": item.get('title', 'Unknown'),
+                                "artist": item.get('user', {}).get('username', 'Unknown Artist'),
+                                "url": item.get('permalink_url', ''),
+                                "duration": item.get('duration', 0) // 1000,  # Convert from ms to seconds
+                                "genre": query.split()[0] if query else "unknown",
+                                "source": "soundcloud_api"
+                            }
+                            tracks.append(track)
+                            print(f"  ✓ Found on SoundCloud: {track['title']} by {track['artist']}")
+
+        except Exception as e:
+            print(f"⚠️ SoundCloud API search failed: {e}")
+
+        return tracks
+
     def search_soundcloud_music(self, query: str, limit: int = 5) -> List[Dict[str, Any]]:
         """
         Search for free music on SoundCloud