Update app.py

app.py

@@ -4,6 +4,7 @@ import yt_dlp
 from pydub import AudioSegment
 import re
 import subprocess
+import time
 
 if not os.path.exists("downloads"):
     os.makedirs("downloads")
@@ -59,79 +60,29 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
 
         # Convert to m4r format for iPhone
         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)
-
-        # Rename to m4r format for iPhone ringtones
-        m4r_filename = f"downloads/{song_name}.m4r"
-        os.rename(m4a_filename, m4r_filename)
-
-        print(f"Files saved: {mp3_filename}, {m4r_filename}")  # Debug
+        try:
+            subprocess.run([
+                'ffmpeg', '-i', mp3_filename, '-vn', '-acodec', 'aac', '-b:a', '192k', m4a_filename
+            ], stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=30, check=True)
+        except subprocess.TimeoutExpired:
+            print("Error: ffmpeg took too long to process the file.")
+            return None, None
+        except subprocess.CalledProcessError as e:
+            print(f"ffmpeg error: {e}")
+            return None, None
 
-        return os.path.abspath(mp3_filename), os.path.abspath(m4r_filename)
+        # Ensure the file exists before renaming
+        if os.path.exists(m4a_filename):
+            m4r_filename = f"downloads/{song_name}.m4r"
+            os.rename(m4a_filename, m4r_filename)
+            print(f"Files saved: {mp3_filename}, {m4r_filename}")  # Debug
+            return os.path.abspath(mp3_filename), os.path.abspath(m4r_filename)
+        else:
+            print("Error: M4A file not created.")
+            return None, None
 
     except Exception as e:
         print(f"Error: {e}")  # Debug: Print the error
         return None, None
 
 
-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: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 upload audio to create ringtones.</p>
-        <p>
-            <a href="https://ringtones.JesseJesse.xyz" target="_blank">Ringtones</a>&nbsp;&nbsp;&nbsp;
-            <a href="https://pub-c1de1cb456e74d6bbbee111ba9e6c757.r2.dev/iphone.png" target="_blank">iPhone xfers</a>&nbsp;&nbsp;&nbsp;
-            <a href="https://youtube.com" target="_blank">YouTube</a>
-        </p>
-    """)
-
-    with gr.Row():
-        with gr.Column(scale=1, min_width=250):
-            gr.HTML('<label><i class="fas fa-link"></i>&nbsp;YouTube URL</label>')
-            youtube_url = gr.Textbox(placeholder="Enter the URL here...", show_label=False)
-
-        with gr.Column(scale=1, min_width=250):
-            gr.HTML('<label><i class="fas fa-upload"></i>&nbsp;Upload Audio</label>')
-            # Change type to 'filepath' for Hugging Face compatibility
-            audio_upload = gr.File(label="Upload Audio", type="filepath", show_label=False)
-
-    with gr.Row():
-        gr.HTML("<h3>Trim Audio (Optional)</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")
-
-    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="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="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)
-
-    process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_upload, start_time, end_time], outputs=[mp3_download, iphone_ringtone])
-
-interface.launch(share=True)
-