Add `save_file`

app.py

@@ -138,6 +138,7 @@ class App:
             dd_uvr_model_size = gr.Dropdown(label="Model", value=uvr_params["model_size"],
                                             choices=self.whisper_inf.music_separator.available_models)
             nb_uvr_segment_size = gr.Number(label="Segment Size", value=uvr_params["segment_size"], precision=0)
+            cb_uvr_save_file = gr.Checkbox(label="Save separate files to output", value=uvr_params["save_file"])
 
         with gr.Accordion("VAD", open=False):
             cb_vad_filter = gr.Checkbox(label="Enable Silero VAD Filter", value=vad_params["vad_filter"],
@@ -187,6 +188,7 @@ class App:
                 language_detection_segments=nb_language_detection_segments,
                 prompt_reset_on_temperature=sld_prompt_reset_on_temperature, is_bgm_separate=cb_bgm_separation,
                 uvr_device=dd_uvr_device, uvr_model_size=dd_uvr_model_size, uvr_segment_size=nb_uvr_segment_size,
+                uvr_save_file=cb_uvr_save_file
             ),
             dd_file_format,
             cb_timestamp

configs/default_parameters.yaml

@@ -45,9 +45,10 @@ diarization:
   hf_token: ""
 
 bgm_separation:
-    is_separate_bgm: false
-    model_size: "UVR-MDX-NET-Inst_HQ_1"
-    segment_size: 256
+  is_separate_bgm: false
+  model_size: "UVR-MDX-NET-Inst_HQ_1"
+  segment_size: 256
+  save_file: true
 
 translation:
   deepl:

modules/whisper/whisper_parameter.py

@@ -51,6 +51,7 @@ class WhisperParameters:
     uvr_model_size: gr.Dropdown
     uvr_device: gr.Dropdown
     uvr_segment_size: gr.Number
+    uvr_save_file: gr.Checkbox
     """
     A data class for Gradio components of the Whisper Parameters. Use "before" Gradio pre-processing.
     This data class is used to mitigate the key-value problem between Gradio components and function parameters.
@@ -219,6 +220,9 @@ class WhisperParameters:
         
     uvr_segment_size: gr.Number
         This parameter is related to UVR. Segment size for UVR model.
+        
+    uvr_save_file: gr.Checkbox
+        This parameter is related to UVR. Boolean value that determines whether to save the file or not.
     """
 
     def as_list(self) -> list:
@@ -293,6 +297,7 @@ class WhisperValues:
     uvr_model_size: str
     uvr_device: str
     uvr_segment_size: int
+    uvr_save_file: bool
     """
     A data class to use Whisper parameters.
     """
@@ -348,6 +353,7 @@ class WhisperValues:
                 "is_separate_bgm": self.is_bgm_separate,
                 "model_size": self.uvr_model_size,
                 "segment_size": self.uvr_segment_size,
+                "save_file": self.uvr_save_file,
             },
         }
         return data