Update app.py

app.py

@@ -28,16 +28,16 @@ def process_youtube_url(url, uploaded_file):
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 info = ydl.extract_info(url, download=True)
                 filename = os.path.join('downloads', f"{info['id']}.webm")
-                song_name = info['title']
+                song_name = sanitize_filename(info['title'])
 
         # Use uploaded file if URL is not provided
         elif uploaded_file:
             filename = uploaded_file.name
-            song_name = os.path.splitext(uploaded_file.name)[0]
+            song_name = sanitize_filename(os.path.splitext(uploaded_file.name)[0])
 
         # Ensure file exists
         if not filename or not os.path.exists(filename):
-            return None, None
+            return None, None, None
 
         # Convert to MP3 if not already converted
         mp3_filename = f"downloads/{song_name}.mp3"
@@ -51,12 +51,12 @@ def process_youtube_url(url, uploaded_file):
             ringtone_audio = AudioSegment.from_file(mp3_filename)[:20000]  # 20 seconds
             ringtone_audio.export(ringtone_filename_m4r, format="mp4")
 
-        # Return file paths and song name for download
-        return song_name, mp3_filename, ringtone_filename_m4r
+        # Return file names for download
+        return mp3_filename, ringtone_filename_m4r
 
     except Exception as e:
         print(f"Error: {e}")
-        return None, None, None
+        return None, None
 
 # Gradio UI
 with gr.Blocks() as interface:
@@ -70,12 +70,11 @@ with gr.Blocks() as interface:
         mp3_upload = gr.File(label="Upload MP3 (Optional)")
 
     with gr.Row():
-        song_name_output = gr.Textbox(label="Song Name", interactive=False)
         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=[song_name_output, mp3_download, iphone_ringtone])
+    process_button.click(process_youtube_url, inputs=[youtube_url, mp3_upload], outputs=[mp3_download, iphone_ringtone])
 
 interface.launch(share=True)
 
@@ -94,3 +93,4 @@ interface.launch(share=True)
 
 
 
+