Upload 2 files

src/main.py

@@ -18,6 +18,7 @@ import yt_dlp
 from pedalboard import Pedalboard, Reverb, Compressor, HighpassFilter
 from pedalboard.io import AudioFile
 from pydub import AudioSegment
+from my_utils import add_stem_name
 
 from mdx import run_mdx
 from rvc import Config, load_hubert, get_vc, rvc_infer
@@ -361,9 +362,15 @@ def song_cover_pipeline(song_input, voice_model, pitch_change, keep_files,
                     if file and os.path.exists(file):
                         os.remove(file)
 
+            # Add _stemname to each stem
+            ai_cover_path = add_stem_name(ai_cover_path, "_cover")
+            vocals_path = add_stem_name(vocals_path, "_origvocals")
+            instrumentals_path = add_stem_name(instrumentals_path, "_originstr")
+            ai_vocals_mixed_path = add_stem_name(ai_vocals_mixed_path, "_covervocals")
+
             # Returning the stems: AI cover, original vocal, original instrumental, AI generated vocal
 
-            return ai_cover_path, vocals_path, instrumentals_path, ai_vocals_path
+            return ai_cover_path, vocals_path, instrumentals_path, ai_vocals_mixed_path
                 
         except Exception as e:
             raise_exception(str(e), is_webui)

src/my_utils.py

@@ -1,6 +1,6 @@
 import ffmpeg
 import numpy as np
-
+import os
 
 def load_audio(file, sr):
     try:
@@ -19,3 +19,17 @@ def load_audio(file, sr):
         raise RuntimeError(f"Failed to load audio: {e}")
 
     return np.frombuffer(out, np.float32).flatten()
+
+
+def add_stem_name(path, stemname):
+
+    directory, filename_with_ext = os.path.split(path)
+    filename, extension = os.path.splitext(filename_with_ext)
+
+    # Concatenate the stem name with desired suffix and extension
+    new_filename = f"{filename}{stemname}{extension}"
+
+    # Construct the new path
+    new_path = os.path.join(directory, new_filename)
+
+    return new_path