update message

app/constants.py

@@ -118,7 +118,7 @@ VIETNAMESE_STOP_WORDS = {
     "mà", "mỗi", "một", "nên", "nếu", "ngay", "như", "nhưng", "những",
     "nơi", "nữa", "phải", "qua", "ra", "rằng", "rất", "rồi", "sau", "sẽ",
     "thì", "trên", "trước", "từ", "từng", "và", "vẫn", "vào", "vậy", "về",
-    "vì", "việc", "với", "xong", "phạt", "xử", "xe", "vi phạm"
+    "vì", "việc", "với", "xong", "phạt", "xử", "xe"
     # ... thêm các từ khác bạn muốn loại bỏ
 }
 
@@ -156,4 +156,32 @@ SUMMARY_STATUS_MESSAGES = [
     "Mình đang chốt lại các điểm chính để trả lời một cách trọn vẹn"
 ]
 
+# Processing status messages for Facebook notifications when processing/searching
+PROCESSING_STATUS_MESSAGES = [
+    "Okie, mình kiểm tra lại thông tin liền nha!",
+    "Để mình rà lại chút, xong mình phản hồi ngay.",
+    "Mình đang xem lại phần này, xử lý xong báo bạn liền.",
+    "Mình check lại một chút cho chắc nhé!",
+    "Ok, mình đang xem lại để đảm bảo chính xác.",
+    "Mình xem lại thông tin rồi cập nhật bạn ngay.",
+    "Để mình xử lý phần này trước đã nhé!",
+    "Mình đang kiểm tra lại, sẽ báo liền khi xong.",
+    "Ok bạn, để mình coi lại rồi chốt luôn.",
+    "Mình đang coi lại nội dung này, có gì mình nói liền."
+]
+
+# Found regulations status messages for Facebook notifications when regulations are found
+FOUND_REGULATIONS_MESSAGES = [
+    "Mình có thấy một vài quy định liên quan, đang xác minh lại và sẽ phản hồi sớm.",
+    "Tạm thời đã tra được một số quy định, mình đang kiểm tra thêm rồi cập nhật liền nhé.",
+    "Mình đang đối chiếu lại thông tin vừa tìm được, sẽ báo bạn ngay khi rõ.",
+    "Có một vài chỗ trong quy định mình đang xem lại cho chắc, sẽ phản hồi sớm thôi.",
+    "Mình đã tìm thấy một số điểm có liên quan, đang xác thực thêm để phản hồi cụ thể.",
+    "Thông tin bước đầu đã có, mình đang kiểm tra thêm rồi gửi lại ngay khi hoàn tất.",
+    "Mình đang kiểm tra kỹ hơn những gì vừa tìm được, sẽ gửi bạn sớm nhất có thể.",
+    "Một số quy định đang được mình rà lại lần nữa, sẽ phản hồi lại ngay khi rõ ràng.",
+    "Mình có dữ liệu sơ bộ rồi, đang xác minh thêm để đảm bảo chính xác trước khi trả lời.",
+    "Thông tin đang được mình kiểm tra lại lần cuối, mình sẽ cập nhật sớm nhất nhé."
+]
+
 SHEET_RANGE = 'chat!A2:L'

app/message_processor.py

@@ -2,8 +2,8 @@ from typing import Dict, Any, List
 import asyncio
 import traceback
 from loguru import logger
-import random
-from .constants import SUMMARY_STATUS_MESSAGES
+from .constants import SUMMARY_STATUS_MESSAGES, PROCESSING_STATUS_MESSAGES, FOUND_REGULATIONS_MESSAGES
+from .utils import get_random_message
 
 class MessageProcessor:
     def __init__(self, channel):
@@ -120,14 +120,14 @@ class MessageProcessor:
             return
         # Gửi message Facebook, nếu lỗi token expired thì invalidate và thử lại một lần
         try:
-            await self.channel.facebook.send_message(message="Ok, để mình check. Bạn chờ mình chút xíu nhé!")
+            await self.channel.facebook.send_message(message=get_random_message(PROCESSING_STATUS_MESSAGES))
         except Exception as e:
             if "expired" in str(e).lower():
                 logger.warning("[FACEBOOK] Token expired, invalidate and refresh")
                 self.channel.invalidate_page_token()
                 page_token = self.channel.get_page_token(force_refresh=True)
                 self.channel.facebook.update_context(page_id=page_id, page_token=page_token, sender_id=sender_id)
-                await self.channel.facebook.send_message(message="Ok, để mình check. Bạn chờ mình chút xíu nhé!")
+                # await self.channel.facebook.send_message(message="Ok, để mình check. Bạn chờ mình chút xíu nhé!")
             else:
                 raise
 
@@ -256,7 +256,7 @@ class MessageProcessor:
     async def format_search_results(self, question: str, matches: List[Dict[str, Any]], page_token: str, sender_id: str) -> str:
         if not matches:
             return "Không tìm thấy kết quả phù hợp."
-        await self.channel.facebook.send_message(message=f"... đã tìm thấy một số quy định .....")
+        await self.channel.facebook.send_message(message=get_random_message(FOUND_REGULATIONS_MESSAGES))
         try:
             reranked = await self.channel.reranker.rerank(question, matches, top_k=5)
             if reranked:
@@ -328,7 +328,7 @@ class MessageProcessor:
             "\n\nHãy trả lời ngắn gọn, dễ hiểu, trích dẫn rõ ràng thông tin từ các đoạn luật nếu cần."
             f"\n\nCâu hỏi của người dùng: {question}\n"
         )
-        await self.channel.facebook.send_message(message=f"... {random.choice(SUMMARY_STATUS_MESSAGES)} ...")
+        await self.channel.facebook.send_message(message=f"... {get_random_message(SUMMARY_STATUS_MESSAGES)} ...")
         try:
             answer = await self.channel.llm.generate_text(prompt)
             if answer and answer.strip():

app/reranker.py

@@ -3,10 +3,10 @@ from .config import get_settings
 from .gemini_client import GeminiClient
 from loguru import logger
 import asyncio
-import random
 import hashlib
 import time
 from .constants import BATCH_STATUS_MESSAGES
+from .utils import get_random_message
 
 class Reranker:
     def __init__(self, facebook_client=None):
@@ -245,12 +245,12 @@ class Reranker:
                     scored.append(doc)
         
         # Gửi Facebook message sau khi hoàn thành
-        # if self.facebook_client:
-        #     try:
-        #         message = random.choice(BATCH_STATUS_MESSAGES)
-        #         await self.facebook_client.send_message(message=f"... {message} ...")
-        #     except Exception as e:
-        #         logger.error(f"[RERANK][FACEBOOK] Error sending batch message: {e}")
+        if self.facebook_client:
+            try:
+                message = get_random_message(BATCH_STATUS_MESSAGES)
+                await self.facebook_client.send_message(message=f"... {message} ...")
+            except Exception as e:
+                logger.error(f"[RERANK][FACEBOOK] Error sending batch message: {e}")
         
         # Sort theo score và trả về top_k
         scored = sorted(scored, key=lambda x: x['rerank_score'], reverse=True)

app/utils.py

@@ -1,7 +1,8 @@
 import time
+import random
 from functools import wraps
 from loguru import logger
-from typing import Any, Callable
+from typing import Any, Callable, List
 import os
 import asyncio
 import httpx
@@ -136,4 +137,24 @@ async def call_endpoint_with_retry(client, url, payload, max_retries=3, base_tim
             raise
         except Exception as e:
             logger.error(f"Other error: {e}")
-            raise 
+            raise 
+
+def get_random_message(message_list: List[str]) -> str:
+    """
+    Lấy ngẫu nhiên một message từ danh sách messages.
+    
+    Args:
+        message_list (List[str]): Danh sách các messages có sẵn
+        
+    Returns:
+        str: Message được chọn ngẫu nhiên, hoặc message mặc định nếu danh sách rỗng
+        
+    Example:
+        >>> messages = ["Message 1", "Message 2", "Message 3"]
+        >>> get_random_message(messages)
+        "Message 2"  # hoặc message khác ngẫu nhiên
+    """
+    if not message_list:
+        return "Đang xử lý..."
+    
+    return random.choice(message_list)