Upload 7 files

app.py

@@ -83,6 +83,20 @@ _AUTOCALIB_LEGIT_URLS: List[str] = []
 _KNOWN_LEGIT_HOSTS: List[str] = []
 _KNOWN_PHISH_HOSTS: List[str] = []
 
+def _normalize_host(value: str) -> str:
+    v = value.strip().lower()
+    if v.startswith("www."):
+        v = v[4:]
+    return v
+
+def _host_matches_any(host: str, known: List[str]) -> bool:
+    base = _normalize_host(host)
+    for item in known:
+        k = _normalize_host(item)
+        if base == k or base.endswith("." + k):
+            return True
+    return False
+
 # -------------------------
 # CSV configuration support (optional)
 # -------------------------
@@ -440,9 +454,9 @@ def predict_url(payload: PredictUrlPayload):
         host = (urlparse(url_str).hostname or "").lower()
         if host:
             override_label: Optional[str] = None
-            if host in _KNOWN_LEGIT_HOSTS:
+            if _host_matches_any(host, _KNOWN_LEGIT_HOSTS):
                 override_label = "LEGIT"
-            elif host in _KNOWN_PHISH_HOSTS:
+            elif _host_matches_any(host, _KNOWN_PHISH_HOSTS):
                 override_label = "PHISH"
             if override_label is not None:
                 # Map numeric label according to resolved polarity