Sync from GitHub: eebedee86b0a587bb5ba0d09a2c9b9c613e53af4

main.py

@@ -158,7 +158,7 @@ def process_audio(file_id: str, background_tasks: BackgroundTasks, mode: str = "
         def progress_cb(step, prog):
             update_progress(file_id, step, prog)
             
-        stems = separator.separate(input_path, output_dir, callback=progress_cb, mode=mode)
+        stems, duration = separator.separate(input_path, output_dir, callback=progress_cb, mode=mode)
         
         update_progress(file_id, "Analyzing Rhythm...", 50)
         
@@ -197,7 +197,8 @@ def process_audio(file_id: str, background_tasks: BackgroundTasks, mode: str = "
             "stems": stems_url,
             "midi": midi_files,
             "bpm": bpm,
-            "beats": beats
+            "beats": beats,
+            "duration": duration
         }
         print(f"Final Data for {file_id}: {final_data}") # Debug
         update_progress(file_id, "Completed", 100, status="completed", data=final_data)

services/audio_separator.py

@@ -124,7 +124,8 @@ class AudioSeparator:
                 self._save_audio(source, sr, stem_path)
                 results[name] = stem_path
                 
-        return results
+        input_duration = len(wav_np) / sr
+        return results, input_duration
 
     def _process_guitar(self, source, sr, output_dir):
         """
@@ -226,7 +227,12 @@ class AudioSeparator:
         # source is tensor (channels, samples) on device
         # Move to cpu
         source = source.cpu()
-        source = source.cpu()
+
+        # Normalize to prevent clipping (limit to -1dB peak)
+        peak = source.abs().max()
+        if peak > 0.89: # approx -1dB
+            source = source / peak * 0.89
+            
         # Save using soundfile
         # source is (channels, samples) -> need (samples, channels)
         import soundfile as sf