Update handler.py

handler.py

@@ -89,34 +89,31 @@ class EndpointHandler:
         print("[INFO] Model loaded successfully")
 
         # ============================================
-        # Threshold Config (tunable)
+        # Threshold Config — TIGHTENED
         # ============================================
         self.config = {
-            "HIGH_ATTACK": 0.75,      # attack type softmax threshold
-            "MEDIUM_ATTACK": 0.55,    # attack type softmax threshold
-            "HIGH_CONF": 0.85,        # binary head dangerous threshold
-            "LOW_CONF": 0.30,         # binary head safe threshold
-            "EXTREME_CONF": 0.95      # override for medium attack scores
+            "HIGH_ATTACK": 0.80,      # ← RAISED from 0.75
+            "MEDIUM_ATTACK": 0.55,
+            "HIGH_CONF": 0.85,
+            "LOW_CONF": 0.30,
+            "EXTREME_CONF": 0.95
         }
 
     # =====================================================
-    # Decision Logic — FIXED
+    # Decision Logic
     # =====================================================
     def decide(self, danger_prob, fine_score, family_score):
         """
         Binary head (danger_prob) is the authority.
         Attack type scores only matter if binary head is already suspicious.
-        This prevents false positives where softmax picks a random attack class
-        on benign input.
         """
         cfg = self.config
 
-        # 1. Definite safe zone — binary head is confident it's safe
+        # 1. Definite safe zone
         if danger_prob <= cfg["LOW_CONF"]:
             return False
 
         # 2. Danger zone — binary head must be confident it's dangerous
-        #    BEFORE we even look at attack type scores
         if danger_prob >= cfg["HIGH_CONF"]:
             # Strong attack type confidence → dangerous
             if fine_score >= cfg["HIGH_ATTACK"] or family_score >= cfg["HIGH_ATTACK"]:
@@ -127,8 +124,7 @@ class EndpointHandler:
                 if danger_prob >= cfg["EXTREME_CONF"]:
                     return True
 
-        # 3. Gray zone (LOW_CONF < danger_prob < HIGH_CONF) → always safe
-        #    This catches miscalibrated binary head outputs
+        # 3. Gray zone — always safe
         return False
 
     # =====================================================
@@ -185,7 +181,6 @@ class EndpointHandler:
             else:
                 attack_type = "none"
                 attack_family = "none"
-                # Zero out misleading scores when safe
                 fine_score = 0.0
                 family_score = 0.0
 
@@ -204,15 +199,13 @@ class EndpointHandler:
         ]
 
         # ================================
-        # Response — FIXED confidence reporting
+        # Response
         # ================================
         return {
             "status": "DANGEROUS" if is_dangerous else "SAFE",
 
-            # Raw binary head probability — always meaningful
             "binary_confidence": round(danger_prob, 4),
 
-            # Decision confidence: how confident we are in the final status
             "confidence": round(
                 danger_prob if is_dangerous else (1 - danger_prob),
                 4