import os
import joblib

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
MODEL_PATH = os.path.join(BASE_DIR, "artifacts", "questioning_model.joblib")


def main():
    model = joblib.load(MODEL_PATH)

    tests = [
        "How does this system work?",
        "This document explains the system architecture.",
        "Can you explain?",
        "The paper describes the methodology."
    ]

    for t in tests:
        pred = model.predict(t)
        label = "Questioning" if pred == 1 else "Declarative"
        print(f"{label} → {t}")


if __name__ == "__main__":
    main()
