Update app.py

app.py

@@ -1,4 +1,46 @@
-m4r):
+import gradio as gr
+import yt_dlp
+import os
+from pydub import AudioSegment
+import re
+
+os.makedirs("downloads", exist_ok=True)
+
+def sanitize_filename(filename):
+    """Sanitize the filename by removing or replacing special characters."""
+    return re.sub(r'[^a-zA-Z0-9_-]', '_', filename)
+
+def process_youtube_or_audio(url, recorded_audio):
+    """Process YouTube URL or recorded audio to create ringtones."""
+    try:
+        filename = None
+        song_name = None
+
+        if url:
+            ydl_opts = {
+                '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 = os.path.join('downloads', f"{info['id']}.webm")
+                song_name = sanitize_filename(info['title'])
+
+        elif recorded_audio:
+            filename = recorded_audio
+            song_name = "recorded_audio"
+        
+        if not filename or not os.path.exists(filename):
+            return None, None
+
+        mp3_filename = f"downloads/{song_name}.mp3"
+        if not os.path.exists(mp3_filename):
+            audio = AudioSegment.from_file(filename)
+            audio.export(mp3_filename, format="mp3")
+
+        ringtone_filename_m4r = f"downloads/{song_name}.m4r"
+        if not os.path.exists(ringtone_filename_m4r):
             ringtone_audio = AudioSegment.from_file(mp3_filename)[:20000]  # 20 seconds
             ringtone_audio.export(ringtone_filename_m4r, format="mp4")
 
@@ -8,22 +50,27 @@ m4r):
         print(f"Error: {e}")
         return None, None
 
-# Gradio UI
+# Gradio UI with FontAwesome
 with gr.Blocks() as interface:
     gr.HTML("""
-        <h1>Python YouTube Ringtones</h1>
+        <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
+        <h1><i class="fa fa-music"></i> Python YouTube Ringtones</h1>
         <p>Insert a URL to create ringtones or record your own audio.</p>
     """)
 
     with gr.Row():
-        youtube_url = gr.Textbox(label="Enter YouTube URL", placeholder="Paste the URL here...")
-        audio_record = gr.Audio(sources=["microphone"], type="filepath", label="Record Audio")
+        gr.HTML('<p><i class="fa fa-link"></i> Enter YouTube URL:</p>')
+        youtube_url = gr.Textbox(placeholder="Paste the URL here...")
+
+    with gr.Row():
+        gr.HTML('<p><i class="fa fa-microphone"></i> Record Audio:</p>')
+        audio_record = gr.Audio(sources=["microphone"], type="filepath")
 
     with gr.Row():
         mp3_download = gr.File(label="Download Android Ringtone")
         iphone_ringtone = gr.File(label="Download iPhone Ringtone")
 
-    process_button = gr.Button("Create Ringtones")
+    process_button = gr.Button("🎵 Create Ringtones")
     process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_record], outputs=[mp3_download, iphone_ringtone])
 
 interface.launch(share=True)