Update app.py

app.py

@@ -2,9 +2,9 @@ import os
 import yt_dlp
 import gradio as gr
 import re
-import subprocess
 from pydub import AudioSegment
 
+# Ensure downloads directory exists
 os.makedirs("downloads", exist_ok=True)
 
 def sanitize_filename(filename):
@@ -18,7 +18,7 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
 
         if url:
             ydl_opts = {
-                'format': 'bestaudio/best',
+                'format': 'm4a/bestaudio',
                 'outtmpl': 'downloads/%(title)s.%(ext)s',
                 'cookiefile': 'cookies.txt'
             }
@@ -26,14 +26,23 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
                 info = ydl.extract_info(url, download=True)
                 filename = ydl.prepare_filename(info)
                 song_name = sanitize_filename(info['title'])
-        
+
+                # Ensure correct extension
+                if not filename.endswith(".m4a"):
+                    possible_file = f"downloads/{song_name}.m4a"
+                    if os.path.exists(possible_file):
+                        filename = possible_file
+
         elif uploaded_audio is not None:
             filename = uploaded_audio
             song_name = sanitize_filename(os.path.splitext(os.path.basename(uploaded_audio))[0])
-        
+
+        # Validate file existence
         if not filename or not os.path.exists(filename):
+            print(f"File {filename} not found!")
             return None, None
 
+        # Process audio
         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)))
@@ -41,18 +50,12 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
 
         trimmed_audio = audio[start_ms:end_ms]
 
+        # Save MP3
         mp3_filename = f"downloads/{song_name}.mp3"
         trimmed_audio.export(mp3_filename, format="mp3")
 
-        m4a_filename = f"downloads/{song_name}.m4a"
-        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
-
+        # Rename M4A to M4R for iPhone
+        m4a_filename = filename
         m4r_filename = f"downloads/{song_name}.m4r"
         os.rename(m4a_filename, m4r_filename)
 
@@ -68,39 +71,29 @@ with gr.Blocks() as interface:
         <h1 style="font-size: 2rem; text-align: center;"><i class="fas fa-music"></i>&nbsp;PYTR - Python YouTube Ringtones</h1>
         <p style="text-align: center;">Enter a YouTube URL or upload an audio file to create ringtones.</p>
         <p style="text-align: center;"><a href="https://ringtones.jessejesse.xyz">Ringtones</a> | <a href="https://github.com/sudo-self/pytr">GitHub</a></p>
-
-        <style>
-            #youtube-url, #uploaded-audio { width: 45%; }
-            #process-button { width: 100%; margin-top: 10px; background-color: #6a4cfc; color: white; border: none; border-radius: 5px; }
-            #mp3-download, #iphone-ringtone { width: 45%; }
-            #instructions { width: 100%; margin-top: 20px; }
-            #start-time-slider, #end-time-slider { width: 45%; }
-        </style>
     """)
 
     with gr.Row():
-        youtube_url = gr.Textbox(placeholder="Enter the URL here...", label="YouTube URL", elem_id="youtube-url")
-        uploaded_audio = gr.File(label="Upload Audio File", type="filepath", elem_id="uploaded-audio")
+        youtube_url = gr.Textbox(placeholder="Enter the URL here...", label="YouTube URL")
+        uploaded_audio = gr.File(label="Upload Audio File", type="filepath")
 
     gr.HTML("<h3 style='text-align: center;'>Trim Audio</h3>")
 
     with gr.Row():
-        start_time = gr.Slider(0, 20, value=0, label="Start Time (seconds)", elem_id="start-time-slider")
-        end_time = gr.Slider(1, 20, value=20, label="End Time (seconds)", elem_id="end-time-slider")
-    
-    # Add a class for custom button color
-    process_button = gr.Button("Create Ringtones", elem_id="process-button")
-    
+        start_time = gr.Slider(0, 20, value=0, label="Start Time (seconds)")
+        end_time = gr.Slider(1, 20, value=20, label="End Time (seconds)")
+
+    process_button = gr.Button("Create Ringtones")
+
     with gr.Row():
-        mp3_download = gr.File(label="Android Ringtone", elem_id="mp3-download")
-        iphone_ringtone = gr.File(label="iPhone Ringtone", elem_id="iphone-ringtone")
-    
+        mp3_download = gr.File(label="Android Ringtone")
+        iphone_ringtone = gr.File(label="iPhone Ringtone")
+
     with gr.Row():
         instructions = gr.Textbox(
             value="Android: move the download to the 'Ringtones' folder.\nApple: import download with 'Garage Band' and export to ringtones.",
             label="Install Ringtones",
-            interactive=False,
-            elem_id="instructions"
+            interactive=False
         )
 
     process_button.click(process_youtube_or_audio, inputs=[youtube_url, uploaded_audio, start_time, end_time], outputs=[mp3_download, iphone_ringtone])
@@ -108,3 +101,4 @@ with gr.Blocks() as interface:
 interface.launch(share=True)
 
 
+