Update app.py

app.py

@@ -303,33 +303,37 @@ class CharacterMaker:
         self.model_lock = threading.Lock()
         self.use_chat_format = False
         self.last_loaded_settings = None
+        self.chat_model_settings = None
+        self.gen_model_settings = None
+
+
 
     def load_model(self, model_type):
         with self.model_lock:
-            # 新しいモデルの設定を取得
-            model_path = os.path.join(MODEL_DIR, self.settings[f'DEFAULT_{model_type.upper()}_MODEL'])
-            n_gpu_layers = self.settings[f'{model_type.lower()}_n_gpu_layers']
-
-            # 現在の設定を取得
-            current_settings = self.get_current_settings(model_type)
+            new_settings = self.get_current_settings(model_type)
             
-            # GENモデルとCHATモデルの設定が同じかチェック
-            if self.are_models_identical():
-                if self.llama and self.last_loaded_settings == current_settings:
-                    print(f"GENモデルとCHATモデルの設定が同じで、既にロードされています。モデルの切り替えはスキップします。")
+            if model_type == 'CHAT':
+                if self.chat_model_settings == new_settings:
+                    print("CHATモデルの設定に変更がないため、リロードをスキップします。")
                     return
-                else:
-                    print(f"GENモデルとCHATモデルの設定が同じですが、まだロードされていないかパラメータが変更されました。モデルをロードします。")
-            else:
-                if self.llama and self.current_model == model_type and self.last_loaded_settings == current_settings:
-                    print(f"{model_type}モデルは既にロードされており、設定も変更されていません。")
+                self.chat_model_settings = new_settings
+            else:  # GEN
+                if self.gen_model_settings == new_settings:
+                    print("GENモデルの設定に変更がないため、リロードをスキップします。")
                     return
-                else:
-                    print(f"{model_type}モデルをロードします。")
+                self.gen_model_settings = new_settings
 
-            self.reload_model(model_type, model_path, n_gpu_layers)
+            if self.are_models_identical():
+                if self.llama and self.current_model == 'SHARED':
+                    print("CHATモデルとGENモデルの設定が同じで、既にロードされています。リロードをスキップします。")
+                    return
+                print("CHATモデルとGENモデルの設定が同じです。共有モデルとしてロードします。")
+                self.reload_model('SHARED', new_settings)
+            else:
+                print(f"{model_type}モデルをロードします。")
+                self.reload_model(model_type, new_settings)
 
-    def reload_model(self, model_type, model_path, n_gpu_layers):
+    def reload_model(self, model_type, settings):
         if self.llama:
             del self.llama
             self.llama = None
@@ -337,13 +341,15 @@ class CharacterMaker:
         self.model_loaded.clear()
 
         try:
-            self.llama = LlamaAdapter(model_path, params, n_gpu_layers)
+            model_path = os.path.join(MODEL_DIR, settings['model_path'])
+            self.llama = LlamaAdapter(model_path, params, settings['n_gpu_layers'])
             self.current_model = model_type
             self.model_loaded.set()
-            self.last_loaded_settings = self.get_current_settings(model_type)
-            print(f"{model_type}モデルをロードしました。モデルパス: {model_path}、GPUレイヤー数: {n_gpu_layers}")
+            print(f"{model_type}モデルをロードしました。モデルパス: {model_path}、GPUレイヤー数: {settings['n_gpu_layers']}")
         except Exception as e:
             print(f"{model_type}モデルのロード中にエラーが発生しました: {e}")
+            import traceback
+            traceback.print_exc()
             self.model_loaded.set()
 
     def get_current_settings(self, model_type):
@@ -358,9 +364,7 @@ class CharacterMaker:
         }
 
     def are_models_identical(self):
-        chat_settings = self.get_current_settings('CHAT')
-        gen_settings = self.get_current_settings('GEN')
-        return chat_settings == gen_settings
+        return self.chat_model_settings == self.gen_model_settings
 
     def generate_response(self, input_str):
         self.load_model('CHAT')
@@ -610,9 +614,17 @@ def build_model_settings(config, section, output):
 
 def apply_settings():
     global temp_settings
+    settings_changed = False
+    
     for section, settings in temp_settings.items():
         for key, value in settings.items():
-            ConfigManager.update_setting(section, key, str(value), DEFAULT_INI_FILE)
+            old_value = ConfigManager.load_settings(DEFAULT_INI_FILE)[section].get(key)
+            if str(value) != str(old_value):
+                ConfigManager.update_setting(section, key, str(value), DEFAULT_INI_FILE)
+                settings_changed = True
+    
+    if not settings_changed:
+        return "設定に変更はありませんでした。"
     
     # iniファイルを再読み込み
     new_config = ConfigManager.load_settings(DEFAULT_INI_FILE)
@@ -640,14 +652,15 @@ def apply_settings():
             int(new_config['GenerateParameters'].get('n_ctx', '10000'))
         )
     
-    # モデルを再ロードするためにcurrent_modelをリセット
-    character_maker.current_model = None
-    character_maker.last_loaded_settings = None
+    # モデルのリロードをトリガー（実際のリロードは次の操作時に行われる）
+    character_maker.chat_model_settings = None
+    character_maker.gen_model_settings = None
     
     # temp_settings をクリア
     temp_settings.clear()
     
-    return "設定をiniファイルに保存し、アプリケーションに反映しました。次回の操作時に新しいモデルがロードされます。"
+    return "設定をiniファイルに保存し、アプリケーションに反映しました。次回の操作時に新しい設定が適用されます。"
+
 # Gradioインターフェース
 def build_gradio_interface():
     global temp_settings