File size: 3,509 Bytes
f7b3397
5806219
da8f353
25c7df7
5806219
6c7ed02
4b1708b
6c7ed02
 
 
 
5806219
 
 
 
 
 
6c7ed02
5806219
1d6256c
5806219
 
6c7ed02
 
 
 
 
 
 
da8f353
ec6f947
 
 
bea4793
ec6f947
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5806219
9d38abe
ec6f947
 
 
 
 
 
 
ba69b7e
 
ec6f947
 
1e47cdd
0a1013e
5806219
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
import gradio as gr
import yt_dlp
import os

def downloader(video_url, audio_format, audio_name):
    # Ensure the directory exists
    os.makedirs('audios', exist_ok=True)
    
    # Use a temporary placeholder for the output file
    temp_output_path = f"audios/{audio_name}.%(ext)s"
    
    ydl_opts = {
        'format': 'bestaudio/best',
        'postprocessors': [{
            'key': 'FFmpegExtractAudio',
            'preferredcodec': audio_format,
        }],
        'outtmpl': temp_output_path,
    }

    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
        ydl.download([video_url])
    
    # Find the downloaded file and rename it
    temp_file = temp_output_path.replace('%(ext)s', audio_format)
    final_output_path = f"audios/{audio_name}.{audio_format}"
    os.rename(temp_file, final_output_path)
    
    return final_output_path


    

    logs_chart = """
    ### Changelog

    #### v1.1.0 - 2024-05-16

    ##### Added
    - **Directory Check**: Added a check to ensure the `audios` directory exists before attempting to save files to it.
      - `os.makedirs('audios', exist_ok=True)`

    ##### Changed
    - **Output Template Placeholder**: Updated `outtmpl` to use a temporary placeholder for the file extension.
      - `outtmpl: f"audios/{audio_name}.%(ext)s"`
    - **File Renaming**: Added logic to rename the temporary output file to the final desired name and extension.
      - `temp_file = temp_output_path.replace('%(ext)s', audio_format)`
      - `final_output_path = f"audios/{audio_name}.{audio_format}`
      - `os.rename(temp_file, final_output_path)`
    - **Return Correct Path**: The `downloader` function now returns the final output path to ensure Gradio can find and load the audio file correctly.
      - `return final_output_path`

    ##### Fixed
    - **File Not Found Error**: Resolved the issue where the file could not be found due to incorrect output path handling.

    ### Summary of Changes

    1. **Ensured Directory Exists**: Added code to create the `audios` directory if it doesn't exist to prevent file saving errors.
    2. **Output Path Handling**: Modified the output template to use a placeholder for the file extension, which yt-dlp will replace with the actual extension upon downloading.
    3. **File Renaming Logic**: After downloading the audio, the code now renames the file to match the desired audio name and format.
    4. **Correct File Path Return**: The correct file path is now returned from the `downloader` function, ensuring that the Gradio `gr.Audio` component can properly display and play the downloaded audio file.

    These changes collectively ensure that the audio file is downloaded correctly, renamed appropriately, and made accessible to the Gradio interface for playback.
    """


with gr.Blocks() as demo:
    gr.Markdown("# YouTube Downloader 2.0")
    with gr.Row():
        video_url = gr.Textbox(label="YouTube video link", default="https://youtu.be/yQBGdXGCUbA?si=7avvqH-6OOkGWqFm")
        audio_name = gr.Textbox(label="Audio name of YouTube audio", default="killshot")
        audio_format = gr.Radio(["wav", "flac", "mp3"], label="Select the output format", default=" wav")
    with gr.Row():
        output = gr.Audio(label="Output")
    with gr.Row():
        download_button = gr.Button("Download")
        download_button.click(downloader, inputs=[video_url, audio_format, audio_name], outputs=output)
    with gr.Group():
        with gr.Row():
            gr.Markdown(logs_chart)

demo.launch()