update

finetuned_reply_service.py

@@ -201,6 +201,7 @@ class FinetunedReplyService:
         
         generated_text = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
         
+        # Extract continuation relative to prompt / markers
         if self.model_arch == "encoder_decoder":
             response = generated_text.strip()
         elif "Next appropriate response for Female:" in generated_text:
@@ -208,8 +209,17 @@ class FinetunedReplyService:
         else:
             response = generated_text[len(prompt):].strip()
         
+        # Chỉ lấy dòng đầu tiên
         response = response.split("\n")[0].strip()
         
+        # Loại bỏ các artifact hội thoại
+        if "|||" in response:
+            response = response.split("|||")[0].strip()
+        for prefix in ["Male:", "Nam:", "Female:", "Nữ:"]:
+            if response.lower().startswith(prefix.lower()):
+                response = response[len(prefix):].strip()
+        
+        # Giới hạn 25 từ
         words = response.split()
         if len(words) > 25:
             response = " ".join(words[:25]).rstrip(",.;!?") + "."
@@ -243,9 +253,9 @@ class FinetunedReplyService:
                     persona_text = custom_prompt
             else:
                 persona_text = (
-                    "You are a charming Vietnamese wingman crafting irresistible yet respectful replies for Male. "
-                    "Keep replies concise (<35 words), confident, playful, and always from Male's perspective using 'anh' for self and 'em' for partner. "
-                    "Embed subtle compliments/flirtation while honoring the identified move."
+                    "Bạn là wingman AI tiếng Việt, chuyên giúp Nam soạn đúng một câu trả lời duyên dáng bằng tiếng Việt có dấu. "
+                    "Luôn xưng 'anh' với 'em', phản hồi dựa trên hội thoại, Trigger và Move đã cho. "
+                    "Câu trả lời phải tự nhiên, tinh tế, không dùng tiếng Anh, không lặp lại nguyên văn câu của đối phương."
                 )
             base_prompt.append(persona_text)
         else:

reply_service.py

@@ -29,11 +29,12 @@ MODEL_OPTIONS = [
 MODEL_ID = MODEL_OPTIONS[0]  # Start with vinaLlama
 
 SYSTEM_MSG = (
-    "Bạn là một wingman AI tinh tế, chuyên giúp **Nam** soạn 1 tin nhắn trả lời duy nhất trong hội thoại hẹn hò tiếng Việt. "
+    "Bạn là một wingman AI tinh tế, chuyên giúp **Nam** soạn 1 tin nhắn trả lời duy nhất trong hội thoại hẹn hò **bằng tiếng Việt có dấu**. "
     "Bạn luôn nhìn từ góc nhìn của Nam, xưng 'anh' và gọi đối phương là 'em'. "
     "Bạn được cung cấp hội thoại gần nhất giữa Male/Female cùng với Trigger (intent) và Move (chiến lược). "
     "Nhiệm vụ của bạn: dựa trên hội thoại + Trigger + Move, hãy chọn một hướng phản hồi tự nhiên, duyên dáng, đúng chiến lược (không quá đẩy hay quá lùi so với Move). "
     "Quy tắc cứng: chỉ trả về đúng **1 câu**, ≤25 từ, lịch sự, không giải thích meta, không lặp lại nguyên văn câu của đối phương, không thêm thông tin fact mới. "
+    "Tuyệt đối **không dùng tiếng Anh**, chỉ dùng tiếng Việt có dấu. "
     "Nếu ngữ cảnh chưa rõ (Trigger/Move mâu thuẫn với hội thoại), hãy ưu tiên sự an toàn và mềm mại, có thể hỏi lại nhẹ nhàng nhưng vẫn giữ frame chủ động của Nam. "
     "Phong cách: ấm áp, tự tin nhưng không tự cao; dùng từ đệm tự nhiên (nha/nhé/ạ/dạ) phù hợp; phản chiếu cảm xúc, không phán xét; giữ mạch trò chuyện mở để còn đất tăng tương tác về sau."
 )