File size: 2,486 Bytes
cb8512f
 
73ce86c
 
 
 
 
 
 
130ca0b
73ce86c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
114d786
73ce86c
61941bf
 
82223a5
61941bf
 
 
 
10c5e6b
45a77eb
 
61941bf
73ce86c
 
b1596a8
 
2c706dc
 
 
 
 
61941bf
114d786
 
 
 
 
 
be5866e
61941bf
114d786
 
 
 
61941bf
 
73ce86c
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
print("please wait...")

import gradio as gr
import yt_dlp

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

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

    return output_file

def get_output_component(download_video):
    if download_video:
        return gr.Video(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 form youtube url </div>")
    
    with gr.Row():
        url_input = gr.Textbox(label="YouTube URL")
    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)
    gr.Markdown(f"### <div style='text-align: center;'>made with ❤ by <a href='https://huggingface.co/Gradio-Blocks'>Gradio-Blocks-Party-Member</a></div>")
    
    download_button.click(
        handle_download, 
        inputs=[url_input, download_video_checkbox], 
        outputs=[output_file, output_audio]
    )

demo.launch()