import joblib

vectorizer = joblib.load("xss_vectorizer.joblib")
clf = joblib.load("xss_classifier.joblib")

print("Type text to classify. Ctrl+C or empty line to quit.\n")
while True:
    text = input("> ").strip()
    if not text:
        break
    X = vectorizer.transform([text])
    pred = clf.predict(X)[0]
    verdict = "MALICIOUS" if pred == 1 else "benign"
    print(f"[{verdict}] {text}\n")