from __future__ import annotations

import argparse
from pathlib import Path

from timeout_risk_model import load_model, predict_timeout_probability


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="Predict timeout probability for one order time.")
    parser.add_argument("order_time", help="Order time, for example: 2026-05-24 18:30")
    parser.add_argument("--product-name", default="", help="Optional product name.")
    parser.add_argument(
        "--model",
        type=Path,
        default=Path(__file__).resolve().parent / "timeout_model.json",
        help="Path to timeout_model.json.",
    )
    return parser.parse_args()


def main() -> None:
    args = parse_args()
    model = load_model(args.model)
    result = predict_timeout_probability(model, args.order_time, args.product_name)
    print(f"order_time: {result['order_time']}")
    print(f"probability: {result['probability']:.1%}")
    print(f"risk_level: {result['risk_level']}")
    print(f"mode: {result['mode']}")
    print(f"calibration: {result['calibration']}")


if __name__ == "__main__":
    main()