quick fix timestamp

app/sheets.py

@@ -21,14 +21,12 @@ def generate_conversation_id(user_id: str, page_id: str, timestamp: str) -> str:
     hash_input = f"{user_id}:{page_id}:{timestamp}"
     return hashlib.sha256(hash_input.encode()).hexdigest()[:32]
 
-# --- HÀM TIỆN ÍCH ĐƯỢC THÊM VÀO ĐỂ FIX LỖI ---
 def _flatten_and_unique_timestamps(items: Any) -> List[Any]:
     """
     Hàm tiện ích để làm phẳng danh sách timestamp (xử lý list lồng nhau)
     và loại bỏ các giá trị trùng lặp, giữ nguyên thứ tự.
     """
     if not isinstance(items, list):
-        # Nếu đầu vào không phải list, gói nó vào một list
         return [items]
         
     flat_list = []
@@ -37,7 +35,6 @@ def _flatten_and_unique_timestamps(items: Any) -> List[Any]:
             flat_list.extend(_flatten_and_unique_timestamps(item))
         else:
             flat_list.append(item)
-    # Dùng dict để giữ lại thứ tự và loại bỏ trùng lặp
     return list(dict.fromkeys(flat_list))
 
 
@@ -83,22 +80,23 @@ class SheetsClient:
             ).execute()
             values = result.get('values', [])
             history = []
-            if not values or len(values) < 1:
+            if not values or len(values) < 2: # Cần có header và ít nhất 1 dòng dữ liệu
                 return []
             
             header = values[0]
             for row in values[1:]:
+                # Zip header với row để tạo dict, an toàn hơn
                 row_data = dict(zip(header, row + [""] * (len(header) - len(row))))
                 
                 if row_data.get('recipient_id') == user_id and row_data.get('page_id') == page_id:
                     try:
-                        # Dùng hàm tiện ích để đảm bảo timestamp đọc ra luôn phẳng
-                        timestamps = _flatten_and_unique_timestamps(json.loads(row_data.get('timestamp', '[]')))
+                        timestamps_raw = json.loads(row_data.get('timestamp', '[]'))
+                        timestamps = _flatten_and_unique_timestamps(timestamps_raw)
                     except (json.JSONDecodeError, TypeError):
                         timestamps = []
                     
                     row_data['timestamp'] = timestamps
-                    row_data['originalattachments'] = json.loads(row_data.get('originalattachments', '[]'))
+                    row_data['originalattachments'] = json.loads(row_data.get('originalattachments', '[]')) if row_data.get('originalattachments') else []
                     row_data['isdone'] = str(row_data.get('isdone', 'false')).lower() == 'true'
                     history.append(row_data)
             return history
@@ -121,7 +119,7 @@ class SheetsClient:
         originalpurpose: str = "",
         originalquestion: str = "",
         systemresponse: str = "",
-        timestamp: Optional[Any] = None, # Cho phép timestamp có thể là list
+        timestamp: Optional[Any] = None,
         isdone: bool = False
     ) -> Optional[Dict[str, Any]]:
         """Ghi lại một hội thoại, giữ nguyên logic gốc của người dùng."""
@@ -134,57 +132,57 @@ class SheetsClient:
                 range=SHEET_RANGE
             ).execute()
             values = result.get('values', [])
-            header = values[0] if values else []
 
-            # --- SỬA LỖI LỒNG TIMESTAMP TẠI ĐÂY ---
-            # 1. Luôn đảm bảo `timestamps_list` là một danh sách phẳng, không lồng nhau.
-            timestamps_list = _flatten_and_unique_timestamps(timestamp or [])
-
-            # 2. Xác định timestamp cho sự kiện hiện tại (dùng để kiểm tra trùng lặp).
-            #    Lấy timestamp mới nhất (cuối cùng trong list). Nếu list rỗng thì tạo mới.
-            if not timestamps_list:
-                current_ts = datetime.now().isoformat()
-                timestamps_list.append(current_ts)
+            # ==================================================================
+            # >>>>> SỬA LỖI LỒNG TIMESTAMP TẠI ĐÂY <<<<<
+            # Bước 1: Xử lý `timestamp` đầu vào để nó LUÔN LUÔN là một danh sách phẳng.
+            processed_timestamps = _flatten_and_unique_timestamps(timestamp or [])
+            
+            # Bước 2: Lấy timestamp mới nhất để kiểm tra trùng lặp. Nếu list rỗng thì tạo mới.
+            if not processed_timestamps:
+                current_ts_for_check = datetime.now().isoformat()
+                processed_timestamps.append(current_ts_for_check)
             else:
-                current_ts = timestamps_list[-1]
-            # --- KẾT THÚC SỬA LỖI ---
+                current_ts_for_check = processed_timestamps[-1]
+            # ==================================================================
 
             # Logic kiểm tra trùng lặp của bạn được giữ nguyên
-            if header:
-                for row in values[1:]:
-                    row_data = dict(zip(header, row + [""] * (len(header) - len(row))))
-                    try:
-                        row_timestamps = json.loads(row_data.get('timestamp', '[]'))
-                        if not isinstance(row_timestamps, list):
-                            row_timestamps = [row_timestamps]
-                    except Exception:
-                        row_timestamps = []
-
-                    if str(current_ts) in [str(ts) for ts in row_timestamps] and \
-                       row_data.get('recipient_id') == str(recipient_id) and \
-                       row_data.get('page_id') == str(page_id):
-                        logger.info(f"Found duplicate conversation for user {recipient_id}, page {page_id}, timestamp {current_ts}")
-                        # Trả về dữ liệu tìm thấy, logic không đổi
-                        return {
-                            'conversation_id': row_data.get('conversation_id'),
-                            'originalcommand': row_data.get('originalcommand'),
-                            'originalcontent': row_data.get('originalcontent'),
-                            'originalattachments': json.loads(row_data.get('originalattachments', '[]')),
-                            'recipient_id': row_data.get('recipient_id'),
-                            'page_id': row_data.get('page_id'),
-                            'originaltext': row_data.get('originaltext'),
-                            'originalvehicle': row_data.get('originalvehicle'),
-                            'originalaction': row_data.get('originalaction'),
-                            'originalpurpose': row_data.get('originalpurpose'),
-                            'originalquestion': row_data.get('originalquestion'),
-                            'systemresponse': row_data.get('systemresponse'),
-                            'timestamp': row_timestamps,
-                            'isdone': str(row_data.get('isdone', 'false')).lower() == 'true'
-                        }
+            for row in values:
+                row = row + [""] * 14
+                try:
+                    row_timestamps = json.loads(row[12]) if row[12] else []
+                    if not isinstance(row_timestamps, list):
+                        row_timestamps = [row_timestamps]
+                except Exception:
+                    row_timestamps = []
+
+                row_recipient_id = row[4]
+                row_page_id = row[5]
+
+                if str(current_ts_for_check) in [str(ts) for ts in row_timestamps] and \
+                   str(row_recipient_id) == str(recipient_id) and \
+                   str(row_page_id) == str(page_id):
+                    logger.info(f"Found duplicate conversation for user {recipient_id}, page {page_id}, timestamp {current_ts_for_check}")
+                    return {
+                        '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],
+                        'originalaction': row[8],
+                        'originalpurpose': row[9],
+                        'originalquestion': row[10],
+                        'systemresponse': row[11],
+                        'timestamp': row_timestamps,
+                        'isdone': row[13].lower() == 'true' if len(row) > 13 else False
+                    }
 
             # Logic tạo dòng mới và append vào sheet được giữ nguyên
             if not conversation_id:
-                conversation_id = generate_conversation_id(recipient_id, page_id, current_ts)
+                conversation_id = generate_conversation_id(recipient_id, page_id, current_ts_for_check)
             
             new_row = [
                 conversation_id,
@@ -199,7 +197,7 @@ class SheetsClient:
                 originalpurpose,
                 originalquestion,
                 systemresponse,
-                json.dumps(timestamps_list), # Lưu danh sách đã được làm phẳng
+                json.dumps(processed_timestamps), # <<<< SỬ DỤNG BIẾN ĐÃ ĐƯỢC XỬ LÝ
                 str(isdone).lower()
             ]
 
@@ -213,7 +211,6 @@ class SheetsClient:
 
             logger.info(f"Thêm mới conversation: {conversation_id}")
 
-            # Trả về dữ liệu đã được ghi, logic không đổi
             return {
                 'conversation_id': conversation_id,
                 'originalcommand': originalcommand,
@@ -227,7 +224,7 @@ class SheetsClient:
                 'originalpurpose': originalpurpose,
                 'originalquestion': originalquestion,
                 'systemresponse': systemresponse,
-                'timestamp': timestamps_list, # Trả về danh sách đã được làm phẳng
+                'timestamp': processed_timestamps, # <<<< TRẢ VỀ DANH SÁCH ĐÃ ĐƯỢC XỬ LÝ
                 'isdone': isdone
             }
         except Exception as e: