import os
import sys

ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, ROOT)

from skala import SkalaClassifier


def main() -> int:
    default = "I have this Python codebase, can you port it to C++"
    text = " ".join(sys.argv[1:]) or default
    classifier = SkalaClassifier(os.path.join(ROOT, "artifacts"))
    classifier.load()
    classifier.warmup()

    label, scores = classifier.classify(text)
    print(f"{text!r} -> {label}")
    for name in ("simple", "medium", "complex"):
        print(f"  {name}: {scores[name]:.3f}")

    return 0


if __name__ == "__main__":
    sys.exit(main())