Spaces:
Sleeping
Sleeping
import gradio as gr | |
import yt_dlp | |
import os | |
from pydub import AudioSegment | |
import re | |
os.makedirs("downloads", exist_ok=True) | |
def sanitize_filename(filename): | |
"""Sanitize the filename by removing or replacing special characters.""" | |
return re.sub(r'[^a-zA-Z0-9_-]', '_', filename) | |
def process_youtube_or_audio(url, recorded_audio): | |
"""Process YouTube URL or recorded audio to create ringtones.""" | |
try: | |
filename = None | |
song_name = None | |
if url: | |
ydl_opts = { | |
'format': 'bestaudio/best', | |
'outtmpl': 'downloads/%(id)s.%(ext)s', | |
'cookiefile': 'cookies.txt' #cookies.txt | |
} | |
with yt_dlp.YoutubeDL(ydl_opts) as ydl: | |
info = ydl.extract_info(url, download=True) | |
filename = os.path.join('downloads', f"{info['id']}.webm") | |
song_name = sanitize_filename(info['title']) | |
elif recorded_audio: | |
filename = recorded_audio | |
song_name = "recorded_audio" | |
if not filename or not os.path.exists(filename): | |
return None, None | |
mp3_filename = f"downloads/{song_name}.mp3" | |
if not os.path.exists(mp3_filename): | |
audio = AudioSegment.from_file(filename) | |
audio.export(mp3_filename, format="mp3") | |
ringtone_filename_m4r = f"downloads/{song_name}.m4r" | |
if not os.path.exists(ringtone_filename_m4r): | |
ringtone_audio = AudioSegment.from_file(mp3_filename)[:20000] # 20 seconds | |
ringtone_audio.export(ringtone_filename_m4r, format="mp4") | |
return mp3_filename, ringtone_filename_m4r | |
except Exception as e: | |
print(f"Error: {e}") | |
return None, None | |
# Gradio UI | |
with gr.Blocks() as interface: | |
gr.HTML(""" | |
<h1>Python YouTube Ringtones</h1> | |
<p>Insert a URL to create ringtones or record your own audio.</p> | |
""") | |
with gr.Row(): | |
youtube_url = gr.Textbox(label="Enter YouTube URL", placeholder="Paste the URL here...") | |
audio_record = gr.Audio(sources=["microphone"], type="filepath", label="Record Audio") | |
with gr.Row(): | |
mp3_download = gr.File(label="Download Android Ringtone") | |
iphone_ringtone = gr.File(label="Download iPhone Ringtone") | |
process_button = gr.Button("Create Ringtones") | |
process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_record], outputs=[mp3_download, iphone_ringtone]) | |
interface.launch(share=True) | |