Upload 2 files

app.py

@@ -0,0 +1,52 @@
+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()

requirements.txt

@@ -0,0 +1,3 @@
+pytube
+gradio
+validators