import streamlit as st
from model_predictor import predict_next_service

st.title("Predictive AMC Notifier")
equipment_id = st.text_input("Equipment ID")
last_service = st.date_input("Last Service Date")

if st.button("Predict"):
    if equipment_id and last_service:
        date = predict_next_service(str(last_service))
        st.success(f"Next Predicted Service Date: {date}")
    else:
        st.warning("Enter Equipment ID and Date")