Create app.py

app.py

@@ -0,0 +1,69 @@
+import gradio as gr
+from pytube import YouTube
+import os
+import time
+import random
+
+def download_video(url, output_path='.'):
+    max_retries = 5
+    retries = 0
+    backoff_factor = 1
+
+    while retries < max_retries:
+        try:
+            # Create a YouTube object with a custom user-agent
+            yt = YouTube(url, use_oauth=False, allow_oauth_cache=True, on_progress_callback=on_progress)
+
+            # Get the highest resolution stream
+            video = yt.streams.get_highest_resolution()
+
+            # Download the video
+            print(f"Downloading: {yt.title}")
+            video.download(output_path=output_path)
+            print(f"Download completed: {yt.title}")
+            return f"Download completed: {yt.title}"
+
+        except Exception as e:
+            retries += 1
+            if retries == max_retries:
+                return f"An error occurred after {max_retries} retries: {e}"
+            else:
+                delay = backoff_factor * (2 ** (retries - 1)) + random.uniform(0, 1)
+                print(f"Retrying in {delay:.2f} seconds... (Attempt {retries}/{max_retries})")
+                time.sleep(delay)
+
+def on_progress(stream, chunk, bytes_remaining):
+    total_size = stream.filesize
+    bytes_downloaded = total_size - bytes_remaining
+    percentage_of_completion = bytes_downloaded / total_size * 100
+    print(f"Download progress: {percentage_of_completion:.2f}%")
+
+def validate_url(url):
+    try:
+        # Check if the URL is a valid YouTube video URL
+        yt = YouTube(url)
+        return True
+    except Exception as e:
+        return False
+
+def gradio_interface(url, output_directory):
+    if not validate_url(url):
+        return "Invalid YouTube URL"
+
+    result = download_video(url, output_directory)
+    return result
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=[
+        gr.inputs.Textbox(label="Enter the YouTube video URL"),
+        gr.inputs.Textbox(label="Enter the output directory (leave blank for current directory)")
+    ],
+    outputs=gr.outputs.Textbox(label="Download Status"),
+    title="YouTube Video Downloader",
+    description="Enter a YouTube video URL and optionally specify an output directory to download the video."
+)
+
+# Launch the Gradio interface
+iface.launch()