From af0bb11c4e5af0cd5281861d576a91d75ba50a75 Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Mon, 10 Jun 2024 21:47:01 +0100
Subject: [PATCH] Path normalisation when splitting files

---
 finetune.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/finetune.py b/finetune.py
index 4d0561f..8ae265c 100644
--- a/finetune.py
+++ b/finetune.py
@@ -435,10 +435,16 @@ def format_audio_list(target_language, whisper_model, out_path, max_sample_lengt
                 while audio.size(-1) > max_duration * sr:
                     split_audio = audio[:, :int(max_duration * sr)]
                     audio = audio[:, int(max_duration * sr):]
-                    split_file_name = f"wavs/{audio_file_name}_{str(i).zfill(8)}.wav"
-                    split_absolute_path = os.path.join(audio_folder, split_file_name)
+                    # Normalize the file path
+                    split_file_name = f"{audio_file_name}_{str(i).zfill(8)}.wav"
+                    split_relative_path = os.path.join('wavs', split_file_name)
+                    split_absolute_path = os.path.normpath(os.path.join(audio_folder, split_relative_path))
+                    # Ensure the directory exists
+                    os.makedirs(os.path.dirname(split_absolute_path), exist_ok=True)
+                    # Save the split audio
                     torchaudio.save(split_absolute_path, split_audio, sr)
-                    metadata["audio_file"].append(split_file_name)
+                    # Update metadata
+                    metadata["audio_file"].append(split_relative_path)
                     metadata["text"].append(sentence)
                     metadata["speaker_name"].append(speaker_name)
                     i += 1