Update app.py

app.py

@@ -211,21 +211,56 @@ def apply_masking(text, keyword_str, replace_word):
 
     return final_text, mapping_table
 
+def refactored_mask_names(text, names):
+    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)
+    return text
+
+def sanitize_sensitive_info(text, keywords, replace_word):
+    # 기관 키워드 치환
+    for kw in keywords:
+        pattern = rf"{re.escape(kw)}(?=\W|$)"  # 조사 대응
+        text = re.sub(pattern, replace_word, text, flags=re.IGNORECASE)
+
+    # 이메일 마스킹 예시
+    text = re.sub(r"\b[\w\.-]+@", "******@", 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()]
 
-# 🔹 Gradio UI
+    names = extract_names(text)
+    masked_text, name_mapping = refactored_mask_names(text, names)
+    sanitized_text = sanitize_sensitive_info(masked_text, keywords, 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=15, label="📅 원본 텍스트 입력")
+
+    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=15, label="🔐 마스킹 결과")
-    mapping_output = gr.Textbox(lines=10, label="🏷️ 이름 태그 매핑", interactive=False)
+    masked_output = gr.Textbox(lines=10, label="🔐 마스킹 결과")
+    mapping_output = gr.Textbox(lines=5, label="🏷️ 이름 태그 매핑", interactive=False)
 
     run_button.click(
         fn=apply_masking,