wrote response to foreground

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-# from mental_health_raqa import mh_assistant
+from mental_health_raqa import mh_assistant
 
 #---------------------------------#
 import pandas as pd
@@ -63,18 +63,17 @@ def mh_assistant(openai_key,query):
 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')
-# key_submit = st.button('Submit')
-# # Display the key when the user presses the 'Submit' button
-# if key_submit and openai_key:
-query = st.text_input('Enter your query', type='default')
-if query:
-    try:
-        # with st.spinner('Processing your query...'):
-        response = mh_assistant("jbkuyb",query)
-        print("response",response)
-        st.write(response)
-    except Exception as e:
-        st.error(f'An error occurred: {e}',icon=':no_entry_sign:')
-# elif key_submit and not openai_key:
-#     st.error('Please enter your OpenAI key',icon="🚨")
+openai_key = st.text_input('Enter your OpenAI Key', type='password')
+key_submit = st.button('Submit')
+# Display the key when the user presses the 'Submit' button
+if key_submit and openai_key:
+    query = st.text_input('Enter your query', type='default')
+    if query:
+        try:
+            with st.spinner('Processing your query...'):
+                response = mh_assistant(openai_key,query)
+                st.write(response)
+        except Exception as e:
+            st.error(f'An error occurred: {e}',icon=':no_entry_sign:')
+elif key_submit and not openai_key:
+    st.error('Please enter your OpenAI key',icon="🚨")