Update app.py

app.py

@@ -123,7 +123,7 @@ def prepare_output_dir(input_file, output_dir):
         raise RuntimeError(f"Failed to prepare output directory {out_dir}: {e}")
     return out_dir
 
-def roformer_separator(audio, model_key, seg_size, override_seg_size, batch_size, overlap, pitch_shift, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True, desc="Audio separated...")):
+def roformer_separator(audio, model_key, seg_size, override_seg_size, batch_size, overlap, pitch_shift, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     """Separate audio using Roformer model."""
     base_name = os.path.splitext(os.path.basename(audio))[0]
     print_message(audio, model_key)
@@ -147,7 +147,10 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, batch_size
             }
         )
 
+        progress(0.2, desc="Model loaded...")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated...")
         separation = separator.separate(audio, f"{base_name}_(Stem1)", f"{base_name}_(Stem2)")
         print(f"Separation complete!\nResults: {', '.join(separation)}")
 
@@ -156,7 +159,7 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, batch_size
     except Exception as e:
         raise RuntimeError(f"Roformer separation failed: {e}") from e
 
-def mdx23c_separator(audio, model, seg_size, override_seg_size, batch_size, overlap, pitch_shift, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True, desc="Audio separated...")):
+def mdx23c_separator(audio, model, seg_size, override_seg_size, batch_size, overlap, pitch_shift, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     """Separate audio using MDX23C model."""
     base_name = os.path.splitext(os.path.basename(audio))[0]
     print_message(audio, model)
@@ -179,7 +182,10 @@ def mdx23c_separator(audio, model, seg_size, override_seg_size, batch_size, over
             }
         )
 
+        progress(0.2, desc="Model loaded...")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated...")
         separation = separator.separate(audio, f"{base_name}_(Stem1)", f"{base_name}_(Stem2)")
         print(f"Separation complete!\nResults: {', '.join(separation)}")
 
@@ -188,7 +194,7 @@ def mdx23c_separator(audio, model, seg_size, override_seg_size, batch_size, over
     except Exception as e:
         raise RuntimeError(f"MDX23C separation failed: {e}") from e
 
-def mdx_separator(audio, model, hop_length, seg_size, overlap, batch_size, denoise, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True, desc="Audio separated...")):
+def mdx_separator(audio, model, hop_length, seg_size, overlap, batch_size, denoise, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     """Separate audio using MDX-NET model."""
     base_name = os.path.splitext(os.path.basename(audio))[0]
     print_message(audio, model)
@@ -211,7 +217,10 @@ def mdx_separator(audio, model, hop_length, seg_size, overlap, batch_size, denoi
             }
         )
 
+        progress(0.2, desc="Model loaded...")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated...")
         separation = separator.separate(audio, f"{base_name}_(Stem1)", f"{base_name}_(Stem2)")
         print(f"Separation complete!\nResults: {', '.join(separation)}")
 
@@ -220,7 +229,7 @@ def mdx_separator(audio, model, hop_length, seg_size, overlap, batch_size, denoi
     except Exception as e:
         raise RuntimeError(f"MDX-NET separation failed: {e}") from e
 
-def vr_separator(audio, model, batch_size, 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(track_tqdm=True, desc="Audio separated...")):
+def vr_separator(audio, model, batch_size, 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(track_tqdm=True)):
     """Separate audio using VR ARCH model."""
     base_name = os.path.splitext(os.path.basename(audio))[0]
     print_message(audio, model)
@@ -245,7 +254,10 @@ def vr_separator(audio, model, batch_size, window_size, aggression, tta, post_pr
             }
         )
 
+        progress(0.2, desc="Model loaded...")
         separator.load_model(model_filename=model)
+
+        progress(0.7, desc="Audio separated...")
         separation = separator.separate(audio, f"{base_name}_(Stem1)", f"{base_name}_(Stem2)")
         print(f"Separation complete!\nResults: {', '.join(separation)}")
 
@@ -254,7 +266,7 @@ def vr_separator(audio, model, batch_size, window_size, aggression, tta, post_pr
     except Exception as e:
         raise RuntimeError(f"VR ARCH separation failed: {e}") from e
 
-def demucs_separator(audio, model, seg_size, shifts, overlap, segments_enabled, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True, desc="Audio separated...")):
+def demucs_separator(audio, model, seg_size, shifts, overlap, segments_enabled, model_dir, out_dir, out_format, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     """Separate audio using Demucs model."""
     print_message(audio, model)
     try:
@@ -275,7 +287,10 @@ 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)
         print(f"Separation complete!\nResults: {', '.join(separation)}")