From 330d60edc6d94049c2af8c681b4ddeb5541d32ed Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Wed, 10 Jan 2024 22:05:43 +0900
Subject: [PATCH] Fix autolaunch bug

---
 webui_style_vectors.py | 12 +++++++++---
 webui_train.py         |  2 +-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/webui_style_vectors.py b/webui_style_vectors.py
index 951f845..e935023 100644
--- a/webui_style_vectors.py
+++ b/webui_style_vectors.py
@@ -6,6 +6,7 @@ import shutil
 import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
+import yaml
 from scipy.spatial.distance import pdist, squareform
 from sklearn.cluster import DBSCAN, AgglomerativeClustering, KMeans
 from sklearn.manifold import TSNE
@@ -15,6 +16,12 @@ from common.constants import DEFAULT_STYLE
 from common.log import logger
 from config import config
 
+# Get path settings
+with open(os.path.join("configs", "paths.yml"), "r", encoding="utf-8") as f:
+    path_config: dict[str, str] = yaml.safe_load(f.read())
+    dataset_root = path_config["dataset_root"]
+    # assets_root = path_config["assets_root"]
+
 MAX_CLUSTER_NUM = 10
 MAX_AUDIO_NUM = 10
 
@@ -31,7 +38,7 @@ centroids = []
 
 def load(model_name, reduction_method):
     global wav_files, x, x_reduced, mean
-    wavs_dir = os.path.join("Data", model_name, "wavs")
+    wavs_dir = os.path.join(dataset_root, model_name, "wavs")
     style_vector_files = [
         os.path.join(wavs_dir, f) for f in os.listdir(wavs_dir) if f.endswith(".npy")
     ]
@@ -103,7 +110,6 @@ def representative_wav_files(cluster_id, num_files=1):
     # 平均距離が最も小さい順にnum_files個のインデックスを取得
     closest_indices = np.argsort(mean_distances)[:num_files]
 
-    # 最も近いメドイドの元のインデックスを取得
     return cluster_indices[closest_indices]
 
 
@@ -237,7 +243,7 @@ def save_style_vectors_from_files(
         return f"スタイル名が重複しています。"
     style_vectors = [mean]
 
-    wavs_dir = os.path.join("Data", model_name, "wavs")
+    wavs_dir = os.path.join(dataset_root, model_name, "wavs")
     for audio_file in audio_files:
         path = os.path.join(wavs_dir, audio_file)
         if not os.path.exists(path):
diff --git a/webui_train.py b/webui_train.py
index 8694215..ac461bb 100644
--- a/webui_train.py
+++ b/webui_train.py
@@ -511,4 +511,4 @@ if __name__ == "__main__":
     )
     args = parser.parse_args()
 
-    app.launch(inbrowser=args.no_autolaunch, server_name=args.server_name)
+    app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name)