Update app.py

app.py

@@ -3,6 +3,7 @@ from pydub import AudioSegment, effects
 import numpy as np
 import requests
 import io
+import os
 
 app = Flask(__name__)
 
@@ -10,10 +11,10 @@ app = Flask(__name__)
 def process_song():
     url = request.args.get("song")
     if not url:
-        return jsonify({"error": "Please provide a song link with ?song=<url>"}), 400
-
+        return jsonify({"error": "Please provide a song link with ?song=<url>"})
+    
     try:
-        # --- Download audio with safe buffering ---
+        # --- Download audio ---
         headers = {"User-Agent": "Mozilla/5.0"}
         resp = requests.get(
             url,
@@ -24,15 +25,25 @@ def process_song():
         )
         resp.raise_for_status()
 
-        # Collect into BytesIO (avoid truncated MP3)
+        # Buffer to BytesIO
         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(data, format="mp3")
+        # --- Detect format from URL extension ---
+        ext = os.path.splitext(url.split("?")[0])[-1].lstrip(".").lower()
+        fmt = ext if ext in ["mp3", "wav", "flac", "ogg", "m4a"] else None
+
+        # --- Load with auto-detect ---
+        try:
+            sound = AudioSegment.from_file(data, format=fmt)
+        except Exception:
+            data.seek(0)  # retry without forcing format
+            sound = AudioSegment.from_file(data)
+
+        # --- Normalize ---
         sound = effects.normalize(sound)
 
         # --- Split channels ---