Update app.py

app.py

@@ -4,76 +4,73 @@ 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_url(url, uploaded_file):
-    """Process the YouTube URL or uploaded file to create ringtones."""
+def process_youtube_or_audio(url, recorded_audio):
+    """Process YouTube URL or recorded audio to create ringtones."""
     try:
         filename = None
         song_name = None
 
-     
+        # Download from YouTube if URL is provided
         if url:
             ydl_opts = {
                 'format': 'bestaudio/best',
                 'outtmpl': 'downloads/%(id)s.%(ext)s',
-                'cookiefile': 'cookies.txt'  #cookies 
+                'cookiefile': 'cookies.txt'  # Use cookies for authentication if needed
             }
-
             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 uploaded_file:
-            filename = uploaded_file.name
-            song_name = sanitize_filename(os.path.splitext(uploaded_file.name)[0])
-
-  
+        # If audio is recorded, use it
+        elif recorded_audio:
+            filename = recorded_audio
+            song_name = "recorded_audio"
+        
         if not filename or not os.path.exists(filename):
-            return None, None, None
+            return None, None
 
+        # Convert to MP3
         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")
 
-    
+        # Convert to iPhone ringtone (M4R, 20 sec clip)
         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")
 
-      
         return mp3_filename, ringtone_filename_m4r
 
     except Exception as e:
         print(f"Error: {e}")
         return None, None
 
-#UI
+# Gradio UI
 with gr.Blocks() as interface:
     gr.HTML("""
         <h1>Python YouTube Ringtones</h1>
-        <p>Insert a URL to create ringtones or Upload an MP3 to convert.</p>
+        <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...")
-        mp3_upload = gr.File(label="Upload MP3 (Optional)")
+        audio_record = gr.Audio(sources=["microphone"], type="filepath", label="Record Audio")
 
     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.click(process_youtube_url, inputs=[youtube_url, mp3_upload], outputs=[mp3_download, iphone_ringtone])
+    process_button.click(process_youtube_or_audio, inputs=[youtube_url, audio_record], outputs=[mp3_download, iphone_ringtone])
 
 interface.launch(share=True)
 
@@ -93,3 +90,4 @@ interface.launch(share=True)
 
 
 
+