JUNGU's picture
Update app.py
e43bdc1 verified
import gradio as gr
import yt_dlp as youtube_dl
import os
def download_youtube_video(youtube_url):
"""Downloads a YouTube video using yt_dlp and returns the downloaded file path."""
try:
ydl_opts = {
'format': 'best',
'outtmpl': '%(title)s.%(ext)s',
}
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
result = ydl.extract_info(youtube_url, download=True)
video_title = result.get('title', None)
video_ext = result.get('ext', None)
downloaded_file = f"{video_title}.{video_ext}"
return downloaded_file
except Exception as e:
print("An error occurred:", e)
return None
def app(video_link):
video_path = download_youtube_video(video_link)
if video_path:
return video_path
else:
return "An error occurred during the download."
with gr.Blocks() as interface:
gr.Markdown("## YouTube Video Downloader")
video_link = gr.Textbox(label="Enter YouTube link 🔗 To Download Video⬇️")
download_button = gr.Button("Download")
output = gr.File(label="Downloaded Video Path")
download_button.click(fn=app, inputs=video_link, outputs=output)
if __name__ == "__main__":
interface.launch(debug=True)