Update app.py

app.py

@@ -235,6 +235,107 @@ def replace_in_hwpx(hwpx_bytes: bytes, mapping: Dict[str,str]) -> Tuple[bytes, d
     zin.close()
     return out_buf.getvalue(), dbg
 
+def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
+    """두 개의 HWPX 파일을 병합하여 하나의 HWPX로 만듦"""
+    import time
+    
+    base_zip = zipfile.ZipFile(io.BytesIO(base_hwpx), "r")
+    add_zip = zipfile.ZipFile(io.BytesIO(additional_hwpx), "r")
+    
+    out_buf = io.BytesIO()
+    out_zip = zipfile.ZipFile(out_buf, "w", compression=zipfile.ZIP_DEFLATED, compresslevel=6)
+    
+    now = time.localtime()
+    
+    # mimetype 먼저 처리
+    if "mimetype" in base_zip.namelist():
+        zi = zipfile.ZipInfo("mimetype")
+        zi.compress_type = zipfile.ZIP_STORED
+        zi.external_attr = 0o100666 << 16
+        zi.create_system = 0
+        zi.date_time = now[:6]
+        zi.flag_bits = 0
+        out_zip.writestr(zi, base_zip.read("mimetype"))
+    
+    # 베이스 HWPX에서 섹션 파일들 읽기
+    base_sections = {}
+    add_sections = {}
+    
+    for filename in base_zip.namelist():
+        if filename == "mimetype":
+            continue
+        if filename.startswith("Contents/section") and filename.endswith(".xml"):
+            base_sections[filename] = base_zip.read(filename).decode("utf-8", errors="ignore")
+        else:
+            # 다른 파일들은 그대로 복사
+            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, base_zip.read(filename))
+    
+    # 추가할 HWPX에서 섹션 파일들 읽기
+    for filename in add_zip.namelist():
+        if filename.startswith("Contents/section") and filename.endswith(".xml"):
+            add_sections[filename] = add_zip.read(filename).decode("utf-8", errors="ignore")
+    
+    # 섹션 병합 처리
+    merged_sections = merge_sections(base_sections, add_sections)
+    
+    # 병합된 섹션들을 ZIP에 쓰기
+    for filename, content in merged_sections.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, content.encode("utf-8"))
+    
+    base_zip.close()
+    add_zip.close()
+    out_zip.close()
+    out_buf.seek(0)
+    
+    return out_buf.getvalue()
+
+def merge_sections(base_sections: dict, add_sections: dict) -> dict:
+    """섹션 XML들을 병합"""
+    merged = base_sections.copy()
+    
+    for filename, add_content in add_sections.items():
+        if filename in merged:
+            # 기존 섹션에 페이지 추가
+            merged[filename] = merge_section_content(merged[filename], add_content)
+        else:
+            # 새로운 섹션 추가
+            merged[filename] = add_content
+    
+    return merged
+
+def merge_section_content(base_xml: str, add_xml: str) -> str:
+    """단일 섹션 XML 내용을 병합"""
+    # 추가할 XML에서 페이지들 추출
+    page_pattern = re.compile(r'(<(?P<prefix>[a-zA-Z0-9_]+):page[^>]*>.*?</(?P=prefix):page>)', re.DOTALL)
+    add_pages = page_pattern.findall(add_xml)
+    
+    if not add_pages:
+        return base_xml
+    
+    # 베이스 XML의 마지막 페이지 뒤에 새 페이지들 삽입
+    # </hp:pages> 태그 앞에 삽입
+    pages_end_pattern = re.compile(r'(</[a-zA-Z0-9_]+:pages>)')
+    
+    # 추가할 페이지들을 문자열로 변환
+    pages_to_add = ''.join([page[0] for page in add_pages])
+    
+    # 삽입
+    merged_xml = pages_end_pattern.sub(pages_to_add + r'\1', base_xml)
+    
+    return merged_xml
+
 # -------------------- UI --------------------
 with st.expander("사용법", expanded=True):
     st.markdown("""
@@ -290,11 +391,13 @@ if tpl and data:
     st.dataframe(pd.DataFrame([{"키":k, "값 앞부분":str(v)[:120]} for k,v in sorted(mapping_preview.items())]),
                  use_container_width=True, height=320)
 
-    if st.button("🚀 라벨 생성 (페이지별 HWPX ZIP)"):
-        mem = io.BytesIO(); zout = zipfile.ZipFile(mem, "w", zipfile.ZIP_DEFLATED)
+    if st.button("🚀 통합 라벨 생성 (단일 HWPX 파일)"):
         pages = (len(records) + int(n_per_page) - 1) // int(n_per_page)
         debug_all = []
-
+        
+        # 첫 페이지로 시작
+        merged_hwpx = None
+        
         for p in range(pages):
             chunk = records[p*int(n_per_page):(p+1)*int(n_per_page)]
             mapping = {}
@@ -313,12 +416,21 @@ if tpl and data:
                 else:
                     for k in keys: mapping[f"{k}{i+1}"] = ""
 
-            out_hwpx, dbg = replace_in_hwpx(tpl_bytes, mapping)
+            if p == 0:
+                # 첫 페이지: 템플릿 기반으로 생성
+                merged_hwpx, dbg = replace_in_hwpx(tpl_bytes, mapping)
+            else:
+                # 두 번째 페이지부터: 기존 HWPX에 페이지 추가
+                page_hwpx, dbg = replace_in_hwpx(tpl_bytes, mapping)
+                merged_hwpx = merge_hwpx_pages(merged_hwpx, page_hwpx)
+            
             debug_all.append({"page": p+1, "stats": dbg})
-            name = "_".join([r.get("박스번호","") for r in chunk]) if chunk else f"empty_{p+1}"
-            zout.writestr(f"label_{name}.hwpx", out_hwpx)
 
-        zout.close(); mem.seek(0)
-        st.download_button("⬇️ ZIP 다운로드", data=mem, file_name="labels_by_page.zip", mime="application/zip")
+        # 박스번호 범위로 파일명 생성
+        first_box = records[0].get("박스번호", "0000") if records else "0000"
+        last_box = records[-1].get("박스번호", "0000") if records else "0000"
+        filename = f"labels_{first_box}to{last_box}.hwpx" if first_box != last_box else f"labels_{first_box}.hwpx"
+        
+        st.download_button("⬇️ 통합 HWPX 다운로드", data=merged_hwpx, file_name=filename, mime="application/zip")
         st.download_button("⬇️ 디버그(JSON)", data=json.dumps(debug_all, ensure_ascii=False, indent=2),
                            file_name="debug.json", mime="application/json")