Upload 4 files

LICENSE

@@ -1,5 +1 @@
-MIT License (내부용)
-본 앱은 개인 및 비영리 목적으로 자유롭게 사용할 수 있으며,
-치환 결과에 대한 법적 책임은 사용자에게 있습니다.
-임의 수정 후 재배포를 금지합니다.
-출처 명시: blueradiance / masking-app
+비영리/개인 내부용. 무단 수정 및 재배포 금지. 출처 명시 필수 (blueradiance / masking-app)

README.md

@@ -1,8 +1,8 @@
 ---
-title: 민감정보 마스킹기
+title: 마스킹 앱 완전체
 emoji: 🛡️
-colorFrom: indigo
-colorTo: blue
+colorFrom: blue
+colorTo: indigo
 sdk: gradio
 sdk_version: 4.16.0
 app_file: app.py

app.py

@@ -49,11 +49,41 @@ def refactored_mask_names(original_text, names, start_counter=100):
             counter += 1
     return masked, mapping
 
-def sanitize_sensitive_info(text, keyword_string):
+def to_chosung(text):
+    CHOSUNG_LIST = [chr(i) for i in range(0x1100, 0x1113)]
+    result = ""
+    for ch in text:
+        if '가' <= ch <= '힣':
+            code = ord(ch) - ord('가')
+            cho = code // 588
+            result += CHOSUNG_LIST[cho]
+        else:
+            result += ch
+    return result
+
+def mask_school_names(text):
+    school_patterns = [
+        (r"(\b[가-힣]{2,20})(초등학교|중학교|고등학교)", True),
+        (r"(\b[가-힣]{2,20})\s(초등학교|중학교|고등학교)", False),
+    ]
+    for pattern, attach in school_patterns:
+        text = re.sub(pattern, lambda m: to_chosung(m.group(1)) + (" " if not attach else "") + m.group(2), text)
+    return text
+
+def mask_department(text):
+    text = re.sub(r"([가-힣]{2,20}학과)", lambda m: to_chosung(m.group(1)[:-2]) + "학과", text)
+    text = re.sub(r"([가-힣]{2,20})과", lambda m: to_chosung(m.group(1)) + "과" if not re.search(r"정신|의무|외래", m.group(1)) else m.group(0), text)
+    return text
+
+def sanitize_sensitive_info(text, keyword_string, replace_word):
+    text = mask_school_names(text)
+    text = mask_department(text)
+    text = re.sub(r"(\d)학년\s?(\d)반", r"*학년 *반", text)
+
     keywords = [k.strip() for k in keyword_string.split(",") if k.strip()]
     for kw in keywords:
         pattern = rf"\b{re.escape(kw)}\b"
-        text = re.sub(pattern, "우리기관", text, flags=re.IGNORECASE)
+        text = re.sub(pattern, replace_word, text, flags=re.IGNORECASE)
     text = re.sub(r"(\d{3})-(\d{4})-(\d{4})", r"\1-****-\3", text)
     text = re.sub(r"(\d{4})년 (\d{1,2})월 (\d{1,2})일", r"19**년 \2월 *일", text)
     text = re.sub(r"(\d{1,3})번지", r"***번지", text)
@@ -74,24 +104,36 @@ def final_name_remask_exact_only(text, mapping_dict):
         text = re.sub(pattern, tag, text)
     return text
 
-def full_pipeline(text, keywords):
+def apply_masking(text, keywords, replace_word):
     names = extract_names(text)
     masked, mapping = refactored_mask_names(text, names)
-    sanitized = sanitize_sensitive_info(masked, keywords)
+    sanitized = sanitize_sensitive_info(masked, keywords, replace_word)
     sanitized = final_name_remask_exact_only(sanitized, mapping)
     mapping_table = "\n".join([f"{k} → {v}" for k, v in mapping.items()])
     return sanitized, mapping_table
 
-gr.Interface(
-    fn=full_pipeline,
-    inputs=[
-        gr.Textbox(lines=15, label="📥 원본 텍스트 입력"),
-        gr.Textbox(lines=1, label="기관 키워드 (쉼표로 구분)", value="굿네이버스, good neighbors, gn, 사회복지법인 굿네이버스")
-    ],
-    outputs=[
-        gr.Textbox(lines=15, label="🔐 마스킹된 텍스트"),
-        gr.Textbox(lines=10, label="🏷️ 이름 태그 매핑 (NXXX → 이름)")
-    ],
-    title="🛡️ 민감정보 마스킹 [땡땡이 마스킹] : 이름 + 민감정보 + 초/중/고 마스킹기 (초성 기반)",
-    description=""
-).launch()
+def remask_with_mapping(text, mapping_string):
+    mapping = {}
+    for line in mapping_string.strip().split("\n"):
+        if "→" in line:
+            tag, name = line.split("→")
+            mapping[tag.strip()] = name.strip()
+    for tag, name in mapping.items():
+        pattern = rf'(?<![\w가-힣]){re.escape(name)}(?![\w가-힣])'
+        text = re.sub(pattern, tag, text)
+    return text
+
+with gr.Blocks() as demo:
+    gr.Markdown("🛡️ 민감정보 마스킹 [땡땡이 마스킹] : 이름 + 민감정보 + 초/중/고 마스킹기 (초성 기반)")
+    input_text = gr.Textbox(lines=15, label="📥 원본 텍스트 입력")
+    keyword_input = gr.Textbox(lines=1, label="기관 키워드 (쉼표로 구분)", value="굿네이버스, good neighbors, gn, 사회복지법인 굿네이버스")
+    replace_input = gr.Textbox(lines=1, label="치환할 텍스트", value="우리기관")
+    run_button = gr.Button("🚀 마스킹 실행")
+    masked_output = gr.Textbox(lines=15, label="🔐 마스킹된 텍스트")
+    mapping_output = gr.Textbox(lines=10, label="🏷️ 이름 태그 매핑 (수정 가능)", interactive=True)
+    reapply_button = gr.Button("🔁 매핑 재반영")
+
+    run_button.click(fn=apply_masking, inputs=[input_text, keyword_input, replace_input], outputs=[masked_output, mapping_output])
+    reapply_button.click(fn=remask_with_mapping, inputs=[masked_output, mapping_output], outputs=masked_output)
+
+demo.launch()

requirements.txt

@@ -1,3 +1,3 @@
-gradio>=4.16.0
+gradio
 torch
 transformers