Update app.py

app.py

@@ -13,26 +13,33 @@ def process_song():
         return jsonify({"error": "Please provide a song link with ?song=<url>"}), 400
 
     try:
-        # Download audio file with headers & redirects
+        # --- Download audio with safe buffering ---
         headers = {"User-Agent": "Mozilla/5.0"}
-        resp = requests.get(url, stream=True, headers=headers, allow_redirects=True, timeout=30)
+        resp = requests.get(
+            url,
+            stream=True,
+            headers=headers,
+            allow_redirects=True,
+            timeout=30
+        )
         resp.raise_for_status()
 
-        # Check if it's really audio
-        if "audio" not in resp.headers.get("Content-Type", ""):
-            return jsonify({"error": f"URL did not return audio, got {resp.headers.get('Content-Type')}"}), 400
+        # Collect into BytesIO (avoid truncated MP3)
+        data = io.BytesIO()
+        for chunk in resp.iter_content(chunk_size=1024 * 1024):
+            if chunk:
+                data.write(chunk)
+        data.seek(0)
 
-        # Load into pydub
-        sound = AudioSegment.from_file(io.BytesIO(resp.content), format="mp3")
+        # --- Load into pydub ---
+        sound = AudioSegment.from_file(data, format="mp3")
         sound = effects.normalize(sound)
 
-        # Split channels
+        # --- Split channels ---
         left, right = sound.split_to_mono()
-
-        # Empty output
         output = AudioSegment.silent(duration=len(sound), frame_rate=sound.frame_rate)
 
-        # Settings
+        # --- Pan & effect settings ---
         chunk_ms = 10
         pan_period = 6.0
         base_depth = 0.7
@@ -54,18 +61,18 @@ def process_song():
             stereo_chunk = AudioSegment.from_mono_audiosegments(left_chunk, right_chunk)
             output += stereo_chunk
 
-        # Stereo widening
+        # --- Stereo widening ---
         widened = output.overlay(output.pan(-0.5), gain_during_overlay=-6)
         widened = widened.overlay(output.pan(0.5), gain_during_overlay=-6)
 
-        # Subtle reverb
+        # --- Subtle reverb ---
         reverb = widened.overlay(widened - 8, delay=reverb_delay)
         reverb = reverb.overlay(widened - 14, delay=reverb_delay * 2)
 
-        # Final mastering
+        # --- Final mastering ---
         final = effects.normalize(reverb)
 
-        # Export to memory
+        # --- Export to memory ---
         buf = io.BytesIO()
         final.export(buf, format="mp3")
         buf.seek(0)