| from joblib import load | |
| from datetime import datetime, timedelta | |
| model = load("amc_model.pkl") | |
| def predict_next_service(last_service_date): | |
| try: | |
| today = datetime.today() | |
| delta_days = (today - datetime.strptime(last_service_date, "%Y-%m-%d")).days | |
| predicted_days = int(model.predict([[delta_days]])[0]) | |
| return (today + timedelta(days=predicted_days)).strftime("%Y-%m-%d") | |
| except Exception as e: | |
| raise ValueError(f"Prediction error: {str(e)}") |