From 9b456c83bbca89c2b0072db2e6274364f4e81b68 Mon Sep 17 00:00:00 2001
From: wataru <wataru@fdev.local.com>
Date: Mon, 26 Jun 2023 00:51:25 +0900
Subject: [PATCH] save server config

---
 .gitignore                                  |  2 +-
 server/const.py                             |  1 +
 server/voice_changer/VoiceChangerManager.py | 40 +++++++++++++++++----
 3 files changed, 36 insertions(+), 7 deletions(-)

diff --git a/.gitignore b/.gitignore
index e601b654f..ae62d30a9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -16,7 +16,7 @@ server/in.wav
 server/out.wav
 server/G_*.pth
 server/train_config.json
-
+server/stored_setting.json
 # v.1.3.xテスト用モデルフォルダ
 server/v13
 
diff --git a/server/const.py b/server/const.py
index ed4ae5ef5..bb5ac0a15 100644
--- a/server/const.py
+++ b/server/const.py
@@ -21,6 +21,7 @@
     "RVC",
 ]
 
+STORED_SETTING_FILE = "stored_setting.json"
 
 tmpdir = tempfile.TemporaryDirectory()
 SSL_KEY_DIR = os.path.join(tmpdir.name, "keys") if hasattr(sys, "_MEIPASS") else "keys"
diff --git a/server/voice_changer/VoiceChangerManager.py b/server/voice_changer/VoiceChangerManager.py
index 9ca928c64..94631228b 100644
--- a/server/voice_changer/VoiceChangerManager.py
+++ b/server/voice_changer/VoiceChangerManager.py
@@ -7,7 +7,7 @@
 from voice_changer.ModelSlotManager import ModelSlotManager
 from voice_changer.RVC.RVCModelMerger import RVCModelMerger
 from voice_changer.VoiceChanger import VoiceChanger
-from const import UPLOAD_DIR, ModelType
+from const import STORED_SETTING_FILE, UPLOAD_DIR, ModelType
 from voice_changer.utils.LoadModelParams import LoadModelParamFile, LoadModelParams
 from voice_changer.utils.ModelMerger import MergeElement, ModelMergerRequest
 from voice_changer.utils.VoiceChangerModel import AudioInOut
@@ -69,6 +69,28 @@ def __init__(self, params: VoiceChangerParams):
         # thread = threading.Thread(target=self.serverDevice.start, args=())
         # thread.start()
 
+        # 設定保存用情報
+        self.stored_setting: dict[str, str | int | float] = {}
+        if os.path.exists(STORED_SETTING_FILE):
+            self.stored_setting = json.load(open(STORED_SETTING_FILE, "r", encoding="utf-8"))
+        for key, val in self.stored_setting.items():
+            self.update_settings(key, val)
+
+    def store_setting(self, key: str, val: str | int | float):
+        saveItemForServerDevice = ["enableServerAudio", "serverInputDeviceId", "serverOutputDeviceId", "serverReadChunkSize", "serverInputAudioGain", "serverOutputAudioGain"]
+        saveItemForVoiceChanger = ["crossFadeOffsetRate", "crossFadeEndRate", "crossFadeOverlapSize"]
+        saveItemForVoiceChangerManager = ["modelSlotIndex"]
+        saveItemForRVC = ["extraConvertSize", "gpu", "silentThreshold"]
+
+        saveItem = []
+        saveItem.extend(saveItemForServerDevice)
+        saveItem.extend(saveItemForVoiceChanger)
+        saveItem.extend(saveItemForVoiceChangerManager)
+        saveItem.extend(saveItemForRVC)
+        if key in saveItem:
+            self.stored_setting[key] = val
+            json.dump(self.stored_setting, open(STORED_SETTING_FILE, "w"))
+
     def _get_gpuInfos(self):
         devCount = torch.cuda.device_count()
         gpus = []
@@ -83,7 +105,7 @@ def _get_gpuInfos(self):
     def get_instance(cls, params: VoiceChangerParams):
         if cls._instance is None:
             cls._instance = cls(params)
-            cls._instance.voiceChanger = VoiceChanger(params)
+            # cls._instance.voiceChanger = VoiceChanger(params)
         return cls._instance
 
     def loadModel(self, params: LoadModelParams):
@@ -149,12 +171,11 @@ def get_info(self):
         info = self.serverDevice.get_info()
         data.update(info)
 
-        if hasattr(self, "voiceChanger"):
+        if self.voiceChanger is not None:
             info = self.voiceChanger.get_info()
             data.update(info)
-            return data
-        else:
-            return {"status": "ERROR", "msg": "no model loaded"}
+
+        return data
 
     def get_performance(self):
         if hasattr(self, "voiceChanger"):
@@ -209,12 +230,19 @@ def generateVoiceChanger(self, val: int):
             return
 
     def update_settings(self, key: str, val: str | int | float):
+        self.store_setting(key, val)
+
         if key in self.settings.intData:
             newVal = int(val)
             if key == "modelSlotIndex":
                 newVal = newVal % 1000
                 print(f"[Voice Changer] model slot is changed {self.settings.modelSlotIndex} -> {newVal}")
                 self.generateVoiceChanger(newVal)
+                # キャッシュ設定の反映
+                for k, v in self.stored_setting.items():
+                    if k != "modelSlotIndex":
+                        self.update_settings(k, v)
+
             setattr(self.settings, key, newVal)
 
         else: