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)}")