Update app.py

app.py

@@ -108,7 +108,7 @@ def rename_stems(input_file, output_dir, stems, output_format):
         renamed_stems.append(new_path)
     return renamed_stems
 
-def roformer_separator(audio, model_key, seg_size, overlap, model_dir, out_dir, out_format, norm_thresh, amp_thresh):
+def roformer_separator(audio, model_key, seg_size, overlap, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress()):
     """Separate audio using Roformer model."""
     model = ROFORMER_MODELS[model_key]
     with tempfile.TemporaryDirectory() as tmp_dir:
@@ -125,14 +125,18 @@ def roformer_separator(audio, model_key, seg_size, overlap, model_dir, out_dir,
             }
         )
 
+        progress(0.2, desc="Model loaded")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated")
         separation = separator.separate(audio)
 
+        progress(0.9, desc="Stems renamed")
         stems = rename_stems(audio, out_dir, separation, out_format)
 
         return stems[0], stems[1]
 
-def mdx23c_separator(audio, model, seg_size, overlap, model_dir, out_dir, out_format, norm_thresh, amp_thresh):
+def mdx23c_separator(audio, model, seg_size, overlap, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress()):
     """Separate audio using MDX23C model."""
     with tempfile.TemporaryDirectory() as tmp_dir:
         separator = Separator(
@@ -148,14 +152,18 @@ def mdx23c_separator(audio, model, seg_size, overlap, model_dir, out_dir, out_fo
             }
         )
 
+        progress(0.2, desc="Model loaded")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated")
         separation = separator.separate(audio)
 
+        progress(0.9, desc="Stems renamed")
         stems = rename_stems(audio, out_dir, separation, out_format)
 
         return stems[0], stems[1]
 
-def mdx_separator(audio, model, hop_length, seg_size, overlap, denoise, model_dir, out_dir, out_format, norm_thresh, amp_thresh):
+def mdx_separator(audio, model, hop_length, seg_size, overlap, denoise, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress()):
     """Separate audio using MDX-NET model."""
     with tempfile.TemporaryDirectory() as tmp_dir:
         separator = Separator(
@@ -173,14 +181,18 @@ def mdx_separator(audio, model, hop_length, seg_size, overlap, denoise, model_di
             }
         )
 
+        progress(0.2, desc="Model loaded")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated")
         separation = separator.separate(audio)
 
+        progress(0.9, desc="Stems renamed")
         stems = rename_stems(audio, out_dir, separation, out_format)
 
         return stems[0], stems[1]
 
-def vr_separator(audio, model, window_size, aggression, tta, post_process, post_process_threshold, high_end_process, model_dir, out_dir, out_format, norm_thresh, amp_thresh):
+def vr_separator(audio, model, window_size, aggression, tta, post_process, post_process_threshold, high_end_process, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress()):
     """Separate audio using VR ARCH model."""
     with tempfile.TemporaryDirectory() as tmp_dir:
         separator = Separator(
@@ -200,14 +212,18 @@ def vr_separator(audio, model, window_size, aggression, tta, post_process, post_
             }
         )
 
+        progress(0.2, desc="Model loaded")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated")
         separation = separator.separate(audio)
 
+        progress(0.9, desc="Stems renamed")
         stems = rename_stems(audio, out_dir, separation, out_format)
 
         return stems[0], stems[1]
 
-def demucs_separator(audio, model, seg_size, shifts, overlap, segments_enabled, model_dir, out_dir, out_format, norm_thresh, amp_thresh):
+def demucs_separator(audio, model, seg_size, shifts, overlap, segments_enabled, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress()):
     """Separate audio using Demucs model."""
     with tempfile.TemporaryDirectory() as tmp_dir:
         separator = Separator(
@@ -224,9 +240,13 @@ def demucs_separator(audio, model, seg_size, shifts, overlap, segments_enabled,
             }
         )
 
+        progress(0.2, desc="Model loaded")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated")
         separation = separator.separate(audio)
 
+        progress(0.9, desc="Stems renamed")
         stems = rename_stems(audio, out_dir, separation, out_format)
 
         return stems[0], stems[1], stems[2], stems[3]