import os
import joblib

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


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

    tests = [
        "You must complete this task today.",
        "If you don't mind, could you review this?",
        "Make sure the report is submitted.",
        "I think maybe we should wait."
    ]

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


if __name__ == "__main__":
    main()
