Search for sliced audio recursively and show directory tree in errors

Applio may store sliced audio in subdirectories within sliced_audios/.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

pipeline/training.py

@@ -88,24 +88,51 @@ def preprocess(model_name: str, audio_path: str, sample_rate: int = 40000):
         else:
             logger.info(f"  {item} ({os.path.getsize(full)} bytes)")
 
-    # Check multiple possible output directories
+    # Count sliced audio files (Applio may nest them in subdirectories)
+    def count_wav_files(directory):
+        """Count .wav files recursively."""
+        count = 0
+        if os.path.exists(directory):
+            for root, dirs, files in os.walk(directory):
+                for f in files:
+                    if f.endswith(".wav"):
+                        count += 1
+        return count
+
     sliced_dir = os.path.join(exp_dir, "sliced_audios")
     sliced_16k_dir = os.path.join(exp_dir, "sliced_audios_16k")
 
-    if os.path.exists(sliced_dir) and len(os.listdir(sliced_dir)) > 0:
-        n_slices = len(os.listdir(sliced_dir))
-        logger.info(f"Preprocessing complete: {n_slices} slices created.")
-        return n_slices
-    elif os.path.exists(sliced_16k_dir) and len(os.listdir(sliced_16k_dir)) > 0:
-        n_slices = len(os.listdir(sliced_16k_dir))
-        logger.info(f"Preprocessing complete: {n_slices} 16k slices created (no native rate slices).")
-        return n_slices
+    n_slices = count_wav_files(sliced_dir)
+    n_slices_16k = count_wav_files(sliced_16k_dir)
+
+    # Debug: show exact directory structure
+    def dir_tree(path, depth=2):
+        """Show directory tree for debugging."""
+        items = []
+        if os.path.exists(path):
+            for item in os.listdir(path):
+                full = os.path.join(path, item)
+                if os.path.isdir(full) and depth > 0:
+                    sub_items = os.listdir(full)
+                    items.append(f"{item}/({len(sub_items)} items: {sub_items[:3]})")
+                else:
+                    items.append(item)
+        return items
+
+    logger.info(f"sliced_audios tree: {dir_tree(sliced_dir)}")
+    logger.info(f"sliced_audios_16k tree: {dir_tree(sliced_16k_dir)}")
+    logger.info(f"WAV counts: sliced={n_slices}, 16k={n_slices_16k}")
+
+    if n_slices > 0 or n_slices_16k > 0:
+        total = max(n_slices, n_slices_16k)
+        logger.info(f"Preprocessing complete: {total} slices created.")
+        return total
     else:
         raise RuntimeError(
             f"Preprocessing produced no audio slices. "
-            f"exp_dir contents: {os.listdir(exp_dir)}. "
-            f"stdout: {result.stdout[-300:] if result.stdout else 'empty'}. "
-            f"stderr: {result.stderr[-300:] if result.stderr else 'empty'}"
+            f"sliced_audios: {dir_tree(sliced_dir)}. "
+            f"sliced_audios_16k: {dir_tree(sliced_16k_dir)}. "
+            f"stdout: {result.stdout[-200:] if result.stdout else 'empty'}"
         )