Update app.py

app.py

@@ -5,19 +5,17 @@ import re
 import subprocess
 from pydub import AudioSegment
 
-
 os.makedirs("downloads", exist_ok=True)
 
 def sanitize_filename(filename):
     """Sanitize filenames by replacing special characters."""
     return re.sub(r'[^a-zA-Z0-9_-]', '_', filename)
 
-def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
+def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
     """Downloads or processes audio, trims it, and exports ringtones."""
     try:
         filename, song_name = None, None
 
-   
         if url:
             ydl_opts = {
                 'format': 'bestaudio/best',
@@ -28,31 +26,24 @@ def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
                 info = ydl.extract_info(url, download=True)
                 filename = ydl.prepare_filename(info)
                 song_name = sanitize_filename(info['title'])
-
-       
-        elif recorded_audio and isinstance(recorded_audio, dict):
-            filename = recorded_audio["name"]
-            song_name = "recorded_audio"
-
-       
+        
+        elif uploaded_audio is not None:
+            filename = uploaded_audio
+            song_name = sanitize_filename(os.path.splitext(os.path.basename(uploaded_audio))[0])
+        
         if not filename or not os.path.exists(filename):
             return None, None
 
-
         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)))
         end_ms = max(start_ms, min(end_ms, len(audio)))
 
         trimmed_audio = audio[start_ms:end_ms]
 
-    
         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
@@ -60,9 +51,8 @@ def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
 
         if result.returncode != 0 or not os.path.exists(m4a_filename):
             print("FFmpeg Error:", result.stderr)
-            return mp3_filename, None  #
+            return mp3_filename, None
 
-      
         m4r_filename = f"downloads/{song_name}.m4r"
         os.rename(m4a_filename, m4r_filename)
 
@@ -72,53 +62,28 @@ def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
         print(f"Error: {e}")
         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:
-
+with gr.Blocks() 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 - Python YouTube Ringtones</h1>
-        <p>Enter a YouTube URL or record audio to create ringtones.</p>
-        <p>
-            <a href="https://ringtones.JesseJesse.xyz" target="_blank">Ringtones API</a>&nbsp;&nbsp;&nbsp;
-            <a href="https://pub-c1de1cb456e74d6bbbee111ba9e6c757.r2.dev/iphone.png" target="_blank">iPhone Transfers</a>&nbsp;&nbsp;&nbsp;
-            <a href="https://youtube.com" target="_blank">YouTube</a>
-        </p>
+        <p>Enter a YouTube URL or upload an audio file to create ringtones.</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-microphone"></i>&nbsp;Record Audio</label>')
-            audio_record = gr.Audio(sources=["microphone"], type="filepath", show_label=False)
-
+        youtube_url = gr.Textbox(placeholder="Enter the URL here...", label="YouTube URL")
+        uploaded_audio = gr.File(label="Upload Audio File", type="file")
+    
     gr.HTML("<h3>Trim Audio</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)")
-
-    process_button = gr.Button("Create Ringtones", elem_classes="light-btn")
-
+    
+    process_button = gr.Button("Create Ringtones")
+    
     with gr.Row():
-        with gr.Column(scale=1, min_width=250):
-            gr.HTML('<label>Android Ringtone</label>')
-            mp3_download = gr.File(label="Android", interactive=False)
-            gr.Textbox(label="Install", value="Copy the MP3 file to your ringtones folder", lines=1, interactive=False)
-
-        with gr.Column(scale=1, min_width=250):
-            gr.HTML('<label>iPhone Ringtone</label>')
-            iphone_ringtone = gr.File(label="Apple", interactive=False)
-            gr.Textbox(label="Install", value="Open GarageBand on your iPhone, import the .m4r file, and set it as a ringtone.", lines=1, interactive=False)
-
-    process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_record, start_time, end_time], outputs=[mp3_download, iphone_ringtone])
+        mp3_download = gr.File(label="Android Ringtone")
+        iphone_ringtone = gr.File(label="iPhone Ringtone")
+    
+    process_button.click(process_youtube_or_audio, inputs=[youtube_url, uploaded_audio, start_time, end_time], outputs=[mp3_download, iphone_ringtone])
 
 interface.launch(share=True)