Spaces:
Running
Running
import gradio as gr | |
import yt_dlp | |
import subprocess | |
import os | |
from pydub import AudioSegment | |
def process_youtube_url(url, uploaded_file, state): | |
log = [] | |
try: | |
filename = None | |
if url: | |
log.append("Starting download...") | |
ydl_opts = { | |
'format': 'bestaudio/best', | |
'outtmpl': 'downloads/%(id)s.%(ext)s', | |
'cookiefile': 'cookies.txt' #YT | |
} | |
with yt_dlp.YoutubeDL(ydl_opts) as ydl: | |
info = ydl.extract_info(url, download=True) | |
log.append(f"Downloaded: {info['title']}") | |
filename = os.path.join('downloads', f"{info['id']}.webm") | |
elif uploaded_file: | |
filename = uploaded_file.name | |
log.append(f"Uploaded MP3 file: {filename}") | |
else: | |
raise ValueError("No YouTube URL or MP3 file provided.") | |
if not os.path.exists(filename): | |
raise FileNotFoundError(f"File not found: {filename}") | |
log.append("Converting audio to MP3 manually...") | |
mp3_filename = 'Ringtone.mp3' | |
audio = AudioSegment.from_file(filename) | |
audio.export(mp3_filename, format="mp3") | |
log.append(f"MP3 saved: {mp3_filename}") | |
log.append("Creating ringtone...") | |
ringtone_filename_m4a = 'temp_ringtone.m4a' | |
ringtone_audio = AudioSegment.from_file(mp3_filename)[:20000] # 20 sec clip | |
ringtone_audio.export(ringtone_filename_m4a, format="mp4") | |
log.append(f"Ringtone saved as M4A: {ringtone_filename_m4a}") | |
ringtone_filename_m4r = 'Apple.m4r' | |
os.rename(ringtone_filename_m4a, ringtone_filename_m4r) | |
log.append(f"Ringtone renamed to .m4r: {ringtone_filename_m4r}") | |
return '\n'.join(log), mp3_filename, ringtone_filename_m4r, state | |
except Exception as e: | |
log.append(f"Error: {str(e)}") | |
return '\n'.join(log), None, None, state | |
with gr.Blocks() as interface: | |
gr.HTML(""" | |
<h1>P.Y.T.R.</h1> | |
<p>Python YouTube Ringtones<br/>insert a URL to make ringtones. Upload an MP3 to convert it too.</p> | |
""") | |
with gr.Row(): | |
youtube_url = gr.Textbox(label="Enter YouTube URL ", placeholder="Paste the URL here...") | |
mp3_upload = gr.File(label="Upload MP3 (Optional)", type="filepath") | |
state = gr.State() | |
with gr.Row(): | |
status_log = gr.Textbox(label="Status") | |
mp3_download = gr.File(label="Android Ringtone") | |
iphone_ringtone = gr.File(label="iPhone Ringtone") | |
process_button = gr.Button("Create Ringtones") | |
process_button.click(process_youtube_url, inputs=[youtube_url, mp3_upload, state], outputs=[status_log, mp3_download, iphone_ringtone]) | |
interface.launch() | |