import streamlit as st
from model_predictor import predict_next_service

st.set_page_config(page_title="Predictive AMC Notifier", page_icon="🧪")
st.title("🧪 Predictive AMC Notification Engine")

equipment_id = st.text_input("Enter Equipment ID")
last_service = st.date_input("Select Last Service Date")

if st.button("Predict Next Service Date"):
    if equipment_id and last_service:
        predicted = predict_next_service(str(last_service))
        st.success(f"✅ Predicted Next AMC Service Date: **{predicted}**")
    else:
        st.warning("Please enter all fields.")