app.py

print("please wait...")

import gradio as gr
import yt_dlp

def download_media(url, download_video):
    ydl_opts = {
        'format': 'bestvideo+bestaudio/best' if download_video else 'bestaudio/best',
        'outtmpl': 'downloads/%(title)s.%(ext)s',
        'postprocessors': [{
            'key': 'FFmpegExtractAudio',
            'preferredcodec': 'mp3',
            'preferredquality': '192',
        }] if not download_video else [],
        'noplaylist': True,
        'quiet': True,
        'progress_hooks': [progress_hook],
    }

    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
        info_dict = ydl.extract_info(url, download=True)
        file_title = ydl.prepare_filename(info_dict)
        
        if not download_video:
            file_title = file_title.rsplit('.', 1)[0] + '.mp3'

    return file_title

def progress_hook(d):
    if d['status'] == 'downloading':
        print(f"Downloading: {d['_percent_str']} of {d['_total_bytes_str']}")
    elif d['status'] == 'finished':
        print("Download complete, now converting...")

def get_output_component(download_video):
    if download_video:
        return gr.File(label="Downloaded Media")
    else:
        return gr.Audio(label="Downloaded Media")

# Create the Gradio interface
with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange")) as demo:
    gr.Markdown(f"# <div style='text-align: center;'>YOUTUBE Downloader</div>")
    gr.Markdown(f"## <div style='text-align: center;'>Download MP3/MP4 from YouTube URL</div>")
    
    with gr.Row():
        url_input = gr.Textbox(label="YouTube URL", placeholder="Enter YouTube video URL here")
    with gr.Row():
        download_video_checkbox = gr.Checkbox(label="Download Video", value=False)
    with gr.Row():    
        download_button = gr.Button("Download")
    with gr.Row():
        output_audio = gr.Audio(label="Downloaded Media", visible=False)
        output_file = gr.File(label="Downloaded Media", visible=False)

    def handle_download(url, download_video):
        output_file = download_media(url, download_video)
        if download_video:
            return gr.update(value=output_file, visible=True), gr.update(visible=False)
        else:
            return gr.update(visible=False), gr.update(value=output_file, visible=True)

    download_button.click(
        handle_download, 
        inputs=[url_input, download_video_checkbox], 
        outputs=[output_file, output_audio]
    )

    gr.Markdown(f"### <div style='text-align: center;'>Made with ❤ by <a href='https://huggingface.co/Gradio-Blocks'>Gradio-Blocks-Party-Member</a></div>")

demo.launch()