Upload 4 files

app.py

@@ -70,13 +70,16 @@ def mask_school_names(text):
         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)
     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)반)?", lambda m: "*학년" + (" *반" if m.group(3) else ""), text)
     text = re.sub(r"(\d)학년\s?(\d)반", r"*학년 *반", text)
 
     keywords = [k.strip() for k in keyword_string.split(",") if k.strip()]
@@ -125,14 +128,12 @@ def remask_with_mapping(text, mapping_string):
 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, 사회복지법인굿네이버스")
+    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("🔁 매핑 재반영")
-
+    mapping_output = gr.Textbox(lines=10, label="🏷️ 이름 태그 매핑", interactive=False)
+    
     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()