Update app.py

app.py

@@ -1,34 +1,54 @@
 import gradio as gr
-import yt_dlp
-import tempfile
+from yt_dlp import YoutubeDL
 import os
+from pydub import AudioSegment
 
-def get_snippet(url):
-    # Create a temporary file
-    temp_file = tempfile.NamedTemporaryFile(suffix=".mp3", delete=False)
-    temp_file.close()
-    
+DOWNLOADS_FOLDER = "downloads"
+os.makedirs(DOWNLOADS_FOLDER, exist_ok=True)
+
+def download_soundcloud(url, file_format, duration_sec):
+    # Download best audio first (usually m4a)
     ydl_opts = {
         'format': 'bestaudio/best',
-        'outtmpl': temp_file.name,
-        'postprocessor_args': ['-t', '30'],  # first 30 seconds
-        'quiet': True,
+        'outtmpl': os.path.join(DOWNLOADS_FOLDER, '%(title)s.%(ext)s')
     }
+    
+    with YoutubeDL(ydl_opts) as ydl:
+        info = ydl.extract_info(url, download=True)
+    
+    original_file = os.path.join(DOWNLOADS_FOLDER, f"{info['title']}.{info['ext']}")
+    
+    # Load audio and trim to duration
+    audio = AudioSegment.from_file(original_file)
+    duration_ms = min(len(audio), int(duration_sec * 1000))
+    trimmed_audio = audio[:duration_ms]
+    
+    # Determine output file path
+    if file_format.lower() == "mp3":
+        output_file = os.path.splitext(original_file)[0] + ".mp3"
+        trimmed_audio.export(output_file, format="mp3")
+    elif file_format.lower() == "opus":
+        output_file = os.path.splitext(original_file)[0] + ".opus"
+        trimmed_audio.export(output_file, format="opus")
+    else:  # keep original format
+        output_file = os.path.splitext(original_file)[0] + f".{info['ext']}"
+        trimmed_audio.export(output_file, format=info['ext'])
+    
+    return output_file
 
-    try:
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([url])
-        return temp_file.name
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("### Minimal SoundCloud 30s Stream")
+# Gradio interface
+with gr.Blocks() as iface:
     url_input = gr.Textbox(label="SoundCloud URL", value="https://soundcloud.com/antonio-antetomaso/mutiny-on-the-bounty-closing-titles-cover")
-    audio_out = gr.Audio(label="Audio Snippet")
-    btn = gr.Button("Get 30s Snippet")
+    format_choice = gr.Dropdown(choices=["mp3", "m4a", "opus"], value="mp3", label="Select format")
+    duration_slider = gr.Slider(minimum=5, maximum=300, value=30, step=5, label="Duration (seconds)")
+    download_button = gr.Button("Download")
+    download_file = gr.File(label="Download your track")
     
-    btn.click(get_snippet, inputs=url_input, outputs=audio_out)
+    download_button.click(
+        fn=download_soundcloud,
+        inputs=[url_input, format_choice, duration_slider],
+        outputs=download_file
+    )
+
+iface.launch(show_error=True)
 
-demo.launch()