Update main.py

main.py

@@ -243,21 +243,21 @@ async def evaluate_xgboost_url(payload: XGBoostRequest):
         features = extract_url_features(processed_url)
         dmatrix_payload = xgb.DMatrix(np.array([features]))
 
-        safeness_prob = float(bst.predict(dmatrix_payload)[0])
+        risk_prob = float(bst.predict(dmatrix_payload)[0])
         latency_ms = (time.perf_counter() - start_time) * 1000
 
-        if safeness_prob < 0.50:
+        if risk_prob >= 0.50:
             verdict = "🚨 PHISHING DETECTED"
-            confidence = (1.0 - safeness_prob) * 100
+            confidence = risk_prob * 100
         else:
             verdict = "✅ LEGITIMATE SAFE"
-            confidence = safeness_prob * 100
+            confidence = (1.0 - risk_prob) * 100
 
         return {
             "processed_url": processed_url,
             "verdict": verdict,
             "confidence_percentage": round(confidence, 2),
-            "raw_probability": safeness_prob,
+            "raw_risk_probability": risk_prob,
             "latency_ms": round(latency_ms, 3)
         }
     except Exception as e:
@@ -539,8 +539,7 @@ async def unified_scan_endpoint(payload: UnifiedRequest):
             features = extract_url_features(url)
             dmatrix_payload = xgb.DMatrix(np.array([features]))
 
-            raw_safeness = float(bst.predict(dmatrix_payload)[0])
-            v4_score = 1.0 - raw_safeness
+            v4_score = float(bst.predict(dmatrix_payload)[0])
             v4_msg = f"XGBoost Match: {round(v4_score * 100, 1)}% Phishing Risk"
     except Exception as e:
         v4_msg = f"V4 Error: {str(e)}"