update message flow

app/main.py

@@ -119,11 +119,6 @@ async def webhook(request: Request):
 
 @timing_decorator_async
 async def process_message(message_data: Dict[str, Any]):
-    """
-    Xử lý message từ người dùng Facebook, phân tích, truy vấn, gửi phản hồi và log lại.
-    Input: message_data (dict) - thông tin message đã parse từ Facebook.
-    Output: None (gửi message và log hội thoại).
-    """
     sender_id = message_data["sender_id"]
     page_id = message_data["page_id"]
     message_text = message_data["text"]
@@ -150,51 +145,111 @@ async def process_message(message_data: Dict[str, Any]):
     logger.info(f"[DEBUG] history: {history}")
 
     response = ""
-    if command == "xong":
-        if not keywords:
-            response = "Vui lòng cho biết loại phương tiện bạn cần tìm (xe máy, ô tô...)"
+    log_needed = True
+    log_kwargs = {
+        'user_id': sender_id,
+        'page_id': page_id,
+        'message': message_text,
+        'command': command,
+        'content': remaining_text,
+        'attachments': message_data.get('attachments', []),
+        'vehicle': ','.join(keywords),
+        'is_done': False
+    }
+
+    # 1. Nếu history rỗng (conversation mới)
+    if not history:
+        if not command:
+            if keywords:
+                # Có thông tin phương tiện
+                embedding = await embedding_client.create_embedding(message_text)
+                logger.info(f"[DEBUG] embedding: {embedding[:5]} ... (total {len(embedding)})")
+                matches = supabase_client.match_documents(embedding)
+                logger.info(f"[DEBUG] matches: {matches}")
+                if matches:
+                    response = format_search_results(matches)
+                else:
+                    response = "Xin lỗi, tôi không tìm thấy thông tin phù hợp."
+                log_kwargs['is_done'] = True
+            else:
+                # Không có thông tin phương tiện
+                response = "Vui lòng cho biết loại phương tiện bạn cần tìm (xe máy, ô tô...)"
+                log_kwargs['is_done'] = False
         else:
-            # Create embedding from message
+            # Có command
+            response = "Vui lòng cung cấp thêm thông tin và gõ lệnh \\xong khi hoàn tất."
+            log_kwargs['is_done'] = False
+        await facebook_client.send_message(page_token, sender_id, response)
+        await loop.run_in_executor(executor, lambda: sheets_client.log_conversation(**log_kwargs))
+        return
+
+    # 2. Nếu history có conversation (conversation cũ)
+    last_conv = history[-1] if history else None
+    last_command = last_conv['originalcommand'] if last_conv else ''
+    last_isdone = last_conv['isdone'] if last_conv else False
+
+    if not last_command:
+        # Lịch sử không có command
+        if keywords:
+            # Có thông tin phương tiện
             embedding = await embedding_client.create_embedding(message_text)
             logger.info(f"[DEBUG] embedding: {embedding[:5]} ... (total {len(embedding)})")
-            
-            # Search for similar documents (sync)
             matches = supabase_client.match_documents(embedding)
             logger.info(f"[DEBUG] matches: {matches}")
-            
             if matches:
                 response = format_search_results(matches)
             else:
                 response = "Xin lỗi, tôi không tìm thấy thông tin phù hợp."
+            log_kwargs['is_done'] = True
+            await facebook_client.send_message(page_token, sender_id, response)
+            await loop.run_in_executor(executor, lambda: sheets_client.log_conversation(**log_kwargs))
+            return
+        else:
+            # Không có thông tin phương tiện
+            response = "Vui lòng cho biết loại phương tiện bạn cần tìm (xe máy, ô tô...)"
+            log_needed = False
+            await facebook_client.send_message(page_token, sender_id, response)
+            return
     else:
-        response = "Vui lòng cung cấp thêm thông tin và gõ lệnh \\xong khi hoàn tất."
-
-    # Send response
-    await facebook_client.send_message(page_token, sender_id, response)
-
-    # Log conversation (run in thread pool)
-    await loop.run_in_executor(
-        executor, lambda: sheets_client.log_conversation(sender_id, page_id, message_text, keywords, response)
-    )
+        # Lịch sử có command
+        if command == "xong":
+            # Tạo bài viết mới trên page (placeholder)
+            # TODO: Thay thế hàm này bằng logic thực tế
+            post_url = await create_facebook_post(page_token, sender_id, history)
+            if post_url:
+                response = f"Bài viết đã được tạo thành công! Bạn có thể xem tại: {post_url}"
+            else:
+                response = "Đã xảy ra lỗi khi tạo bài viết. Vui lòng thử lại sau."
+            log_kwargs['is_done'] = True
+            await facebook_client.send_message(page_token, sender_id, response)
+            await loop.run_in_executor(executor, lambda: sheets_client.log_conversation(**log_kwargs))
+            return
+        else:
+            response = "Vui lòng cung cấp thêm thông tin và gõ lệnh \\xong khi hoàn tất."
+            log_kwargs['is_done'] = False
+            await facebook_client.send_message(page_token, sender_id, response)
+            await loop.run_in_executor(executor, lambda: sheets_client.log_conversation(**log_kwargs))
+            return
 
 def format_search_results(matches: List[Dict[str, Any]]) -> str:
-    """
-    Format kết quả truy vấn vector search thành chuỗi gửi về user.
-    Input: matches (list[dict]) - danh sách kết quả từ Supabase.
-    Output: Chuỗi kết quả đã format.
-    """
     if not matches:
         return "Không tìm thấy kết quả phù hợp."
-
     result = "Đây là một số kết quả phù hợp:\n\n"
     for i, match in enumerate(matches, 1):
         result += f"{i}. {match['content']}\n"
         if match.get('metadata', {}).get('url'):
             result += f"   Link: {match['metadata']['url']}\n"
         result += "\n"
-
     return result.strip()
 
+async def create_facebook_post(page_token: str, sender_id: str, history: List[Dict[str, Any]]) -> str:
+    """
+    Placeholder: Tạo bài viết mới trên page Facebook. Trả về URL bài viết nếu thành công, None nếu thất bại.
+    """
+    # TODO: Thay thế bằng logic thực tế
+    logger.info(f"[MOCK] Creating Facebook post for sender_id={sender_id} with history={history}")
+    return "https://facebook.com/mock_post_url"
+
 if __name__ == "__main__":
     import uvicorn
     logger.info("[STARTUP] Bắt đầu chạy uvicorn server...")

app/sheets.py

@@ -8,6 +8,7 @@ import pickle
 from datetime import datetime
 from loguru import logger
 import json
+import hashlib
 from google.oauth2.service_account import Credentials
 
 from .utils import timing_decorator_sync
@@ -15,6 +16,13 @@ from .constants import SHEET_RANGE
 
 SCOPES = ['https://www.googleapis.com/auth/spreadsheets']
 
+def generate_conversation_id(user_id: str, page_id: str, timestamp: str) -> str:
+    """
+    Tạo conversation_id duy nhất dựa trên user_id, page_id và timestamp.
+    """
+    hash_input = f"{user_id}:{page_id}:{timestamp}"
+    return hashlib.sha256(hash_input.encode()).hexdigest()[:32]
+
 class SheetsClient:
     def __init__(self, credentials_file: str, token_file: str, sheet_id: str):
         """
@@ -62,15 +70,16 @@ class SheetsClient:
     @timing_decorator_sync
     def get_conversation_history(self, user_id: str, page_id: str) -> List[Dict[str, Any]]:
         """
-        Lấy lịch sử hội thoại của user từ Google Sheets.
+        Lấy lịch sử hội thoại chưa hoàn thành của user từ Google Sheets.
         Input: user_id (str), page_id (str)
-        Output: list[dict] các dòng hội thoại (nếu có).
+        Output: list[dict] các dòng hội thoại chưa hoàn thành.
         """
         try:
             if not self.service:
                 self.authenticate()
             if not self.service:
                 raise RuntimeError("Google Sheets service not initialized")
+            
             range_name = SHEET_RANGE
             result = self.service.spreadsheets().values().get(
                 spreadsheetId=self.sheet_id,
@@ -81,14 +90,17 @@ class SheetsClient:
             history = []
 
             for row in values:
-                if len(row) >= 6 and row[0] == user_id and row[1] == page_id:
+                if len(row) >= 9 and row[4] == user_id and row[5] == page_id and row[8].lower() == 'false':
                     history.append({
-                        'user_id': row[0],
-                        'page_id': row[1],
-                        'timestamp': row[2],
-                        'message': row[3],
-                        'keywords': row[4],
-                        'response': row[5]
+                        'conversation_id': row[0],
+                        'originalcommand': row[1],
+                        'originalcontent': row[2],
+                        'originalattachments': json.loads(row[3]) if row[3] else [],
+                        'recipient_id': row[4],
+                        'page_id': row[5],
+                        'originaltext': row[6],
+                        'originalvehicle': row[7],
+                        'isdone': row[8].lower() == 'true'
                     })
 
             return history
@@ -102,12 +114,15 @@ class SheetsClient:
         user_id: str,
         page_id: str,
         message: str,
-        keywords: List[str],
-        response: str
+        command: str = "",
+        content: str = "",
+        attachments: Optional[List[str]] = None,
+        vehicle: str = "",
+        is_done: bool = False
     ) -> bool:
         """
         Ghi log hội thoại vào Google Sheets.
-        Input: user_id (str), page_id (str), message (str), keywords (list[str]), response (str)
+        Input: user_id, page_id, message, command, content, attachments, vehicle, is_done
         Output: bool (True nếu thành công, False nếu lỗi)
         """
         try:
@@ -115,27 +130,81 @@ class SheetsClient:
                 self.authenticate()
             if not self.service:
                 raise RuntimeError("Google Sheets service not initialized")
+
+            # Get existing conversations
+            existing_conversations = self.get_conversation_history(user_id, page_id)
             timestamp = datetime.now().isoformat()
-            values = [[
-                user_id,
-                page_id,
-                timestamp,
-                message,
-                ','.join(keywords),
-                response
-            ]]
-
-            body = {
-                'values': values
-            }
 
-            range_name = SHEET_RANGE
-            self.service.spreadsheets().values().append(
-                spreadsheetId=self.sheet_id,
-                range=range_name,
-                valueInputOption='RAW',
-                body=body
-            ).execute()
+            if not existing_conversations:
+                # Create new conversation
+                conversation_id = generate_conversation_id(user_id, page_id, timestamp)
+                values = [[
+                    conversation_id,
+                    command,
+                    content,
+                    json.dumps(attachments or []),
+                    user_id,
+                    page_id,
+                    message,
+                    vehicle,
+                    str(is_done).lower()
+                ]]
+
+                body = {
+                    'values': values
+                }
+
+                range_name = SHEET_RANGE
+                self.service.spreadsheets().values().append(
+                    spreadsheetId=self.sheet_id,
+                    range=range_name,
+                    valueInputOption='RAW',
+                    body=body
+                ).execute()
+            else:
+                # Update existing conversation
+                conversation = existing_conversations[0]
+                conversation_id = conversation['conversation_id']
+                
+                # Find the row with this conversation_id
+                result = self.service.spreadsheets().values().get(
+                    spreadsheetId=self.sheet_id,
+                    range=SHEET_RANGE
+                ).execute()
+                
+                values = result.get('values', [])
+                row_index = None
+                
+                for i, row in enumerate(values):
+                    if row[0] == conversation_id:
+                        row_index = i
+                        break
+                
+                if row_index is not None:
+                    # Update the row
+                    update_range = f"{SHEET_RANGE.split('!')[0]}!A{row_index + 1}"
+                    update_values = [[
+                        conversation_id,
+                        command or conversation['originalcommand'],
+                        content or conversation['originalcontent'],
+                        json.dumps(attachments or json.loads(conversation['originalattachments'])),
+                        user_id,
+                        page_id,
+                        message,
+                        vehicle or conversation['originalvehicle'],
+                        str(is_done).lower()
+                    ]]
+                    
+                    body = {
+                        'values': update_values
+                    }
+                    
+                    self.service.spreadsheets().values().update(
+                        spreadsheetId=self.sheet_id,
+                        range=update_range,
+                        valueInputOption='RAW',
+                        body=body
+                    ).execute()
 
             return True
         except Exception as e: