| import gradio as gr | |
| from pytube import YouTube | |
| import validators | |
| import pathlib | |
| import glob | |
| def download_video(url): | |
| with gr.spinner("Downloading..."): | |
| local_file = ( | |
| YouTube(url) | |
| .streams.filter(progressive=True, file_extension="mp4") | |
| .first() | |
| .download() | |
| ) | |
| gr.success("Downloaded") | |
| return local_file | |
| def validate_url(url): | |
| if not validators.url(url): | |
| gr.error("Hi there 👋 URL seems invalid 👽") | |
| gr.stop() | |
| def cleanup(): | |
| junks = glob.glob("*.mp4") | |
| for junk in junks: | |
| pathlib.Path(junk).unlink() | |
| def helper_message(): | |
| gr.write( | |
| "> To save the video to the local computer, " | |
| "click the vertical ... icon (aka hamburger button) in the bottom-right corner (in the video above) and click download." | |
| ) | |
| def youtube_downloader(url): | |
| gr.header("YouTube Video Downloader") | |
| url = gr.text_input("Enter YouTube URL to download:") | |
| start_button = gr.button("Start Download") | |
| if start_button: | |
| if url: | |
| validate_url(url) | |
| with gr.expander("preview video"): | |
| gr.video(url) | |
| cleanup() | |
| file_ = download_video(url) | |
| gr.video(file_) | |
| helper_message() | |
| gr.markdown("YouTube Video Download Help") | |
| iface = gr.Interface(fn=youtube_downloader, live=True) | |
| iface.launch() | |