Upload app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import re
 from docx import Document
-from docx.shared import Cm
+from docx.shared import Cm, Pt
 from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 from docx.oxml.ns import qn
 from docx.oxml import OxmlElement
@@ -49,10 +49,27 @@ def format_docx(file, chapter_keywords):
             
             # 3. 檢查是否為章節標題
             if re.search(combined_pattern, paragraph.text):
-                # 設定為標題 1 樣式
-                paragraph.style = doc.styles['Heading 1']
+                # 嘗試設定標題 1 樣式（處理不同語言版本的樣式名稱）
+                try:
+                    # 英文版本
+                    paragraph.style = doc.styles['Heading 1']
+                except KeyError:
+                    try:
+                        # 中文版本
+                        paragraph.style = doc.styles['標題 1']
+                    except KeyError:
+                        try:
+                            # 其他可能的名稱
+                            paragraph.style = doc.styles['Heading1']
+                        except KeyError:
+                            # 如果都找不到，手動設定樣式
+                            paragraph.style = doc.styles['Normal']
+                            # 手動設定標題樣式
+                            for run in paragraph.runs:
+                                run.font.bold = True
+                                run.font.size = Pt(16)  # 16pt
                 
-                # 設定粗體
+                # 確保粗體設定
                 for run in paragraph.runs:
                     run.font.bold = True