Spaces:
Sleeping
Sleeping
File size: 3,664 Bytes
50d6eaf |
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 |
import gradio as gr
import yt_dlp
import os
import time
import threading
def download_video(url, resolution):
# Define resolution options
resolutions = {
"2160p": "bestvideo[height<=2160]+bestaudio/best",
"1440p": "bestvideo[height<=1440]+bestaudio/best",
"1080p": "bestvideo[height<=1080]+bestaudio/best",
"720p": "bestvideo[height<=720]+bestaudio/best",
"480p": "bestvideo[height<=480]+bestaudio/best",
}
try:
# Set yt-dlp options
ydl_opts = {
'format': resolutions[resolution],
'outtmpl': '%(title)s.%(ext)s',
'noplaylist': True,
'merge_output_format': 'mp4',
'postprocessors': [{
'key': 'FFmpegVideoConvertor',
'preferedformat': 'mp4', # Ensure output is in MP4 format
}],
}
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info_dict = ydl.extract_info(url, download=True)
filename = ydl.prepare_filename(info_dict)
# Check if the file already exists
if os.path.exists(filename):
os.remove(filename) # Remove the existing file
ydl.download([url])
yield filename # Yield the filename to display the video
time.sleep(10) # Wait for 20 seconds before deleting the file
if os.path.exists(filename):
os.remove(filename) # Remove the file after 20 seconds
except Exception as e:
yield None
print("An error occurred:", e)
def download_audio(url):
try:
# Set yt-dlp options for audio download
ydl_opts = {
'format': 'bestaudio/best',
'outtmpl': '%(title)s.%(ext)s',
'noplaylist': True,
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
}],
}
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info_dict = ydl.extract_info(url, download=True)
filename = ydl.prepare_filename(info_dict).replace('.webm', '.mp3').replace('.m4a', '.mp3')
# Check if the file already exists
if os.path.exists(filename):
os.remove(filename) # Remove the existing file
ydl.download([url])
yield filename # Yield the filename to display the audio
time.sleep(20) # Wait for 20 seconds before deleting the file
if os.path.exists(filename):
os.remove(filename) # Remove the file after 20 seconds
except Exception as e:
yield None
print("An error occurred:", e)
# Create Gradio interface
with gr.Blocks() as iface:
with gr.Tab("MP4 Downloader"):
gr.Markdown("### Download YouTube videos as MP4")
url_input_mp4 = gr.Textbox(label="YouTube URL", placeholder="Enter YouTube URL here")
resolution_input = gr.Dropdown(label="Resolution", choices=["2160p", "1440p", "1080p", "720p", "480p"], value="1080p")
download_button_mp4 = gr.Button("Download MP4")
output_mp4 = gr.Video(label="Downloaded Video")
download_button_mp4.click(download_video, inputs=[url_input_mp4, resolution_input], outputs=output_mp4)
with gr.Tab("MP3 Downloader"):
gr.Markdown("### Download YouTube videos as MP3")
url_input_mp3 = gr.Textbox(label="YouTube URL", placeholder="Enter YouTube URL here")
download_button_mp3 = gr.Button("Download MP3")
output_mp3 = gr.Audio(label="Downloaded Audio")
download_button_mp3.click(download_audio, inputs=[url_input_mp3], outputs=output_mp3)
iface.launch()
|