import streamlit as st


def main():
    st.title("Health Chatbot")
    
    if 'messages' not in st.session_state:
        st.session_state.messages = []
    

    user_input = st.chat_input("You", key="user_input")
    if user_input:
        st.session_state.messages.append({"sender": "user", "message": user_input})
        # display all the messages already in the session state
        for message in st.session_state.messages:
            if message["sender"] == "user":
                st.write("You: ", message["message"])
            else:
                st.write("Bot: ", message["message"])

if __name__ == "__main__":
    main()