Upload app.py

app.py

@@ -2,14 +2,9 @@ import gradio as gr
 import re
 from docx import Document
 from docx.shared import Cm, Pt
-from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
-from docx.enum.style import WD_STYLE_TYPE
-from docx.oxml.ns import qn
-from docx.oxml import OxmlElement
 import tempfile
 import os
 
-
 def format_docx(file, chapter_keywords):
     """
     處理上傳的 Word 文件
@@ -21,13 +16,17 @@ def format_docx(file, chapter_keywords):
         return None, "請輸入章節分段方式（例如：章,節,話）"
 
     try:
-        from docx import Document
-        from docx.shared import Cm, Pt
-        import re
-        import tempfile
-
         doc = Document(file.name)
 
+        # 確保有 Heading 1 樣式
+        from docx.enum.style import WD_STYLE_TYPE
+        styles = doc.styles
+        if 'Heading 1' not in styles:
+            heading_style = styles.add_style('Heading 1', WD_STYLE_TYPE.PARAGRAPH)
+            heading_style.base_style = styles['Normal']
+            heading_style.font.bold = True
+            heading_style.font.size = Pt(16)
+
         # 解析章節關鍵字
         keywords = [keyword.strip() for keyword in chapter_keywords.split(',')]
 
@@ -86,10 +85,6 @@ def format_docx(file, chapter_keywords):
         return None, f"❌ 處理失敗：{str(e)}"
 
 def create_interface():
-
-    """
-    建立 Gradio 介面
-    """
     with gr.Blocks(title="Word 文件格式化工具", theme=gr.themes.Soft()) as demo:
         gr.HTML("""
         <div style="text-align: center; margin-bottom: 20px;">
@@ -97,44 +92,33 @@ def create_interface():
             <p>自動格式化您的 Word 文件，設定章節樣式和分頁</p>
         </div>
         """)
-        
+
         with gr.Row():
             with gr.Column(scale=1):
-                # 檔案上傳
                 file_input = gr.File(
                     label="上傳 Word 文件 (.docx)",
                     file_types=[".docx"],
                     file_count="single"
                 )
-                
-                # 章節關鍵字輸入
                 chapter_input = gr.Textbox(
                     label="章節分段方式",
                     placeholder="章,節,話",
                     info="請輸入章節關鍵字，用逗號分隔（例如：章,節,話）",
                     value="章,節,話"
                 )
-                
-                # 處理按鈕
                 process_btn = gr.Button("🔄 開始處理", variant="primary", size="lg")
-                
             with gr.Column(scale=1):
-                # 狀態顯示
                 status_output = gr.Textbox(
                     label="處理狀態",
                     interactive=False,
                     lines=3
                 )
-                
-                # 下載連結
                 download_output = gr.File(
                     label="下載處理後的文件",
                     interactive=False
                 )
-        
-        # 功能說明
-        gr.HTML("""
-        <div style="margin-top: 30px; padding: 20px; background-color: #f8f9fa; border-radius: 10px;">
+
+        gr.HTML("""<div style="margin-top: 30px; padding: 20px; background-color: #f8f9fa; border-radius: 10px;">
             <h3>🛠️ 處理功能說明</h3>
             <ul>
                 <li><strong>樣式調整：</strong>將所有段落設定為無間距樣式</li>
@@ -144,12 +128,9 @@ def create_interface():
                 <li><strong>標題格式：</strong>章節標題設定為粗體標題 1 樣式（無縮排）</li>
                 <li><strong>分頁設定：</strong>每個章節前自動分頁</li>
             </ul>
-        </div>
-        """)
-        
-        # 使用範例
-        gr.HTML("""
-        <div style="margin-top: 20px; padding: 20px; background-color: #e8f4f8; border-radius: 10px;">
+        </div>""")
+
+        gr.HTML("""<div style="margin-top: 20px; padding: 20px; background-color: #e8f4f8; border-radius: 10px;">
             <h3>💡 使用範例</h3>
             <p><strong>章節關鍵字設定：</strong></p>
             <ul>
@@ -167,23 +148,16 @@ def create_interface():
                 <li>第I章、第II章、第III章...</li>
                 <li>第二十三章、第三十四節...</li>
             </ul>
-        </div>
-        """)
-        
-        # 綁定事件
+        </div>""")
+
         process_btn.click(
             fn=format_docx,
             inputs=[file_input, chapter_input],
             outputs=[download_output, status_output]
         )
-    
+
     return demo
 
-# 建立並啟動介面
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)