Upload 4 files

app.py

@@ -49,7 +49,11 @@ def refactored_mask_names(original_text, names, start_counter=100):
             counter += 1
     return masked, mapping
 
-def sanitize_sensitive_info(text):
+def sanitize_sensitive_info(text, keyword_string):
+    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(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)
@@ -62,7 +66,6 @@ def sanitize_sensitive_info(text):
     text = re.sub(r"(\d{4})[- ]?(\d{4})[- ]?(\d{4})[- ]?(\d{4})", lambda m: f"{m.group(1)}-****-****-{m.group(4)}", text)
     text = re.sub(r"(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})", lambda m: f"{m.group(1)}.{m.group(2)}.*.*", text)
     text = re.sub(r"([가-힣]{1,10})(은행|동|로|길)\s?([\d\-]{4,})", lambda m: m.group(1) + m.group(2) + " " + re.sub(r"\d", "*", m.group(3)), text)
-    text = re.sub(r"\b(good neighbors|굿네이버스|사회복지법인 굿네이버스|gn)\b", "우리기관", text, flags=re.IGNORECASE)
     return text
 
 def final_name_remask_exact_only(text, mapping_dict):
@@ -71,17 +74,20 @@ def final_name_remask_exact_only(text, mapping_dict):
         text = re.sub(pattern, tag, text)
     return text
 
-def full_pipeline(text):
+def full_pipeline(text, keywords):
     names = extract_names(text)
     masked, mapping = refactored_mask_names(text, names)
-    sanitized = sanitize_sensitive_info(masked)
+    sanitized = sanitize_sensitive_info(masked, keywords)
     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="📥 원본 텍스트 입력"),
+    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 → 이름)")