add translate sheet_name

excel/xlsx.py

@@ -10,6 +10,8 @@ import gridfs
 from io import BytesIO
 import shutil
 import io
+import re
+from typing import Dict
 
 
 NS_MAIN = {'main': 'http://schemas.openxmlformats.org/spreadsheetml/2006/main'}
@@ -96,16 +98,7 @@ def extract_text_from_sheet(unzipped_folder_path: str) -> Optional[Tuple[List[Di
                         'source_file': os.path.join("xl", "sharedStrings.xml"),
                         'sheet_name': None
                     })
-                elif t_elements: #  Trường hợp có <t> nhưng không có <r> nào (simple shared string)
-                    # Tìm <t> đầu tiên trực tiếp dưới <si> nếu không phải rich text
-                    # Hoặc nếu cấu trúc là <si><t>...</t></si>
-                    # Trong trường hợp này, element nên là si_element để khi apply_changes,
-                    # ta sẽ tạo cấu trúc <si><r><t>...</t></r></si> nếu có định dạng
-                    # hoặc <si><t>...</t></si> nếu không.
-                    # Tuy nhiên, để đơn giản hóa, nếu không có <r>, ta coi element là <t> đầu tiên
-                    # và không áp dụng "first_format" (vì nó sẽ là None).
-                    # Hoặc, ta có thể luôn coi <si> là element cho shared strings.
-                    # Lựa chọn hiện tại: nếu không có <r>, element là <t> đầu tiên tìm thấy.
+                elif t_elements:
                     direct_t = si_element.find('./main:t', NS_MAIN)
                     if direct_t is not None:
                          modifiable_nodes.append({
@@ -284,17 +277,10 @@ def apply_and_save_changes(modified_nodes_data: List[Dict[str, Any]], global_dat
                     # Tạo thẻ text mới <a:t>
                     new_t = ET.Element(f"{{{NS_A['a']}}}t")
                     new_t.text = modified_text
-                    # Trong DrawingML, xml:space="preserve" thường không cần thiết cho <a:t>
-                    # vì việc xuống dòng được kiểm soát bởi <a:br> hoặc các paragraph <a:p> riêng biệt.
-                    # Tuy nhiên, việc thêm nó không gây hại.
 
                     new_r.append(new_t) # Thêm <a:t> vào <a:r>
                     p_element.append(new_r) # Thêm <a:r> vào <a:p>
 
-                    # Một số text box có thể có <a:endParaRPr> để định dạng cuối paragraph.
-                    # Nếu muốn giữ lại, cần logic phức tạp hơn.
-                    # Hiện tại, chúng ta chỉ tạo lại với một run duy nhất.
-
                     updated_files.add(node_info['source_file'])
                     # print(f"Applied first format to Drawing Text in {node_info['source_file']}")
 
@@ -329,13 +315,6 @@ def apply_and_save_changes(modified_nodes_data: List[Dict[str, Any]], global_dat
             if drawing_relative_path in updated_files:
                 drawing_file_path = os.path.join(drawings_folder, drawing_filename)
                 try:
-                    # Đảm bảo namespaces được đăng ký đúng cách TRƯỚC KHI GHI
-                    # register_namespaces(drawing_file_path) # Có thể không cần nếu đã làm ở extract
-                    # Hoặc đăng ký cứng các namespace cần thiết:
-                    # ET.register_namespace('xdr', NS_DRAWING['xdr'])
-                    # ET.register_namespace('a', NS_A['a'])
-                    # (Đã chuyển lên đầu hàm apply_and_save_changes)
-
                     drawing_tree.write(drawing_file_path, encoding='utf-8', xml_declaration=True)
                 except Exception as e: print(f"Lỗi lưu {drawing_file_path}: {e}"); success = False
 
@@ -344,6 +323,89 @@ def apply_and_save_changes(modified_nodes_data: List[Dict[str, Any]], global_dat
     elif not updated_files: print("Không có file XML nào cần cập nhật.") ; return True # Vẫn coi là success nếu không có gì thay đổi
     return success
 
+def translate_sheet_names_via_regex(
+    workbook_xml_path: str,
+    source_lang: str = 'chinese',
+    target_lang: str = 'vietnamese'
+):
+
+    original_to_translated_map: Dict[str, str] = {}
+    modified_content: str = ""
+    file_changed_flag: bool = False # Sử dụng tên biến rõ ràng hơn
+
+    try:
+        with open(workbook_xml_path, 'r', encoding='utf-8-sig') as f:
+            content = f.read()
+
+        current_content = content
+
+        def replace_name_callback(match_obj):
+            nonlocal file_changed_flag # Để sửa đổi biến bên ngoài
+            nonlocal original_to_translated_map
+
+            attr_prefix = match_obj.group(1)  # Ví dụ: '<sheet other_attr="foo" name='
+            opening_quote = match_obj.group(2) # Ví dụ: '"'
+            original_name_xml_encoded = match_obj.group(3)
+            attr_suffix = match_obj.group(5)  # Ví dụ: ' sheetId="12" r:id="rId1"/>'
+
+            original_name = original_name_xml_encoded # Tạm thời bỏ qua unescape/escape cho đơn giản ví dụ
+
+            if not original_name.strip():
+                return match_obj.group(0) # Trả về chuỗi gốc nếu tên rỗng
+
+            translated_name = original_name # Mặc định giữ nguyên
+            if original_name in original_to_translated_map and original_to_translated_map[original_name] != original_name:
+                translated_name = original_to_translated_map[original_name]
+                # Nếu đã dịch và có thay đổi, không cần gọi API dịch nữa
+                if translated_name != original_name: # Cần kiểm tra lại vì map có thể lưu tên gốc nếu dịch lỗi
+                     print(f"Regex: Sử dụng bản dịch đã có cho '{original_name}' -> '{translated_name}'")
+                     file_changed_flag = True # Đảm bảo cờ được set nếu sử dụng bản dịch đã có mà khác gốc
+            else:
+                try:
+                    translated_name_raw = translate_single_text(original_name, source_lang, target_lang)
+
+                    if translated_name_raw and translated_name_raw.strip() and translated_name_raw != original_name:
+                        translated_name = translated_name_raw[:31]
+                        original_to_translated_map[original_name] = translated_name
+                        file_changed_flag = True
+                        print(f"Regex: Đã dịch sheet: '{original_name}' -> '{translated_name}'")
+                    else:
+                        original_to_translated_map[original_name] = original_name # Lưu tên gốc nếu dịch lỗi/không đổi
+                        # translated_name vẫn là original_name
+                        if translated_name_raw and translated_name_raw.strip() and translated_name_raw == original_name:
+                            print(f"Bản dịch cho '{original_name}' giống hệt bản gốc, không thay đổi.")
+                        elif not (translated_name_raw and translated_name_raw.strip()):
+                             print(f"Bản dịch cho '{original_name}' trống hoặc không hợp lệ, giữ nguyên.")
+
+                except Exception as e_translate:
+                    print(f"Lỗi khi gọi hàm dịch cho '{original_name}': {e_translate}")
+                    original_to_translated_map[original_name] = original_name
+
+            translated_name_xml_encoded = translated_name # Tạm thời bỏ qua escape
+
+            return f"{attr_prefix}{opening_quote}{translated_name_xml_encoded}{opening_quote}{attr_suffix}"
+
+        sheet_name_pattern = re.compile(
+            r'(<sheet[^>]*?\sname=)(["\'])((?:(?!\2).)*?)(\2)([^>]*?>)'
+        )
+        modified_content = sheet_name_pattern.sub(replace_name_callback, current_content)
+
+        if file_changed_flag:
+            with open(workbook_xml_path, 'w', encoding='utf-8') as f:
+                f.write(modified_content)
+            print(f"Regex: Đã cập nhật thành công file: {workbook_xml_path}")
+            return original_to_translated_map, True
+        else:
+            print(f"Regex: Không có tên sheet nào được thay đổi trong file: {workbook_xml_path}")
+
+    except FileNotFoundError:
+        print(f"Lỗi: Không tìm thấy file tại '{workbook_xml_path}'")
+    except Exception as e:
+        print(f"Đã xảy ra lỗi không mong muốn khi xử lý file '{workbook_xml_path}' bằng regex: {e}")
+        import traceback
+        traceback.print_exc()
+
+
 def zip_folder_to_excel_file(folder_path, file_name):
     try:
         # Nén thư mục thành file .xlsx trong RAM
@@ -421,19 +483,7 @@ def _translate_batch_helper(segments_to_translate, original_indices_1based, sour
 
 
 def translate_xlsx(file_id, file_name, source_lang='en', target_lang='vi', batch_size_segments=50, max_words_per_segment=100, delay_between_requests=1):
-    """
-    Dịch file XLSX, chia thành batch động, dịch riêng các segment quá dài.
-
-    Args:
-        input_filepath (str): Đường dẫn đến file XLSX đầu vào.
-        output_filepath (str): Đường dẫn để lưu file XLSX đã dịch.
-        source_lang (str): Mã ngôn ngữ nguồn.
-        target_lang (str): Mã ngôn ngữ đích.
-        batch_size_segments (int): Số lượng đoạn text tối đa MONG MUỐN trong mỗi lần gọi API.
-        max_words_per_segment (int): Giới hạn từ tối đa cho một segment để được dịch theo batch.
-                                     Các segment dài hơn sẽ được dịch riêng lẻ.
-        delay_between_requests (int): Thời gian chờ (giây) giữa các lần gọi API dịch.
-    """
+    
     client = MongoClient("mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0")
     db = client['excel']
     fs = gridfs.GridFS(db, collection='root_file')
@@ -442,6 +492,8 @@ def translate_xlsx(file_id, file_name, source_lang='en', target_lang='vi', batch
     excel_file = BytesIO(ppt_file.read())
 
     xml_folder = unzip_office_file(excel_file)
+    path_to_workbook_xml = os.path.join(xml_folder, "xl", "workbook.xml")
+    translate_sheet_names_via_regex(path_to_workbook_xml, source_lang, target_lang)
 
     modifiable_nodes, global_data = extract_text_from_sheet(xml_folder)
 

test.ipynb

@@ -30,7 +30,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -125,11 +125,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
-    "def download_input_from_mongodb(file_id, save_name, db_name=\"ppt\", collection_name=\"root_file\"):\n",
+    "def download_input_from_mongodb(file_id, save_name, db_name=\"excel\", collection_name=\"root_file\"):\n",
     "    os.makedirs(\"D:\\Show_me_everything\\Machine Translation\\input\", exist_ok=True)\n",
     "\n",
     "    full_file_path = os.path.join(\"D:\\Show_me_everything\\Machine Translation\\input\", save_name)\n",
@@ -160,7 +160,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "def download_output_from_mongodb(file_id, save_name, db_name=\"ppt\", collection_name=\"root_file\"):\n",
+    "def download_output_from_mongodb(file_id, save_name, db_name=\"excel\", collection_name=\"root_file\"):\n",
     "    os.makedirs(\"D:\\Show_me_everything\\Machine Translation\\output\", exist_ok=True)\n",
     "\n",
     "    full_file_path = os.path.join(\"D:\\Show_me_everything\\Machine Translation\\output\", save_name)\n",
@@ -187,36 +187,36 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 18,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "✅ File đã được tải về: D:\\Show_me_everything\\Machine Translation\\input\\syllabus_big_data_analytics.docx\n"
+      "✅ File đã được tải về: D:\\Show_me_everything\\Machine Translation\\input\\sample(1).xlsx\n"
      ]
     }
    ],
    "source": [
-    "download_input_from_mongodb(file_id=\"67f725830f818c0a963a9421\", save_name=\"syllabus_big_data_analytics.docx\", db_name=\"word\", collection_name=\"root_file\")"
+    "download_input_from_mongodb(file_id=\"6836967a57ffb0e2132ae372\", save_name=\"sample(1).xlsx\", db_name=\"excel\", collection_name=\"root_file\")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "✅ File đã được tải về: D:\\Show_me_everything\\Machine Translation\\output\\syllabus_big_data_analytics.docx\n"
+      "✅ File đã được tải về: D:\\Show_me_everything\\Machine Translation\\output\\file_rangdong_26052025.xlsx\n"
      ]
     }
    ],
    "source": [
-    "download_output_from_mongodb(file_id=\"67f725a10f818c0a963a9425\", save_name=\"syllabus_big_data_analytics.docx\", db_name=\"word\", collection_name=\"final_file\")"
+    "download_output_from_mongodb(file_id=\"6833d51e97f852b6a2084ea8\", save_name=\"file_rangdong_26052025.xlsx\", db_name=\"excel\", collection_name=\"final_file\")"
    ]
   },
   {
@@ -228,17 +228,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "✅ Đã xóa 4 file trong collection 'root_file'\n",
+      "✅ Đã xóa 14 file trong collection 'root_file'\n",
       "✅ Đã xóa 0 file trong collection 'final_pptx'\n",
-      "✅ Đã xóa 1 file trong collection 'original_xml'\n",
-      "✅ Đã xóa 1 file trong collection 'final_xml'\n"
+      "✅ Đã xóa 0 file trong collection 'original_xml'\n",
+      "✅ Đã xóa 0 file trong collection 'final_xml'\n"
      ]
     }
    ],
@@ -250,21 +250,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 40,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "✅ Đã xóa 3 file trong collection 'root_file'\n",
-      "✅ Đã xóa 2 file trong collection 'final_file'\n"
+      "✅ Đã xóa 16 file trong collection 'root_file'\n",
+      "✅ Đã xóa 14 file trong collection 'final_file'\n"
      ]
     }
    ],
    "source": [
     "for i in ['root_file', 'final_file']:\n",
-    "    delete_all_files_in_collection(i, db_name=\"word\")"
+    "    delete_all_files_in_collection(i, db_name=\"excel\")"
    ]
   },
   {
@@ -324,6 +324,404 @@
    "source": [
     "delete_pptx_from_mongodb(file_id='67d383681d4db191e51b0bd8')"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from io import BytesIO\n",
+    "from openpyxl import load_workbook\n",
+    "from utils.utils import translate_single_text\n",
+    "from pymongo import MongoClient\n",
+    "from bson import ObjectId\n",
+    "import gridfs\n",
+    "from utils.utils import unzip_office_file\n",
+    "\n",
+    "\n",
+    "# 1. Kết nối tới MongoDB\n",
+    "client = MongoClient(\"mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0\")\n",
+    "db = client['excel']\n",
+    "fs = gridfs.GridFS(db, collection='root_file')\n",
+    "\n",
+    "# 2. Tải file Excel từ MongoDB\n",
+    "file_obj = fs.get(ObjectId('6836c69e59530f034bd86576'))\n",
+    "excel_file = BytesIO(file_obj.read())\n",
+    "\n",
+    "xml_folder = unzip_office_file(r\"D:\\Show_me_everything\\Machine Translation\\input\\sample.xlsx\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 31,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import xml.etree.ElementTree as ET\n",
+    "from typing import List\n",
+    "from utils.utils import translate_single_text\n",
+    "\n",
+    "def _register_all_namespaces(xml_file_path):\n",
+    "    try:\n",
+    "        namespaces = dict([\n",
+    "            node for _, node in ET.iterparse(xml_file_path, events=['start-ns'])\n",
+    "        ])\n",
+    "        for prefix, uri in namespaces.items():\n",
+    "            ET.register_namespace(prefix, uri)\n",
+    "        if '' in namespaces and namespaces[''] == \"http://schemas.openxmlformats.org/spreadsheetml/2006/main\":\n",
+    "             ET.register_namespace('', \"http://schemas.openxmlformats.org/spreadsheetml/2006/main\")\n",
+    "        if 'r' not in namespaces:\n",
+    "             ET.register_namespace('r', \"http://schemas.openxmlformats.org/officeDocument/2006/relationships\")\n",
+    "        if 'mc' not in namespaces:\n",
+    "            ET.register_namespace('mc', \"http://schemas.openxmlformats.org/markup-compatibility/2006\")\n",
+    "        if 'x15' not in namespaces:\n",
+    "            ET.register_namespace('x15', \"http://schemas.microsoft.com/office/spreadsheetml/2010/11/main\")\n",
+    "        return namespaces # Trả về để có thể dùng trong findall nếu cần\n",
+    "    except Exception as e:\n",
+    "        print(f\"Lỗi khi đăng ký namespace từ {xml_file_path}: {e}\")\n",
+    "        return {}\n",
+    "\n",
+    "def get_all_sheet_names_from_file(workbook_xml_path, source_lang='chinese', target_lang='vietnamese') -> List[str]:\n",
+    "    sheet_names = {}\n",
+    "    try:\n",
+    "        # Phân tích trực tiếp từ file\n",
+    "        tree = ET.parse(workbook_xml_path)\n",
+    "        root = tree.getroot()\n",
+    "\n",
+    "        namespaces = {'main': 'http://schemas.openxmlformats.org/spreadsheetml/2006/main'}\n",
+    "\n",
+    "        for sheet_element in root.findall('.//main:sheet', namespaces):\n",
+    "            name = sheet_element.get('name')\n",
+    "            if name:\n",
+    "                sheet_names[name] = name\n",
+    "        for k in sheet_names:\n",
+    "            sheet_names[k] = translate_single_text(k, source_lang, target_lang)\n",
+    "\n",
+    "    except FileNotFoundError:\n",
+    "        print(f\"Lỗi: Không tìm thấy file tại '{workbook_xml_path}'\")\n",
+    "    except ET.ParseError as e:\n",
+    "        print(f\"Lỗi khi phân tích XML từ file '{workbook_xml_path}': {e}\")\n",
+    "    except Exception as e:\n",
+    "        print(f\"Đã xảy ra lỗi không mong muốn khi xử lý file '{workbook_xml_path}': {e}\")\n",
+    "\n",
+    "    return sheet_names\n",
+    "\n",
+    "def translate_sheet_names_in_file(workbook_xml_path, source_lang = 'chinese', target_lang = 'vietnamese'):\n",
+    "\n",
+    "    original_to_translated_map: Dict[str, str] = {}\n",
+    "    file_modified = False\n",
+    "\n",
+    "    try:\n",
+    "        parsed_namespaces = _register_all_namespaces(workbook_xml_path)\n",
+    "        tree = ET.parse(workbook_xml_path)\n",
+    "        root = tree.getroot()\n",
+    "        search_namespaces = {'main': 'http://schemas.openxmlformats.org/spreadsheetml/2006/main'}\n",
+    "        if '' in parsed_namespaces and parsed_namespaces[''] == search_namespaces['main']:\n",
+    "            pass # search_namespaces đã ổn\n",
+    "\n",
+    "\n",
+    "        # Bước 3: Lặp qua các sheet, dịch và cập nhật tên\n",
+    "        for sheet_element in root.findall('.//main:sheet', search_namespaces):\n",
+    "            original_name = sheet_element.get('name')\n",
+    "            if original_name and original_name.strip(): # Chỉ dịch nếu tên có nội dung\n",
+    "                try:\n",
+    "                    # Gọi hàm dịch của bạn\n",
+    "                    translated_name = translate_single_text(original_name, source_lang, target_lang)\n",
+    "                    os.wait(5)\n",
+    "\n",
+    "                    if translated_name and translated_name.strip() and translated_name != original_name:\n",
+    "                        # Cập nhật thuộc tính 'name' của element trong cây XML\n",
+    "                        sheet_element.set('name', translated_name)\n",
+    "                        original_to_translated_map[original_name] = translated_name\n",
+    "                        file_modified = True\n",
+    "                        print(f\"Đã dịch sheet: '{original_name}' -> '{translated_name}'\")\n",
+    "                    else:\n",
+    "                        # Nếu dịch thất bại hoặc không thay đổi, giữ lại tên gốc trong map\n",
+    "                        original_to_translated_map[original_name] = original_name\n",
+    "                        if translated_name and translated_name != original_name :\n",
+    "                             print(f\"Bản dịch cho '{original_name}' trống hoặc không hợp lệ, không cập nhật XML.\")\n",
+    "                        elif not translated_name:\n",
+    "                             print(f\"Dịch thất bại cho '{original_name}', không cập nhật XML.\")\n",
+    "\n",
+    "\n",
+    "                except Exception as e_translate:\n",
+    "                    print(f\"Lỗi khi dịch tên sheet '{original_name}': {e_translate}\")\n",
+    "                    original_to_translated_map[original_name] = original_name # Ghi nhận lỗi, giữ tên gốc\n",
+    "\n",
+    "        # Bước 4: Nếu có thay đổi, ghi lại toàn bộ cây XML vào file\n",
+    "        if file_modified:\n",
+    "            # encoding='utf-8' và xml_declaration=True là quan trọng\n",
+    "            tree.write(workbook_xml_path, encoding='utf-8', xml_declaration=True)\n",
+    "            print(f\"Đã cập nhật thành công file: {workbook_xml_path}\")\n",
+    "        else:\n",
+    "            print(f\"Không có tên sheet nào được thay đổi trong file: {workbook_xml_path}\")\n",
+    "\n",
+    "    except FileNotFoundError:\n",
+    "        print(f\"Lỗi: Không tìm thấy file tại '{workbook_xml_path}'\")\n",
+    "    except ET.ParseError as e:\n",
+    "        print(f\"Lỗi khi phân tích XML từ file '{workbook_xml_path}': {e}\")\n",
+    "    except Exception as e:\n",
+    "        print(f\"Đã xảy ra lỗi không mong muốn khi xử lý file '{workbook_xml_path}': {e}\")\n",
+    "        import traceback\n",
+    "        traceback.print_exc()\n",
+    "\n",
+    "\n",
+    "    return original_to_translated_map"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 24,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Đã dịch sheet: '流程图' -> 'Lưu đồ'\n",
+      "Đã dịch sheet: '分割光源板' -> 'Tấm dẫn sáng phân đoạn'\n",
+      "Đã dịch sheet: '支架外观检验' -> 'Kiểm tra ngoại quan giá đỡ'\n",
+      "Đã dịch sheet: '固定驱动' -> 'Ổ đĩa cố định'\n",
+      "Đã dịch sheet: '固定接地线' -> 'Dây nối đất cố định'\n",
+      "Đã dịch sheet: '整理电源线' -> 'Sắp xếp dây nguồn'\n",
+      "Đã dịch sheet: '固定光源板' -> 'Bảng nguồn sáng cố định'\n",
+      "Đã dịch sheet: '连接光源板' -> 'Kết nối bảng nguồn sáng'\n",
+      "Đã dịch sheet: '焊接光源板' -> 'Hàn bảng nguồn sáng'\n",
+      "Đã dịch sheet: '连接驱动' -> 'Kết nối trình điều khiển'\n",
+      "Đã dịch sheet: '安装端头 ' -> 'Lắp đặt đầu cuối'\n",
+      "Đã dịch sheet: '试 亮' -> 'Thử Sáng'\n",
+      "Đã dịch sheet: '绝缘、接地 ' -> 'Cách điện, tiếp đất'\n",
+      "Đã dịch sheet: '安装透光罩 (2)' -> 'Lắp chụp đèn (2)'\n",
+      "Đã dịch sheet: '老练' -> 'Lão luyện'\n",
+      "Đã dịch sheet: '二次试亮' -> 'Thử sáng lần hai'\n",
+      "Đã cập nhật thành công file: D:\\Show_me_everything\\MT deploy\\MT_deploy\\unzipped_office\\xl\\workbook.xml\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'流程图': 'Lưu đồ',\n",
+       " '分割光源板': 'Tấm dẫn sáng phân đoạn',\n",
+       " '支架外观检验': 'Kiểm tra ngoại quan giá đỡ',\n",
+       " '固定驱动': 'Ổ đĩa cố định',\n",
+       " '固定接地线': 'Dây nối đất cố định',\n",
+       " '整理电源线': 'Sắp xếp dây nguồn',\n",
+       " '固定光源板': 'Bảng nguồn sáng cố định',\n",
+       " '连接光源板': 'Kết nối bảng nguồn sáng',\n",
+       " '焊接光源板': 'Hàn bảng nguồn sáng',\n",
+       " '连接驱动': 'Kết nối trình điều khiển',\n",
+       " '安装端头 ': 'Lắp đặt đầu cuối',\n",
+       " '试 亮': 'Thử Sáng',\n",
+       " '绝缘、接地 ': 'Cách điện, tiếp đất',\n",
+       " '安装透光罩 (2)': 'Lắp chụp đèn (2)',\n",
+       " '老练': 'Lão luyện',\n",
+       " '二次试亮': 'Thử sáng lần hai'}"
+      ]
+     },
+     "execution_count": 24,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "translate_sheet_names_in_file(r'D:\\Show_me_everything\\MT deploy\\MT_deploy\\unzipped_office\\xl\\workbook.xml')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'流程图': 'Lưu đồ',\n",
+       " '分割光源板': 'Tấm dẫn sáng phân đoạn',\n",
+       " '支架外观检验': 'Kiểm tra ngoại quan của giá đỡ',\n",
+       " '固定驱动': 'Ổ đĩa cố định',\n",
+       " '固定接地线': 'Dây nối đất cố định',\n",
+       " '整理电源线': 'Sắp xếp dây nguồn',\n",
+       " '固定光源板': 'Tấm nền nguồn sáng cố định',\n",
+       " '连接光源板': 'Kết nối bảng nguồn sáng',\n",
+       " '焊接光源板': 'Hàn bảng nguồn sáng',\n",
+       " '连接驱动': 'Kết nối trình điều khiển',\n",
+       " '安装端头 ': 'Lắp đặt đầu cuối',\n",
+       " '试 亮': 'Thử Lượng',\n",
+       " '绝缘、接地 ': 'Cách điện, tiếp đất',\n",
+       " '安装透光罩 (2)': 'Lắp đặt chụp đèn trong suốt (2)',\n",
+       " '老练': 'Lão luyện',\n",
+       " '二次试亮': 'Thử sáng lần hai'}"
+      ]
+     },
+     "execution_count": 39,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "sheets_name_d"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import re\n",
+    "from typing import Dict, Callable\n",
+    "# Giả sử utils.utils.translate_single_text đã được import và hoạt động đúng\n",
+    "# từ file trước: from utils.utils import translate_single_text\n",
+    "\n",
+    "def translate_sheet_names_via_regex(\n",
+    "    workbook_xml_path: str,\n",
+    "    source_lang: str = 'chinese',\n",
+    "    target_lang: str = 'vietnamese'\n",
+    ") -> (Dict[str, str], bool):\n",
+    "\n",
+    "    original_to_translated_map: Dict[str, str] = {}\n",
+    "    modified_content: str = \"\"\n",
+    "    file_changed_flag: bool = False # Sử dụng tên biến rõ ràng hơn\n",
+    "\n",
+    "    try:\n",
+    "        with open(workbook_xml_path, 'r', encoding='utf-8') as f:\n",
+    "            content = f.read()\n",
+    "\n",
+    "        current_content = content\n",
+    "\n",
+    "        def replace_name_callback(match_obj):\n",
+    "            nonlocal file_changed_flag # Để sửa đổi biến bên ngoài\n",
+    "            nonlocal original_to_translated_map\n",
+    "\n",
+    "            attr_prefix = match_obj.group(1)  # Ví dụ: '<sheet other_attr=\"foo\" name='\n",
+    "            opening_quote = match_obj.group(2) # Ví dụ: '\"'\n",
+    "            original_name_xml_encoded = match_obj.group(3)\n",
+    "            attr_suffix = match_obj.group(5)  # Ví dụ: ' sheetId=\"12\" r:id=\"rId1\"/>'\n",
+    "\n",
+    "            original_name = original_name_xml_encoded # Tạm thời bỏ qua unescape/escape cho đơn giản ví dụ\n",
+    "\n",
+    "            if not original_name.strip():\n",
+    "                return match_obj.group(0) # Trả về chuỗi gốc nếu tên rỗng\n",
+    "\n",
+    "            translated_name = original_name # Mặc định giữ nguyên\n",
+    "            if original_name in original_to_translated_map and original_to_translated_map[original_name] != original_name:\n",
+    "                translated_name = original_to_translated_map[original_name]\n",
+    "                # Nếu đã dịch và có thay đổi, không cần gọi API dịch nữa\n",
+    "                if translated_name != original_name: # Cần kiểm tra lại vì map có thể lưu tên gốc nếu dịch lỗi\n",
+    "                     print(f\"Regex: Sử dụng bản dịch đã có cho '{original_name}' -> '{translated_name}'\")\n",
+    "                     file_changed_flag = True # Đảm bảo cờ được set nếu sử dụng bản dịch đã có mà khác gốc\n",
+    "            else:\n",
+    "                try:\n",
+    "                    translated_name_raw = translate_single_text(original_name, source_lang, target_lang)\n",
+    "\n",
+    "                    if translated_name_raw and translated_name_raw.strip() and translated_name_raw != original_name:\n",
+    "                        translated_name = translated_name_raw[:31]\n",
+    "                        original_to_translated_map[original_name] = translated_name\n",
+    "                        file_changed_flag = True\n",
+    "                        print(f\"Regex: Đã dịch sheet: '{original_name}' -> '{translated_name}'\")\n",
+    "                    else:\n",
+    "                        original_to_translated_map[original_name] = original_name # Lưu tên gốc nếu dịch lỗi/không đổi\n",
+    "                        # translated_name vẫn là original_name\n",
+    "                        if translated_name_raw and translated_name_raw.strip() and translated_name_raw == original_name:\n",
+    "                            print(f\"Bản dịch cho '{original_name}' giống hệt bản gốc, không thay đổi.\")\n",
+    "                        elif not (translated_name_raw and translated_name_raw.strip()):\n",
+    "                             print(f\"Bản dịch cho '{original_name}' trống hoặc không hợp lệ, giữ nguyên.\")\n",
+    "\n",
+    "                except Exception as e_translate:\n",
+    "                    print(f\"Lỗi khi gọi hàm dịch cho '{original_name}': {e_translate}\")\n",
+    "                    original_to_translated_map[original_name] = original_name\n",
+    "\n",
+    "            translated_name_xml_encoded = translated_name # Tạm thời bỏ qua escape\n",
+    "\n",
+    "            return f\"{attr_prefix}{opening_quote}{translated_name_xml_encoded}{opening_quote}{attr_suffix}\"\n",
+    "\n",
+    "        sheet_name_pattern = re.compile(\n",
+    "            r'(<sheet[^>]*?\\sname=)([\"\\'])((?:(?!\\2).)*?)(\\2)([^>]*?>)'\n",
+    "        )\n",
+    "        modified_content = sheet_name_pattern.sub(replace_name_callback, current_content)\n",
+    "\n",
+    "        if file_changed_flag:\n",
+    "            with open(workbook_xml_path, 'w', encoding='utf-8') as f:\n",
+    "                f.write(modified_content)\n",
+    "            print(f\"Regex: Đã cập nhật thành công file: {workbook_xml_path}\")\n",
+    "            return original_to_translated_map, True\n",
+    "        else:\n",
+    "            print(f\"Regex: Không có tên sheet nào được thay đổi trong file: {workbook_xml_path}\")\n",
+    "\n",
+    "    except FileNotFoundError:\n",
+    "        print(f\"Lỗi: Không tìm thấy file tại '{workbook_xml_path}'\")\n",
+    "    except Exception as e:\n",
+    "        print(f\"Đã xảy ra lỗi không mong muốn khi xử lý file '{workbook_xml_path}' bằng regex: {e}\")\n",
+    "        import traceback\n",
+    "        traceback.print_exc()\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 38,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Regex: Đã dịch sheet: '流程图' -> 'Lưu đồ'\n",
+      "Regex: Đã dịch sheet: '分割光源板' -> 'Tấm dẫn sáng phân đoạn'\n",
+      "Regex: Đã dịch sheet: '支架外观检验' -> 'Kiểm tra ngoại quan giá đỡ'\n",
+      "Regex: Đã dịch sheet: '固定驱动' -> 'Ổ đĩa cố định'\n",
+      "Regex: Đã dịch sheet: '固定接地线' -> 'Dây tiếp đất cố định'\n",
+      "Regex: Đã dịch sheet: '整理电源线' -> 'Sắp xếp dây nguồn'\n",
+      "Regex: Đã dịch sheet: '固定光源板' -> 'Bảng nguồn sáng cố định'\n",
+      "Regex: Đã dịch sheet: '连接光源板' -> 'Kết nối bảng đèn nền'\n",
+      "Regex: Đã dịch sheet: '焊接光源板' -> 'Bảng nguồn sáng hàn'\n",
+      "Regex: Đã dịch sheet: '连接驱动' -> 'Kết nối trình điều khiển'\n",
+      "Regex: Đã dịch sheet: '安装端头 ' -> 'Lắp đặt đầu cuối'\n",
+      "Regex: Đã dịch sheet: '试 亮' -> 'Thử Sáng'\n",
+      "Regex: Đã dịch sheet: '绝缘、接地 ' -> 'Cách điện, tiếp đất'\n",
+      "Regex: Đã dịch sheet: '安装透光罩 (2)' -> 'Lắp chụp đèn (2)'\n",
+      "Regex: Đã dịch sheet: '老练' -> 'Lão luyện'\n",
+      "Regex: Đã dịch sheet: '二次试亮' -> 'Thử sáng lần hai'\n",
+      "Regex: Đã cập nhật thành công file: D:\\Show_me_everything\\MT deploy\\MT_deploy\\unzipped_office\\xl\\workbook.xml\n"
+     ]
+    }
+   ],
+   "source": [
+    "translation_map_regex, success_regex = translate_sheet_names_via_regex(r'D:\\Show_me_everything\\MT deploy\\MT_deploy\\unzipped_office\\xl\\workbook.xml')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n",
+    "<workbook xmlns=\"http://schemas.openxmlformats.org/spreadsheetml/2006/main\" xmlns:r=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships\" xmlns:mc=\"http://schemas.openxmlformats.org/markup-compatibility/2006\" mc:Ignorable=\"x15 xr xr6 xr10 xr2\" xmlns:x15=\"http://schemas.microsoft.com/office/spreadsheetml/2010/11/main\" xmlns:xr=\"http://schemas.microsoft.com/office/spreadsheetml/2014/revision\" xmlns:xr6=\"http://schemas.microsoft.com/office/spreadsheetml/2016/revision6\" xmlns:xr10=\"http://schemas.microsoft.com/office/spreadsheetml/2016/revision10\" xmlns:xr2=\"http://schemas.microsoft.com/office/spreadsheetml/2015/revision2\"><fileVersion appName=\"xl\" lastEdited=\"7\" lowestEdited=\"5\" rupBuild=\"28324\"/><workbookPr codeName=\"ThisWorkbook\"/><mc:AlternateContent xmlns:mc=\"http://schemas.openxmlformats.org/markup-compatibility/2006\"><mc:Choice Requires=\"x15\"><x15ac:absPath url=\"G:\\Machine_Learning\\machine_translation\\temp\\\" xmlns:x15ac=\"http://schemas.microsoft.com/office/spreadsheetml/2010/11/ac\"/></mc:Choice></mc:AlternateContent><xr:revisionPtr revIDLastSave=\"0\" documentId=\"13_ncr:1_{0BB03666-2754-4098-B7E4-88837ADAC82E}\" xr6:coauthVersionLast=\"47\" xr6:coauthVersionMax=\"47\" xr10:uidLastSave=\"{00000000-0000-0000-0000-000000000000}\"/><bookViews><workbookView xWindow=\"-110\" yWindow=\"-110\" windowWidth=\"25820\" windowHeight=\"15500\" firstSheet=\"15\" activeTab=\"15\" xr2:uid=\"{00000000-000D-0000-FFFF-FFFF00000000}\"/></bookViews><sheets><sheet name=\"流程图\" sheetId=\"12\" r:id=\"rId1\"/><sheet name=\"分割光源板\" sheetId=\"2\" r:id=\"rId2\"/><sheet name=\"支架外观检验\" sheetId=\"24\" r:id=\"rId3\"/><sheet name=\"固定驱动\" sheetId=\"21\" r:id=\"rId4\"/><sheet name=\"固定接地线\" sheetId=\"4\" r:id=\"rId5\"/><sheet name=\"整理电源线\" sheetId=\"22\" r:id=\"rId6\"/><sheet name=\"固定光源板\" sheetId=\"3\" r:id=\"rId7\"/><sheet name=\"连接光源板\" sheetId=\"5\" r:id=\"rId8\"/><sheet name=\"焊接光源板\" sheetId=\"29\" r:id=\"rId9\"/><sheet name=\"连接驱动\" sheetId=\"30\" r:id=\"rId10\"/><sheet name=\"安装端头 \" sheetId=\"31\" r:id=\"rId11\"/><sheet name=\"试 亮\" sheetId=\"20\" r:id=\"rId12\"/><sheet name=\"绝缘、接地 \" sheetId=\"9\" r:id=\"rId13\"/><sheet name=\"安装透光罩 (2)\" sheetId=\"32\" state=\"hidden\" r:id=\"rId14\"/><sheet name=\"老练\" sheetId=\"10\" r:id=\"rId15\"/><sheet name=\"二次试亮\" sheetId=\"6\" r:id=\"rId16\"/></sheets><definedNames><definedName name=\"_xlnm.Print_Area\" localSheetId=\"15\">二次试亮!$A$1:$L$39</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"1\">分割光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"6\">固定光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"4\">固定接地线!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"3\">固定驱动!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"10\">'安装端头 '!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"5\">整理电源线!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"0\">流程图!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"8\">焊接光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"12\">'绝缘、接地 '!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"14\">老练!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"11\">'试 亮'!$A$1:$L$39</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"7\">连接光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"9\">连接驱动!$A$1:$L$40</definedName></definedNames><calcPr calcId=\"191029\"/><extLst><ext uri=\"{B58B0392-4F1F-4190-BB64-5DF3571DCE5F}\" xmlns:xcalcf=\"http://schemas.microsoft.com/office/spreadsheetml/2018/calcfeatures\"><xcalcf:calcFeatures><xcalcf:feature name=\"microsoft.com:RD\"/><xcalcf:feature name=\"microsoft.com:Single\"/><xcalcf:feature name=\"microsoft.com:FV\"/><xcalcf:feature name=\"microsoft.com:CNMTM\"/><xcalcf:feature name=\"microsoft.com:LET_WF\"/><xcalcf:feature name=\"microsoft.com:LAMBDA_WF\"/><xcalcf:feature name=\"microsoft.com:ARRAYTEXT_WF\"/></xcalcf:calcFeatures></ext></extLst></workbook>\n",
+    "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n",
+    "<workbook xmlns=\"http://schemas.openxmlformats.org/spreadsheetml/2006/main\" xmlns:r=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships\" xmlns:mc=\"http://schemas.openxmlformats.org/markup-compatibility/2006\" mc:Ignorable=\"x15 xr xr6 xr10 xr2\" xmlns:x15=\"http://schemas.microsoft.com/office/spreadsheetml/2010/11/main\" xmlns:xr=\"http://schemas.microsoft.com/office/spreadsheetml/2014/revision\" xmlns:xr6=\"http://schemas.microsoft.com/office/spreadsheetml/2016/revision6\" xmlns:xr10=\"http://schemas.microsoft.com/office/spreadsheetml/2016/revision10\" xmlns:xr2=\"http://schemas.microsoft.com/office/spreadsheetml/2015/revision2\"><fileVersion appName=\"xl\" lastEdited=\"7\" lowestEdited=\"5\" rupBuild=\"28324\"/><workbookPr codeName=\"ThisWorkbook\"/><mc:AlternateContent xmlns:mc=\"http://schemas.openxmlformats.org/markup-compatibility/2006\"><mc:Choice Requires=\"x15\"><x15ac:absPath url=\"G:\\Machine_Learning\\machine_translation\\temp\\\" xmlns:x15ac=\"http://schemas.microsoft.com/office/spreadsheetml/2010/11/ac\"/></mc:Choice></mc:AlternateContent><xr:revisionPtr revIDLastSave=\"0\" documentId=\"13_ncr:1_{0BB03666-2754-4098-B7E4-88837ADAC82E}\" xr6:coauthVersionLast=\"47\" xr6:coauthVersionMax=\"47\" xr10:uidLastSave=\"{00000000-0000-0000-0000-000000000000}\"/><bookViews><workbookView xWindow=\"-110\" yWindow=\"-110\" windowWidth=\"25820\" windowHeight=\"15500\" firstSheet=\"15\" activeTab=\"15\" xr2:uid=\"{00000000-000D-0000-FFFF-FFFF00000000}\"/></bookViews><sheets><sheet name=\"Lưu đồ\" sheetId=\"12\" r:id=\"rId1\"/><sheet name=\"Tấm dẫn sáng phân đoạn\" sheetId=\"2\" r:id=\"rId2\"/><sheet name=\"Kiểm tra ngoại quan giá đỡ\" sheetId=\"24\" r:id=\"rId3\"/><sheet name=\"Ổ đĩa cố định\" sheetId=\"21\" r:id=\"rId4\"/><sheet name=\"Dây tiếp đất cố định\" sheetId=\"4\" r:id=\"rId5\"/><sheet name=\"S���p xếp dây nguồn\" sheetId=\"22\" r:id=\"rId6\"/><sheet name=\"Tấm nền nguồn sáng cố định\" sheetId=\"3\" r:id=\"rId7\"/><sheet name=\"Kết nối bảng đèn.\" sheetId=\"5\" r:id=\"rId8\"/><sheet name=\"Hàn bảng nguồn sáng\" sheetId=\"29\" r:id=\"rId9\"/><sheet name=\"Kết nối trình điều khiển\" sheetId=\"30\" r:id=\"rId10\"/><sheet name=\"Lắp đặt đầu cuối\" sheetId=\"31\" r:id=\"rId11\"/><sheet name=\"Thử Sáng\" sheetId=\"20\" r:id=\"rId12\"/><sheet name=\"Cách điện, tiếp đất\" sheetId=\"9\" r:id=\"rId13\"/><sheet name=\"Lắp chụp đèn (2)\" sheetId=\"32\" state=\"hidden\" r:id=\"rId14\"/><sheet name=\"Lão luyện\" sheetId=\"10\" r:id=\"rId15\"/><sheet name=\"Thử sáng lần hai\" sheetId=\"6\" r:id=\"rId16\"/></sheets><definedNames><definedName name=\"_xlnm.Print_Area\" localSheetId=\"15\">二次试亮!$A$1:$L$39</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"1\">分割光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"6\">固定光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"4\">固定接地线!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"3\">固定驱动!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"10\">'安装端头 '!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"5\">整理电源线!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"0\">流程图!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"8\">焊接光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"12\">'绝缘、接地 '!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"14\">老练!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"11\">'试 亮'!$A$1:$L$39</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"7\">连接光源板!$A$1:$L$40</definedName><definedName name=\"_xlnm.Print_Area\" localSheetId=\"9\">连接驱动!$A$1:$L$40</definedName></definedNames><calcPr calcId=\"191029\"/><extLst><ext uri=\"{B58B0392-4F1F-4190-BB64-5DF3571DCE5F}\" xmlns:xcalcf=\"http://schemas.microsoft.com/office/spreadsheetml/2018/calcfeatures\"><xcalcf:calcFeatures><xcalcf:feature name=\"microsoft.com:RD\"/><xcalcf:feature name=\"microsoft.com:Single\"/><xcalcf:feature name=\"microsoft.com:FV\"/><xcalcf:feature name=\"microsoft.com:CNMTM\"/><xcalcf:feature name=\"microsoft.com:LET_WF\"/><xcalcf:feature name=\"microsoft.com:LAMBDA_WF\"/><xcalcf:feature name=\"microsoft.com:ARRAYTEXT_WF\"/></xcalcf:calcFeatures></ext></extLst></workbook>\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 39,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "26"
+      ]
+     },
+     "execution_count": 39,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "str = 'Kiểm tra ngoại quan giá đỡ'\n",
+    "len(str)"
+   ]
   }
  ],
  "metadata": {

utils/utils.py

@@ -4,11 +4,12 @@ import google.generativeai as genai
 import tempfile
 import io
 import json
+import time
+from google.api_core.exceptions import ResourceExhausted
 
 genai.configure(api_key="AIzaSyBH8O5IfqYrJ5wtWnmUC21IfMjzJCrTm3I")
 
 
-
 def unzip_office_file(pptx_file: io.BytesIO):
     """
     Giải nén nội dung từ file PPTX (dạng BytesIO) vào thư mục tạm thời.
@@ -24,36 +25,57 @@ def unzip_office_file(pptx_file: io.BytesIO):
     return output_dir
 
     
-def translate_single_text(text, source_lang='English', target_lang="Vietnamese"):
-    if not text:
-        return ""  # Bỏ qua nếu chuỗi rỗng
-
-    try:
-        model = genai.GenerativeModel('gemini-2.0-flash') # Sử dụng model từ code gốc nếu hoạt động tốt
-
-        # --- Prompt đơn giản chỉ yêu cầu dịch thuật ---
-        system_prompt_simple = f"""You are a translation engine.
-        Translate the following text accurately from {source_lang} to {target_lang}.
-        Provide *only* the translated text as a single string.
-        Do NOT add any extra formatting, delimiters like '#', introductory phrases, or explanations."""
+def translate_single_text(text: str, source_lang: str = 'English', target_lang: str = "Vietnamese",
+                          max_retries: int = 5, retry_delay_seconds: int = 3) -> str:
+    if not text or not text.strip():
+        return ""  # Bỏ qua nếu chuỗi rỗng hoặc chỉ chứa khoảng trắng
 
-        user_prompt = f"Source language: {source_lang}. Target language: {target_lang}. Text to translate: {text}"
-        full_prompt = system_prompt_simple.strip() + "\n\n" + user_prompt.strip()
+    retries = 0
+    while retries <= max_retries:
+        try:
+            model = genai.GenerativeModel('gemini-2.0-flash') # Hoặc 'gemini-1.0-pro', 'gemini-1.5-flash' tùy bạn chọn
+
+            system_prompt_simple = f"""You are a translation engine.
+            Translate the following text accurately from {source_lang} to {target_lang}.
+            Provide *only* the translated text as a single string.
+            Do NOT add any extra formatting, delimiters like '#', introductory phrases, or explanations."""
+
+            user_prompt = f"Source language: {source_lang}. Target language: {target_lang}. Text to translate: {text}"
+            full_prompt = system_prompt_simple.strip() + "\n\n" + user_prompt.strip()
+
+            response = model.generate_content(
+                contents=full_prompt,
+                generation_config={
+                    'temperature': 0.2, 
+                    'top_p': 1.0,       
+                    'top_k': 1,         
+                    # 'max_output_tokens': 2048, 
+                }
+            )
+            # Kiểm tra xem response có text không và có ứng viên không
+            if response.candidates and response.candidates[0].content.parts:
+                translated_text = "".join(part.text for part in response.candidates[0].content.parts if hasattr(part, 'text')).strip()
+                return translated_text
+            else:
+                print(f"Không nhận được nội dung hợp lệ từ API cho văn bản: '{text[:50]}...'")
+                # Không thử lại với lỗi này, trả về rỗng
+                return ""
+
+
+        except ResourceExhausted as e: # Bắt lỗi 429 (Too Many Requests / Quota Exceeded)
+            print(f"Lỗi quota (429) khi dịch '{text[:50]}...': {e}. Đang thử lại sau {retry_delay_seconds} giây. Lần thử {retries + 1}/{max_retries +1 }.")
+            if retries < max_retries:
+                time.sleep(retry_delay_seconds)
+                retries += 1
+            else:
+                print(f"Đã vượt quá số lần thử lại tối đa ({max_retries + 1}) cho '{text[:50]}...'. Bỏ qua.")
+                return "" # Trả về rỗng sau khi đã thử lại tối đa số lần
 
-        response = model.generate_content(
-            contents=full_prompt,
-            generation_config={
-                'temperature': 0.7, # Nhiệt độ phù hợp cho dịch thuật (có thể điều chỉnh)
-                'top_p': 1,
-                'top_k': 1,
-            }
-        )
-        translated_text = response.text.strip()
-        return translated_text
+        except Exception as e:
+            print(f"Lỗi không mong muốn trong quá trình dịch (translate_single_text) cho '{text[:50]}...': {e}")
+            return ""
 
-    except Exception as e:
-        print(f"Lỗi trong quá trình dịch (translate_single_text): {e}")
-        return "" # Trả về chuỗi rỗng nếu có lỗi
+    return "" # Trường hợp không bao giờ nên xảy ra nếu logic vòng lặp đúng
     
 
 def preprocess_text(text_list):