Update app.py

app.py

@@ -72,7 +72,11 @@ class LlamaCppAdapter:
     def generate(self, prompt, max_new_tokens=4096, temperature=0.5, top_p=0.7, top_k=80, stop=["<END>"]):
         if self.llama is None:
             return {"choices": [{"text": "モデルの初期化に失敗しました。"}]}
-        return self._generate(prompt, max_new_tokens, temperature, top_p, top_k, stop)
+        try:
+            return self._generate(prompt, max_new_tokens, temperature, top_p, top_k, stop)
+        except Exception as e:
+            print(f"生成中にエラーが発生しました: {e}")
+            return {"choices": [{"text": "生成中にエラーが発生しました。"}]}
 
     def _generate(self, prompt: str, max_new_tokens: int, temperature: float, top_p: float, top_k: int, stop: list):
         return self.llama(
@@ -88,7 +92,11 @@ class LlamaCppAdapter:
 # キャラクターメーカークラス
 class CharacterMaker:
     def __init__(self):
-        self.llama = LlamaCppAdapter(model_path)
+        try:
+            self.llama = LlamaCppAdapter(model_path)
+        except Exception as e:
+            print(f"LlamaCppAdapter の初期化中にエラーが発生しました: {e}")
+            self.llama = None
         self.history = []
         self.settings = load_settings_from_ini()
         if not self.settings:
@@ -159,7 +167,11 @@ class CharacterMaker:
 
     def reset(self):
         self.history = []
-        self.llama = LlamaCppAdapter(model_path)
+        try:
+            self.llama = LlamaCppAdapter(model_path)
+        except Exception as e:
+            print(f"LlamaCppAdapter の再初期化中にエラーが発生しました: {e}")
+            self.llama = None
 
 character_maker = CharacterMaker()