Spaces:
Sleeping
Sleeping
| import os | |
| import yt_dlp | |
| import gradio as gr | |
| import re | |
| from pydub import AudioSegment | |
| os.makedirs("downloads", exist_ok=True) | |
| def sanitize_filename(filename): | |
| """Sanitize filenames by replacing special characters.""" | |
| return re.sub(r'[^a-zA-Z0-9_-]', '_', filename) | |
| def process_youtube_or_audio(url, uploaded_audio, start_time, end_time): | |
| """Downloads or processes audio, trims it, and exports ringtones.""" | |
| try: | |
| filename, song_name = None, None | |
| if url: | |
| ydl_opts = { | |
| 'format': 'm4a/bestaudio', | |
| 'outtmpl': 'downloads/%(title)s.%(ext)s', | |
| 'cookiefile': 'cookies.txt' #yt | |
| } | |
| with yt_dlp.YoutubeDL(ydl_opts) as ydl: | |
| info = ydl.extract_info(url, download=True) | |
| filename = ydl.prepare_filename(info) | |
| song_name = sanitize_filename(info['title']) | |
| if not filename.endswith(".m4a"): | |
| possible_file = f"downloads/{song_name}.m4a" | |
| if os.path.exists(possible_file): | |
| filename = possible_file | |
| else: | |
| filename = None | |
| elif uploaded_audio is not None: | |
| filename = uploaded_audio | |
| song_name = sanitize_filename(os.path.splitext(os.path.basename(uploaded_audio))[0]) | |
| if not filename or not os.path.exists(filename): | |
| print(f"File not found: {filename}") | |
| return None, None | |
| audio = AudioSegment.from_file(filename) | |
| start_ms, end_ms = int(start_time * 1000), int(end_time * 1000) | |
| start_ms = max(0, min(start_ms, len(audio))) | |
| end_ms = max(start_ms, min(end_ms, len(audio))) | |
| trimmed_audio = audio[start_ms:end_ms] | |
| mp3_filename = f"downloads/{song_name}.mp3" | |
| trimmed_audio.export(mp3_filename, format="mp3") | |
| m4r_filename = f"downloads/{song_name}.m4r" | |
| trimmed_audio.export(m4r_filename, format="m4r") | |
| return mp3_filename, m4r_filename | |
| except Exception as e: | |
| print(f"Error: {e}") | |
| return None, None | |
| # Gradio UI | |
| with gr.Blocks() as interface: | |
| gr.Markdown(""" | |
| # 🎵 PYTR - Python YouTube Ringtones | |
| Create custom ringtones from YouTube or uploaded audio. | |
| - **Enter a YouTube URL** to extract audio. | |
| - **Upload a file** to trim an existing audio. | |
| """) | |
| with gr.Row(): | |
| youtube_url = gr.Textbox(placeholder="Enter YouTube URL...", label="YouTube URL") | |
| uploaded_audio = gr.File(label="Upload Audio File", type="filepath") | |
| gr.Markdown("## Trim Audio") | |
| with gr.Row(): | |
| start_time = gr.Slider(0, 30, value=0, label="Start Time (seconds)") | |
| end_time = gr.Slider(1, 30, value=10, label="End Time (seconds)") | |
| process_button = gr.Button("Create Ringtones") | |
| with gr.Row(): | |
| mp3_download = gr.File(label="Android Ringtone (MP3)") | |
| iphone_ringtone = gr.File(label="iPhone Ringtone (M4R)") | |
| process_button.click(process_youtube_or_audio, inputs=[youtube_url, uploaded_audio, start_time, end_time], outputs=[mp3_download, iphone_ringtone]) | |
| interface.launch(share=True) | |