타입별 필터링 추가

Author: YH0223

main.py

@@ -193,6 +193,17 @@ def mask_text(text: str, mask_char: str = "0") -> str:
 def filter_sensitive_info(text: str, fix_spelling: bool = False) -> Dict[str, Any]:
     """민감 정보 필터링 (000으로 마스킹) + 오타 교정"""
 
+        # 기본 마스킹 매핑 설정
+    if mask_mapping is None:
+        mask_mapping = {
+            "이름": "*",
+            "지명": "#",
+            "조직명": "@",
+            "전화번호": "0",
+            "이메일": "X",
+            "default": "0"  # 기본값
+        }
+        
     # 1. 오타 교정 (옵션)
     spelling_result = None
     corrected_text = None
@@ -278,13 +289,15 @@ def filter_sensitive_info(text: str, fix_spelling: bool = False) -> Dict[str, An
         original_text = filtered_text[start:end]
 
         # 000으로 마스킹
-        masked = mask_text(original_text)
+        mask_char = mask_mapping.get(entity['type'], mask_mapping['default'])
+        masked = mask_text(original_text, mask_char)
         filtered_text = filtered_text[:start] + masked + filtered_text[end:]
 
     result = {
         "original_text": text,
         "filtered_text": filtered_text,
-        "detected_entities": sorted(unique_entities, key=lambda x: x['start'])
+        "detected_entities": sorted(unique_entities, key=lambda x: x['start']),
+        "mask_mapping_used": mask_mapping
     }
 
     # 오타 교정 결과 추가