Spaces:
Running
Running
File size: 1,857 Bytes
73ce86c 61941bf 73ce86c 61941bf 73ce86c 61941bf 73ce86c 61941bf 73ce86c 61941bf 73ce86c 61941bf 73ce86c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import gradio as gr
import yt_dlp
def download_media(url, download_video):
if download_video:
ydl_opts = {
'format': 'bestvideo+bestaudio/best',
'outtmpl': 'downloads/%(title)s.%(ext)s',
}
else:
ydl_opts = {
'format': 'bestaudio/best',
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '192',
}],
'outtmpl': 'downloads/%(title)s.%(ext)s',
}
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info_dict = ydl.extract_info(url, download=True)
file_title = ydl.prepare_filename(info_dict)
if download_video:
output_file = file_title
else:
output_file = file_title.rsplit('.', 1)[0] + '.mp3'
return output_file, download_video
def get_output_component(download_video):
if download_video:
return gr.File(label="Downloaded Media")
else:
return gr.Audio(label="Downloaded Media")
# Create the Gradio interface
with gr.Blocks() as demo:
gr.Markdown("# YouTube Media Downloader")
with gr.Row():
url_input = gr.Textbox(label="YouTube URL")
download_video_checkbox = gr.Checkbox(label="Download Video", value=False)
with gr.Row():
download_button = gr.Button("Download")
with gr.Row():
dynamic_output = gr.State()
def update_output_component(download_video):
return get_output_component(download_video)
dynamic_output = download_video_checkbox.change(
update_output_component, inputs=download_video_checkbox, outputs=dynamic_output
)
download_button.click(download_media, inputs=[url_input, download_video_checkbox], outputs=[dynamic_output, download_video_checkbox])
demo.launch()
|