dseditor commited on
Commit
af86850
·
verified ·
1 Parent(s): b565e7e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -59
app.py CHANGED
@@ -9,111 +9,84 @@ from docx.oxml import OxmlElement
9
  import tempfile
10
  import os
11
 
 
12
  def format_docx(file, chapter_keywords):
13
  """
14
  處理上傳的 Word 文件
15
  """
16
  if file is None:
17
  return None, "請上傳一個 Word 文件"
18
-
19
  if not chapter_keywords.strip():
20
  return None, "請輸入章節分段方式(例如:章,節,話)"
21
-
22
  try:
23
- # 讀取上傳的文件
 
 
 
 
24
  doc = Document(file.name)
25
-
26
  # 解析章節關鍵字
27
  keywords = [keyword.strip() for keyword in chapter_keywords.split(',')]
28
-
29
  # 建立正規表示式模式
30
  patterns = []
31
  for keyword in keywords:
32
- # 匹配多種數字格式:阿拉伯數字、中文數字、羅馬數字等
33
- # 例如:第1章、第一章、第二十三章、第I章等
34
- pattern = rf'第\s*[0-9一二三四五六七八九十百千萬壹貳參肆伍陸柒捌玖拾佰仟萬IVXLCDMivxlcdm]+\s*{keyword}'
35
  patterns.append(pattern)
36
-
37
- # 合併所有模式
38
  combined_pattern = '|'.join(patterns)
39
-
40
- # 處理文件:先收集所有段落內容和類型
41
  content_list = []
42
-
43
  for paragraph in doc.paragraphs:
44
  text = paragraph.text.strip()
45
-
46
- # 檢查是否為章節標題
47
  if text and re.search(combined_pattern, text):
48
  content_list.append(('heading', text))
49
- elif text: # 有內容的一般段落
50
  content_list.append(('paragraph', text))
51
- else: # 空段落
52
- # 只有在前一個不是空段落時才添加空段落
53
  if not content_list or content_list[-1][0] != 'empty':
54
  content_list.append(('empty', ''))
55
-
56
- # 清空整個文件
57
  for paragraph in doc.paragraphs:
58
  p = paragraph._element
59
  p.getparent().remove(p)
60
-
61
- # 重新建立文件內容
62
  for content_type, text in content_list:
63
  if content_type == 'heading':
64
- # 直接添加段落,然後手動設定為標題格式
65
- heading = doc.add_paragraph(text)
66
-
67
- # 手動設定標題格式
68
- for run in heading.runs:
69
- run.font.bold = True
70
- run.font.size = Pt(16)
71
- run.font.name = '新細明體' # 設定中文字體
72
-
73
- # 設定段落格式
74
  heading.paragraph_format.space_before = Cm(0)
75
- heading.paragraph_format.space_after = Cm(0.3) # 標題後稍微間距
76
  heading.paragraph_format.line_spacing = 1.0
77
-
78
- # 在標題前分頁
79
- heading.paragraph_format.page_break_before = True
80
-
81
- # 重置標題的縮排
82
  heading.paragraph_format.left_indent = Cm(0)
83
  heading.paragraph_format.first_line_indent = Cm(0)
84
-
85
- # 嘗試設定大綱層級(這樣可以在導覽窗格中顯示)
86
- try:
87
- # 直接設定大綱層級,不依賴樣式
88
- heading.paragraph_format.outline_level = 0 # 0 = 層級1
89
- except:
90
- pass # 如果失敗就跳過
91
-
92
  elif content_type == 'paragraph':
93
- # 添加一般段落
94
- para = doc.add_paragraph(text)
95
- # 設定樣式
96
- para.style.paragraph_format.space_before = Cm(0)
97
- para.style.paragraph_format.space_after = Cm(0)
98
- para.style.paragraph_format.line_spacing = 1.0
99
- # 首行縮排
100
  para.paragraph_format.left_indent = Cm(0)
101
  para.paragraph_format.first_line_indent = Cm(0.7)
102
-
 
103
  elif content_type == 'empty':
104
- # 添加空段落
105
  doc.add_paragraph('')
106
-
107
- # 儲存處理後的文件
108
  output_path = tempfile.mktemp(suffix='.docx')
109
  doc.save(output_path)
110
-
111
  return output_path, f"✅ 處理完成!找到章節關鍵字:{', '.join(keywords)}"
112
-
113
  except Exception as e:
114
  return None, f"❌ 處理失敗:{str(e)}"
115
 
116
  def create_interface():
 
117
  """
118
  建立 Gradio 介面
119
  """
 
9
  import tempfile
10
  import os
11
 
12
+
13
  def format_docx(file, chapter_keywords):
14
  """
15
  處理上傳的 Word 文件
16
  """
17
  if file is None:
18
  return None, "請上傳一個 Word 文件"
19
+
20
  if not chapter_keywords.strip():
21
  return None, "請輸入章節分段方式(例如:章,節,話)"
22
+
23
  try:
24
+ from docx import Document
25
+ from docx.shared import Cm, Pt
26
+ import re
27
+ import tempfile
28
+
29
  doc = Document(file.name)
30
+
31
  # 解析章節關鍵字
32
  keywords = [keyword.strip() for keyword in chapter_keywords.split(',')]
33
+
34
  # 建立正規表示式模式
35
  patterns = []
36
  for keyword in keywords:
37
+ pattern = f'第\s*[0-9一二三四五六七八九十百千萬壹貳參肆伍陸柒捌玖拾佰仟萬IVXLCDMivxlcdm]+\s*{keyword}'
 
 
38
  patterns.append(pattern)
 
 
39
  combined_pattern = '|'.join(patterns)
40
+
 
41
  content_list = []
 
42
  for paragraph in doc.paragraphs:
43
  text = paragraph.text.strip()
 
 
44
  if text and re.search(combined_pattern, text):
45
  content_list.append(('heading', text))
46
+ elif text:
47
  content_list.append(('paragraph', text))
48
+ else:
 
49
  if not content_list or content_list[-1][0] != 'empty':
50
  content_list.append(('empty', ''))
51
+
 
52
  for paragraph in doc.paragraphs:
53
  p = paragraph._element
54
  p.getparent().remove(p)
55
+
 
56
  for content_type, text in content_list:
57
  if content_type == 'heading':
58
+ heading = doc.add_paragraph(text, style='Heading 1')
59
+ heading.paragraph_format.page_break_before = True
 
 
 
 
 
 
 
 
60
  heading.paragraph_format.space_before = Cm(0)
61
+ heading.paragraph_format.space_after = Cm(0.3)
62
  heading.paragraph_format.line_spacing = 1.0
 
 
 
 
 
63
  heading.paragraph_format.left_indent = Cm(0)
64
  heading.paragraph_format.first_line_indent = Cm(0)
65
+ for run in heading.runs:
66
+ run.font.name = '新細明體'
67
+ run.font.size = Pt(16)
 
 
 
 
 
68
  elif content_type == 'paragraph':
69
+ lines = [line.strip() for line in text.splitlines() if line.strip()]
70
+ clean_text = ' '.join(lines)
71
+ para = doc.add_paragraph(clean_text)
72
+ para.paragraph_format.space_before = Cm(0)
73
+ para.paragraph_format.space_after = Cm(0)
74
+ para.paragraph_format.line_spacing = 1.0
 
75
  para.paragraph_format.left_indent = Cm(0)
76
  para.paragraph_format.first_line_indent = Cm(0.7)
77
+ for run in para.runs:
78
+ run.font.name = '新細明體'
79
  elif content_type == 'empty':
 
80
  doc.add_paragraph('')
81
+
 
82
  output_path = tempfile.mktemp(suffix='.docx')
83
  doc.save(output_path)
 
84
  return output_path, f"✅ 處理完成!找到章節關鍵字:{', '.join(keywords)}"
 
85
  except Exception as e:
86
  return None, f"❌ 處理失敗:{str(e)}"
87
 
88
  def create_interface():
89
+
90
  """
91
  建立 Gradio 介面
92
  """