from joblib import load
from datetime import datetime, timedelta

# Load pretrained model
model = load("amc_model.pkl")

def predict_next_service(last_service_date):
    today = datetime.today()
    delta_days = (today - datetime.strptime(last_service_date, "%Y-%m-%d")).days
    predicted_days = int(model.predict([[delta_days]])[0])
    next_service_date = today + timedelta(days=predicted_days)
    return next_service_date.strftime("%Y-%m-%d")