Create app.py

app.py

@@ -0,0 +1,135 @@
+import gradio as gr
+from yt_dlp import YoutubeDL
+import os
+from pydub import AudioSegment
+import re
+
+DOWNLOADS_FOLDER = "downloads"
+os.makedirs(DOWNLOADS_FOLDER, exist_ok=True)
+
+def sanitize_filename(filename):
+    """Remove invalid characters from filename"""
+    # Remove characters that are invalid in filenames
+    return re.sub(r'[<>:"/\\|?*]', '', filename)
+
+def download_audio(url, file_format, platform):
+    try:
+        # Set download options
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'outtmpl': os.path.join(DOWNLOADS_FOLDER, '%(title)s.%(ext)s'),
+            'quiet': True,
+            'no_warnings': True
+        }
+        
+        # Platform-specific handling if needed
+        if platform == "youtube":
+            # You could add YouTube-specific options here
+            pass
+        elif platform == "soundcloud":
+            # You could add SoundCloud-specific options here
+            pass
+        
+        with YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            original_filename = ydl.prepare_filename(info)
+        
+        # Get the actual downloaded filename
+        if os.path.exists(original_filename):
+            downloaded_file = original_filename
+        else:
+            # Sometimes the extension might be different
+            base_name = os.path.splitext(original_filename)[0]
+            for ext in ['.m4a', '.webm', '.mp3', '.opus']:
+                possible_file = base_name + ext
+                if os.path.exists(possible_file):
+                    downloaded_file = possible_file
+                    break
+            else:
+                downloaded_file = original_filename
+        
+        # If user wants specific format, convert if needed
+        if file_format.lower() != "original":
+            target_ext = f".{file_format.lower()}"
+            current_ext = os.path.splitext(downloaded_file)[1].lower()
+            
+            if current_ext != target_ext:
+                # Create sanitized output filename
+                base_name = os.path.splitext(downloaded_file)[0]
+                sanitized_base = sanitize_filename(os.path.basename(base_name))
+                output_file = os.path.join(DOWNLOADS_FOLDER, sanitized_base + target_ext)
+                
+                try:
+                    # Convert audio format
+                    audio = AudioSegment.from_file(downloaded_file)
+                    audio.export(output_file, format=file_format.lower())
+                    downloaded_file = output_file
+                except Exception as e:
+                    print(f"Conversion failed: {e}")
+                    # Return original file if conversion fails
+                    pass
+        
+        return downloaded_file
+    
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Gradio interface
+with gr.Blocks(title="Audio Downloader", theme=gr.themes.Soft()) as iface:
+    gr.Markdown("# 🎵 Audio Downloader")
+    gr.Markdown("Download audio from YouTube or SoundCloud")
+    
+    with gr.Row():
+        platform_choice = gr.Dropdown(
+            choices=["youtube", "soundcloud"], 
+            value="youtube", 
+            label="Select Platform"
+        )
+    
+    with gr.Row():
+        url_input = gr.Textbox(
+            label="URL",
+            placeholder="Enter YouTube or SoundCloud URL here...",
+            scale=4
+        )
+    
+    with gr.Row():
+        format_choice = gr.Dropdown(
+            choices=["original", "mp3", "m4a", "wav", "opus"], 
+            value="mp3", 
+            label="Select Output Format"
+        )
+    
+    with gr.Row():
+        download_button = gr.Button("Download Audio", variant="primary", size="lg")
+    
+    with gr.Row():
+        download_file = gr.File(label="Download Your Audio")
+    
+    with gr.Row():
+        status_output = gr.Textbox(label="Status", visible=False)
+    
+    def download_wrapper(url, file_format, platform):
+        result = download_audio(url, file_format, platform)
+        if isinstance(result, str) and result.startswith("Error:"):
+            return gr.update(value=None, visible=True), gr.update(value=result, visible=True)
+        else:
+            return gr.update(value=result, visible=True), gr.update(value="Download complete!", visible=True)
+    
+    download_button.click(
+        fn=download_wrapper,
+        inputs=[url_input, format_choice, platform_choice],
+        outputs=[download_file, status_output]
+    )
+    
+    # Examples
+    gr.Examples(
+        examples=[
+            ["https://www.youtube.com/watch?v=dQw4w9WgXcQ", "mp3", "youtube"],
+            ["https://soundcloud.com/artist/track", "mp3", "soundcloud"],
+        ],
+        inputs=[url_input, format_choice, platform_choice],
+        label="Try these examples:"
+    )
+
+iface.launch(share=False, show_error=True)