Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -371,15 +371,15 @@ async def download_track(track_id: str):
|
|
371 |
title = track_data["title"]
|
372 |
artist = track_data["artist"]
|
373 |
query = f'{title}+{artist}'
|
374 |
-
|
375 |
-
|
376 |
-
|
377 |
-
|
378 |
-
|
379 |
-
|
380 |
-
|
381 |
-
|
382 |
-
|
383 |
else:
|
384 |
yt_url = track_data['url']
|
385 |
d_data = get_download_url(yt_url)
|
|
|
371 |
title = track_data["title"]
|
372 |
artist = track_data["artist"]
|
373 |
query = f'{title}+{artist}'
|
374 |
+
logger.info(f"search query: {query}")
|
375 |
+
search_results = ytmusic.search(query, filter="songs")
|
376 |
+
first_song = next((song for song in search_results if 'videoId' in song and song['videoId']), {}) if search_results else {}
|
377 |
+
if 'videoId' in first_song:
|
378 |
+
videoId = first_song["videoId"]
|
379 |
+
ym_url = f'https://www.youtube.com/watch?v={videoId}'
|
380 |
+
d_data = get_download_url(ym_url)
|
381 |
+
track_data['download_url'] = d_data
|
382 |
+
return track_data
|
383 |
else:
|
384 |
yt_url = track_data['url']
|
385 |
d_data = get_download_url(yt_url)
|