Spaces:
Paused
Paused
File size: 1,778 Bytes
27c6f93 69339e7 c91f6d5 69339e7 27c6f93 895bc24 69339e7 e692106 64ed562 69339e7 64ed562 895bc24 69339e7 e692106 895bc24 69339e7 64ed562 69339e7 | 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 | import os
import re
import subprocess
import tempfile
import gradio as gr
FORMAT_OPTS = {
"mp3": ["-f", "ba/b", "-x", "--audio-format", "mp3"], # MP3 format
"mp4": ["-f", "bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best", "--merge-output-format", "mp4"], # MP4 video + audio
"flac": ["-f", "ba/b", "-x", "--audio-format", "flac"], # FLAC lossless audio
"aac": ["-f", "ba/b", "-x", "--audio-format", "aac"], # AAC lossy high-quality audio
}
def fetch(url, name, ext):
url = (url or "").strip()
if not url.startswith(("http://", "https://")):
raise gr.Error("Please enter a valid http(s) link.")
name = re.sub(r"[^\w\-. ]", "_", (name or "").strip()) or "download"
out_dir = tempfile.mkdtemp()
filename = os.path.join(out_dir, f"{name}.{ext}")
# Argument list (no shell) so the URL can't inject commands
cmd = ["yt-dlp", "--no-playlist", *FORMAT_OPTS[ext], "-o", filename, url]
result = subprocess.run(cmd, capture_output=True, text=True)
if result.returncode != 0 or not os.path.exists(filename):
err = (result.stderr or result.stdout or "").strip().splitlines()
raise gr.Error("Download failed: " + (err[-1] if err else "unknown error"))
return filename
# Create the Gradio interface
interface = gr.Interface(
fn=fetch,
inputs=[
gr.Textbox(label="Link Address", placeholder="Enter the URL here ..."),
gr.Textbox(label="File Name", placeholder="Example: file_name"),
gr.Dropdown(["mp3", "mp4", "flac", "aac"], value="mp3", label="File Type"),
],
outputs=gr.File(label="Download!"),
description="Download from X, Youtube, Reddit, etc..",
flagging_mode="never",
)
# Launch the interface
interface.launch(theme="Nymbo/Nymbo_Theme")
|