whisper-webui / download.py
aadnk's picture
Add support for downloading files from YouTube using yt-dlp
8d120bf
raw
history blame
974 Bytes
import os
from tempfile import mkdtemp
from yt_dlp import YoutubeDL
from yt_dlp.postprocessor import PostProcessor
class FilenameCollectorPP(PostProcessor):
def __init__(self):
super(FilenameCollectorPP, self).__init__(None)
self.filenames = []
def run(self, information):
self.filenames.append(information["filepath"])
return [], information
def downloadUrl(url: str):
destinationDirectory = mkdtemp()
ydl_opts = {
"format": "bestaudio/best",
'playlist_items': '1',
'paths': {
'home': destinationDirectory
}
}
filename_collector = FilenameCollectorPP()
with YoutubeDL(ydl_opts) as ydl:
ydl.add_post_processor(filename_collector)
ydl.download([url])
if len(filename_collector.filenames) <= 0:
raise Exception("Cannot download " + url)
result = filename_collector.filenames[0]
print("Downloaded " + result)
return result