Update app.py
Browse files
app.py
CHANGED
|
@@ -1,210 +1,307 @@
|
|
| 1 |
-
import io, os, zipfile, re
|
| 2 |
-
import streamlit as st
|
| 3 |
-
import pandas as pd
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
from reportlab.
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
for
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
def
|
| 92 |
-
|
| 93 |
-
|
| 94 |
-
|
| 95 |
-
|
| 96 |
-
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
""")
|
| 134 |
-
|
| 135 |
-
|
| 136 |
-
|
| 137 |
-
|
| 138 |
-
|
| 139 |
-
|
| 140 |
-
|
| 141 |
-
|
| 142 |
-
if
|
| 143 |
-
|
| 144 |
-
|
| 145 |
-
|
| 146 |
-
|
| 147 |
-
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
|
| 165 |
-
|
| 166 |
-
|
| 167 |
-
|
| 168 |
-
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
| 176 |
-
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
| 190 |
-
|
| 191 |
-
|
| 192 |
-
|
| 193 |
-
|
| 194 |
-
|
| 195 |
-
|
| 196 |
-
|
| 197 |
-
|
| 198 |
-
|
| 199 |
-
|
| 200 |
-
|
| 201 |
-
|
| 202 |
-
|
| 203 |
-
|
| 204 |
-
|
| 205 |
-
|
| 206 |
-
|
| 207 |
-
|
| 208 |
-
|
| 209 |
-
|
| 210 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import io, os, zipfile, re
|
| 2 |
+
import streamlit as st
|
| 3 |
+
import pandas as pd
|
| 4 |
+
|
| 5 |
+
# (์ ํ) PDF ๋ฏธ๋ฆฌ๋ณด๊ธฐ์ฉ
|
| 6 |
+
from reportlab.pdfgen import canvas
|
| 7 |
+
from reportlab.lib.pagesizes import A4
|
| 8 |
+
from reportlab.lib.units import mm
|
| 9 |
+
|
| 10 |
+
# =========================
|
| 11 |
+
# HWPX ํ
ํ๋ฆฟ ์นํ ์ ํธ (ํ๋์ปจํธ๋กค & {{ํ ํฐ}} ์ง์)
|
| 12 |
+
# =========================
|
| 13 |
+
def replace_placeholders_in_hwpx(hwpx_bytes: bytes, mapping: dict) -> bytes:
|
| 14 |
+
mem_in = io.BytesIO(hwpx_bytes)
|
| 15 |
+
mem_out = io.BytesIO()
|
| 16 |
+
zin = zipfile.ZipFile(mem_in, "r")
|
| 17 |
+
zout = zipfile.ZipFile(mem_out, "w", zipfile.ZIP_DEFLATED)
|
| 18 |
+
|
| 19 |
+
# ํ
ํ๋ฆฟ์ ์๋ ์๋ฌธ ๋ณ์นญ โ ํ๊ธ ํ๋ ๋์ ์ฃผ์
|
| 20 |
+
alias = {
|
| 21 |
+
# ์ธํธ2
|
| 22 |
+
"box2": "๋ฐ์ค๋ฒํธ2","year2": "์ข
๋ฃ์ฐ๋2","duration2": "๋ณด์กด๊ธฐ๊ฐ2","task2": "๋จ์์
๋ฌด2","file2": "๊ธฐ๋ก๋ฌผ์ฒ 2","list2": "๋ชฉ๋ก2",
|
| 23 |
+
# ์ธํธ3
|
| 24 |
+
"box3": "๋ฐ์ค๋ฒํธ3","year3": "์ข
๋ฃ์ฐ๋3","duration3": "๋ณด์กด๊ธฐ๊ฐ3","task3": "๋จ์์
๋ฌด3","file3": "๊ธฐ๋ก๋ฌผ์ฒ 3","list3": "๋ชฉ๋ก3",
|
| 25 |
+
}
|
| 26 |
+
expanded_map = dict(mapping)
|
| 27 |
+
for k, v in list(mapping.items()):
|
| 28 |
+
for a, tgt in alias.items():
|
| 29 |
+
if k == tgt and a not in expanded_map:
|
| 30 |
+
expanded_map[a] = v
|
| 31 |
+
|
| 32 |
+
def repl_xml(txt: str, kv: dict) -> str:
|
| 33 |
+
# 1) {{ํค}} ์นํ
|
| 34 |
+
any_token = any(("{{"+k+"}}" in txt) for k in kv.keys())
|
| 35 |
+
if any_token:
|
| 36 |
+
for k, v in kv.items():
|
| 37 |
+
txt = txt.replace(f"{{{{{k}}}}}", "" if v is None else str(v))
|
| 38 |
+
return txt
|
| 39 |
+
# 2) ํ๋์ปจํธ๋กค ๋ณธ๋ฌธ ์นํ
|
| 40 |
+
for k, v in kv.items():
|
| 41 |
+
pattern = re.compile(
|
| 42 |
+
rf'(<hp:fieldBegin[^>]*name="{re.escape(k)}"[^>]*>.*?</hp:fieldBegin>)(.*?)(<hp:fieldEnd[^>]*/>)',
|
| 43 |
+
re.DOTALL
|
| 44 |
+
)
|
| 45 |
+
def _repl(m):
|
| 46 |
+
val = "" if v is None else str(v)
|
| 47 |
+
return f'{m.group(1)}<hp:run><hp:t>{val}</hp:t></hp:run>{m.group(3)}'
|
| 48 |
+
txt, _ = pattern.subn(_repl, txt, count=50)
|
| 49 |
+
return txt
|
| 50 |
+
|
| 51 |
+
for item in zin.infolist():
|
| 52 |
+
data = zin.read(item.filename)
|
| 53 |
+
if item.filename.startswith("Contents/") and item.filename.endswith(".xml"):
|
| 54 |
+
try:
|
| 55 |
+
txt = data.decode("utf-8", errors="ignore")
|
| 56 |
+
txt = repl_xml(txt, expanded_map)
|
| 57 |
+
data = txt.encode("utf-8")
|
| 58 |
+
except Exception:
|
| 59 |
+
pass
|
| 60 |
+
zout.writestr(item, data)
|
| 61 |
+
|
| 62 |
+
zin.close()
|
| 63 |
+
zout.close()
|
| 64 |
+
mem_out.seek(0)
|
| 65 |
+
return mem_out.getvalue()
|
| 66 |
+
|
| 67 |
+
# =========================
|
| 68 |
+
# PDF ๋ฏธ๋ฆฌ๋ณด๊ธฐ (ํผํ
3203 ์์)
|
| 69 |
+
# =========================
|
| 70 |
+
LABEL_W, LABEL_H = 89*mm, 38*mm
|
| 71 |
+
MARGIN_L, MARGIN_T = 10*mm, 10*mm
|
| 72 |
+
COL_GAP, ROW_GAP = 5*mm, 5*mm
|
| 73 |
+
COLS, ROWS = 2, 5
|
| 74 |
+
|
| 75 |
+
def draw_label(c, x, y, row_dict):
|
| 76 |
+
c.setFont("Helvetica", 9)
|
| 77 |
+
lines = [
|
| 78 |
+
f"๋ฐ์ค๋ฒํธ: {row_dict.get('๋ฐ์ค๋ฒํธ','')}",
|
| 79 |
+
f"์ข
๋ฃ์ฐ๋(๋ฒ์): {row_dict.get('์์ฐ์ฐ๋','')}",
|
| 80 |
+
f"๋ณด์กด๊ธฐ๊ฐ: {row_dict.get('๋ณด์กด๊ธฐ๊ฐ','')}",
|
| 81 |
+
f"๋จ์์
๋ฌด: {row_dict.get('๋จ์์
๋ฌด','')}",
|
| 82 |
+
f"๊ธฐ๋ก๋ฌผ์ฒ : {row_dict.get('๊ธฐ๋ก๋ฌผ์ฒ ','')}",
|
| 83 |
+
f"๋ชฉ๋ก(์์ฝ): {str(row_dict.get('๋ชฉ๋ก',''))[:40]}",
|
| 84 |
+
]
|
| 85 |
+
ty = y + LABEL_H - 8*mm
|
| 86 |
+
for ln in lines:
|
| 87 |
+
c.drawString(x + 5*mm, ty, ln)
|
| 88 |
+
ty -= 5*mm
|
| 89 |
+
c.rect(x, y, LABEL_W, LABEL_H)
|
| 90 |
+
|
| 91 |
+
def make_preview_pdf(rows: list) -> bytes:
|
| 92 |
+
buf = io.BytesIO()
|
| 93 |
+
c = canvas.Canvas(buf, pagesize=A4)
|
| 94 |
+
page_i, i = 0, 0
|
| 95 |
+
while i < len(rows):
|
| 96 |
+
c.setFont("Helvetica-Bold", 10)
|
| 97 |
+
c.drawString(15*mm, 287*mm, f"Formtec 3203 Preview โ Page {page_i+1}")
|
| 98 |
+
for r in range(ROWS):
|
| 99 |
+
for col in range(COLS):
|
| 100 |
+
if i >= len(rows): break
|
| 101 |
+
x = MARGIN_L + col*(LABEL_W + COL_GAP)
|
| 102 |
+
y = (A4[1] - MARGIN_T - LABEL_H) - r*(LABEL_H + ROW_GAP)
|
| 103 |
+
draw_label(c, x, y, rows[i])
|
| 104 |
+
i += 1
|
| 105 |
+
c.showPage()
|
| 106 |
+
page_i += 1
|
| 107 |
+
c.save()
|
| 108 |
+
buf.seek(0)
|
| 109 |
+
return buf.read()
|
| 110 |
+
|
| 111 |
+
# =========================
|
| 112 |
+
# ์
๋ก๋ ๋ชจ๋: ์์
๋ฐ์ดํฐ ๋ณํฉ ๋ก์ง
|
| 113 |
+
# =========================
|
| 114 |
+
def compute_year_range(series: pd.Series):
|
| 115 |
+
s = series.astype(str).fillna("")
|
| 116 |
+
valid = s[~s.isin(["", "0", "0000"])]
|
| 117 |
+
if len(valid) == 0:
|
| 118 |
+
return "0000-0000"
|
| 119 |
+
valid_int = pd.to_numeric(valid, errors="coerce").dropna().astype(int)
|
| 120 |
+
if len(valid_int) == 0:
|
| 121 |
+
return "0000-0000"
|
| 122 |
+
return f"{valid_int.min():04d}-{valid_int.max():04d}"
|
| 123 |
+
|
| 124 |
+
def build_merged_df(df: pd.DataFrame) -> pd.DataFrame:
|
| 125 |
+
df = df.copy()
|
| 126 |
+
if "์ ๏ฟฝ๏ฟฝ" in df.columns:
|
| 127 |
+
df["์ ๋ชฉ"] = df["์ ๋ชฉ"].astype(str)
|
| 128 |
+
if "๋ฐ์ค๋ฒํธ" in df.columns:
|
| 129 |
+
df["๋ฐ์ค๋ฒํธ"] = df["๋ฐ์ค๋ฒํธ"].astype(str).str.zfill(4)
|
| 130 |
+
|
| 131 |
+
# ์์ฐ์ฐ๋(๋ฒ์) ๊ณ์ฐ
|
| 132 |
+
if "์ข
๋ฃ์ฐ๋" in df.columns:
|
| 133 |
+
prod_df = df.groupby("๋ฐ์ค๋ฒํธ")["์ข
๋ฃ์ฐ๋"].apply(compute_year_range).reset_index()
|
| 134 |
+
prod_df.columns = ["๋ฐ์ค๋ฒํธ", "์์ฐ์ฐ๋"]
|
| 135 |
+
else:
|
| 136 |
+
prod_df = pd.DataFrame({"๋ฐ์ค๋ฒํธ": df["๋ฐ์ค๋ฒํธ"].unique(), "์์ฐ์ฐ๋": "0000-0000"})
|
| 137 |
+
|
| 138 |
+
# ๋ชฉ๋ก(๊ด๋ฆฌ๋ฒํธ + ์ ๋ชฉ)
|
| 139 |
+
has_mgmt = "๊ด๋ฆฌ๋ฒํธ" in df.columns
|
| 140 |
+
list_data = []
|
| 141 |
+
for box_num, group in df.groupby("๋ฐ์ค๋ฒํธ"):
|
| 142 |
+
if has_mgmt:
|
| 143 |
+
lines = [f"- {row['๊ด๋ฆฌ๋ฒํธ']} {row['์ ๋ชฉ']}" for _, row in group.iterrows()]
|
| 144 |
+
else:
|
| 145 |
+
lines = [f"- {row['์ ๋ชฉ']}" for _, row in group.iterrows()]
|
| 146 |
+
list_data.append({"๋ฐ์ค๋ฒํธ": box_num, "๋ชฉ๋ก": "\r\n".join(lines)})
|
| 147 |
+
list_df = pd.DataFrame(list_data)
|
| 148 |
+
|
| 149 |
+
meta_cols = ["๋ฐ์ค๋ฒํธ","์ข
๋ฃ์ฐ๋","๋ณด์กด๊ธฐ๊ฐ","๋จ์์
๋ฌด","๊ธฐ๋ก๋ฌผ์ฒ ","์ ๋ชฉ"]
|
| 150 |
+
meta_exist = [c for c in meta_cols if c in df.columns]
|
| 151 |
+
meta_df = df.groupby("๋ฐ์ค๋ฒํธ", as_index=False).first()[meta_exist]
|
| 152 |
+
|
| 153 |
+
merged = meta_df.merge(list_df, on="๋ฐ์ค๋ฒํธ").merge(prod_df, on="๋ฐ์ค๋ฒํธ")
|
| 154 |
+
return merged
|
| 155 |
+
|
| 156 |
+
# =========================
|
| 157 |
+
# Streamlit UI
|
| 158 |
+
# =========================
|
| 159 |
+
st.set_page_config(page_title="๋ฐ์ค๋ผ๋ฒจ(.HWPX) ์์ฑ๊ธฐ โ Hugging Face", layout="wide")
|
| 160 |
+
st.title("๐ฆ ๋ฐ์ค๋ผ๋ฒจ ์๋ ์์ฑ๊ธฐ (.HWPX ํ
ํ๋ฆฟ)")
|
| 161 |
+
|
| 162 |
+
with st.expander("์ฌ์ฉ ๋ฐฉ๋ฒ", expanded=True):
|
| 163 |
+
st.markdown("""
|
| 164 |
+
- **์์(.HWPX ํ
ํ๋ฆฟ)** ์ ์
๋ก๋ํฉ๋๋ค. (ํ๊ธ์์ '๋ค๋ฅธ ์ด๋ฆ์ผ๋ก ์ ์ฅ' โ HWPX)
|
| 165 |
+
- **๋ฐ์ดํฐ ์
๋ ฅ ๋ฐฉ์**์ ์ ํํฉ๋๋ค.
|
| 166 |
+
1) **ํ์ผ ์
๋ก๋(์์
/CSV)**: ๊ธฐ์กด ๋ฐ์ดํฐ๋ก ์ผ๊ด ์์ฑ
|
| 167 |
+
2) **์ง์ ์
๋ ฅ(๋ชฉ๋ก & ๋ฉํ๋ฐ์ดํฐ)**: ํ๋ฉด์์ ๋ฐ์ค๋ณ ์ ๋ณด๋ฅผ ์๋์ผ๋ก ์
๋ ฅ
|
| 168 |
+
- ํ
ํ๋ฆฟ์ ์ธํธ๋ณ ํ๋(์: `๋ฐ์ค๋ฒํธ1~3`, `์ข
๋ฃ์ฐ๋1~3`, `๋ณด์กด๊ธฐ๊ฐ1~3`, `๋จ์์
๋ฌด1~3`, `๊ธฐ๋ก๋ฌผ์ฒ 1~3`, `๋ชฉ๋ก1~3`)๊ฐ ์์ด์ผ ํฉ๋๋ค.
|
| 169 |
+
""")
|
| 170 |
+
|
| 171 |
+
tpl_file = st.file_uploader("HWPX ํ
ํ๋ฆฟ ์
๋ก๋", type=["hwpx"])
|
| 172 |
+
batch_size = st.number_input("ํ
ํ๋ฆฟ์ ๋ผ๋ฒจ ์ธํธ ๊ฐ์", min_value=1, max_value=12, value=3, step=1)
|
| 173 |
+
preview_pdf = st.checkbox("PDF ๋ฏธ๋ฆฌ๋ณด๊ธฐ๋ ์์ฑ", value=False)
|
| 174 |
+
|
| 175 |
+
mode = st.radio("๋ฐ์ดํฐ ์
๋ ฅ ๋ฐฉ์", ["ํ์ผ ์
๋ก๋(์์
/CSV)", "์ง์ ์
๋ ฅ(๋ชฉ๋ก & ๋ฉํ๋ฐ์ดํฐ)"], horizontal=True)
|
| 176 |
+
default_fields = ["๋ฐ์ค๋ฒํธ","์ข
๋ฃ์ฐ๋","๋ณด์กด๊ธฐ๊ฐ","๋จ์์
๋ฌด","๊ธฐ๋ก๋ฌผ์ฒ ","๋ชฉ๋ก"]
|
| 177 |
+
|
| 178 |
+
rows = None # ์ต์ข
์์ฑ์ ์ฌ์ฉํ row dict ๋ฆฌ์คํธ
|
| 179 |
+
|
| 180 |
+
# ----- ๋ชจ๋ 1: ํ์ผ ์
๋ก๋ -----
|
| 181 |
+
if mode == "ํ์ผ ์
๋ก๋(์์
/CSV)":
|
| 182 |
+
data_file = st.file_uploader("๋ฐ์ดํฐ ์
๋ก๋ (Excel/CSV)", type=["xlsx","xls","csv"])
|
| 183 |
+
if data_file:
|
| 184 |
+
if data_file.name.lower().endswith(".csv"):
|
| 185 |
+
df = pd.read_csv(data_file)
|
| 186 |
+
else:
|
| 187 |
+
df = pd.read_excel(data_file)
|
| 188 |
+
st.write("๋ฐ์ดํฐ ๋ฏธ๋ฆฌ๋ณด๊ธฐ")
|
| 189 |
+
st.dataframe(df.head(10), use_container_width=True)
|
| 190 |
+
|
| 191 |
+
# ์ปฌ๋ผ ๋งคํ
|
| 192 |
+
st.subheader("์ปฌ๋ผ ๋งคํ (์์
/CSV โ ๋ผ๋ฒจ ํ๋)")
|
| 193 |
+
col_map = {}
|
| 194 |
+
for f in ["๋ฐ์ค๋ฒํธ","์ข
๋ฃ์ฐ๋","๋ณด์กด๊ธฐ๊ฐ","๋จ์์
๋ฌด","๊ธฐ๋ก๋ฌผ์ฒ ","๊ด๋ฆฌ๋ฒํธ","์ ๋ชฉ"]:
|
| 195 |
+
options = ["(์์)"] + list(df.columns)
|
| 196 |
+
default_idx = options.index(f) if f in df.columns else 0
|
| 197 |
+
col_map[f] = st.selectbox(f"{f} โ", options, index=default_idx, key=f"map_{f}")
|
| 198 |
+
renames = {v: k for k, v in col_map.items() if v != "(์์)"}
|
| 199 |
+
df_std = df.rename(columns=renames)
|
| 200 |
+
|
| 201 |
+
if "๋ฐ์ค๋ฒํธ" not in df_std.columns:
|
| 202 |
+
st.error("ํ์ ์ปฌ๋ผ '๋ฐ์ค๋ฒํธ'๊ฐ ๋งคํ๋์ง ์์์ต๋๋ค.")
|
| 203 |
+
else:
|
| 204 |
+
merged_df = build_merged_df(df_std)
|
| 205 |
+
st.text_input("์ถ๋ ฅํ ๋ฐ์ค๋ฒํธ (์ผํ ๊ตฌ๋ถ, ๋น์ฐ๋ฉด ์ ์ฒด)", key="sel_boxes")
|
| 206 |
+
if st.button("๋ผ๋ฒจ ์์ฑ (HWPX ZIP)", key="btn_upload"):
|
| 207 |
+
if not tpl_file:
|
| 208 |
+
st.error("ํ
ํ๋ฆฟ(.hwpx)์ ์
๋ก๋ํด ์ฃผ์ธ์.")
|
| 209 |
+
else:
|
| 210 |
+
tpl_bytes = tpl_file.read()
|
| 211 |
+
user_input = st.session_state.get("sel_boxes", "").strip()
|
| 212 |
+
work_df = merged_df.copy()
|
| 213 |
+
if user_input:
|
| 214 |
+
targets = [s.strip().zfill(4) for s in user_input.split(",") if s.strip()]
|
| 215 |
+
work_df = work_df[work_df["๋ฐ์ค๋ฒํธ"].isin(targets)]
|
| 216 |
+
if len(work_df) == 0:
|
| 217 |
+
st.warning("์กฐ๊ฑด์ ํด๋นํ๋ ๋ฐ์ค๋ฒํธ๊ฐ ์์ต๋๋ค.")
|
| 218 |
+
else:
|
| 219 |
+
rows = work_df.to_dict(orient="records")
|
| 220 |
+
|
| 221 |
+
# ----- ๋ชจ๋ 2: ์ง์ ์
๋ ฅ -----
|
| 222 |
+
else:
|
| 223 |
+
st.info("๋ฐ์ค๋ณ ์ ๋ณด๋ฅผ ์ง์ ์
๋ ฅํ์ธ์. **๋ชฉ๋ก**์ ์ฌ๋ฌ ์ค๋ก ์
๋ ฅํ๋ฉด ์ค๋ฐ๊ฟ ์ ์ง๋ฉ๋๋ค.")
|
| 224 |
+
num_boxes = st.number_input("๋ฐ์ค ๊ฐ์", min_value=1, max_value=100, value=3, step=1)
|
| 225 |
+
manual_rows = []
|
| 226 |
+
for i in range(num_boxes):
|
| 227 |
+
st.subheader(f"๋ฐ์ค #{i+1}")
|
| 228 |
+
c1, c2, c3 = st.columns(3)
|
| 229 |
+
with c1:
|
| 230 |
+
box = st.text_input("๋ฐ์ค๋ฒํธ (์ซ์, ์๋ 4์๋ฆฌ)", key=f"box_{i}")
|
| 231 |
+
with c2:
|
| 232 |
+
dur = st.text_input("๋ณด์กด๊ธฐ๊ฐ", key=f"dur_{i}")
|
| 233 |
+
with c3:
|
| 234 |
+
task = st.text_input("๋จ์์
๋ฌด", key=f"task_{i}")
|
| 235 |
+
c4, c5 = st.columns(2)
|
| 236 |
+
with c4:
|
| 237 |
+
folder = st.text_input("๊ธฐ๋ก๋ฌผ์ฒ ", key=f"file_{i}")
|
| 238 |
+
with c5:
|
| 239 |
+
prod = st.text_input("์์ฐ์ฐ๋(๋ฒ์) ์: 2019-2022", key=f"prod_{i}")
|
| 240 |
+
memo = st.text_area("๋ชฉ๋ก (ํ ์ค๋น ํ ํญ๋ชฉ, ์์ '- '๋ ์์ด๋ ๋ฉ๋๋ค)", key=f"list_{i}", height=120,
|
| 241 |
+
placeholder="์)\n2020-001 ์ด๋ฌด๋ถ ํ์๋ก\n2020-002 ์์ฐ๊ฒฐ์ฐ์\n...")
|
| 242 |
+
|
| 243 |
+
# ๋ชฉ๋ก ์ ๋ฆฌ: '- ' ์ ๋ ์์ผ๋ฉด ๋ถ์ฌ์ค, ๋น ์ค ์ ์ธ
|
| 244 |
+
lines = [ln.strip() for ln in memo.splitlines() if ln.strip()]
|
| 245 |
+
norm_lines = [ln if ln.startswith("- ") else f"- {ln}" for ln in lines]
|
| 246 |
+
manual_rows.append({
|
| 247 |
+
"๋ฐ์ค๋ฒํธ": (box or "").strip().zfill(4) if box else "",
|
| 248 |
+
"๋ณด์กด๊ธฐ๊ฐ": dur.strip(),
|
| 249 |
+
"๋จ์์
๋ฌด": task.strip(),
|
| 250 |
+
"๊ธฐ๋ก๋ฌผ์ฒ ": folder.strip(),
|
| 251 |
+
"์์ฐ์ฐ๋": prod.strip(),
|
| 252 |
+
"๋ชฉ๋ก": "\r\n".join(norm_lines)
|
| 253 |
+
})
|
| 254 |
+
if st.button("๋ผ๋ฒจ ์์ฑ (HWPX ZIP)", key="btn_manual"):
|
| 255 |
+
if not tpl_file:
|
| 256 |
+
st.error("ํ
ํ๋ฆฟ(.hwpx)์ ์
๋ก๋ํด ์ฃผ์ธ์.")
|
| 257 |
+
else:
|
| 258 |
+
rows = manual_rows
|
| 259 |
+
|
| 260 |
+
# ===== ์ค์ ์์ฑ ๊ณตํต ๋ฃจํด =====
|
| 261 |
+
if rows is not None and len(rows) > 0:
|
| 262 |
+
tpl_bytes = tpl_file.read() if tpl_file else None
|
| 263 |
+
if not tpl_bytes:
|
| 264 |
+
st.error("ํ
ํ๋ฆฟ(.hwpx)์ ์
๋ก๋ํด ์ฃผ์ธ์.")
|
| 265 |
+
else:
|
| 266 |
+
# ZIP์ผ๋ก ํ์ด์ง ๋จ์ HWPX ์์ฑ
|
| 267 |
+
default_fields = ["๋ฐ์ค๋ฒํธ","์ข
๋ฃ์ฐ๋","๋ณด์กด๊ธฐ๊ฐ","๋จ์์
๋ฌด","๊ธฐ๋ก๋ฌผ์ฒ ","๋ชฉ๋ก"]
|
| 268 |
+
zip_buf = io.BytesIO()
|
| 269 |
+
with zipfile.ZipFile(zip_buf, "w", zipfile.ZIP_DEFLATED) as zout:
|
| 270 |
+
num_boxes = len(rows)
|
| 271 |
+
num_pages = (num_boxes + batch_size - 1) // batch_size
|
| 272 |
+
for i in range(num_pages):
|
| 273 |
+
mapping_all = {}
|
| 274 |
+
page_rows = []
|
| 275 |
+
for j in range(batch_size):
|
| 276 |
+
idx = i * batch_size + j
|
| 277 |
+
if idx >= num_boxes:
|
| 278 |
+
# ๋จ๋ ์นธ์ ๊ณต๋ฐฑ
|
| 279 |
+
for prefix in default_fields:
|
| 280 |
+
mapping_all[f"{prefix}{j+1}"] = ""
|
| 281 |
+
continue
|
| 282 |
+
r = rows[idx]
|
| 283 |
+
page_rows.append(r)
|
| 284 |
+
for prefix in default_fields:
|
| 285 |
+
if prefix == "์ข
๋ฃ์ฐ๋":
|
| 286 |
+
# ํ
ํ๋ฆฟ์ "์ข
๋ฃ์ฐ๋" ์๋ฆฌ์๋ '์์ฐ์ฐ๋(๋ฒ์)'๋ฅผ ๋ฃ๋ ์ค๊ณ
|
| 287 |
+
mapping_all[f"{prefix}{j+1}"] = r.get("์์ฐ์ฐ๋","")
|
| 288 |
+
else:
|
| 289 |
+
mapping_all[f"{prefix}{j+1}"] = r.get(prefix,"")
|
| 290 |
+
out_hwpx = replace_placeholders_in_hwpx(tpl_bytes, mapping_all)
|
| 291 |
+
page_nums = [str(x.get("๋ฐ์ค๋ฒํธ","")) for x in page_rows]
|
| 292 |
+
safe_name = "_".join(page_nums) if page_nums else f"empty_{i+1}"
|
| 293 |
+
zout.writestr(f"label_{safe_name}.hwpx", out_hwpx)
|
| 294 |
+
|
| 295 |
+
zip_buf.seek(0)
|
| 296 |
+
st.download_button("๐ฅ HWPX ์ผ๊ด ๋ค์ด๋ก๋ (ZIP)", data=zip_buf,
|
| 297 |
+
file_name="boxlabels_hwpx.zip", mime="application/zip")
|
| 298 |
+
|
| 299 |
+
if preview_pdf:
|
| 300 |
+
try:
|
| 301 |
+
pdf_bytes = make_preview_pdf(rows)
|
| 302 |
+
st.download_button("๐ PDF ๋ฏธ๋ฆฌ๋ณด๊ธฐ ๋ค์ด๋ก๋", data=pdf_bytes,
|
| 303 |
+
file_name="preview.pdf", mime="application/pdf")
|
| 304 |
+
except Exception as e:
|
| 305 |
+
st.warning(f"PDF ๋ฏธ๋ฆฌ๋ณด๊ธฐ ์์ฑ ์คํจ: {e}")
|
| 306 |
+
|
| 307 |
+
st.caption("โป ํ
ํ๋ฆฟ์ ๋ฐ๋์ .HWPX ํ์ผ์ด์ด์ผ ํฉ๋๋ค. (.HWP ์ฌ์ฉ ๋ถ๊ฐ) โข '์ข
๋ฃ์ฐ๋' ์๋ฆฌ์๋ '์์ฐ์ฐ๋(๋ฒ์)'๊ฐ ์ฑ์์ง๋๋ก ์ค๊ณ๋์ด ์์ต๋๋ค.")
|