Spaces:
Sleeping
Sleeping
File size: 2,438 Bytes
29ded09 53b8e30 3b852a1 53b8e30 0014839 53b8e30 0014839 3b852a1 0014839 3b852a1 d1c57b1 0014839 90be0a7 0014839 90be0a7 3b852a1 29ded09 0014839 |
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 62 63 64 65 66 67 |
import gradio as gr
import yt_dlp
import os
os.system('yt-dlp -U')
def download_media(url, output_format):
ydl_opts = {}
if output_format == "Audio":
ydl_opts = {
'format': 'bestaudio/best',
'outtmpl': 'downloads/%(title)s.%(ext)s',
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '192',
}],
}
elif output_format == "Video":
ydl_opts = {
'format': 'bestvideo+bestaudio/best',
'outtmpl': 'downloads/%(title)s.%(ext)s',
}
try:
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info_dict = ydl.extract_info(url, download=True)
title = info_dict.get('title', 'Unknown')
ext = info_dict.get('ext', 'mp4' if output_format == "Video" else 'mp3')
file_path = f'downloads/{title}.{ext}'
return file_path, f"Download successful: {title}.{ext}"
except Exception as e:
return None, f"Error: {str(e)}"
# Gradio Blocks UI
with gr.Blocks() as demo:
gr.Markdown("# Video and Audio Downloader using yt-dlp")
with gr.Row():
url_input = gr.Textbox(label="Video/Audio URL")
format_dropdown = gr.Dropdown(choices=["Audio", "Video"], label="Output Format")
download_button = gr.Button("Download")
output_text = gr.Textbox(label="Output Message")
# Placeholder for media output
audio_output = gr.Audio(label="Audio Output", visible=False)
video_output = gr.Video(label="Video Output", visible=False)
def handle_download(url, output_format):
file_path, message = download_media(url, output_format)
if file_path and output_format == "Audio":
return file_path, None, audio_output.update(visible=True), video_output.update(visible=False), message
elif file_path and output_format == "Video":
return None, file_path, audio_output.update(visible=False), video_output.update(visible=True), message
else:
return None, None, audio_output.update(visible=False), video_output.update(visible=False), message
download_button.click(handle_download,
inputs=[url_input, format_dropdown],
outputs=[audio_output, video_output, audio_output, video_output, output_text])
demo.launch()
|