Update app.py

app.py

@@ -236,7 +236,7 @@ def replace_in_hwpx(hwpx_bytes: bytes, mapping: Dict[str,str]) -> Tuple[bytes, d
     return out_buf.getvalue(), dbg
 
 def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
-    """두 개의 HWPX 파일을 병합하여 하나의 HWPX로 만듦"""
+    """HWPX에 새 페이지 추가 (템플릿 복제 방식)"""
     import time
     
     base_zip = zipfile.ZipFile(io.BytesIO(base_hwpx), "r")
@@ -257,15 +257,15 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
         zi.flag_bits = 0
         out_zip.writestr(zi, base_zip.read("mimetype"))
     
-    # 모든 파일을 처리 - 섹션 파일들은 별도 병합
+    # 섹션 파일들을 따로 처리
     section_files = {}
     
-    # 베이스에서 모든 파일 수집
+    # 베이스에서 모든 파일 처리
     for filename in base_zip.namelist():
         if filename == "mimetype":
             continue
         if filename.startswith("Contents/section") and filename.endswith(".xml"):
-            section_files[filename] = [base_zip.read(filename).decode("utf-8", errors="ignore")]
+            section_files[filename] = base_zip.read(filename).decode("utf-8", errors="ignore")
         else:
             zi = zipfile.ZipInfo(filename)
             zi.compress_type = zipfile.ZIP_DEFLATED
@@ -275,29 +275,22 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
             zi.flag_bits = 0
             out_zip.writestr(zi, base_zip.read(filename))
     
-    # 추가 파일에서 섹션만 수집
+    # 추가 파일에서 페이지 추출해서 기존 섹션에 추가
     for filename in add_zip.namelist():
         if filename.startswith("Contents/section") and filename.endswith(".xml"):
-            content = add_zip.read(filename).decode("utf-8", errors="ignore")
+            add_content = add_zip.read(filename).decode("utf-8", errors="ignore")
             if filename in section_files:
-                section_files[filename].append(content)
-            else:
-                section_files[filename] = [content]
+                section_files[filename] = add_page_to_section(section_files[filename], add_content)
     
-    # 섹션 파일들 병합해서 쓰기
-    for filename, contents in section_files.items():
-        if len(contents) == 1:
-            merged_content = contents[0]
-        else:
-            merged_content = merge_section_xml_list(contents)
-        
+    # 수정된 섹션 파일들 저장
+    for filename, content in section_files.items():
         zi = zipfile.ZipInfo(filename)
         zi.compress_type = zipfile.ZIP_DEFLATED
         zi.external_attr = 0o100666 << 16
         zi.create_system = 0
         zi.date_time = now[:6]
         zi.flag_bits = 0
-        out_zip.writestr(zi, merged_content.encode("utf-8"))
+        out_zip.writestr(zi, content.encode("utf-8"))
     
     base_zip.close()
     add_zip.close()
@@ -306,33 +299,67 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
     
     return out_buf.getvalue()
 
+def add_page_to_section(base_xml: str, add_xml: str) -> str:
+    """섹션에 새 페이지 추가"""
+    # 추가할 XML에서 첫 번째 페이지 추출
+    page_pattern = re.compile(
+        r'<(?P<prefix>[a-zA-Z0-9_]+):page\b[^>]*>.*?</(?P=prefix):page>', 
+        re.DOTALL
+    )
+    
+    page_match = page_pattern.search(add_xml)
+    if not page_match:
+        return base_xml
+    
+    new_page = page_match.group(0)
+    
+    # 베이스 XML에서 </prefix:pages> 태그 찾아서 그 앞에 새 페이지 삽입
+    pages_close_pattern = re.compile(r'(</[a-zA-Z0-9_]+:pages>)')
+    
+    # 페이지 ID 업데이트 (중복 방지)
+    new_page = update_page_id(base_xml, new_page)
+    
+    # 페이지 삽입
+    result = pages_close_pattern.sub(new_page + r'\1', base_xml)
+    
+    return result
+
+def update_page_id(base_xml: str, new_page: str) -> str:
+    """페이지 ID를 중복되지 않게 업데이트"""
+    # 기존 페이지들의 ID 추출
+    existing_ids = set()
+    id_pattern = re.compile(r'<[a-zA-Z0-9_]+:page\b[^>]*\bid="([^"]*)"')
+    
+    for match in id_pattern.finditer(base_xml):
+        existing_ids.add(match.group(1))
+    
+    # 새 페이지의 ID 찾기
+    new_page_id_match = id_pattern.search(new_page)
+    if not new_page_id_match:
+        return new_page
+    
+    original_id = new_page_id_match.group(1)
+    
+    # 중복되지 않는 새 ID 생성
+    counter = 2
+    new_id = f"{original_id}_{counter}"
+    while new_id in existing_ids:
+        counter += 1
+        new_id = f"{original_id}_{counter}"
+    
+    # ID 교체
+    updated_page = new_page.replace(f'id="{original_id}"', f'id="{new_id}"')
+    
+    return updated_page
+
 def merge_section_xml_list(xml_list: list) -> str:
-    """여러 섹션 XML을 하나로 병합"""
+    """여러 섹션 XML을 하나로 병합 (사용하지 않지만 호환성 유지)"""
     if len(xml_list) <= 1:
         return xml_list[0] if xml_list else ""
     
     base_xml = xml_list[0]
-    
     for additional_xml in xml_list[1:]:
-        # 각 XML에서 페이지 추출
-        page_pattern = re.compile(
-            r'<(?P<prefix>[a-zA-Z0-9_]+):page\b[^>]*>.*?</(?P=prefix):page>', 
-            re.DOTALL
-        )
-        
-        add_pages = page_pattern.findall(additional_xml)
-        if not add_pages:
-            continue
-        
-        # 페이지 전체 매치
-        page_matches = page_pattern.finditer(additional_xml)
-        pages_to_add = [match.group(0) for match in page_matches]
-        
-        if pages_to_add:
-            # 베이스 XML의 </prefix:pages> 태그 찾아서 그 앞에 삽입
-            pages_close_pattern = re.compile(r'(</[a-zA-Z0-9_]+:pages>)')
-            pages_str = ''.join(pages_to_add)
-            base_xml = pages_close_pattern.sub(pages_str + r'\1', base_xml)
+        base_xml = add_page_to_section(base_xml, additional_xml)
     
     return base_xml
 
@@ -377,6 +404,8 @@ def merge_section_content(base_xml: str, add_xml: str) -> str:
 # -------------------- UI --------------------
 with st.expander("사용법", expanded=True):
     st.markdown("""
+- **템플릿은 1페이지에 N개 라벨**이 있는 표 형태를 사용합니다.
+- **박스 수가 N개를 초과하면 새 페이지가 자동 추가**됩니다.
 - **다중 줄(목록/제목)은 부모 문단을 여러 문단으로 교체**하여 겹침 없이 표시합니다.
 - **업무명은 단일줄로 처리**하여 원본 폰트 스타일을 유지합니다.
 - **생성된 HWPX 파일의 읽기전용 속성이 해제**되어 편집 가능합니다.