Update app.py

app.py

@@ -4,10 +4,10 @@ import yt_dlp
 from pydub import AudioSegment
 import re
 import subprocess
-import time
 
-if not os.path.exists("downloads"):
-    os.makedirs("downloads")
+# Ensure static directory exists for storing audio files
+DOWNLOAD_DIR = "static"
+os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
 def sanitize_filename(filename):
     """Sanitize filenames to avoid special characters."""
@@ -18,48 +18,48 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
         filename = None
         song_name = None
 
-        print(f"URL: {url}")  # Debug: Check URL input
-        print(f"Uploaded audio: {uploaded_audio}")  # Debug: Check if audio is uploaded
-        print(f"Start time: {start_time}, End time: {end_time}")  # Debug: Check times
+        print(f"Processing URL: {url}")
+        print(f"Uploaded audio: {uploaded_audio}")
+        print(f"Start time: {start_time}, End time: {end_time}")
 
-        # Process YouTube URL
+        # Download from YouTube if URL is provided
         if url:
-            print("Processing YouTube URL...")
+            print("Downloading from YouTube...")
             ydl_opts = {
                 'format': 'bestaudio/best',
-                'outtmpl': 'downloads/%(id)s.%(ext)s',
-                'cookiefile': 'cookies.txt'
+                'outtmpl': f'{DOWNLOAD_DIR}/%(id)s.%(ext)s',
+                'cookiefile': 'cookies.txt',
+                'noplaylist': True,
             }
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 info = ydl.extract_info(url, download=True)
-                filename = os.path.join('downloads', f"{info['id']}.webm")
+                filename = os.path.join(DOWNLOAD_DIR, f"{info['id']}.webm")
                 song_name = sanitize_filename(info['title'])
 
-        # Process uploaded audio file
+        # Use uploaded audio if available
         elif uploaded_audio:
             print("Processing uploaded audio...")
-            filename = uploaded_audio.name  # Using the uploaded file's name
+            filename = uploaded_audio
             song_name = "uploaded_audio"
 
+        # If no valid file exists, return an error
         if not filename or not os.path.exists(filename):
-            print(f"Error: No valid file found at {filename}")  # Debug
+            print(f"Error: No valid file found at {filename}")
             return None, None
 
-        # Load audio and trim it
+        # Load and trim audio
         audio = AudioSegment.from_file(filename)
-        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)))
+        start_time_ms = max(0, min(start_time * 1000, len(audio)))
+        end_time_ms = max(start_time_ms, min(end_time * 1000, len(audio)))
 
         trimmed_audio = audio[start_time_ms:end_time_ms]
 
-        # Export trimmed audio
-        mp3_filename = f"downloads/{song_name}.mp3"
+        # Export trimmed audio as MP3
+        mp3_filename = f"{DOWNLOAD_DIR}/{song_name}.mp3"
         trimmed_audio.export(mp3_filename, format="mp3")
 
-        # Convert to m4r format for iPhone
-        m4a_filename = f"downloads/{song_name}.m4a"
+        # Convert MP3 to M4R for iPhone
+        m4a_filename = f"{DOWNLOAD_DIR}/{song_name}.m4a"
         try:
             subprocess.run([
                 'ffmpeg', '-i', mp3_filename, '-vn', '-acodec', 'aac', '-b:a', '192k', m4a_filename
@@ -71,18 +71,73 @@ def process_youtube_or_audio(url, uploaded_audio, start_time, end_time):
             print(f"ffmpeg error: {e}")
             return None, None
 
-        # Ensure the file exists before renaming
+        # Rename to M4R format for iPhone ringtones
+        m4r_filename = f"{DOWNLOAD_DIR}/{song_name}.m4r"
         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)
+            print(f"Files saved: {mp3_filename}, {m4r_filename}")
+            return mp3_filename, m4r_filename
         else:
             print("Error: M4A file not created.")
             return None, None
 
     except Exception as e:
-        print(f"Error: {e}")  # Debug: Print the error
+        print(f"Error: {e}")
         return None, None
 
 
+# === GRADIO UI ===
+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>')
+            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. Import the .m4r file and set it as a ringtone.", lines=4)
+
+    process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_upload, start_time, end_time], outputs=[mp3_download, iphone_ringtone])
+
+# Launch with queue for stability
+interface.queue().launch(share=True, server_port=7860)
+
+
+