Update app.py

app.py

@@ -13,12 +13,12 @@ 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):
+def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
     try:
         filename = None
         song_name = None
 
-      
+        # Download YouTube audio
         if url:
             ydl_opts = {
                 'format': 'bestaudio/best',
@@ -38,24 +38,27 @@ def process_youtube_or_audio(url, recorded_audio):
         if not filename or not os.path.exists(filename):
             return None, None
 
-     
+        # Load audio file
         audio = AudioSegment.from_file(filename)
 
-      
-        if len(audio) > 20000:
-            audio = audio[:20000]
+        # Ensure valid trimming times
+        start_time = max(0, min(start_time, len(audio)))
+        end_time = max(start_time, min(end_time, len(audio)))
 
-    
+        # Trim the audio
+        trimmed_audio = audio[start_time:end_time]
+
+        # Convert to MP3
         mp3_filename = f"downloads/{song_name}.mp3"
-        audio.export(mp3_filename, format="mp3")
+        trimmed_audio.export(mp3_filename, format="mp3")
 
-       
+        # Convert to M4R (iPhone format)
         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
         m4r_filename = f"downloads/{song_name}.m4r"
         os.rename(m4a_filename, m4r_filename)
 
@@ -99,6 +102,13 @@ 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>")
+
+    with gr.Row():
+        start_time = gr.Slider(0, 20000, value=0, label="Start Time (ms)")
+        end_time = gr.Slider(1000, 20000, value=20000, label="End Time (ms)")
+
     with gr.Row():
         process_button = gr.Button("Create Ringtones", elem_classes="light-btn")
 
@@ -113,7 +123,7 @@ with gr.Blocks(css="""
             iphone_ringtone = gr.File(label="Apple")
             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])
+    process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_record, start_time, end_time], outputs=[mp3_download, iphone_ringtone])
 
 interface.launch(share=True)