Update app.py

app.py

@@ -215,8 +215,20 @@ def sanitize_sensitive_info(text, keyword_string, replace_word):
 
 # 🔹 마스킹 함수 (정리된 최종본)
 def extract_names(text):
-    # 이름이 '홍길동'이면 테스트용으로 감지
-    return ["홍길동"] if "홍길동" in text else []
+    try:
+        results = ner_pipeline(text)
+    except Exception as e:
+        print("NER 오류 발생:", e)
+        return []
+
+    names = []
+    for entity in results:
+        if entity.get("entity_group") == "PS":
+            name = entity["word"].replace("##", "").strip()
+            if len(name) >= 2 and name not in NAME_ENTITY_EXCEPTIONS:
+                names.append(name)
+    return names
+
 
 def refactored_mask_names(text, names):
     counter = 1