File size: 1,837 Bytes
77f0587 eb94b9d 4d0fc4b 77f0587 4d0fc4b 77f0587 4d0fc4b 77f0587 4d0fc4b 77f0587 4d0fc4b 77f0587 4d0fc4b 77f0587 4d0fc4b 77f0587 4d0fc4b 77f0587 4d0fc4b |
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 |
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': 'bestaudio/best',
'format': 'm4a/bestaudio/best',
# 'outtmpl': '%(id)s.%(ext)s',
'outtmpl': f'{path_to_folder_audio_mp3}%(title)s',
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
# 'preferredquality': '192',
}]
}
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info_dict = ydl.extract_info(url, download=True)
video_title = info_dict['title']
# ydl.download([video_link])
# 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
# def video_to_audio(url):
# video_url = download_video(url)
# video = mp.VideoFileClip(video_url)
# audio = video.audio
# audio_file = "output_audio.mp3"
# audio.write_audiofile(audio_file)
# return audio_file
iface = gr.Interface(fn=download_audio,
inputs=gr.Textbox(label="YouTube Video URL"),
outputs=gr.File(label="Output Audio"),
allow_flagging="never"
)
iface.launch(debug=True)
|