Update services/chat_service.py

services/chat_service.py

@@ -12,58 +12,85 @@ class ChatService:
         self.tts_service = tts_service
         self.multilingual_manager = rag_system.multilingual_manager
 
-    def respond(self, message: str,chat_history: List[Tuple[str, str]]) -> tuple:
+    def respond(self, message: str, chat_history: List[Tuple[str, str]]) -> tuple:
         """Tạo phản hồi cho tin nhắn đầu vào dựa trên lịch sử trò chuyện và ngôn ngữ."""
         if chat_history is None:
             chat_history = []
 
-        language = self.multilingual_manager.detect_language(message)
-        llm_model = self.multilingual_manager.get_llm_model(language)
-
-        messages = []
-        for user_msg, assistant_msg in chat_history:
-            messages.append({"role": "user", "content": user_msg})
-            messages.append({"role": "assistant", "content": assistant_msg})
-        messages.append({"role": "user", "content": message})
+        # Kiểm tra message hợp lệ
+        if not message or message.strip() == "":
+            return "", chat_history, chat_history, None, "unknown"
 
         try:
+            language = self.multilingual_manager.detect_language(message)
+            llm_model = self.multilingual_manager.get_llm_model(language)
+
+            # Xây dựng messages từ chat history
+            messages = []
+            for user_msg, assistant_msg in chat_history:
+                messages.append({"role": "user", "content": user_msg})
+                messages.append({"role": "assistant", "content": assistant_msg})
+            messages.append({"role": "user", "content": message})
+
+            # Tìm kiếm RAG
             rag_results = self.rag_system.semantic_search(message, top_k=3)
             context_text = ""
             if rag_results:
-                context_text = "\n Thông tin tham khảo:\n +".join([result.document for result in rag_results])
+                context_text = "\n".join([f"- {result.text}" for result in rag_results])
 
+            # Tạo system message
             if language == 'vi':
-                system_message = {
-                    "role": "system",
-                    "content": f"Bạn là trợ lý AI hữu ích chuyên về tiếng Việt. Sử dụng thông tin từ cơ sở kiến thức khi có liên quan. Luôn trả lời bằng tiếng Việt tự nhiên.{context_text}"
-                }
+                system_content = f"""Bạn là trợ lý AI hữu ích chuyên về tiếng Việt. Hãy trả lời một cách tự nhiên và thân thiện.
+                
+Thông tin tham khảo:
+{context_text if context_text else "Không có thông tin tham khảo cụ thể."}
+
+Nếu thông tin tham khảo có liên quan, hãy sử dụng nó. Nếu không, dựa vào kiến thức chung của bạn. Luôn trả lời bằng tiếng Việt."""
             else:
-                system_message = {
-                    "role": "system",
-                    "content": f"You are a helpful AI assistant. Use information from the knowledge base when relevant. Always respond in natural language matching the user's language.{context_text}"
-                }
+                system_content = f"""You are a helpful AI assistant. Please respond naturally and friendly.
+
+Reference information:
+{context_text if context_text else "No specific reference information available."}
+
+If reference information is relevant, use it. Otherwise, rely on your general knowledge. Always respond in the same language as the user."""
+
+            system_message = {"role": "system", "content": system_content}
+            
+            # Tạo messages với context
             messages_with_context = [system_message] + messages
 
+            # Gọi Groq API
             completion = self.groq_client.chat.completions.create(
                 model=llm_model,
                 messages=messages_with_context,
                 max_tokens=512,
                 temperature=0.7,
             )
-            assistant_message  = completion.choices[0].message['content'].strip()
+            
+            # SỬA QUAN TRỌNG: Cách truy cập response đúng
+            assistant_message = completion.choices[0].message.content.strip()
 
+            # Cập nhật chat history
             chat_history.append((message, assistant_message))
 
+            # Tạo TTS
             tts_audio_path = None
             if assistant_message and not assistant_message.startswith("Error"):
-                tts_bytes = self.tts_service.text_to_speech(assistant_message, language)
-                if tts_bytes:
-                    tts_audio_path = self.tts_service.save_tts_audio(tts_bytes)
+                try:
+                    tts_bytes = self.tts_service.text_to_speech(assistant_message, language)
+                    if tts_bytes:
+                        tts_audio_path = self.tts_service.save_tts_audio(tts_bytes)
+                except Exception as tts_error:
+                    print(f"⚠️ Lỗi TTS: {tts_error}")
+                    # Vẫn tiếp tục nếu TTS fail
+
+            return "", chat_history, chat_history, tts_audio_path, language
+
         except Exception as e:
-            assistant_message = f"Error trong quá trình tạo phản hồi: {e}"
-            chat_history.append((message, assistant_message))
-            tts_audio_path = None
-        return  "", chat_history, tts_audio_path, language
-    def clear_chat_history(self, chat_history: List[Tuple[str,str]]) -> tuple:
-        """Xóa lịch sử trò chuyện (nếu lưu trữ lịch sử trong phiên làm việc)."""
-        return [],[]
+            error_msg = f"❌ Lỗi trong quá trình tạo phản hồi: {str(e)}"
+            chat_history.append((message, error_msg))
+            return "", chat_history, chat_history, None, "unknown"
+
+    def clear_chat_history(self, chat_history: List[Tuple[str, str]]) -> tuple:
+        """Xóa lịch sử trò chuyện."""
+        return [], []