Update model_utils.py

model_utils.py

@@ -124,8 +124,15 @@ def generate_answer(question: str) -> str:
         )
 
     generated_ids = outputs[0][inputs["input_ids"].shape[1]:]
-    answer = tokenizer.decode(generated_ids, skip_special_tokens=True)
-    return answer.strip()
+    answer = tokenizer.decode(generated_ids, skip_special_tokens=True).strip()
+
+    # ✅ Enforce 2–3 short sentences
+    # `re` is already imported at the top of this file
+    sentences = re.split(r"(?<=[\.?!…])\s+", answer)
+    short_answer = " ".join(sentences[:3]).strip()
+
+    return short_answer if short_answer else answer
+
 
 
 def answer_from_qa(question: str) -> Optional[str]:
@@ -172,11 +179,16 @@ def laos_history_bot(message: str, history: List) -> str:
 
     direct = answer_from_qa(message)
     if direct:
-        return direct
+        # later you can make this dynamic from the dataset
+        meta = "[ຊັ້ນ M1, ບົດ 1]"  
+        return f"{meta} {direct}"
+
 
     try:
         answer = generate_answer(message)
     except Exception as e:  # noqa: BLE001
         return f"ລະບົບມີບັນຫາ: {e}"
 
-    return answer
+    meta = "[ຊັ້ນ M1, ບົດ 1]"  # placeholder, later make dynamic
+    return f"{meta} {answer}"
+