File size: 3,445 Bytes
9312825 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import gradio as gr
from yt_dlp import YoutubeDL
import os
import tempfile
import shutil
from pathlib import Path
def download_for_browser(url, mode='audio', quality='high'):
if not url:
return None, "Please enter a valid URL"
# Create temporary directory that persists until file is served
temp_dir = Path(tempfile.mkdtemp())
try:
# Configure download options
opts = {
'format': 'bestaudio/best' if mode == 'audio' else 'bestvideo+bestaudio/best',
'outtmpl': str(temp_dir / '%(title)s.%(ext)s'),
'restrictfilenames': True,
'windowsfilenames': True,
'quiet': True,
'no_warnings': True
}
# Add format-specific options
if mode == 'audio':
opts.update({
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '320' if quality == 'high' else '192',
}],
'prefer_ffmpeg': True,
'keepvideo': False
})
else:
opts.update({
'format': 'bestvideo+bestaudio/best' if quality == 'high' else 'best[height<=720]',
'merge_output_format': 'mp4'
})
# Download file
with YoutubeDL(opts) as ydl:
info = ydl.extract_info(url, download=True)
# Find downloaded file
files = list(temp_dir.glob('*'))
if not files:
return None, "Download failed - no files found"
download_file = files[0]
if not download_file.exists():
return None, f"File not found after download: {download_file.name}"
# Return file while it exists in temp directory
return str(download_file), f"Successfully converted: {download_file.name}"
except Exception as e:
if temp_dir.exists():
shutil.rmtree(temp_dir)
return None, f"Error during download: {str(e)}"
def create_browser_ui():
with gr.Blocks(title="YouTube Downloader", theme=gr.themes.Soft()) as demo:
gr.Markdown("## YouTube Downloader")
with gr.Row():
url_input = gr.Textbox(
label="YouTube URL",
placeholder="https://youtube.com/watch?v=...",
scale=2
)
with gr.Row():
with gr.Column(scale=1):
mode_input = gr.Radio(
choices=["audio", "video"],
value="audio",
label="Format"
)
quality_input = gr.Radio(
choices=["high", "medium"],
value="high",
label="Quality"
)
download_button = gr.Button("Convert", variant="primary")
with gr.Column(scale=2):
status_text = gr.Textbox(label="Converting Status", interactive=False)
output_file = gr.File(label="Download to your device ...")
download_button.click(
fn=download_for_browser,
inputs=[url_input, mode_input, quality_input],
outputs=[output_file, status_text]
)
return demo
demo = create_browser_ui()
demo.launch(
share=False,
debug=True,
show_error=True
) |