Update app.py

app.py

@@ -1,71 +1,72 @@
 import os
-import gradio as gr
 import yt_dlp
-from pydub import AudioSegment
+import gradio as gr
 import re
 import subprocess
+from pydub import AudioSegment
 
 
-if not os.path.exists("downloads"):
-    os.makedirs("downloads")
+os.makedirs("downloads", exist_ok=True)
 
 def sanitize_filename(filename):
-    """Sanitize filenames to avoid special characters."""
+    """Sanitize filenames by replacing special characters."""
     return re.sub(r'[^a-zA-Z0-9_-]', '_', filename)
 
 def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
+    """Downloads or processes audio, trims it, and exports ringtones."""
     try:
-        filename = None
-        song_name = None
+        filename, song_name = None, None
 
-      
+   
         if url:
             ydl_opts = {
                 'format': 'bestaudio/best',
-                'outtmpl': 'downloads/%(id)s.%(ext)s',
+                'outtmpl': 'downloads/%(title)s.%(ext)s',
                 'cookiefile': '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")
+                filename = ydl.prepare_filename(info)
                 song_name = sanitize_filename(info['title'])
 
-      
-        elif recorded_audio:
-            filename = recorded_audio
+       
+        elif recorded_audio and isinstance(recorded_audio, dict):
+            filename = recorded_audio["name"]
             song_name = "recorded_audio"
-        
+
+       
         if not filename or not os.path.exists(filename):
             return None, None
 
-   
-        audio = AudioSegment.from_file(filename)
 
-     
-        start_time_ms = start_time * 1000
-        end_time_ms = end_time * 1000
+        audio = AudioSegment.from_file(filename)
+        start_ms, end_ms = int(start_time * 1000), int(end_time * 1000)
 
       
-        start_time_ms = max(0, min(start_time_ms, len(audio)))
-        end_time_ms = max(start_time_ms, min(end_time_ms, len(audio)))
+        start_ms = max(0, min(start_ms, len(audio)))
+        end_ms = max(start_ms, min(end_ms, len(audio)))
 
-       
-        trimmed_audio = audio[start_time_ms:end_time_ms]
+        trimmed_audio = audio[start_ms:end_ms]
 
-      
+    
         mp3_filename = f"downloads/{song_name}.mp3"
         trimmed_audio.export(mp3_filename, format="mp3")
 
-      
+
         m4a_filename = f"downloads/{song_name}.m4a"
-        subprocess.run([
-            'ffmpeg', '-i', mp3_filename, '-vn', '-acodec', 'aac', '-b:a', '192k', m4a_filename
-        ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        result = subprocess.run([
+            'ffmpeg', '-y', '-i', mp3_filename, '-vn', '-acodec', 'aac', '-b:a', '192k', m4a_filename
+        ], capture_output=True, text=True)
+
+        if result.returncode != 0 or not os.path.exists(m4a_filename):
+            print("FFmpeg Error:", result.stderr)
+            return mp3_filename, None  #
 
+      
         m4r_filename = f"downloads/{song_name}.m4r"
         os.rename(m4a_filename, m4r_filename)
 
-        return os.path.abspath(mp3_filename), os.path.abspath(m4r_filename)
+        return mp3_filename, m4r_filename
 
     except Exception as e:
         print(f"Error: {e}")
@@ -74,24 +75,17 @@ def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
 
 with gr.Blocks(css="""
     body { font-family: Arial, sans-serif; text-align: center; }
-    .light-btn { 
-        background-color: #ADD8E6; 
-        color: #333; 
-        border: 2px solid #ccc; 
-        padding: 10px 20px; 
-        font-size: 16px; 
-        cursor: pointer; 
-    }
+    .light-btn { background-color: #ADD8E6; color: #333; border: 2px solid #ccc; padding: 10px 20px; font-size: 16px; cursor: pointer; }
     .light-btn:hover { background-color: #87CEFA; }
 """) as interface:
 
     gr.HTML("""
         <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
-        <h1><i class="fas fa-music"></i>&nbsp;PYTR</h1>
-        <p>Python YouTube Ringtones. Enter a YouTube URL or record audio to create ringtones.</p>
+        <h1><i class="fas fa-music"></i>&nbsp;PYTR - Python YouTube Ringtones</h1>
+        <p>Enter a YouTube URL or record audio to create ringtones.</p>
         <p>
-            <a href="https://ringtones.JesseJesse.xyz" target="_blank">slim UI</a>&nbsp;&nbsp;&nbsp;
-            <a href="https://pub-c1de1cb456e74d6bbbee111ba9e6c757.r2.dev/iphone.png" target="_blank">iPhone xfers</a>&nbsp;&nbsp;&nbsp;
+            <a href="https://ringtones.JesseJesse.xyz" target="_blank">Ringtones API</a>&nbsp;&nbsp;&nbsp;
+            <a href="https://pub-c1de1cb456e74d6bbbee111ba9e6c757.r2.dev/iphone.png" target="_blank">iPhone Transfers</a>&nbsp;&nbsp;&nbsp;
             <a href="https://youtube.com" target="_blank">YouTube</a>
         </p>
     """)
@@ -105,26 +99,24 @@ with gr.Blocks(css="""
             gr.HTML('<label><i class="fas fa-microphone"></i>&nbsp;Record Audio</label>')
             audio_record = gr.Audio(sources=["microphone"], type="filepath", show_label=False)
 
-    with gr.Row():
-        gr.HTML("<h3>Trim Audio (Optional)</h3>")
+    gr.HTML("<h3>Trim Audio</h3>")
 
     with gr.Row():
         start_time = gr.Slider(0, 20, value=0, label="Start Time (seconds)")
         end_time = gr.Slider(1, 20, value=20, label="End Time (seconds)")
 
-    with gr.Row():
-        process_button = gr.Button("Create Ringtones", elem_classes="light-btn")
+    process_button = gr.Button("Create Ringtones", elem_classes="light-btn")
 
     with gr.Row():
         with gr.Column(scale=1, min_width=250):
-            gr.HTML('<label>&nbsp;Android Ringtone</label>')
-            mp3_download = gr.File(label="Android")
-            android_instructions = gr.Textbox(label="Install", placeholder="Copy the mp3 file to the ringtones folder", lines=1)
+            gr.HTML('<label>Android Ringtone</label>')
+            mp3_download = gr.File(label="Android", interactive=False)
+            gr.Textbox(label="Install", value="Copy the MP3 file to your ringtones folder", lines=1, interactive=False)
 
         with gr.Column(scale=1, min_width=250):
-            gr.HTML('<label>&nbsp;iPhone Ringtone</label>')
-            iphone_ringtone = gr.File(label="Apple")
-            iphone_instructions = gr.Textbox(label="Install", placeholder="Open GarageBand on your iPhone. Create a new project with Audio Recorder. Tap on the Tracks view (≡ icon in top-left). Tap the Loop icon (top-right corner). Select 'Files', then tap 'Browse items from the Files app'. Locate your .m4r file and tap it to import into GarageBand.", lines=4)
+            gr.HTML('<label>iPhone Ringtone</label>')
+            iphone_ringtone = gr.File(label="Apple", interactive=False)
+            gr.Textbox(label="Install", value="Open GarageBand on your iPhone, import the .m4r file, and set it as a ringtone.", lines=1, interactive=False)
 
     process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_record, start_time, end_time], outputs=[mp3_download, iphone_ringtone])