Update app.py

app.py

@@ -3,6 +3,9 @@ import yt_dlp
 import os
 from pydub import AudioSegment
 
+# Ensure downloads directory exists
+os.makedirs("downloads", exist_ok=True)
+
 def process_youtube_url(url, uploaded_file):
     try:
         filename = None
@@ -11,39 +14,36 @@ def process_youtube_url(url, uploaded_file):
             ydl_opts = {
                 'format': 'bestaudio/best',
                 'outtmpl': 'downloads/%(id)s.%(ext)s',
-                'cookiefile': 'cookies.txt'  # YT
+                'cookiefile': 'cookies.txt'  # YTcookies 
             }
 
             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('downloads', f"{info['id']}.webm")
 
         elif uploaded_file:
-            filename = uploaded_file
-
-        else:
-            return None, None
+            filename = uploaded_file.name  
 
-        if not os.path.exists(filename):
-            return None, None
+        if not filename or not os.path.exists(filename):
+            return None, None  # No file, no output
 
-        # Convert to MP3
-        mp3_filename = "Ringtone.mp3"
+    
+        mp3_filename = "downloads/Ringtone.mp3"
         audio = AudioSegment.from_file(filename)
         audio.export(mp3_filename, format="mp3")
 
-        # Create ringtone (first 20 seconds)
-        ringtone_filename_m4r = "Apple.m4r"
+        
+        ringtone_filename_m4r = "downloads/Apple.m4r"
         ringtone_audio = AudioSegment.from_file(mp3_filename)[:20000]
         ringtone_audio.export(ringtone_filename_m4r, format="mp4")
 
-        return mp3_filename, ringtone_filename_m4r
-
+        return mp3_filename, ringtone_filename_m4r  
+        
     except Exception as e:
         print(f"Error: {e}")
         return None, None
 
+# Gradio UI
 with gr.Blocks() as interface:
     gr.HTML("""
         <h1>Python YouTube Ringtones</h1>
@@ -55,8 +55,8 @@ with gr.Blocks() as interface:
         mp3_upload = gr.File(label="Upload MP3 (Optional)")
 
     with gr.Row():
-        mp3_download = gr.File(label="Android Ringtone")
-        iphone_ringtone = gr.File(label="iPhone Ringtone")
+        mp3_download = gr.File(label="Download Android Ringtone")
+        iphone_ringtone = gr.File(label="Download iPhone Ringtone")
 
     process_button = gr.Button("Create Ringtones")
     process_button.click(process_youtube_url, inputs=[youtube_url, mp3_upload], outputs=[mp3_download, iphone_ringtone])
@@ -75,3 +75,4 @@ interface.launch()
 
 
 
+