Upload 4 files

requirements.txt

@@ -0,0 +1,2 @@
+yt-dlp
+gradio==4.38.1

youtube_downloader/__init__.py

@@ -0,0 +1,4 @@
+import os
+
+base_dir = os.path.dirname(os.path.abspath(__file__))
+main_dir = os.path.dirname(base_dir)

youtube_downloader/app.py

@@ -0,0 +1,80 @@
+import io
+import tempfile
+from youtube_downloader.yt_download import download_video
+import gradio as gr
+
+video_choices = ["best", "1080", "720", "360"]
+
+
+print("Setting up Gradio interface...")
+with gr.Blocks(theme=gr.themes.Soft(), title=" youtube downloader") as demo:
+    with gr.Tabs():
+        with gr.TabItem("youtube downloader"):
+            with tempfile.TemporaryDirectory() as tmpdirname:
+                with gr.Row():
+                    with gr.Column(scale=4):
+                        url_input = gr.Textbox(
+                            value="https://www.youtube.com/shorts/43BhDHYBG0o",
+                            label="🔗 Paste YouTube / Shorts URL here",
+                            placeholder="e.g., https://www.youtube.com/watch?v=.",
+                            max_lines=1,
+                        )
+                    with gr.Column(scale=3):
+                        resolution_dropdown = gr.Dropdown(
+                            choices=video_choices, value="best", label="video resolution", info="choose video resolution", interactive=True
+                        )
+
+                    with gr.Column(scale=2):
+                        download_button = gr.Button("download", variant="primary")
+
+                with gr.Row():
+                    og_video = gr.Video(
+                        visible=True,
+                        show_download_button=True,
+                        show_label=True,
+                        label="your video",
+                        format="mp4",
+                        width="50vw",
+                        height="50vw",
+                    )
+
+                @download_button.click(inputs=[url_input, resolution_dropdown], outputs=[og_video])
+                def download_this(url_input, resolution_dropdown):
+                    temporary_video_location = download_video(url_input, tmpdirname, resolution_dropdown)
+
+                    filename = open(temporary_video_location, "rb")
+                    byte_file = io.BytesIO(filename.read())
+                    with open(temporary_video_location, "wb") as out:
+                        out.write(byte_file.read())
+
+                    new_og_video = gr.Video(
+                        value=temporary_video_location,
+                        visible=True,
+                        show_download_button=True,
+                        show_label=True,
+                        label="your video",
+                        format="mp4",
+                        width="50vw",
+                        height="50vw",
+                    )
+
+                    return new_og_video
+
+        with gr.TabItem("💡 About"):
+            with gr.Blocks() as about:
+                gr.Markdown(
+                    (
+                        "### About \n"
+                        "Some notes on how this works: \n\n"
+                        "1.  **youtube / google login**: you do **not** need to be logged into a google account to use the app, with one exception: age restricted videos"
+                        "2.  **age restricted videos**: this app cannot fetch age restricted videos yet, which requires a user login to google / youtube - this feature is not yet available"
+                        "3.  **video resolution**: not all videos have all possible resolutions, so you may not be able to fetch the resolution you want for some videos (as they don't exist) \n"
+                        "4.  **recommended hardware**: this is a very light weight app, so minimum specs should work fine"
+                        "5.  **proxies**: there is an option in the yt_download module to enter proxy server ips"
+                    )
+                )
+
+
+if __name__ == "__main__":
+    print("Launching Gradio interface...")
+    demo.launch()  # allow_flagging="never"

youtube_downloader/yt_download.py

@@ -0,0 +1,63 @@
+import yt_dlp
+from yt_dlp import YoutubeDL
+import re
+
+
+def is_valid_youtube_url(url: str) -> bool:
+    if not isinstance(url, str):
+        return False
+    pattern = r"^https://www\.youtube\.com/watch\?v=[A-Za-z0-9_-]{11}$"  # youtube vido ids are always 11 chars long
+    if "shorts" in url:
+        pattern = r"^https://www\.youtube\.com/shorts/[A-Za-z0-9_-]{11}$"  # youtube vido ids are always 11 chars long
+    return re.match(pattern, url) is not None
+
+
+def download_video(url: str, savedir: str, resolution_dropdown: str, my_proxies: dict = {}) -> str:
+    try:
+        print("Downloading video from youtube...")
+        if is_valid_youtube_url(url):
+            with YoutubeDL() as ydl:
+                info_dict = ydl.extract_info(url, download=False)
+                video_url = info_dict.get("url", None)
+                video_id = info_dict.get("id", None)
+                video_title = info_dict.get("title", None)
+                if video_title is None:
+                    savepath = savedir + "/" + video_id + ".mp4"
+                else:
+                    savepath = savedir + "/" + video_title + ".mp4"
+
+            ydl_opts = {
+                "format": "bestvideo+bestaudio/best",
+                "merge_output_format": "mp4",
+                "outtmpl": savepath,
+            }
+            if resolution_dropdown == "1080":
+                ydl_opts = {
+                    "format": "bestvideo[height<=1080]+bestaudio/best",
+                    "merge_output_format": "mp4",
+                    "outtmpl": savepath,
+                }
+
+            if resolution_dropdown == "720":
+                ydl_opts = {
+                    "format": "bestvideo[height<=720]+bestaudio/best",
+                    "merge_output_format": "mp4",
+                    "outtmpl": savepath,
+                }
+
+            if resolution_dropdown == "360":
+                ydl_opts = {
+                    "format": "bestvideo[height<=360]+bestaudio/best",
+                    "merge_output_format": "mp4",
+                    "outtmpl": savepath,
+                }
+
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                ydl.download([url])
+
+            print("...done!")
+            return savepath
+        else:
+            raise ValueError(f"invalid input url: {url}")
+    except Exception as e:
+        raise ValueError(f"yt_download failed with exception {e}")