Spaces:
Running
Running
fix ytdlp
Browse files- src/main.py +10 -9
src/main.py
CHANGED
@@ -63,21 +63,22 @@ def get_youtube_video_id(url, ignore_playlist=True): # this code unused here!
|
|
63 |
def yt_download(link):
|
64 |
ydl_opts = {
|
65 |
'format': 'bestaudio',
|
66 |
-
'outtmpl': '%(title)s',
|
67 |
-
|
68 |
-
'ignoreerrors': True,
|
69 |
-
'no_warnings': True,
|
70 |
-
'quiet': True,
|
71 |
'extractaudio': True,
|
72 |
'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'mp3'}],
|
73 |
}
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
|
|
|
|
|
|
78 |
return download_path
|
79 |
|
80 |
|
|
|
81 |
def raise_exception(error_msg, is_webui):
|
82 |
if is_webui:
|
83 |
raise gr.Error(error_msg)
|
|
|
63 |
def yt_download(link):
|
64 |
ydl_opts = {
|
65 |
'format': 'bestaudio',
|
66 |
+
'outtmpl': '%(title)s.%(ext)s',
|
67 |
+
|
|
|
|
|
|
|
68 |
'extractaudio': True,
|
69 |
'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'mp3'}],
|
70 |
}
|
71 |
+
try:
|
72 |
+
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
73 |
+
result = ydl.extract_info(link, download=True)
|
74 |
+
download_path = ydl.prepare_filename(result, outtmpl='%(title)s.mp3')
|
75 |
+
except Exception as e:
|
76 |
+
raise Exception(f"Error downloading video: {e}")
|
77 |
+
|
78 |
return download_path
|
79 |
|
80 |
|
81 |
+
|
82 |
def raise_exception(error_msg, is_webui):
|
83 |
if is_webui:
|
84 |
raise gr.Error(error_msg)
|