Upload app.py

app.py

@@ -75,6 +75,23 @@ def _host_matches_any(host: str, known: List[str]) -> bool:
     return False
 
 
+_URL_EXTRACT_RE = re.compile(r"(https?://[^\s<>\"'\)\]]+)", re.IGNORECASE)
+
+def _sanitize_input_url(text: str) -> str:
+    v = (text or "").strip()
+    if v.startswith("@"):
+        v = v.lstrip("@").strip()
+    m = _URL_EXTRACT_RE.search(v)
+    if m:
+        v = m.group(1)
+    v = v.strip("<>[]()")
+    return v
+
+_SCHEME_RE = re.compile(r"^[a-zA-Z][a-zA-Z0-9+\-.]*://")
+def _ensure_scheme(u: str) -> str:
+    u = (u or "").strip()
+    return u if _SCHEME_RE.match(u) else ("http://" + u)
+
 def _read_urls_from_csv(path: str) -> List[str]:
     urls: List[str] = []
     try:
@@ -144,7 +161,7 @@ def _engineer_features(urls: List[str], feature_cols: List[str]) -> pd.DataFrame
     out["ends_with_zip"] = s.str.endswith(".zip").astype(int)
 
     # Host/SLD/TLD derived features used by newer models
-    hosts = s.apply(lambda x: (urlparse(x).hostname or "").lower())
+    hosts = s.apply(lambda x: (urlparse(_ensure_scheme(x)).hostname or "").lower())
     out["host_len"] = hosts.str.len().fillna(0)
 
     # Subdomain count: number of labels minus 2 (for sld.tld); never below 0
@@ -282,7 +299,8 @@ def predict_url(payload: PredictUrlPayload):
         url_col: str = bundle.get("url_col") or "url"
         model_type: str = bundle.get("model_type") or ""
 
-        url_str = (payload.url or "").strip()
+        raw_input = (payload.url or "").strip()
+        url_str = _sanitize_input_url(raw_input)
         if not url_str:
             return JSONResponse(status_code=400, content={"error": "Empty url"})
 
@@ -309,7 +327,7 @@ def predict_url(payload: PredictUrlPayload):
             }
 
         # Known-host override (suffix match)
-        host = (urlparse(url_str).hostname or "").lower()
+        host = (urlparse(_ensure_scheme(url_str)).hostname or "").lower()
         if host and host_map:
             for h, lbl in host_map.items():
                 if _host_matches_any(host, [h]):