File size: 1,472 Bytes
77f0587
eb94b9d
4d0fc4b
77f0587
4d0fc4b
 
 
 
 
 
 
77f0587
 
4d0fc4b
77f0587
 
4d0fc4b
77f0587
 
 
 
 
 
 
4d0fc4b
 
 
 
 
 
 
 
 
 
 
87dc9c9
77f0587
 
87dc9c9
77f0587
 
87dc9c9
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
import gradio as gr
import re, unidecode
from unidecode import unidecode
import yt_dlp
import os

# no space, punctuation, accent in lower string
def cleanString(string):
    cleanString = unidecode(string)
    cleanString = re.sub('\W+','_', cleanString)
    return cleanString.lower()

def download_audio(url):
    path_to_folder_audio_mp3 = "./audio/"
    ydl_opts = {
        'format': 'm4a/bestaudio/best',
        'outtmpl': f'{path_to_folder_audio_mp3}%(title)s',
        'postprocessors': [{
            'key': 'FFmpegExtractAudio',
            'preferredcodec': 'mp3',
        }]
    }
    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
        info_dict = ydl.extract_info(url, download=True)
        video_title = info_dict['title']

        # Rename the audio file
        local_link = video_title + ".mp3"
        new_local_link = cleanString(video_title) + ".mp3"
        for filename in os.listdir(path_to_folder_audio_mp3):
            if cleanString(local_link) == cleanString(filename):
                os.rename(os.path.join(path_to_folder_audio_mp3, filename),os.path.join(path_to_folder_audio_mp3, new_local_link)) 

    return path_to_folder_audio_mp3 + new_local_link 

# Gradio interface
iface = gr.Interface(fn=download_audio, 
                     inputs=gr.Textbox(label="YouTube Video URL"),
                     outputs=gr.Audio(label="Output Audio", type="filepath"),
                     allow_flagging="never"
                     )
iface.launch()