From 83f09d5278c27b4d86bf53fdb37579277d9eb679 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Mon, 9 Sep 2024 17:28:10 +0900
Subject: [PATCH] Fix: style_by_dirs in WebUI

---
 README.md                    | 2 +-
 gradio_tabs/style_vectors.py | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 6126c81..910d3d5 100644
--- a/README.md
+++ b/README.md
@@ -79,9 +79,9 @@ powershell -c "irm https://astral.sh/uv/install.ps1 | iex"
 git clone https://github.com/litagin02/Style-Bert-VITS2.git
 cd Style-Bert-VITS2
 uv venv venv
+venv\Scripts\activate
 uv pip install "torch<2.4" "torchaudio<2.4" --index-url https://download.pytorch.org/whl/cu118
 uv pip install -r requirements.txt
-venv\Scripts\activate
 python initialize.py  # 必要なモデルとデフォルトTTSモデルをダウンロード
 ```
 最後を忘れずに。
diff --git a/gradio_tabs/style_vectors.py b/gradio_tabs/style_vectors.py
index f6ba949..c9014cd 100644
--- a/gradio_tabs/style_vectors.py
+++ b/gradio_tabs/style_vectors.py
@@ -336,7 +336,12 @@ def save_style_vectors_by_dirs(model_name: str, audio_dir_str: str):
     if style_vector_path.exists():
         logger.info(f"Backup {style_vector_path} to {style_vector_path}.bak")
         shutil.copy(style_vector_path, f"{style_vector_path}.bak")
-    save_styles_by_dirs(audio_dir, result_dir)
+    save_styles_by_dirs(
+        wav_dir=audio_dir,
+        output_dir=result_dir,
+        config_path=config_path,
+        config_output_path=config_path,
+    )
     return f"成功!\n{result_dir}にスタイルベクトルを保存しました。"