Detectar dinámicamente carpeta de stems de Demucs 6stems

audio_pipeline.py

@@ -86,32 +86,29 @@ def reducir_ruido(input_file, output_file, noise_duration=0.5):
 import os
 import subprocess
 import sys
-import warnings
 import torch
-
+import warnings
 import librosa
 import numpy as np
 import soundfile as sf
 import noisereduce as nr
 
-# Suprime warnings de runtime (p.ej. invalid value encountered in divide)
 warnings.filterwarnings("ignore", category=RuntimeWarning)
 
-# Directorio base donde guardaremos todos los stems
 BASE_STEMS_DIR = "data/stems"
 
 def separar_audio_demucs_6stems(input_file, model="htdemucs_6s"):
     """
     Separa 6 stems con Demucs (vocals, drums, bass, guitar, piano, other),
-    los guarda en data/stems/<model>/<track_folder>/ y devuelve la ruta de ese folder.
+    busca recursivamente la carpeta con .wav y la devuelve.
     """
+    base = os.path.splitext(os.path.basename(input_file))[0]
     out_root = os.path.join(BASE_STEMS_DIR, model)
     os.makedirs(out_root, exist_ok=True)
 
     device = "cuda" if torch.cuda.is_available() else "cpu"
     cmd = [
-        sys.executable,
-        "-m", "demucs",
+        sys.executable, "-m", "demucs",
         "-n", model,
         "--out", out_root,
         "--device", device,
@@ -119,17 +116,15 @@ def separar_audio_demucs_6stems(input_file, model="htdemucs_6s"):
     ]
     subprocess.run(cmd, check=True)
 
-    # Demucs crea un subdirectorio con el nombre de la pista dentro de out_root
-    # Encuentra el primer subdirectorio que contenga archivos .wav
-    for entry in os.listdir(out_root):
-        candidate = os.path.join(out_root, entry)
-        if os.path.isdir(candidate):
-            # Verifica que tenga stems
-            wavs = [f for f in os.listdir(candidate) if f.endswith('.wav')]
-            if wavs:
-                return candidate
+    # Ahora buscamos recursivamente el primer directorio que contenga .wav
+    for root, dirs, files in os.walk(out_root):
+        if any(f.endswith(".wav") for f in files):
+            return root
+
+    # Si no aparece ninguno, error
     raise FileNotFoundError(f"No se encontró el folder de stems en {out_root}")
 
+
 def limpiar_stems(stems_dir):
     """Aplica reducción de ruido a cada stem (_cleaned.wav)."""
     for archivo in os.listdir(stems_dir):