def chat(self, user_message, chat_history, socratic_mode=False, service_type='openai'):

chatbot.py

@@ -10,7 +10,7 @@ class Chatbot:
         self.jutor_chat_key = config.get('jutor_chat_key')
         self.transcript_text = self.get_transcript_text(config.get('transcript'))
         self.key_moments_text = self.get_key_moments_text(config.get('key_moments'))
-        self.ai_name = config.get('ai_name')
+        self.ai_model_name = config.get('ai_model_name')
         self.ai_client = config.get('ai_client')
         self.instructions = config.get('instructions')
 
@@ -39,15 +39,17 @@ class Chatbot:
         return key_moments_text
 
 
-    def chat(self, user_message, chat_history, socratic_mode=False, service_type='openai'):
-        messages = self.prepare_messages(chat_history, user_message)
-        system_prompt = self.instructions
-        service_type_list = ['openai', 'claude3', 'groq_llama3', 'groq_mixtral']
-        if service_type in service_type_list:
+    def chat(self, user_message, chat_history):
+        try:
+            messages = self.prepare_messages(chat_history, user_message)
+            system_prompt = self.instructions
+            service_type = self.ai_model_name
             response_text = self.chat_with_service(service_type, system_prompt, messages)
-            return response_text
-        else:
-            raise gr.Error("不支持此服務")
+        except Exception as e:
+            print(f"Error: {e}")
+            response_text = "學習精靈有點累，請稍後再試！"
+
+        return response_text
 
     def prepare_messages(self, chat_history, user_message):
         messages = []