Update app.py

app.py

@@ -10,13 +10,18 @@ query = st.text_input('Enter your query', type='default')
 
 submit = st.button('Submit')
 
-st.write(openai_key)
-
-
 
 if submit:
-    response = yg_assistant(openai_key,query)
-    st.write(response)
+   if query and openai_key:
+        try:
+            with st.spinner('Processing your query...'):
+                response = yg_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 and Query both!',icon="🚨")
+