Update app.py

app.py

@@ -5,10 +5,12 @@ from pydub import AudioSegment
 import re
 import subprocess
 
+# Ensure the downloads directory exists
 if not os.path.exists("downloads"):
     os.makedirs("downloads")
 
 def sanitize_filename(filename):
+    """Sanitize filenames to avoid special characters."""
     return re.sub(r'[^a-zA-Z0-9_-]', '_', filename)
 
 def process_youtube_or_audio(url, recorded_audio):
@@ -16,6 +18,7 @@ def process_youtube_or_audio(url, recorded_audio):
         filename = None
         song_name = None
 
+        # Extract and download YouTube audio
         if url:
             ydl_opts = {
                 'format': 'bestaudio/best',
@@ -27,6 +30,7 @@ def process_youtube_or_audio(url, recorded_audio):
                 filename = os.path.join('downloads', f"{info['id']}.webm")
                 song_name = sanitize_filename(info['title'])
 
+        # Handle recorded audio
         elif recorded_audio:
             filename = recorded_audio
             song_name = "recorded_audio"
@@ -34,22 +38,26 @@ def process_youtube_or_audio(url, recorded_audio):
         if not filename or not os.path.exists(filename):
             return None, None
 
+        # Load audio
         audio = AudioSegment.from_file(filename)
-        
-        if len(audio) > 20000: 
+
+        # Trim to 20 seconds (iPhone ringtone limit)
+        if len(audio) > 20000:
             audio = audio[:20000]
 
+        # Convert to MP3 (Android)
         mp3_filename = f"downloads/{song_name}.mp3"
-        if not os.path.exists(mp3_filename):
-            audio.export(mp3_filename, format="mp3")
+        audio.export(mp3_filename, format="mp3")
 
+        # Convert to M4A first (needed for M4R)
         m4a_filename = f"downloads/{song_name}.m4a"
-        if not os.path.exists(m4a_filename):
-            subprocess.run(['ffmpeg', '-i', mp3_filename, '-vn', '-acodec', 'aac', '-b:a', '192k', m4a_filename])
+        subprocess.run([
+            'ffmpeg', '-i', mp3_filename, '-vn', '-acodec', 'aac', '-b:a', '192k', m4a_filename
+        ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
 
+        # Rename to M4R (iPhone)
         m4r_filename = f"downloads/{song_name}.m4r"
-        if not os.path.exists(m4r_filename):
-            os.rename(m4a_filename, m4r_filename)
+        os.rename(m4a_filename, m4r_filename)
 
         return os.path.abspath(mp3_filename), os.path.abspath(m4r_filename)
 
@@ -57,6 +65,7 @@ def process_youtube_or_audio(url, recorded_audio):
         print(f"Error: {e}")
         return None, None
 
+# Gradio UI
 with gr.Blocks(css="""
     body { font-family: Arial, sans-serif; text-align: center; }
     .light-btn { 
@@ -67,20 +76,18 @@ with gr.Blocks(css="""
         font-size: 16px; 
         cursor: pointer; 
     }
-    .light-btn:hover { 
-        background-color: #87CEFA; 
-    }
+    .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>
-    <p>
-        <a href="https://ringtone.JesseJesse.xyz" target="_blank">ringtone.JesseJesse.xyz</a> | 
-        <a href="https://sudo-self.github.io/pytr/" target="_blank">GitHub</a>
-    </p>
-  """)
+    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>
+        <p>
+            <a href="https://ringtone.JesseJesse.xyz" target="_blank">ringtone.JesseJesse.xyz</a> | 
+            <a href="https://sudo-self.github.io/pytr/" target="_blank">GitHub</a>
+        </p>
+    """)
 
     with gr.Row():
         with gr.Column(scale=1, min_width=250):
@@ -98,13 +105,14 @@ with gr.Blocks(css="""
         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="Place the downloaded .mp3 file in the ringtones folder", lines=2)
+            android_instructions = gr.Textbox(label="Install", placeholder="Move the .mp3 file to the ringtones folder", lines=2)
 
         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="Swipe left and open the Garage Band App. Then long-press on the Ringtone file, select share, choose ringtone, and click export to complete the import.", lines=2)
+            iphone_instructions = gr.Textbox(label="Install", placeholder="Swipe left and open GarageBand. Long-press the ringtone file, select share, then choose 'Ringtone' to export.", lines=2)
 
     process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_record], outputs=[mp3_download, iphone_ringtone])
 
 interface.launch(share=True)
+