Nymbo's picture
Update run.py
b65b420 verified
raw
history blame
1.37 kB
import gradio as gr
import yt_dlp
import os
os.system("pip yt_dlp -U")
def download_media(url, media_type):
ydl_opts = {
'format': 'bestaudio/best' if media_type == 'audio' else 'bestvideo+bestaudio',
'outtmpl': 'downloads/%(title)s.%(ext)s',
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '192',
}] if media_type == 'audio' else []
}
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info = ydl.extract_info(url, download=True)
file_path = ydl.prepare_filename(info)
if media_type == 'audio':
file_path = file_path.rsplit('.', 1)[0] + '.mp3'
return file_path
def show_media(url, media_type):
file_path = download_media(url, media_type)
if media_type == 'audio':
return None, file_path
else:
return file_path, None
with gr.Blocks(theme="Nymbo/Nymbo_Theme") as demo:
gr.Markdown(f"<h1><center> Media Downloader")
url = gr.Textbox(label="YouTube URL")
media_type = gr.Radio(label="Media Type", choices=["audio", "video"])
download_button = gr.Button("Download")
video_output = gr.Video()
audio_output = gr.Audio()
download_button.click(show_media, inputs=[url, media_type], outputs=[video_output, audio_output])
demo.launch()