Update app.py

app.py

@@ -1,48 +1,49 @@
-import gradio as gr
-import subprocess
-import tempfile
 import yt_dlp
-import os
-
-def get_soundcloud_clip(url, duration):
-    try:
-        # Get the direct audio URL from SoundCloud
-        ydl_opts = {
-            "format": "bestaudio/best",
-            "quiet": True,
-            "skip_download": True,
-        }
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=False)
-            audio_url = info["url"]
-
-        # Create a temporary file for the clipped audio
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
-        temp_file.close()
-
-        # Use ffmpeg to download and trim the audio
-        cmd = [
-            "ffmpeg",
-            "-y",
-            "-i", audio_url,
-            "-t", str(duration),
-            "-c", "copy",  # keeps original AAC without re-encoding
-            temp_file.name
-        ]
-        subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-
-        return temp_file.name
-
-    except Exception as e:
-        return f"Error: {e}"
-
-# Gradio UI
+import ffmpeg
+import io
+import gradio as gr
+import requests
+
+def get_snippet(url, duration_sec):
+    # Extract the direct audio URL (bypassing file download)
+    ydl_opts = {
+        "format": "bestaudio/best",
+        "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)
+    )
+    # 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:
-    gr.Markdown("## SoundCloud 30-second Clip Cutter")
     url_input = gr.Textbox(label="SoundCloud URL")
-    duration_slider = gr.Slider(1, 300, value=30, step=1, label="Clip Duration (seconds)")
-    output_audio = gr.Audio(label="Clipped Audio", type="filepath")
-    btn = gr.Button("Get Clip")
-    btn.click(get_soundcloud_clip, inputs=[url_input, duration_slider], outputs=[output_audio])
+    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])
 
 demo.launch()
+