quick fix timestamp

app/sheets.py

@@ -38,6 +38,22 @@ def _flatten_and_unique_timestamps(items: Any) -> List[Any]:
             flat_list.append(item)
     return list(dict.fromkeys(flat_list))
 
+def _get_start_row_from_range(range_string: str) -> int:
+    """
+    Phân tích một chuỗi range (ví dụ: 'Sheet1!A2:Z') để lấy ra số của dòng bắt đầu.
+    Hàm này giúp code hoạt động chính xác ngay cả khi range không bắt đầu từ dòng 2.
+    """
+    # Tìm số đầu tiên xuất hiện sau một chữ cái trong chuỗi range
+    match = re.search(r"[A-Z]+([0-9]+)", range_string)
+    if match:
+        try:
+            return int(match.group(1))
+        except (ValueError, IndexError):
+            pass # Bỏ qua nếu không chuyển đổi được và dùng giá trị mặc định
+    
+    logger.warning(f"Không thể xác định dòng bắt đầu từ range '{range_string}'. Mặc định là 2.")
+    return 2 # Mặc định là 2 nếu không phân tích được
+
 
 class SheetsClient:
     def __init__(self, credentials_file: str, token_file: str, sheet_id: str):
@@ -89,10 +105,8 @@ class SheetsClient:
             
             logger.debug(f"[get_conversation_history] Đã lấy được {len(values)} dòng từ sheet. Bắt đầu xử lý...")
             
-            # --- THAY ĐỔI THEO YÊU CẦU ---
-            # Lặp qua toàn bộ `values` vì header đã được loại bỏ trong SHEET_RANGE
-            # start=2 giả định rằng SHEET_RANGE bắt đầu từ dòng 2
-            for i, row in enumerate(values, start=2):
+            start_row = _get_start_row_from_range(range_name)
+            for i, row in enumerate(values, start=start_row):
                 if len(row) < 14:
                     row.extend([""] * (14 - len(row)))
 
@@ -144,9 +158,6 @@ class SheetsClient:
             if not self.service:
                 self.authenticate()
             
-            # --- THAY ĐỔI LOGIC ĐỂ HOẠT ĐỘNG KHI SHEET_RANGE KHÔNG CÓ HEADER ---
-            # 1. Lấy header từ dòng 1 một cách tường minh
-            # Tự động xác định tên sheet từ SHEET_RANGE
             sheet_name_match = re.match(r"([^!]+)!", SHEET_RANGE)
             sheet_name = sheet_name_match.group(1) if sheet_name_match else "Sheet1"
             header_range = f"{sheet_name}!A1:Z1"
@@ -157,23 +168,25 @@ class SheetsClient:
                 logger.error(f"Không thể lấy được header từ range '{header_range}'.")
                 return None
 
-            # 2. Lấy dữ liệu từ SHEET_RANGE (không có header)
             data_result = self.service.spreadsheets().values().get(spreadsheetId=self.sheet_id, range=SHEET_RANGE).execute()
             values = data_result.get('values', [])
 
             conversation_id = kwargs.get('conversation_id')
             row_to_update_index = -1
 
-            # Nếu có conversation_id, tìm dòng để cập nhật
             if conversation_id:
+                logger.debug(f"Đang tìm dòng để cập nhật cho conversation_id: {conversation_id}")
                 try:
                     id_col_index = header.index('conversation_id')
-                    # Lặp qua `values` (không có header)
-                    # start=2 vì sheet index bắt đầu từ 1 và SHEET_RANGE bắt đầu từ dòng 2
-                    for i, row in enumerate(values, start=2):
+                    start_row = _get_start_row_from_range(SHEET_RANGE)
+                    
+                    for i, row in enumerate(values, start=start_row):
                         if len(row) > id_col_index and row[id_col_index] == conversation_id:
                             row_to_update_index = i
+                            logger.success(f"Đã tìm thấy conversation_id tại dòng {row_to_update_index} để cập nhật.")
                             break
+                    if row_to_update_index == -1:
+                        logger.warning(f"Không tìm thấy dòng nào khớp với conversation_id: {conversation_id}. Sẽ tiến hành thêm dòng mới.")
                 except ValueError:
                     logger.error("Không tìm thấy cột 'conversation_id' trong header của sheet.")
                     return None
@@ -192,7 +205,7 @@ class SheetsClient:
 
             if row_to_update_index != -1:
                 logger.info(f"Đang cập nhật conversation_id {conversation_id} tại dòng {row_to_update_index}")
-                range_to_update = f"A{row_to_update_index}"
+                range_to_update = f"{sheet_name}!A{row_to_update_index}"
                 body = {'values': [row_data]}
                 self.service.spreadsheets().values().update(
                     spreadsheetId=self.sheet_id,