import streamlit as st
from chatbot.chatbot import chatbot_response

st.title("Clinical Readmission Assistant")

note = st.text_area("Paste Discharge Summary")
risk_score = st.slider("Predicted Readmission Risk", 0.0, 1.0, 0.65)

question = st.text_input("Ask a question")

if st.button("Ask"):
    answer = chatbot_response(question, note, risk_score)
    st.write(answer)