import openai 
import streamlit as st
from health_insurance_bot import chatbot_response 
st.title("Health Insurance Chatbot")
if "messages" not in st.session_state: 
   st.session_state["messages"] = 
   [{"role": "assistant", "content": "How can I help you?"}] 

for msg in st.session_state.messages: 
   st.chat_message(msg["role"]).write(msg["content"]) 

if prompt := st.chat_input(): 
  openai.api_key = st.secrets['openai_api_key']     
  st.session_state.messages.append({"role": "user", "content": prompt}) 
  st.chat_message(name="user").write(prompt) 
  response = chatbot_response(prompt)   
  st.session_state.messages.append({"role": "assistant", "content": response})  
  st.chat_message(name="assistant").write(response)