Sync from GitHub: 5c4dc2b10bee528ddb7361e9d8a66872ade756e9

services/audio_separator.py

@@ -224,16 +224,25 @@ class AudioSeparator:
         rhythm_stereo = normalize(rhythm_stereo)
         lead_stereo = normalize(lead_stereo)
         
-        # Save files
-        path_rhythm = os.path.join(output_dir, "guitar_rhythm.mp3")
-        path_lead = os.path.join(output_dir, "guitar_lead.mp3")
+        # MERGE TO SINGLE STEREO FILE (L=Rhythm, R=Lead)
+        # We take the Left channel of the Rhythm stereo mix (which is mono-ish)
+        # And the Right channel of the Lead stereo mix (which is mono-ish)
+        # Or better: Just use the Mono mix of each.
         
-        self._save_audio(rhythm_stereo, sr, path_rhythm)
-        self._save_audio(lead_stereo, sr, path_lead)
+        rhythm_mono = rhythm_stereo.mean(dim=0, keepdim=True)
+        lead_mono = lead_stereo.mean(dim=0, keepdim=True)
+        
+        # Combine: Channel 0 = Rhythm, Channel 1 = Lead
+        guitar_split = torch.cat([rhythm_mono, lead_mono], dim=0)
+        
+        guitar_split = normalize(guitar_split)
+        
+        # Save as single file named "guitar.mp3" (Special split)
+        path = os.path.join(output_dir, "guitar.mp3")
+        self._save_audio(guitar_split, sr, path)
         
         return {
-            "guitar_rhythm": path_rhythm,
-            "guitar_lead": path_lead
+            "guitar": path 
         }
 
     def _save_audio(self, source, sr, path):