Spaces:
Sleeping
Sleeping
import yt_dlp | |
import gradio as gr | |
import os | |
# Function to download YouTube video | |
def download_video(url): | |
try: | |
# Temporary file path to save the downloaded video | |
download_folder = 'downloads' | |
if not os.path.exists(download_folder): | |
os.makedirs(download_folder) | |
# Setting the options for yt-dlp | |
ydl_opts = { | |
'format': 'best', # best available quality | |
'noplaylist': True, # avoid downloading playlists | |
'outtmpl': os.path.join(download_folder, '%(title)s.%(ext)s'), # save in 'downloads' folder | |
} | |
# Initialize yt-dlp | |
with yt_dlp.YoutubeDL(ydl_opts) as ydl: | |
# Download the video | |
info_dict = ydl.extract_info(url, download=True) | |
video_title = info_dict.get('title', 'downloaded_video') | |
video_file_path = os.path.join(download_folder, f"{video_title}.mp4") | |
# Return the path to the downloaded video so Gradio can make it available for download | |
return video_file_path | |
except Exception as e: | |
# In case of error | |
return f"An error occurred: {e}" | |
# Create Gradio interface | |
def create_interface(): | |
# Interface Layout | |
interface = gr.Interface( | |
fn=download_video, # Function to call on submit | |
inputs=gr.Textbox(label="YouTube Video URL", placeholder="Enter the video URL here", lines=1), | |
outputs=gr.File(label="Download Your Video"), # Provide a download link | |
title="YouTube Video Downloader", # Title of the Interface | |
description="Enter a YouTube video URL to download it as an MP4 with audio included.", | |
theme="compact", # Make it compact and clean | |
) | |
return interface | |
# Run Gradio Interface | |
interface = create_interface() | |
interface.launch() |