Update app.py

app.py

@@ -1,96 +1,132 @@
 import os
-import yt_dlp
 import gradio as gr
-import re
+import yt_dlp
 from pydub import AudioSegment
+import re
+import subprocess
 
 
-os.makedirs("downloads", exist_ok=True)
+if not os.path.exists("downloads"):
+    os.makedirs("downloads")
 
 def sanitize_filename(filename):
-    """Sanitize filenames by replacing special characters."""
+    """Sanitize filenames to avoid special characters."""
     return re.sub(r'[^a-zA-Z0-9_-]', '_', filename)
 
-def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
-    """Downloads or processes audio, trims it, and exports ringtones."""
+def process_youtube_or_audio(url, recorded_audio, start_time, end_time):
     try:
-        filename, song_name = None, None
+        filename = None
+        song_name = None
 
+      
         if url:
             ydl_opts = {
-                'format': 'm4a/bestaudio',
-                'outtmpl': 'downloads/%(title)s.%(ext)s',
-                'cookiefile': 'cookies.txt'  #yt
+                'format': 'bestaudio/best',
+                'outtmpl': 'downloads/%(id)s.%(ext)s',
+                'cookiefile': 'cookies.txt'
             }
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 info = ydl.extract_info(url, download=True)
-                filename = ydl.prepare_filename(info)
+                filename = os.path.join('downloads', f"{info['id']}.webm")
                 song_name = sanitize_filename(info['title'])
 
-             
-                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
-            song_name = sanitize_filename(os.path.splitext(os.path.basename(uploaded_audio))[0])
-
-       
+      
+        elif recorded_audio:
+            filename = recorded_audio
+            song_name = "recorded_audio"
+        
         if not filename or not os.path.exists(filename):
-            print(f"File not found: {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]
 
      
+        start_time_ms = start_time * 1000
+        end_time_ms = end_time * 1000
+
+      
+        start_time_ms = max(0, min(start_time_ms, len(audio)))
+        end_time_ms = max(start_time_ms, min(end_time_ms, len(audio)))
+
+       
+        trimmed_audio = audio[start_time_ms:end_time_ms]
+
+      
         mp3_filename = f"downloads/{song_name}.mp3"
         trimmed_audio.export(mp3_filename, format="mp3")
 
-    
+      
+        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)
+
         m4r_filename = f"downloads/{song_name}.m4r"
-        trimmed_audio.export(m4r_filename, format="m4r")
+        os.rename(m4a_filename, m4r_filename)
 
-        return mp3_filename, m4r_filename
+        return os.path.abspath(mp3_filename), os.path.abspath(m4r_filename)
 
     except Exception as e:
         print(f"Error: {e}")
         return None, None
 
-# Gradio UI
-with gr.Blocks() as interface:
-    gr.Markdown("""
-        # 🎵 PYTR - Python YouTube Ringtones
-        Create custom ringtones from YouTube or uploaded audio.
-        - **Enter a YouTube URL** to extract audio.
-        - **Upload a file** to trim an existing audio.
+
+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 record audio to create ringtones.</p>
+        <p>
+            <a href="https://ringtones.JesseJesse.xyz" target="_blank">slim UI</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():
-        youtube_url = gr.Textbox(placeholder="Enter YouTube URL...", label="YouTube URL")
-        uploaded_audio = gr.File(label="Upload Audio File", type="filepath")
+        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)
 
-    gr.Markdown("## Trim Audio")
     with gr.Row():
-        start_time = gr.Slider(0, 30, value=0, label="Start Time (seconds)")
-        end_time = gr.Slider(1, 30, value=10, label="End Time (seconds)")
+        gr.HTML("<h3>Trim Audio (Optional)</h3>")
 
-    process_button = gr.Button("Create Ringtones")
+    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():
-        mp3_download = gr.File(label="Android Ringtone (MP3)")
-        iphone_ringtone = gr.File(label="iPhone Ringtone (M4R)")
+        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, uploaded_audio, start_time, end_time], 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)