Update app.py

app.py

@@ -1,49 +1,53 @@
-import yt_dlp
-import ffmpeg
-import io
 import gradio as gr
-import requests
+from yt_dlp import YoutubeDL
+import tempfile
+import os
+
+def download_snippet(url, duration_sec):
+    # Temporary output file
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+    output_path = tmp.name
+    tmp.close()
 
-def get_snippet(url, duration_sec):
-    # Extract the direct audio URL (bypassing file download)
     ydl_opts = {
-        "format": "bestaudio/best",
+        "format": "bestaudio",
+        "outtmpl": output_path,
+        "external_downloader": "ffmpeg",
+        "external_downloader_args": ["-t", str(int(duration_sec))],
         "quiet": True,
         "no_warnings": True,
-        "skip_download": True,
     }
 
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info = ydl.extract_info(url, download=False)
-        audio_url = info['url']
-
-    # Download raw audio data
-    r = requests.get(audio_url, stream=True)
-    r.raise_for_status()
-    input_bytes = io.BytesIO(r.content)
-
-    # Cut the first N seconds with ffmpeg into memory
-    output_bytes = io.BytesIO()
-    (
-        ffmpeg
-        .input('pipe:0')
-        .output('pipe:1', format='mp3', t=duration_sec)
-        .run(input=input_bytes.read(), capture_stdout=True, capture_stderr=True)
+    with YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
+
+    return output_path
+
+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.")
+
+    url = gr.Textbox(
+        label="SoundCloud URL",
+        value="https://soundcloud.com/antonio-antetomaso/mutiny-on-the-bounty-closing-titles-cover"
+    )
+
+    duration = gr.Slider(
+        minimum=1,
+        maximum=300,
+        value=30,
+        step=1,
+        label="Snippet length (seconds)"
     )
-    # Note: ffmpeg.run returns bytes, so we wrap it in BytesIO
-    output_bytes = io.BytesIO(ffmpeg.run(input=input_bytes.read(), capture_stdout=True, capture_stderr=True))
-
-    # Return tuple: file object + filename (for download in Gradio)
-    output_bytes.name = "snippet.mp3"
-    return output_bytes
-
-with gr.Blocks() as demo:
-    url_input = gr.Textbox(label="SoundCloud URL")
-    duration_slider = gr.Slider(1, 60, value=30, step=1, label="Snippet duration (sec)")
-    output_audio = gr.Audio(type="file", label="Playable & downloadable snippet")
-    
-    btn = gr.Button("Generate Snippet")
-    btn.click(get_snippet, inputs=[url_input, duration_slider], outputs=[output_audio])
+
+    btn = gr.Button("Generate snippet")
+
+    audio = gr.Audio(
+        type="filepath",
+        label="Preview & download"
+    )
+
+    btn.click(download_snippet, inputs=[url, duration], outputs=audio)
 
 demo.launch()