Update app.py

app.py

@@ -1,217 +1,33 @@
 import gradio as gr
-from yt_dlp import YoutubeDL
+import yt_dlp
 import tempfile
 import os
-import shutil
 
-def download_snippet(url, duration_sec):
-    # Create temporary directory for better file management
-    temp_dir = tempfile.mkdtemp()
-    
-    # Generate a more descriptive filename
-    try:
-        with YoutubeDL({"quiet": True, "no_warnings": True}) as ydl:
-            info = ydl.extract_info(url, download=False)
-            # Create a safe filename from title
-            safe_title = "".join(c for c in info.get('title', 'audio') if c.isalnum() or c in (' ', '-', '_')).rstrip()
-            filename = f"{safe_title}_{duration_sec}s.mp3"
-    except:
-        filename = f"soundcloud_snippet_{duration_sec}s.mp3"
-    
-    output_path = os.path.join(temp_dir, filename)
+def simple_download(url, duration):
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.mp3') as f:
+        output = f.name
     
     ydl_opts = {
-        "format": "bestaudio/best",
-        "outtmpl": output_path,
-        "external_downloader": "ffmpeg",
-        "external_downloader_args": ["-t", str(int(duration_sec))],
-        "quiet": True,
-        "no_warnings": True,
-        "postprocessors": [{
-            "key": "FFmpegExtractAudio",
-            "preferredcodec": "mp3",
-            "preferredquality": "192",
-        }],
+        'format': 'bestaudio',
+        'outtmpl': output,
+        'quiet': True,
+        'postprocessor_args': ['-t', str(min(duration, 120))],
     }
-
-    try:
-        with YoutubeDL(ydl_opts) as ydl:
-            ydl.download([url])
-        
-        # Verify file was created and has content
-        if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
-            return output_path, filename
-        else:
-            raise Exception("Download failed - empty file")
-            
-    except Exception as e:
-        # Clean up on error
-        if os.path.exists(temp_dir):
-            shutil.rmtree(temp_dir, ignore_errors=True)
-        raise gr.Error(f"Download failed: {str(e)}")
-
-def cleanup_temp_files(filepath):
-    """Clean up temporary files after download"""
-    if filepath and os.path.exists(os.path.dirname(filepath)):
-        temp_dir = os.path.dirname(filepath)
-        try:
-            shutil.rmtree(temp_dir, ignore_errors=True)
-        except:
-            pass
-
-with gr.Blocks(title="SoundCloud Snippet Generator", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    # 🎵 SoundCloud Snippet Generator
-    Download only the first *N seconds* of any SoundCloud track.
-    """)
-    
-    with gr.Row():
-        with gr.Column(scale=3):
-            url = gr.Textbox(
-                label="SoundCloud URL",
-                placeholder="https://soundcloud.com/...",
-                value="https://soundcloud.com/antonio-antetomaso/mutiny-on-the-bounty-closing-titles-cover"
-            )
-            
-            with gr.Row():
-                duration = gr.Slider(
-                    minimum=1,
-                    maximum=300,
-                    value=30,
-                    step=1,
-                    label="Snippet length (seconds)"
-                )
-                duration_text = gr.Number(
-                    value=30,
-                    label="Seconds",
-                    precision=0,
-                    minimum=1,
-                    maximum=300
-                )
-            
-            gr.Markdown("### Output")
-            filename_display = gr.Textbox(
-                label="Download as",
-                interactive=False
-            )
-            
-            download_btn = gr.DownloadButton(
-                "Download MP3 File",
-                visible=False
-            )
-            
-            generate_btn = gr.Button(
-                "Generate Snippet",
-                variant="primary",
-                size="lg"
-            )
-            
-            status = gr.Textbox(
-                label="Status",
-                interactive=False,
-                visible=False
-            )
-        
-        with gr.Column(scale=2):
-            audio_player = gr.Audio(
-                label="Preview",
-                type="filepath",
-                interactive=False
-            )
-    
-    # Link slider and number input
-    duration.change(lambda x: x, inputs=[duration], outputs=[duration_text])
-    duration_text.change(lambda x: x, inputs=[duration_text], outputs=[duration])
-    
-    # Store filename for download
-    file_path_store = gr.State()
-    file_name_store = gr.State()
-    
-    def process_and_preview(url, duration_sec):
-        """Process the download and update UI"""
-        if not url or not url.strip():
-            raise gr.Error("Please enter a SoundCloud URL")
-        
-        # Clear previous
-        yield "", None, None, "⏳ Downloading...", gr.DownloadButton(visible=False), gr.Textbox(visible=True)
-        
-        try:
-            filepath, filename = download_snippet(url, duration_sec)
-            
-            yield (
-                filename,  # filename_display
-                filepath,  # file_path_store
-                filename,  # file_name_store
-                "✅ Ready to download!",  # status
-                gr.DownloadButton(visible=True),  # download_btn
-                gr.Textbox(visible=True),  # status visible
-            )
-            
-            # Return audio preview
-            return filepath
-            
-        except Exception as e:
-            raise gr.Error(f"Error: {str(e)}")
-    
-    # Generate button click
-    generate_btn.click(
-        fn=process_and_preview,
-        inputs=[url, duration],
-        outputs=[
-            filename_display,
-            file_path_store,
-            file_name_store,
-            status,
-            download_btn,
-            status
-        ]
-    ).then(
-        fn=lambda x: x,
-        inputs=[file_path_store],
-        outputs=[audio_player]
-    )
-    
-    # Setup download button
-    download_btn.click(
-        fn=lambda filepath, filename: (filepath, filename) if filepath else None,
-        inputs=[file_path_store, file_name_store],
-        outputs=None
-    )
     
-    # Cleanup on new generation
-    url.change(
-        fn=lambda: (
-            "",  # Clear filename
-            None,  # Clear file_path_store
-            None,  # Clear file_name_store
-            "",  # Clear status
-            gr.DownloadButton(visible=False),  # Hide download
-            gr.Textbox(visible=False),  # Hide status box
-            None  # Clear audio player
-        ),
-        outputs=[
-            filename_display,
-            file_path_store,
-            file_name_store,
-            status,
-            download_btn,
-            status,
-            audio_player
-        ]
-    )
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
     
-    # Add footer with info
-    gr.Markdown("""
-    ---
-    **Note:** Downloads are limited to 5 minutes (300 seconds) maximum.
-    Files are temporarily stored and will be cleaned up after download.
-    """)
+    return output
 
-if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        show_error=True
-    )
+iface = gr.Interface(
+    fn=simple_download,
+    inputs=[
+        gr.Textbox(label="SoundCloud URL"),
+        gr.Slider(5, 120, 30, label="Duration (seconds)")
+    ],
+    outputs=gr.Audio(type="filepath", label="Download"),
+    title="SoundCloud Snippet Downloader",
+    description="Download short snippets from SoundCloud (max 120 seconds)"
+)
 
+iface.launch()