refactor: centralize classifier constants and simplify logic

backend/app/services/integrated_classifier.py

@@ -6,79 +6,14 @@
 import re
 from typing import Dict, List, Optional, Tuple
 from .fitting_classifier import classify_fitting
-
-# Level 1: 명확한 타입 키워드 (최우선)
-LEVEL1_TYPE_KEYWORDS = {
-    "BOLT": ["FLANGE BOLT", "U-BOLT", "U BOLT", "BOLT", "STUD", "NUT", "SCREW", "WASHER", "볼트", "너트", "스터드", "나사", "와셔", "유볼트"],
-    "VALVE": ["VALVE", "GATE", "BALL", "GLOBE", "CHECK", "BUTTERFLY", "NEEDLE", "RELIEF", "SIGHT GLASS", "STRAINER", "밸브", "게이트", "볼", "글로브", "체크", "버터플라이", "니들", "릴리프", "사이트글라스", "스트레이너"],
-    "FLANGE": ["FLG", "FLANGE", "플랜지", "프랜지", "ORIFICE", "SPECTACLE", "PADDLE", "SPACER", "BLIND", "REDUCING FLANGE", "RED FLANGE"],
-    "PIPE": ["PIPE", "TUBE", "파이프", "배관", "SMLS", "SEAMLESS"],
-    "FITTING": [
-        "SOCK-O-LET", "WELD-O-LET", "ELL-O-LET", "THREAD-O-LET", "ELB-O-LET", "NIP-O-LET", "COUP-O-LET", 
-        "SOCKOLET", "WELDOLET", "ELLOLET", "THREADOLET", "ELBOLET", "NIPOLET", "COUPOLET", "OLET", 
-        "ELBOW", "ELL", "TEE", "REDUCER", "CAP", "COUPLING", "NIPPLE", "SWAGE", "PLUG", 
-        "엘보", "티", "리듀서", "캡", "니플", "커플링", "플러그", "CONC", "ECC",
-        "SWAGELOK", "UNION", "CONNECTOR", "FERRULE", "NUT & FERRULE", "MALE CONNECTOR", "FEMALE CONNECTOR"
-    ],
-    "GASKET": ["GASKET", "GASK", "가스켓", "SWG", "SPIRAL"],
-    "INSTRUMENT": ["GAUGE", "TRANSMITTER", "SENSOR", "THERMOMETER", "계기", "게이지", "트랜스미터", "센서"],
-    "SUPPORT": ["URETHANE BLOCK", "URETHANE", "BLOCK SHOE", "CLAMP", "SUPPORT", "HANGER", "SPRING", "우레탄", "블록", "클램프", "서포트", "행거", "스프링"],
-    "PLATE": ["PLATE", "PL", "CHECKER PLATE", "판재", "철판"],
-    "STRUCTURAL": ["H-BEAM", "BEAM", "ANGLE", "CHANNEL", "H-SECTION", "I-BEAM", "형강", "앵글", "채널"]
-}
-
-# Level 2: 서브타입 키워드 (구체화)
-LEVEL2_SUBTYPE_KEYWORDS = {
-    "VALVE": {
-        "GATE": ["GATE VALVE", "GATE", "게이트 밸브"],
-        "BALL": ["BALL VALVE", "BALL", "볼 밸브"],
-        "GLOBE": ["GLOBE VALVE", "GLOBE", "글로브 밸브"],
-        "CHECK": ["CHECK VALVE", "CHECK", "체크 밸브", "역지 밸브"]
-    },
-    "FLANGE": {
-        "WELD_NECK": ["WELD NECK", "WN", "웰드넥"],
-        "SLIP_ON": ["SLIP ON", "SO", "슬립온"],
-        "BLIND": ["BLIND", "BL", "막음", "차단"],
-        "SOCKET_WELD": ["SOCKET WELD", "소켓웰드"]
-    },
-    "BOLT": {
-        "HEX_BOLT": ["HEX BOLT", "HEXAGON", "육각 볼트"],
-        "STUD_BOLT": ["STUD BOLT", "STUD", "스터드 볼트"],
-        "U_BOLT": ["U-BOLT", "U BOLT", "유볼트"]
-    },
-    "SUPPORT": {
-        "URETHANE_BLOCK": ["URETHANE BLOCK", "BLOCK SHOE", "우레탄 블록"],
-        "CLAMP": ["CLAMP", "클램프"],
-        "HANGER": ["HANGER", "SUPPORT", "행거", "서포트"],
-        "SPRING": ["SPRING", "스프링"]
-    }
-}
-
-# Level 3: 연결/압력 키워드 (공용)
-LEVEL3_CONNECTION_KEYWORDS = {
-    "SW": ["SW", "SOCKET WELD", "소켓웰드"],
-    "THD": ["THD", "THREADED", "NPT", "나사"],
-    "FL": ["FL", "FLANGED", "플랜지형"],
-    "BW": ["BW", "BUTT WELD", "맞대기용접"]
-}
-
-LEVEL3_PRESSURE_KEYWORDS = ["150LB", "300LB", "600LB", "900LB", "1500LB", "2500LB", "3000LB", "6000LB"]
-
-# Level 4: 재질 키워드 (최후 판단)
-LEVEL4_MATERIAL_KEYWORDS = {
-    "PIPE": ["A106", "A333", "A312", "A53"],
-    "FITTING": ["A234", "A403", "A420"],
-    "FLANGE": ["A182", "A350"],  # A105 제거 (범용 재질로 이동)
-    "VALVE": ["A216", "A217", "A351", "A352"],
-    "BOLT": ["A193", "A194", "A320", "A325", "A490"]
-}
-
-# 범용 재질 (여러 타입에 사용 가능)
-GENERIC_MATERIALS = {
-    "A105": ["VALVE", "FLANGE", "FITTING"],  # 우선순위 순서
-    "316": ["VALVE", "FLANGE", "FITTING", "PIPE", "BOLT"],
-    "304": ["VALVE", "FLANGE", "FITTING", "PIPE", "BOLT"]
-}
+from .classifier_constants import (
+    LEVEL1_TYPE_KEYWORDS,
+    LEVEL2_SUBTYPE_KEYWORDS,
+    LEVEL3_CONNECTION_KEYWORDS,
+    LEVEL3_PRESSURE_KEYWORDS,
+    LEVEL4_MATERIAL_KEYWORDS,
+    GENERIC_MATERIALS
+)
 
 def classify_material_integrated(description: str, main_nom: str = "", 
                                 red_nom: str = "", length: float = None) -> Dict:
@@ -142,6 +77,18 @@ def classify_material_integrated(description: str, main_nom: str = "",
             "classification_level": "LEVEL0_SUPPORT",
             "reason": "SUPPORT 시스템 키워드 발견"
         }
+        
+    # [신규] Swagelok 스타일 파트 넘버 패턴 확인
+    # 예: SS-400-1-4, SS-810-6, B-400-9, SS-1610-P
+    swagelok_pattern = r'\b(SS|S|B|A|M)-([0-9]{3,4}|[0-9]+M[0-9]*)-([0-9A-Z])'
+    if re.search(swagelok_pattern, desc_upper):
+        return {
+            "category": "TUBE_FITTING",
+            "confidence": 0.98,
+            "evidence": ["SWAGELOK_PART_NO"],
+            "classification_level": "LEVEL0_PARTNO",
+            "reason": "Swagelok 스타일 파트넘버 감지"
+        }
     
     # 쉼표로 구분된 각 부분을 별도로 체크 (예: "NIPPLE, SMLS, SCH 80")
     desc_parts = [part.strip() for part in desc_upper.split(',')]
@@ -351,4 +298,4 @@ def should_exclude_material(description: str) -> bool:
     ]
     
     desc_upper = description.upper()
-    return any(keyword in desc_upper for keyword in exclude_keywords) 
+    return any(keyword in desc_upper for keyword in exclude_keywords)