From ccc7fec7563c3df9787fa219c1c46d18f869bbeb Mon Sep 17 00:00:00 2001
From: Sakamoto Shiina <68018796+ShiinaSakamoto@users.noreply.github.com>
Date: Sat, 2 Nov 2024 18:30:40 +0900
Subject: [PATCH] =?UTF-8?q?[bugfix]=20=E3=83=95=E3=83=AD=E3=83=B3=E3=83=88?=
 =?UTF-8?q?:=20=E9=81=B8=E6=8A=9E=E4=B8=AD=E3=81=AE=E7=BF=BB=E8=A8=B3?=
 =?UTF-8?q?=E3=82=A8=E3=83=B3=E3=82=B8=E3=83=B3=E3=83=87=E3=83=BC=E3=82=BF?=
 =?UTF-8?q?=E5=BD=A2=E5=BC=8F=E5=A4=89=E6=9B=B4=E3=81=A8UI=E3=81=B8?=
 =?UTF-8?q?=E3=81=AE=E6=B5=81=E3=81=97=E8=BE=BC=E3=81=BF=E3=80=81=20?=
 =?UTF-8?q?=E3=83=90=E3=83=83=E3=82=AF:=20UI=E3=81=B8=E3=81=AErun=E6=99=82?=
 =?UTF-8?q?=E3=81=AE=E3=83=87=E3=83=BC=E3=82=BF=E5=BD=A2=E5=BC=8F=E3=82=92?=
 =?UTF-8?q?set/get=E3=81=AB=E3=81=82=E3=82=8F=E3=81=9B=E3=81=A6=E5=A4=89?=
 =?UTF-8?q?=E6=9B=B4=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src-python/webui_controller.py                              | 6 +++++-
 .../TranslatorSelectorOpenButton.jsx                        | 2 +-
 src-ui/logics/main/useLanguageSettings.js                   | 2 +-
 src-ui/store.js                                             | 5 ++++-
 4 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/src-python/webui_controller.py b/src-python/webui_controller.py
index 636dceff..eb900729 100644
--- a/src-python/webui_controller.py
+++ b/src-python/webui_controller.py
@@ -1568,7 +1568,11 @@ class Controller:
         engines[config.SELECTED_TAB_NO] = engine
         config.SELECTED_TRANSLATION_ENGINES = engines
 
-        self.run(200, self.run_mapping["selected_translation_engines"], config.SELECTED_TRANSLATION_ENGINES)
+        data = {
+            "engines":config.SELECTED_TRANSLATION_ENGINES,
+            "weight_type":config.CTRANSLATE2_WEIGHT_TYPE,
+        }
+        self.run(200, self.run_mapping["selected_translation_engines"], data)
         self.run(200, self.run_mapping["translation_engines"], selectable_engines)
 
     def updateDownloadedWhisperModelWeight(self) -> None:
diff --git a/src-ui/app/main_page/sidebar_section/language_settings/translator_selector_open_button/TranslatorSelectorOpenButton.jsx b/src-ui/app/main_page/sidebar_section/language_settings/translator_selector_open_button/TranslatorSelectorOpenButton.jsx
index a82b9f90..8df127ab 100644
--- a/src-ui/app/main_page/sidebar_section/language_settings/translator_selector_open_button/TranslatorSelectorOpenButton.jsx
+++ b/src-ui/app/main_page/sidebar_section/language_settings/translator_selector_open_button/TranslatorSelectorOpenButton.jsx
@@ -16,7 +16,7 @@ export const TranslatorSelectorOpenButton = () => {
     const selected_translator_name = (currentTranslationEngines.state === "pending")
     ? "Loading..."
     : currentTranslationEngines.data.find(
-        translator_data => translator_data.translator_id === currentSelectedTranslationEngines.data[currentSelectedPresetTabNumber.data]
+        translator_data => translator_data.translator_id === currentSelectedTranslationEngines.data.engines[currentSelectedPresetTabNumber.data]
     )?.translator_name;
 
 
diff --git a/src-ui/logics/main/useLanguageSettings.js b/src-ui/logics/main/useLanguageSettings.js
index 067fdcf4..17206fd1 100644
--- a/src-ui/logics/main/useLanguageSettings.js
+++ b/src-ui/logics/main/useLanguageSettings.js
@@ -100,7 +100,7 @@ export const useLanguageSettings = () => {
     const setSelectedTranslationEngines = (selected_translator) => {
         pendingSelectedTranslationEngines();
         let send_obj = currentSelectedTranslationEngines.data;
-        send_obj[currentSelectedPresetTabNumber.data] = selected_translator;
+        send_obj.engines[currentSelectedPresetTabNumber.data] = selected_translator;
         asyncStdoutToPython("/set/data/selected_translation_engines", send_obj);
     };
 
diff --git a/src-ui/store.js b/src-ui/store.js
index 0df5654f..87235b0f 100644
--- a/src-ui/store.js
+++ b/src-ui/store.js
@@ -123,7 +123,10 @@ export const { atomInstance: Atom_SelectedTargetLanguages, useHook: useStore_Sel
 
 
 export const { atomInstance: Atom_TranslationEngines, useHook: useStore_TranslationEngines } = createAtomWithHook(translator_status, "TranslationEngines");
-export const { atomInstance: Atom_SelectedTranslationEngines, useHook: useStore_SelectedTranslationEngines } = createAtomWithHook({}, "SelectedTranslationEngines");
+export const { atomInstance: Atom_SelectedTranslationEngines, useHook: useStore_SelectedTranslationEngines } = createAtomWithHook({
+    engines: {1:"", 2:"", 3:""},
+    weight_type: "small",
+}, "SelectedTranslationEngines");
 
 
 // Designs