Update app.py

app.py

@@ -2,52 +2,216 @@ import gradio as gr
 from yt_dlp import YoutubeDL
 import tempfile
 import os
+import shutil
 
 def download_snippet(url, duration_sec):
-    # Temporary output file
-    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
-    output_path = tmp.name
-    tmp.close()
-
+    # 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)
+    
     ydl_opts = {
-        "format": "bestaudio",
+        "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",
+        }],
     }
 
-    with YoutubeDL(ydl_opts) as ydl:
-        ydl.download([url])
-
-    return output_path
+    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)}")
 
-with gr.Blocks(title="SoundCloud 30s Stream Snippet") as demo:
-    gr.Markdown("## 🎵 SoundCloud Snippet Generator")
-    gr.Markdown("Downloads only the first *N seconds* using yt-dlp + ffmpeg.")
+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
 
-    url = gr.Textbox(
-        label="SoundCloud URL",
-        value="https://soundcloud.com/antonio-antetomaso/mutiny-on-the-bounty-closing-titles-cover"
+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]
     )
-
-    duration = gr.Slider(
-        minimum=1,
-        maximum=300,
-        value=30,
-        step=1,
-        label="Snippet length (seconds)"
+    
+    # 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
     )
-
-    btn = gr.Button("Generate snippet")
-
-    audio = gr.Audio(
-        type="filepath",
-        label="Preview & download"
+    
+    # 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
+        ]
     )
+    
+    # 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.
+    """)
 
-    btn.click(download_snippet, inputs=[url, duration], outputs=audio)
-
-demo.launch()
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_error=True
+    )