Update app.py

app.py

@@ -1,240 +1,212 @@
 import streamlit as st
 import pandas as pd
-from reportlab.pdfgen import canvas
-from reportlab.pdfbase import pdfmetrics
-from reportlab.pdfbase.ttfonts import TTFont
-from reportlab.lib.pagesizes import A4
-from reportlab.lib.units import mm
-from io import BytesIO
-import math
-
-st.set_page_config(page_title="📦 박스라벨 PDF 출력기", layout="wide")
-st.title("📦 박스라벨 PDF 출력기 (라벨 규격 커스텀 / 한국어 폰트 업로드)")
-
-with st.expander("사용 방법", expanded=True):
-    st.markdown("""
-1. **엑셀/CSV 업로드** → 필수 컬럼: `박스번호` / 권장: `종료연도`, `보존기간`, `단위업무`, `기록물철`, `제목`, `관리번호`  
-2. (선택) **TTF 폰트 업로드**(예: 나눔고딕, 본고딕, 맑은 고딕 등). 업로드 안 하면 기본 폰트 사용(영문 위주).
-3. **라벨 규격**(페이지 여백, 라벨 가로/세로, 행/열, 라벨 간격)을 입력.
-4. **텍스트 배치**(라벨 안쪽 패딩, 폰트 크기, 줄 간격 등) 조정.
-5. **PDF 생성** → 라벨 용지(Formtec 등)에 인쇄.
-    """)
-
-# -----------------
-# 데이터 로드
-# -----------------
-file = st.file_uploader("📊 데이터 업로드 (Excel/CSV)", type=["xlsx","xls","csv"])
-df = None
-if file:
-    if file.name.lower().endswith(".csv"):
-        df = pd.read_csv(file)
-    else:
-        df = pd.read_excel(file)
-
-# 필수 컬럼 검사
-if df is not None and "박스번호" not in df.columns:
-    st.error("❌ 필수 컬럼 '박스번호'가 없습니다.")
-    st.stop()
-
-# -----------------
-# 폰트 설정
-# -----------------
-st.subheader("🔤 폰트 설정")
-font_file = st.file_uploader("한국어 폰트(TTF) 업로드 (예: NanumGothic.ttf / MalgunGothic.ttf)", type=["ttf"])
-font_name = "BaseFont"
-if font_file:
-    try:
-        font_bytes = font_file.read()
-        # 메모리 등록: ReportLab은 파일 경로가 필요 → 임시 파일 만들기보다 메모리 레지스터 트릭
-        # 하지만 TTFont는 파일 경로 요구 → 임시파일 저장
-        import tempfile
-        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".ttf")
-        tmp.write(font_bytes); tmp.flush()
-        pdfmetrics.registerFont(TTFont("UserKorean", tmp.name))
-        font_name = "UserKorean"
-        st.success("✅ 폰트 등록 완료: UserKorean")
-    except Exception as e:
-        st.warning(f"폰트 등록 실패. 기본 폰트 사용합니다. (사유: {e})")
-else:
-    # 내장 기본 폰트 (영문 중심)
-    font_name = "Helvetica"
-
-# -----------------
-# 라벨/페이지 레이아웃
-# -----------------
-st.subheader("📐 라벨 규격 (mm 단위)")
-colA, colB, colC = st.columns(3)
-with colA:
-    page_size = st.selectbox("페이지 크기", ["A4"], index=0)
-with colB:
-    margin_left = st.number_input("왼쪽 여백(mm)", 5.0, 50.0, 10.0, 0.5)
-    margin_top = st.number_input("상단 여백(mm)", 5.0, 50.0, 10.0, 0.5)
-with colC:
-    rows = st.number_input("행 수", 1, 20, 10, 1)
-    cols = st.number_input("열 수", 1, 10, 3, 1)
-
-colD, colE, colF = st.columns(3)
-with colD:
-    label_w = st.number_input("라벨 가로(mm)", 20.0, 210.0, 70.0, 0.5)
-with colE:
-    label_h = st.number_input("라벨 세로(mm)", 10.0, 297.0, 25.0, 0.5)
-with colF:
-    gap_x = st.number_input("가로 간격(mm)", 0.0, 20.0, 3.0, 0.5)
-    gap_y = st.number_input("세로 간격(mm)", 0.0, 20.0, 3.0, 0.5)
-
-# -----------------
-# 라벨 내부 텍스트 배치
-# -----------------
-st.subheader("🧱 라벨 내부 레이아웃")
-col1, col2, col3 = st.columns(3)
-with col1:
-    pad_x = st.number_input("내부 패딩 X(mm)", 0.0, 20.0, 2.0, 0.5)
-    pad_y = st.number_input("내부 패딩 Y(mm)", 0.0, 20.0, 2.0, 0.5)
-with col2:
-    fs_big = st.number_input("폰트 크기(큰 제목)", 6, 40, 16, 1)
-    fs_mid = st.number_input("폰트 크기(중간)", 6, 40, 11, 1)
-with col3:
-    fs_small = st.number_input("폰트 크기(작게/목록)", 6, 20, 9, 1)
-    line_gap = st.number_input("줄 간격(배수)", 0.8, 2.0, 1.2, 0.1)
-
-st.caption("💡 Formtec 3203 비슷한 설정 예시: 가로 70, 세로 25, 열 3, 행 10, 여백 10/10, 간격 3/3 (프린터마다 약간 조정)")
-
-# -----------------
-# 텍스트 생성 함수
-# -----------------
-def year_range(series):
+import io, zipfile, re, html, json
+from typing import Dict, Tuple, Optional
+
+st.set_page_config(page_title="📦 박스라벨(HWPX) — 필드 완전 치환", layout="wide")
+st.title("📦 박스라벨 자동 생성기 — HWPX 필드 완전 치환(모든 XML / 접두어 와일드카드)")
+
+# ================== 데이터 유틸 ==================
+def _year_range(series: pd.Series) -> str:
     s = series.astype(str).fillna("")
     v = s[~s.isin(["", "0", "0000"])]
-    if len(v) == 0:
-        return "0000-0000"
+    if v.empty: return "0000-0000"
     nums = pd.to_numeric(v, errors="coerce").dropna().astype(int)
-    if len(nums) == 0:
-        return "0000-0000"
+    if nums.empty: return "0000-0000"
     return f"{nums.min():04d}-{nums.max():04d}"
 
-def build_records(df: pd.DataFrame):
+def build_rows(df: pd.DataFrame) -> pd.DataFrame:
     df = df.copy()
     df["박스번호"] = df["박스번호"].astype(str).str.zfill(4)
-    # 생산연도(범위)
+    if "제목" in df.columns:
+        df["제목"] = df["제목"].astype(str)
+
+    # 생산연도(범위) = 종료연도 그룹 범위
     if "종료연도" in df.columns:
-        yr = df.groupby("박스번호")["종료연도"].apply(year_range).reset_index()
+        yr = df.groupby("박스번호")["종료연도"].apply(_year_range).reset_index()
         yr.columns = ["박스번호", "생산연도"]
     else:
         yr = pd.DataFrame({"박스번호": df["박스번호"].unique(), "생산연도": "0000-0000"})
-    # 목록
+
+    # 목록(관리번호 + 제목)
     has_mgmt = "관리번호" in df.columns
-    list_rows = []
-    for box, g in df.groupby("박스번호"):
+    lists = []
+    for b, g in df.groupby("박스번호"):
         lines = [f"- {r['관리번호']} {r.get('제목','')}" if has_mgmt else f"- {r.get('제목','')}"
                  for _, r in g.iterrows()]
-        list_rows.append({"박스번호": box, "목록": "\n".join(lines)})
-    list_df = pd.DataFrame(list_rows)
+        lists.append({"박스번호": b, "목록": "\r\n".join(lines)})
+    list_df = pd.DataFrame(lists)
+
     # 대표 메타
-    cols = ["박스번호","보존기간","단위업무","기록물철","제목"]
-    meta_exist = [c for c in cols if c in df.columns]
-    meta = df.groupby("박스번호", as_index=False).first()[meta_exist] if meta_exist else pd.DataFrame({"박스번호": df["박스번호"].unique()})
+    meta_cols = ["박스번호","종료연도","보존기간","단위업무","기록물철","제목"]
+    meta_exist = [c for c in meta_cols if c in df.columns]
+    meta = df.groupby("박스번호", as_index=False).first()[meta_exist] if meta_exist \
+          else pd.DataFrame({"박스번호": df["박스번호"].unique()})
+
     merged = meta.merge(list_df, on="박스번호", how="left").merge(yr, on="박스번호", how="left")
-    return merged.sort_values("박스번호").to_dict(orient="records")
-
-def draw_label(c: canvas.Canvas, x, y, w, h, rec, font_name, fs_big, fs_mid, fs_small, line_gap):
-    """
-    좌표계: reportlab은 좌하단이 원점. 
-    x,y = 라벨 좌하단. w,h = 라벨 크기.
-    """
-    # 여백
-    inner_x = x + pad_x * mm
-    inner_y = y + pad_y * mm
-    inner_w = w - 2 * pad_x * mm
-    inner_h = h - 2 * pad_y * mm
-
-    # 상단 굵은 줄: 박스번호
-    c.setFont(font_name, fs_big)
-    boxno = rec.get("박스번호", "")
-    c.drawString(inner_x, inner_y + inner_h - fs_big*1.1, f"{boxno}")
-
-    # 2행: (생산연도/보존기간)
-    c.setFont(font_name, fs_mid)
-    prod = rec.get("생산연도","")
-    keep = rec.get("보존기간","") or ""
-    line_y = inner_y + inner_h - fs_big*1.1 - fs_mid*1.5
-    c.drawString(inner_x, line_y, f"{prod}   {keep}")
-
-    # 3행: 단위업무 / 기록물철 (있으면)
-    line_y -= fs_mid * 1.2
-    unit = rec.get("단위업무","") or ""
-    series = rec.get("기록물철","") or ""
-    if unit or series:
-        c.setFont(font_name, fs_mid)
-        c.drawString(inner_x, line_y, f"{unit}  {series}")
-        line_y -= fs_mid * 1.0
-
-    # 목록(여러 줄, 작은 글씨)
-    c.setFont(font_name, fs_small)
-    list_text = rec.get("목록","") or ""
-    for ln in list_text.split("\n"):
-        if line_y < inner_y + fs_small * 1.2:  # 라벨 하단 넘어가면 중단
-            break
-        c.drawString(inner_x, line_y, ln)
-        line_y -= fs_small * line_gap
-
-def make_pdf(records):
-    buffer = BytesIO()
-    if page_size == "A4":
-        pw, ph = A4
-    else:
-        pw, ph = A4
-
-    c = canvas.Canvas(buffer, pagesize=(pw, ph))
-    c.setAuthor("BoxLabel")
-    c.setTitle("Box Labels")
-
-    pdfmetrics.getFont(font_name)  # ensure registered
-
-    # 좌표/크기(mm → pt)
-    L = margin_left * mm
-    T = margin_top * mm
-    W = label_w * mm
-    H = label_h * mm
-    GX = gap_x * mm
-    GY = gap_y * mm
-
-    per_page = int(rows * cols)
-    total_pages = math.ceil(len(records) / per_page) if records else 1
-
-    idx = 0
-    for p in range(total_pages):
-        for r in range(int(rows)):
-            for ccol in range(int(cols)):
-                if idx >= len(records):
-                    break
-                # 좌표 계산 (좌하단 원점이므로 상단에서 내려오게 Y를 조정)
-                x = L + ccol * (W + GX)
-                y_top = ph - T - r * (H + GY)
-                y = y_top - H
-                draw_label(c, x, y, W, H, records[idx], font_name, fs_big, fs_mid, fs_small, line_gap)
-                idx += 1
-            if idx >= len(records):
-                break
-        c.showPage()
-    c.save()
-    buffer.seek(0)
-    return buffer
-
-# -----------------
-# 메인 동작
-# -----------------
-if df is not None:
-    # 미리보기
-    st.subheader("📋 데이터 미리보기")
-    st.dataframe(df.head(10), use_container_width=True)
+    return merged
+
+# ================== 치환 유틸 ==================
+# 접두어 와일드카드: <hp:..> 뿐 아니라 <hwp:..>, <h:..> 등 모두 허용
+FIELD_PAIR_RE_TMPL = (
+    r'<(?P<prefix>[a-zA-Z0-9_]+):fieldBegin\b[^>]*\bname="{name}"[^>]*/>'
+    r'(.*?)'
+    r'<(?P=prefix):fieldEnd\b[^>]*/>'
+)
+
+# 토큰(백업 경로)
+TOKEN_FMT = "{{{{{key}}}}}"
+
+def _run_for_plain(text: str) -> str:
+    return f"<hp:run><hp:t>{html.escape('' if text is None else str(text))}</hp:t></hp:run>"
+
+def _run_for_list(text: str) -> str:
+    if text is None: return ""
+    lines = str(text).replace("\r\n","\n").split("\n")
+    parts = []
+    for i, ln in enumerate(lines):
+        if i>0: parts.append("<hp:lineBreak/>")
+        parts.append(f"<hp:run><hp:t>{html.escape(ln)}</hp:t></hp:run>")
+    return "".join(parts)
+
+def _apply_to_xml(xml: str, mapping: Dict[str, str], dbg: dict) -> str:
+    changed_any = False
+
+    # 1) 필드쌍 완전 치환 (모든 접두어, 모든 XML 대상)
+    for k, v in mapping.items():
+        is_list = bool(re.match(r"^(목록|list)\d+$", k, re.IGNORECASE))
+        replacement = _run_for_list(v) if is_list else _run_for_plain(v)
+
+        pat = re.compile(FIELD_PAIR_RE_TMPL.format(name=re.escape(k)), re.DOTALL)
+        xml_new, n = pat.subn(replacement, xml)
+        if n:
+            dbg["field_hits"][k] = dbg["field_hits"].get(k, 0) + n
+            xml = xml_new
+            changed_any = True
+
+    # 2) 백업 경로: 토큰 치환 ({{키}}가 어딘가 남아있다면)
+    for k, v in mapping.items():
+        tok = TOKEN_FMT.format(key=k)
+        if tok in xml:
+            rep = _run_for_list(v) if re.match(r"^(목록|list)\d+$", k, re.IGNORECASE) else html.escape("" if v is None else str(v))
+            xml = xml.replace(tok, rep)
+            dbg["token_hits"][k] = dbg["token_hits"].get(k, 0) + 1
+            changed_any = True
+
+    if changed_any:
+        dbg["files_touched"] = True
+    return xml
+
+def replace_in_hwpx(hwpx_bytes: bytes, mapping: Dict[str,str]) -> Tuple[bytes, dict]:
+    dbg = {"field_hits":{}, "token_hits":{}, "touched_files": []}
+    zin = zipfile.ZipFile(io.BytesIO(hwpx_bytes), "r")
+    out_buf = io.BytesIO()
+    zout = zipfile.ZipFile(out_buf, "w")
+
+    # mimetype 무압축 + 맨앞
+    names = zin.namelist()
+    if "mimetype" in names:
+        zi = zipfile.ZipInfo("mimetype"); zi.compress_type = zipfile.ZIP_STORED
+        zout.writestr(zi, zin.read("mimetype"))
+
+    for e in zin.infolist():
+        if e.filename == "mimetype":
+            continue
+        data = zin.read(e.filename)
+        if e.filename.lower().endswith(".xml"):
+            try:
+                s = data.decode("utf-8", errors="ignore")
+                before = s
+                s = _apply_to_xml(s, mapping, {"field_hits": dbg["field_hits"], "token_hits": dbg["token_hits"], "files_touched": False})
+                if s != before:
+                    dbg["touched_files"].append(e.filename)
+                data = s.encode("utf-8")
+            except Exception:
+                pass
+        zi = zipfile.ZipInfo(e.filename); zi.compress_type = zipfile.ZIP_DEFLATED
+        zout.writestr(zi, data)
+
+    zout.close(); out_buf.seek(0); zin.close()
+    return out_buf.getvalue(), dbg
+
+# ================== UI ==================
+with st.expander("사용법", expanded=True):
+    st.markdown("""
+- 템플릿은 **HWPX(한글)**이며, 필드컨트롤 `name="박스번호1"` 등의 이름을 그대로 사용합니다.  
+- 이 앱은 ZIP 내부의 **모든 XML**을 훑으며, 접두어가 무엇이든(`<hp:...>`, `<hwp:...>` 등)  
+  **`fieldBegin(name=키)` ~ `fieldEnd` 사이 전체 블록을 값 run들로 교체**합니다.  
+- 토큰(`{{박스번호1}}`)이 남아 있으면 **백업 경로**로 그 자리도 치환합니다.
+""")
+
+tpl = st.file_uploader("📄 HWPX 템플릿 업로드", type=["hwpx"])
+n_per_page = st.number_input("템플릿의 라벨 세트 개수(한 페이지 N개)", 1, 12, 3, 1)
+data = st.file_uploader("📊 데이터 업로드 (Excel/CSV)", type=["xlsx","xls","csv"])
 
-    records = build_records(df)
-    st.write(f"총 **{len(records)}**개 박스가 감지되었습니다.")
-    default_sel = [r["박스번호"] for r in records]
-    sel = st.multiselect("생성할 박스번호 선택 (비우면 전체)", options=default_sel)
-    if sel:
-        records = [r for r in records if r["박스번호"] in set(sel)]
+if tpl and data:
+    tpl_bytes = tpl.read()
+    df = pd.read_csv(data) if data.name.lower().endswith(".csv") else pd.read_excel(data)
+
+    if "박스번호" not in df.columns:
+        st.error("❌ 필수 컬럼 '박스번호'가 없습니다.")
+        st.stop()
+
+    st.success("✅ 위치 매핑 완료 (엑셀 측)")
+    st.dataframe(df.head(10), use_container_width=True)
 
-    if st.button("🚀 PDF 생성"):
-        pdf = make_pdf(records)
-        st.download_button("⬇️ PDF 다운로드", data=pdf.getvalue(), file_name="box_labels.pdf", mime="application/pdf")
+    merged = build_rows(df)
+    boxes = merged["박스번호"].astype(str).str.zfill(4).unique().tolist()
+
+    st.subheader("🔎 업로드된 박스번호 목록")
+    st.write(f"총 **{len(boxes)}**개")
+    st.dataframe(pd.DataFrame({"박스번호": boxes}), use_container_width=True, height=240)
+
+    sel = st.multiselect("생성할 박스번호 선택 (비우면 전체)", options=boxes)
+    work = merged[merged["박스번호"].isin(sel)] if sel else merged
+    records = work.sort_values("박스번호").to_dict(orient="records")
+
+    # 1페이지 매핑 프리뷰
+    st.subheader("🧪 1페이지 매핑 프리뷰")
+    keys = ["박스번호","종료연도","보존기간","단위업무","기록물철","목록"]
+    mapping_preview = {}
+    for i in range(int(n_per_page)):
+        if i < len(records):
+            r = records[i]
+            for k in keys:
+                mapping_preview[f"{k}{i+1}"] = r.get("생산연도","") if k=="종료연도" else r.get(k,"")
+        else:
+            for k in keys:
+                mapping_preview[f"{k}{i+1}"] = ""
+    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)
+        pages = (len(records) + int(n_per_page) - 1) // int(n_per_page)
+        debug_all = []
+
+        for p in range(pages):
+            chunk = records[p*int(n_per_page):(p+1)*int(n_per_page)]
+            # 매핑 구축
+            mapping = {}
+            for i in range(int(n_per_page)):
+                if i < len(chunk):
+                    r = chunk[i]
+                    mapping[f"박스번호{i+1}"] = r.get("박스번호","")
+                    mapping[f"종료연도{i+1}"] = r.get("생산연도","")
+                    mapping[f"보존기간{i+1}"] = r.get("보존기간","")
+                    mapping[f"단위업무{i+1}"] = r.get("단위업무","")
+                    mapping[f"기록물철{i+1}"] = r.get("기록물철","")
+                    mapping[f"목록{i+1}"]   = r.get("목록","")
+                else:
+                    for k in keys:
+                        mapping[f"{k}{i+1}"] = ""
+
+            out_hwpx, dbg = replace_in_hwpx(tpl_bytes, mapping)
+            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")
+        st.download_button("⬇️ 디버그(JSON)", data=json.dumps(debug_all, ensure_ascii=False, indent=2),
+                           file_name="debug.json", mime="application/json")