Update app.py

app.py

@@ -9,11 +9,13 @@ 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):
+def download_audio(url, file_format):
     try:
+        if not url or not url.strip():
+            return None, "Error: Please enter a URL"
+        
         # Set download options
         ydl_opts = {
             'format': 'bestaudio/best',
@@ -22,14 +24,6 @@ def download_audio(url, file_format, platform):
             '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)
@@ -40,7 +34,7 @@ def download_audio(url, file_format, platform):
         else:
             # Sometimes the extension might be different
             base_name = os.path.splitext(original_filename)[0]
-            for ext in ['.m4a', '.webm', '.mp3', '.opus']:
+            for ext in ['.m4a', '.webm', '.mp3', '.opus', '.flac', '.wav']:
                 possible_file = base_name + ext
                 if os.path.exists(possible_file):
                     downloaded_file = possible_file
@@ -63,73 +57,38 @@ def download_audio(url, file_format, platform):
                     # Convert audio format
                     audio = AudioSegment.from_file(downloaded_file)
                     audio.export(output_file, format=file_format.lower())
+                    # Optionally remove original file after conversion
+                    # os.remove(downloaded_file)
                     downloaded_file = output_file
                 except Exception as e:
                     print(f"Conversion failed: {e}")
                     # Return original file if conversion fails
                     pass
         
-        return downloaded_file
+        return downloaded_file, "Download completed successfully!"
     
     except Exception as e:
-        return f"Error: {str(e)}"
+        return None, 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:"
-    )
+# Create Gradio interface
+iface = gr.Interface(
+    fn=download_audio,
+    inputs=[
+        gr.Textbox(label="YouTube or SoundCloud URL", placeholder="Enter URL here..."),
+        gr.Dropdown(["original", "mp3", "m4a", "wav"], value="mp3", label="Output Format")
+    ],
+    outputs=[
+        gr.File(label="Download Audio"),
+        gr.Textbox(label="Status")
+    ],
+    title="🎵 Audio Downloader",
+    description="Download audio from YouTube or SoundCloud. Supports MP3, M4A, WAV formats.",
+    examples=[
+        ["https://www.youtube.com/watch?v=dQw4w9WgXcQ", "mp3"],
+        ["https://soundcloud.com/aviciiofficial/levels", "mp3"],
+    ]
+)
 
-iface.launch(share=False, show_error=True)
+# Launch the interface
+if __name__ == "__main__":
+    iface.launch(share=False, show_error=True)