Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import re
 from pydub import AudioSegment
 
-#downloads directory
+# Ensure downloads directory exists
 os.makedirs("downloads", exist_ok=True)
 
 def sanitize_filename(filename):
@@ -27,11 +27,13 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
                 filename = ydl.prepare_filename(info)
                 song_name = sanitize_filename(info['title'])
 
-               
+                # Ensure filename is correct
                 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
@@ -39,7 +41,6 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
 
         # Validate file existence
         if not filename or not os.path.exists(filename):
-            print(f"File {filename} not found!")
             return None, None
 
         # Process audio
@@ -50,14 +51,13 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
 
         trimmed_audio = audio[start_ms:end_ms]
 
-        #MP3
+        # Save as MP3 (Android)
         mp3_filename = f"downloads/{song_name}.mp3"
         trimmed_audio.export(mp3_filename, format="mp3")
 
-        #M4R
-        m4a_filename = filename
+        # Convert to M4R (iPhone)
         m4r_filename = f"downloads/{song_name}.m4r"
-        os.rename(m4a_filename, m4r_filename)
+        trimmed_audio.export(m4r_filename, format="m4r")  # Fix: Convert instead of renaming
 
         return mp3_filename, m4r_filename
 
@@ -65,7 +65,7 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
         print(f"Error: {e}")
         return None, None
 
-#Gradio
+# Gradio UI
 with gr.Blocks() as interface:
     gr.HTML("""
         <h1 style="font-size: 2rem; text-align: center;"><i class="fas fa-music"></i>&nbsp;PYTR - Python YouTube Ringtones</h1>