import streamlit as st
from inference.milo_wba_inference import generate_reply

st.title("🚀 Milo WhatsApp Business Assistant")

user_input = st.text_input("You:", placeholder="Type a WhatsApp-style message...")

if user_input:
    reply = generate_reply(user_input)
    st.markdown(f"**Milo:** {reply}")