PYTR / app.py
sudo-soldier's picture
Update app.py
118a5f9 verified
raw
history blame
2.83 kB
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()