Update app.py

app.py

@@ -369,6 +369,13 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
     if not audio:
         raise ValueError("No audio file provided.")
     
+    # If audio is a tuple (sample_rate, data), save it as a temporary file
+    if isinstance(audio, tuple):
+        sample_rate, data = audio
+        temp_audio_path = os.path.join("/tmp", "temp_audio.wav")
+        scipy.io.wavfile.write(temp_audio_path, sample_rate, data)
+        audio = temp_audio_path
+    
     override_seg_size = override_seg_size == "True"
     
     if os.path.exists(output_dir):
@@ -409,6 +416,10 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
     except Exception as e:
         logger.error(f"Separation failed: {e}")
         raise RuntimeError(f"Separation failed: {e}")
+    finally:
+        # Clean up temporary file if created
+        if isinstance(audio, tuple) and os.path.exists(temp_audio_path):
+            os.remove(temp_audio_path)
 
 @spaces.GPU
 def auto_ensemble_process(audio, model_keys, seg_size, overlap, out_format, use_tta, model_dir, output_dir, norm_thresh, amp_thresh, batch_size, ensemble_method, exclude_stems="", weights_str="", progress=gr.Progress()):
@@ -416,6 +427,13 @@ def auto_ensemble_process(audio, model_keys, seg_size, overlap, out_format, use_
     if not audio or not model_keys:
         raise ValueError("Audio or models missing.")
     
+    # If audio is a tuple (sample_rate, data), save it as a temporary file
+    if isinstance(audio, tuple):
+        sample_rate, data = audio
+        temp_audio_path = os.path.join("/tmp", "temp_audio.wav")
+        scipy.io.wavfile.write(temp_audio_path, sample_rate, data)
+        audio = temp_audio_path
+    
     use_tta = use_tta == "True"
     
     if os.path.exists(output_dir):
@@ -478,7 +496,11 @@ def auto_ensemble_process(audio, model_keys, seg_size, overlap, out_format, use_
     
     progress(1.0, desc="Ensemble complete")
     return output_file, f"Ensemble completed with {ensemble_method}, excluded: {exclude_stems if exclude_stems else 'None'}"
-
+    finally:
+        # Clean up temporary file if created
+        if isinstance(audio, tuple) and os.path.exists(temp_audio_path):
+            os.remove(temp_audio_path)
+            
 def update_roformer_models(category):
     """Update Roformer model dropdown based on selected category."""
     choices = list(ROFORMER_MODELS.get(category, {}).keys()) or []