Update app.py

app.py

@@ -192,63 +192,64 @@ def sanitize_sensitive_info(text, keyword_string, replace_word):
 
 # 🔹 마스킹 함수 (정리된 최종본)
 def extract_names(text):
-    return ["홍길동"]  # 예시용
+    # 이름이 '홍길동'이면 테스트용으로 감지
+    return ["홍길동"] if "홍길동" in text else []
 
 def refactored_mask_names(text, names):
+    counter = 1
     mapping = {}
-    for i, name in enumerate(names):
-        tag = f"N{i+1:03}"
-        text = text.replace(name, tag)
-        mapping[name] = tag
-    return text, mapping
-
-def final_name_remask_exact_only(text, mapping):
-    for name, tag in mapping.items():
-        text = text.replace(name, tag)
+    used_names = set()
+    masked = text
+
+    for name in names:
+        # 조사 구분 있는 경우
+        for josa in ["은", "는", "이", "가", "을", "를", "께서", "도", "만", "의", "에서"]:
+            pattern = rf'(?<![\w가-힣]){re.escape(name)}{josa}(?![\w가-힣])'
+            if re.search(pattern, masked):
+                tag = f"{TAG_PREFIX}{counter:03d}"
+                mapping[tag] = name
+                masked = re.sub(pattern, tag + josa, masked)
+                counter += 1
+                used_names.add(name)
+                break
+    for name in names:
+        if name in used_names:
+            continue
+        pattern = rf'(?<![\w가-힣]){re.escape(name)}(?![\w가-힣])'
+        if re.search(pattern, masked):
+            tag = f"{TAG_PREFIX}{counter:03d}"
+            mapping[tag] = name
+            masked = re.sub(pattern, tag, masked)
+            counter += 1
+    return masked, mapping
+
+def final_name_remask_exact_only(text, mapping_dict):
+    for tag, name in mapping_dict.items():
+        pattern = rf'(?<![\w가-힣]){re.escape(name)}(?![\w가-힣])'
+        text = re.sub(pattern, tag, text)
     return text
 
-def sanitize_sensitive_info(text, keywords, replace_word):
-    # 기관 키워드 치환
+def sanitize_sensitive_info(text, keyword_string, replace_word):
+    text = postprocess_sensitive_patterns(text)
+    text = mask_school_names(text)
+    text = mask_department(text)
+    text = re.sub(r"(\d)학년(\s?(\d)반)?", lambda m: "*학년" + (" *반" if m.group(3) else ""), text)
+
+    keywords = [k.strip() for k in keyword_string.split(",") if k.strip()] + list(REGEX_KEYWORDS_TO_MASK)
     for kw in keywords:
-        pattern = rf"{re.escape(kw)}(?=\W|$)"  # 조사 대응
+        pattern = rf"{re.escape(kw)}(?=\W|$)"
         text = re.sub(pattern, replace_word, text, flags=re.IGNORECASE)
 
-    # 이메일 마스킹 예시
-    text = re.sub(r"\b[\w\.-]+@", "******@", text)
+    text = re.sub(r"(\d{6})[-](\d)\d{6}", r"*******-\2*****", text)
+    text = re.sub(r"([가-힣]+(대로|로|길))\s?(\d+)(호|번길|가)?", r"\1 ***", text)
     return text
 
-
-# ✅ 핵심 apply_masking 함수
 def apply_masking(text, keyword_str, replace_word):
     keywords = [kw.strip() for kw in keyword_str.split(",") if kw.strip()]
-
     names = extract_names(text)
     masked_text, name_mapping = refactored_mask_names(text, names)
-    sanitized_text = sanitize_sensitive_info(masked_text, keywords, replace_word)
+    sanitized_text = sanitize_sensitive_info(masked_text, keyword_str, replace_word)
     final_text = final_name_remask_exact_only(sanitized_text, name_mapping)
-
     mapping_table = "\n".join(f"{k} → {v}" for k, v in name_mapping.items())
     return final_text, mapping_table
-
-
-# ✅ Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("""
-    🛡️ **민감정보 마스킹 [ver2]**  
-    이름 + 민감정보 + 기관 키워드 마스킹기 (초성 기반 + 후처리 강화)
-    """)
-
-    input_text = gr.Textbox(lines=10, label="📅 원본 텍스트 입력")
-    keyword_input = gr.Textbox(lines=1, label="기관 키워드 (쉼표 구분)", value="굿네이버스, 사회복지법인 굿네이버스")
-    replace_input = gr.Textbox(lines=1, label="치환할 텍스트", value="우리기관")
-    run_button = gr.Button("🚀 마스킹 실행")
-    masked_output = gr.Textbox(lines=10, label="🔐 마스킹 결과")
-    mapping_output = gr.Textbox(lines=5, label="🏷️ 이름 태그 매핑", interactive=False)
-
-    run_button.click(
-        fn=apply_masking,
-        inputs=[input_text, keyword_input, replace_input],
-        outputs=[masked_output, mapping_output]
-    )
-
 demo.launch()