refactor message flow on purposes

app/facebook.py

@@ -63,27 +63,67 @@ class FacebookClient:
         
         return hmac.compare_digest(signature[7:], expected)
 
+    def format_message(self, text: str) -> str:
+        # 1. Thay bullet markdown bằng ký hiệu khác
+        text = text.replace('\n*   ', '\n- ')
+        text = text.replace('\n    *   ', '\n    + ')
+        text = text.replace('\n* ', '\n- ')
+        text = text.replace('\n    * ', '\n    + ')
+        # 2. Chuyển **text** hoặc __text__ thành *text*
+        import re
+        text = re.sub(r'\*\*([^\*]+)\*\*', r'*\1*', text)
+        text = re.sub(r'__([^_]+)__', r'*\1*', text)
+        # 3. Loại bỏ các tiêu đề markdown kiểu #, ##, ###, ...
+        text = re.sub(r'^#+\s+', '', text, flags=re.MULTILINE)
+        # 4. Rút gọn nhiều dòng trống liên tiếp thành 1 dòng trống
+        text = re.sub(r'\n{3,}', '\n\n', text)
+        # 5. Loại bỏ các markdown không hỗ trợ khác nếu cần
+        return text
+
+    def split_message(self, text: str, max_length: int = 2000) -> list:
+        """
+        Chia message thành các đoạn <= max_length ký tự, ưu tiên chia theo dòng.
+        """
+        lines = text.split('\n')
+        messages = []
+        current = ""
+        for line in lines:
+            # +1 cho ký tự xuống dòng
+            if len(current) + len(line) + 1 > max_length:
+                messages.append(current.rstrip())
+                current = ""
+            current += (line + '\n')
+        if current.strip():
+            messages.append(current.rstrip())
+        return messages
+
     @timing_decorator_async
-    async def send_message(self, page_access_token: Optional[str] = None, recipient_id: Optional[str] = None, message: str = "") -> Dict[str, Any]:
+    async def send_message(self, page_access_token: Optional[str] = None, recipient_id: Optional[str] = None, message: str = "") -> dict:
         page_access_token = page_access_token or self.page_token
         recipient_id = recipient_id or self.sender_id
         if not page_access_token or not recipient_id:
             raise ValueError("FacebookClient: page_access_token and recipient_id must not be None when sending a message.")
-        # Replace all occurrences of '**' with '*' before sending
-        response_to_send = message.replace('**', '*') if isinstance(message, str) else message
-        logger.info(f"Đang gửi tin nhắn đến Facebook Messenger....\n\t{response_to_send}")
-        url = f"https://graph.facebook.com/v18.0/me/messages?access_token={page_access_token}"
-        payload = {
-            "recipient": {"id": recipient_id},
-            "message": {"text": response_to_send}  # Sử dụng response_to_send đã được xử lý
-        }
-        try:
-            response = await self._client.post(url, json=payload)
-            response.raise_for_status()
-            return response.json()
-        except httpx.HTTPError as e:
-            logger.error(f"Error sending message to Facebook: {e}")
-            raise HTTPException(status_code=500, detail="Failed to send message to Facebook")
+        # Format message
+        response_to_send = self.format_message(message.replace('**', '*')) if isinstance(message, str) else message
+        # Chia nhỏ nếu quá dài
+        messages = self.split_message(response_to_send)
+        results = []
+        for msg in messages:
+            if len(msg) > 2000:
+                msg = msg[:2000]  # fallback cắt cứng
+            url = f"https://graph.facebook.com/v18.0/me/messages?access_token={page_access_token}"
+            payload = {
+                "recipient": {"id": recipient_id},
+                "message": {"text": msg}
+            }
+            try:
+                response = await self._client.post(url, json=payload)
+                response.raise_for_status()
+                results.append(response.json())
+            except httpx.HTTPError as e:
+                logger.error(f"Error sending message to Facebook: {e}")
+                raise HTTPException(status_code=500, detail="Failed to send message to Facebook")
+        return results[0] if results else {}
 
     @timing_decorator_sync
     def parse_message(self, body: Dict[str, Any]) -> Optional[Dict[str, Any]]:

app/message_processor.py

@@ -196,52 +196,15 @@ class MessageProcessor:
         response = None
         if not command:
             if muc_dich_to_use == "hỏi về mức phạt":
-                # Giữ nguyên luồng cũ cho hỏi về mức phạt
-                vehicle = conv.get('originalvehicle', '')
-                action = conv.get('originalaction', '')
-                keywords = [kw.strip() for kw in vehicle.split(',') if kw.strip()]
-                if keywords:
-                    if action:
-                        logger.info(f"[DEBUG] tạo embedding: {action}")
-                        embedding = await self.channel.embedder.create_embedding(action)
-                        logger.info(f"[DEBUG] embedding: {embedding[:5]} ... (total {len(embedding)})")
-                        matches = self.channel.supabase.match_documents(
-                            embedding,
-                            vehicle_keywords=keywords,
-                            user_question=action
-                        )
-                        logger.info(f"[DEBUG] matches: {matches}")
-                        if matches:
-                            response = await self.format_search_results(message_text, matches, page_token, sender_id)
-                        else:
-                            response = "Xin lỗi, tôi không tìm thấy thông tin phù hợp."
-                    else:
-                        logger.info(f"[DEBUG] Không có hành vi vi phạm: {message_text}")
-                        response = "Xin lỗi, tôi không tìm thấy thông tin về hành vi vi phạm trong câu hỏi của bạn."
-                    conv['isdone'] = True
-                else:
-                    response = "Vui lòng cho biết loại phương tiện bạn cần tìm (xe máy, ô tô...)"
-                    conv['isdone'] = False
+                response = await self.handle_muc_phat(conv, message_text, page_token, sender_id)
             elif muc_dich_to_use == "hỏi về quy tắc giao thông":
-                # Dùng LLM để trả lời câu hỏi về quy tắc giao thông
-                answer = await self.channel.llm.generate_text(message_text)
-                response = answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng trả lời về quy tắc giao thông sẽ sớm có mặt."
-                conv['isdone'] = True
+                response = await self.handle_quy_tac(conv, message_text)
             elif muc_dich_to_use == "hỏi về báo hiệu đường bộ":
-                # Dùng LLM để trả lời câu hỏi về báo hiệu đường bộ
-                answer = await self.channel.llm.generate_text(message_text)
-                response = answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng trả lời về báo hiệu đường bộ sẽ sớm có mặt."
-                conv['isdone'] = True
+                response = await self.handle_bao_hieu(conv, message_text)
             elif muc_dich_to_use == "hỏi về quy trình xử lý vi phạm giao thông":
-                # Dùng LLM để trả lời câu hỏi về quy trình xử lý vi phạm giao thông
-                answer = await self.channel.llm.generate_text(message_text)
-                response = answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng trả lời về quy trình xử lý vi phạm giao thông sẽ sớm có mặt."
-                conv['isdone'] = True
+                response = await self.handle_quy_trinh(conv, message_text)
             else:
-                # Dùng LLM cho các mục đích khác
-                answer = await self.channel.llm.generate_text(message_text)
-                response = answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng này sẽ sớm có mặt."
-                conv['isdone'] = True
+                response = await self.handle_khac(conv, message_text)
         else:
             # Có command
             if command == "xong":
@@ -256,11 +219,9 @@ class MessageProcessor:
                 conv['isdone'] = False
 
         # 6. Gửi response và cập nhật final state
-        # Replace all occurrences of '**' with '*' before sending
-        response_to_send = format_for_facebook(response.replace('**', '*')) if isinstance(response, str) else response
-        await self.facebook.send_message(message=response_to_send)
+        await self.facebook.send_message(message=response)
         if hasattr(self.channel, 'sheets'):
-            await loop.run_in_executor(None, lambda: self.channel.sheets.log_conversation(**conv))
+            await loop.run_in_executor(None, lambda: sheets_client.log_conversation(**conv))
         return
 
     def flatten_timestamp(self, ts):
@@ -454,14 +415,50 @@ class MessageProcessor:
         logger.info(f"[MOCK] Creating Facebook post for sender_id={sender_id} with history={history}")
         return "https://facebook.com/mock_post_url" 
 
-def format_for_facebook(text: str) -> str:
-    # 1. Thay bullet markdown bằng ký hiệu khác
-    text = text.replace('\n*   ', '\n- ')
-    text = text.replace('\n    *   ', '\n    + ')
-    text = text.replace('\n* ', '\n- ')
-    text = text.replace('\n    * ', '\n    + ')
-    # 2. Chuyển **text** hoặc __text__ thành *text*
-    text = re.sub(r'\*\*([^\*]+)\*\*', r'*\1*', text)
-    text = re.sub(r'__([^_]+)__', r'*\1*', text)
-    # 3. Loại bỏ các markdown không hỗ trợ khác nếu cần
-    return text 
+    async def handle_muc_phat(self, conv, message_text, page_token, sender_id):
+        vehicle = conv.get('originalvehicle', '')
+        action = conv.get('originalaction', '')
+        keywords = [kw.strip() for kw in vehicle.split(',') if kw.strip()]
+        if keywords:
+            if action:
+                logger.info(f"[DEBUG] tạo embedding: {action}")
+                embedding = await self.channel.embedder.create_embedding(action)
+                logger.info(f"[DEBUG] embedding: {embedding[:5]} ... (total {len(embedding)})")
+                matches = self.channel.supabase.match_documents(
+                    embedding,
+                    vehicle_keywords=keywords,
+                    user_question=action
+                )
+                logger.info(f"[DEBUG] matches: {matches}")
+                if matches:
+                    response = await self.format_search_results(message_text, matches, page_token, sender_id)
+                else:
+                    response = "Xin lỗi, tôi không tìm thấy thông tin phù hợp."
+            else:
+                logger.info(f"[DEBUG] Không có hành vi vi phạm: {message_text}")
+                response = "Xin lỗi, tôi không tìm thấy thông tin về hành vi vi phạm trong câu hỏi của bạn."
+            conv['isdone'] = True
+        else:
+            response = "Vui lòng cho biết loại phương tiện bạn cần tìm (xe máy, ô tô...)"
+            conv['isdone'] = False
+        return response
+
+    async def handle_quy_tac(self, conv, message_text):
+        answer = await self.channel.llm.generate_text(message_text)
+        conv['isdone'] = True
+        return answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng trả lời về quy tắc giao thông sẽ sớm có mặt."
+
+    async def handle_bao_hieu(self, conv, message_text):
+        answer = await self.channel.llm.generate_text(message_text)
+        conv['isdone'] = True
+        return answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng trả lời về báo hiệu đường bộ sẽ sớm có mặt."
+
+    async def handle_quy_trinh(self, conv, message_text):
+        answer = await self.channel.llm.generate_text(message_text)
+        conv['isdone'] = True
+        return answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng trả lời về quy trình xử lý vi phạm giao thông sẽ sớm có mặt."
+
+    async def handle_khac(self, conv, message_text):
+        answer = await self.channel.llm.generate_text(message_text)
+        conv['isdone'] = True
+        return answer.strip() if answer and answer.strip() else "[Đang phát triển] Tính năng này sẽ sớm có mặt."