Hide loan application form automatically

app.py

@@ -92,6 +92,10 @@ with st.sidebar:
     client = Client(**client_options)
 
 
+if "expander_toggle" not in st.session_state:
+    st.session_state.expander_toggle = True
+
+
 def submit_and_clear():
     try:
         # Call the explain endpoint as it also includes the prediction
@@ -102,6 +106,7 @@ def submit_and_clear():
         st.session_state.eval_selected = False
         st.session_state.predict_button_clicked = False
         st.session_state.eval_selected = False
+        show_expander()
     except Exception as e:
         logging.error(e)
         st.error(
@@ -112,9 +117,15 @@ def submit_and_clear():
         st.toast(f"Failed to submit feedback: {e}")
 
 
+def hide_expander():
+    st.session_state.expander_toggle = False
+
+def show_expander():
+    st.session_state.expander_toggle = True
+
 # Attributes
 st.subheader("Loan Application")
-with st.expander("Application form", expanded=False):
+with st.expander("Application form", expanded=st.session_state.expander_toggle):
     # Split view in 2 columns
     col1, col2 = st.columns(2)
     with col1:
@@ -194,7 +205,7 @@ if "predict_button_clicked" not in st.session_state:
     st.session_state.predict_button_clicked = False
 if deployment_token != "my-secret-token":
     predict_button = st.button(
-        "Predict", key="predict_button", help="Click to get the AI prediction."
+        "Predict", key="predict_button", help="Click to get the AI prediction.", on_click=hide_expander
     )
     if predict_button:
         st.session_state.predict_button_clicked = True