import streamlit as st
from mental_health_raqa import mh_assistant


st.title('Mental Health Assistant :broken_heart:')

# Create a text input box for the OpenAI key
openai_key = st.text_input('Enter your OpenAI Key', type='password')

# Display the key when the user presses the 'Submit' button
if st.button('Submit'):
    if openai_key:
        query = st.text_input('Enter your query', type='default')
        if st.button('Ask'):
            try:
                response = mh_assistant(openai_key,query)
                st.write(response)
            except Exception as e:
                st.error(f'An error occurred: {e}',icon=':no_entry_sign:')
    else:
        st.error('Please enter your OpenAI key',icon=':no_entry_sign:')