adjust message process for main topics

app/message_processor.py

@@ -189,15 +189,15 @@ class MessageProcessor:
             if muc_dich_to_use == "hỏi về mức phạt":
                 response = await self.handle_muc_phat(conv, conversation_context, page_token, sender_id)
             elif muc_dich_to_use == "hỏi về quy tắc giao thông":
-                response = await self.handle_quy_tac(conv, conversation_context, message_text)
+                response = await self.handle_quy_tac(conv, conversation_context, page_token, sender_id)
             elif muc_dich_to_use == "hỏi về báo hiệu đường bộ":
-                response = await self.handle_bao_hieu(conv, conversation_context, message_text)
+                response = await self.handle_bao_hieu(conv, conversation_context, page_token, sender_id)
             elif muc_dich_to_use == "hỏi về quy trình xử lý vi phạm giao thông":
-                response = await self.handle_quy_trinh(conv, conversation_context, message_text)
+                response = await self.handle_quy_trinh(conv, conversation_context, page_token, sender_id)
             elif muc_dich_to_use == "thông tin cá nhân của AI":
-                response = await self.handle_ca_nhan(conv, conversation_context, message_text)
+                response = await self.handle_ca_nhan(conv, conversation_context, page_token, sender_id)
             else:
-                response = await self.handle_khac(conv, conversation_context, message_text)
+                response = await self.handle_khac(conv, conversation_context, page_token, sender_id)
         else:
             if command == "xong":
                 post_url = await self.create_facebook_post(page_token, conv['recipient_id'], [conv])
@@ -356,30 +356,33 @@ class MessageProcessor:
             logger.error(f"Lỗi khi xử lý chủ đề {topic}: {e}")
             return f"Xin lỗi, tôi đang gặp sự cố khi xử lý câu hỏi về {topic}. Vui lòng thử lại sau."
 
-    async def handle_khac(self, conv, conversation_context, message_text):
+    async def handle_khac(self, conv, conversation_context, page_token, sender_id):
         conv['isdone'] = True
-        return await self._handle_general_question(conversation_context, message_text, "một vấn đề khác")
+        return await self._handle_general_question(conversation_context, conv['originaltext'], "một vấn đề khác")
 
-    async def handle_quy_tac(self, conv, conversation_context, message_text):
+    async def handle_quy_tac(self, conv, conversation_context, page_token, sender_id):
         conv['isdone'] = True
-        return await self._handle_general_question(conversation_context, message_text, "quy tắc giao thông")
+        # return await self._handle_general_question(conversation_context, conv['originaltext'], "quy tắc giao thông")
+        return await self.handle_muc_phat(conv, conversation_context, page_token, sender_id)
 
-    async def handle_bao_hieu(self, conv, conversation_context, message_text):
+    async def handle_bao_hieu(self, conv, conversation_context, page_token, sender_id):
         conv['isdone'] = True
-        return await self._handle_general_question(conversation_context, message_text, "báo hiệu đường bộ")
+        # return await self._handle_general_question(conversation_context, conv['originaltext'], "báo hiệu đường bộ")
+        return await self.handle_muc_phat(conv, conversation_context, page_token, sender_id)
 
-    async def handle_quy_trinh(self, conv, conversation_context, message_text):
+    async def handle_quy_trinh(self, conv, conversation_context, page_token, sender_id):
         conv['isdone'] = True
-        return await self._handle_general_question(conversation_context, message_text, "quy trình xử lý vi phạm giao thông")
+        # return await self._handle_general_question(conversation_context, conv['originaltext'], "quy trình xử lý vi phạm giao thông")
+        return await self.handle_muc_phat(conv, conversation_context, page_token, sender_id)
 
-    async def handle_ca_nhan(self, conv, conversation_context, message_text):
+    async def handle_ca_nhan(self, conv, conversation_context, page_token, sender_id):
         prompt = (
             "Biết rằng bạn đã có lịch sử trao đổi như sau:"
             f"Lịch sử:\n{conversation_context}\n\n"
             'Với các thông tin sau: "Bạn có tên là WeThoong AI, là trợ lý giao thông thông minh. Bạn được anh Viet Cat tạo ra và facebook cá nhân của anh ấy là https://facebook.com/vietcat". '
             'Không được trả lời bạn là AI của Google, OpenAI, hay bất kỳ hãng nào khác. '
             'Hãy trả lời thông minh, hài hước, ngắn gọn cho câu hỏi sau:\n'
-            f'Câu hỏi:\n"{message_text}"'
+            f'Câu hỏi:\n"{conv['originaltext']}"'
         )
         try:
             answer = await self.channel.llm.generate_text(prompt)