correct separation step

app.py

@@ -8,10 +8,6 @@ from typing import Dict
 from pyharp import *
 
 
-# ---------------------------
-# Config
-# ---------------------------
-
 DEMUX_MODELS = ["mdx_extra_q", "mdx_extra", "htdemucs", "mdx_q"]
 
 STEM_CHOICES = {
@@ -22,9 +18,6 @@ STEM_CHOICES = {
     "Instrumental (No Vocals)": "instrumental"
 }
 
-# ---------------------------
-# Stem Separation Logic
-# ---------------------------
 
 def separate_stem(audio_file_path: str, model_name: str, stem_choice: str) -> AudioSignal:
     model = pretrained.get_model(model_name)
@@ -58,26 +51,20 @@ def separate_stem(audio_file_path: str, model_name: str, stem_choice: str) -> Au
 
     return AudioSignal(stem.cpu().numpy().astype('float32'), sample_rate=sr)
 
-# ---------------------------
 # Gradio Callback Function
-# ---------------------------
 
-def process_fn_stem(audio_file_path: str, demucs_model: str, stem_choice: str) -> Dict:
+def process_fn_stem(audio_file_path: str, demucs_model: str, stem_choice: str):
+    """
+    PyHARP process function:
+      - Separates the chosen stem using Demucs.
+      - Saves the stem as a .wav file.
+    """
     stem_signal = separate_stem(audio_file_path, model_name=demucs_model, stem_choice=stem_choice)
-    stem_filename = f"{stem_choice.lower().replace(' ', '_')}.wav"
-    stem_path = save_audio(stem_signal, stem_filename)
-
-    return {
-        "file": stem_path,
-        "stem": stem_choice,
-        "description": f"Separated {stem_choice} stem",
-        "amplitude": 0.7
-    }
+    stem_path = save_audio(stem_signal, f"{stem_choice.lower().replace(' ', '_')}.wav")
+    return stem_path, LabelList(labels=[])
 
-# ---------------------------
-# Model Card (for HARP)
-# ---------------------------
 
+# Model Card 
 model_card = ModelCard(
     name="Demucs Stem Separator",
     description="Uses Demucs to separate a music track into a selected stem.",
@@ -85,10 +72,7 @@ model_card = ModelCard(
     tags=["demucs", "source-separation", "pyharp", "stems"]
 )
 
-# ---------------------------
-# Gradio Interface (Blocks)
-# ---------------------------
-
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# 🎧 Demucs Stem Separator")
     gr.Markdown("Log in with your Hugging Face account to separate music into stems.")