Show prediction after evaluation

app.py

@@ -95,6 +95,8 @@ with st.sidebar:
 if "expander_toggle" not in st.session_state:
     st.session_state.expander_toggle = True
 
+if "evaluation_submitted" not in st.session_state:
+    st.session_state.evaluation_submitted = False
 
 def submit_and_clear():
     try:
@@ -104,9 +106,8 @@ def submit_and_clear():
         )
         st.toast(":green[Feedback submitted successfully.]", icon="✅")
         st.session_state.eval_selected = False
-        st.session_state.predict_button_clicked = False
+        st.session_state.evaluation_submitted = True
         st.session_state.eval_selected = False
-        show_expander()
     except Exception as e:
         logging.error(e)
         st.error(
@@ -203,12 +204,14 @@ with st.expander("Application form", expanded=st.session_state.expander_toggle):
     }
 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(
         "Send loan application", key="predict_button", help="Click to get the AI prediction.", on_click=hide_expander
     )
     if predict_button:
         st.session_state.predict_button_clicked = True
+        st.session_state.evaluation_submitted = False
         # st.session_state.selected = "Loan Decision"
         with st.spinner("Loading prediction and explanation..."):
             # Call the explain endpoint as it also includes the prediction
@@ -342,55 +345,58 @@ if st.session_state.predict_button_clicked:
                 )
 
         st.divider()
-        
-        # Add prediction evaluation
-        st.subheader("Evaluation: Do you agree with the predicted creditworthiness?")
-        st.info(
-            "AI model predictions always come with a certain level of uncertainty.  \nEvaluate the correctness of the prediction based on your expertise and experience."
-        )
-        cols = st.columns(4)
-        col_yes, col_no = cols[:2]
-        with col_yes:
-            yes_button = st.button(
-                "Yes, I agree",
-                key="yes_button",
-                use_container_width=True,
-                help="Click if you agree with the prediction",
-            )
-            ChangeButtonColour("Yes, I agree", "white", "green")
-        with col_no:
-            no_button = st.button(
-                "No, I disagree",
-                key="no_button",
-                use_container_width=True,
-                help="Click if you disagree with the prediction",
-                type="primary",
+
+        if not st.session_state.evaluation_submitted:            
+            # Add prediction evaluation
+            st.subheader("Evaluation: Do you agree with the predicted creditworthiness?")
+            st.info(
+                "AI model predictions always come with a certain level of uncertainty.  \nEvaluate the correctness of the prediction based on your expertise and experience."
             )
-            ChangeButtonColour("No, I disagree", "white", "#DD360C")
-            # ChangeButtonColour("No, I disagree", "#DD360C", "#F0F0F0")
-        if "eval_selected" not in st.session_state:
-            st.session_state["eval_selected"] = False
-        if yes_button:
-            st.session_state.eval_selected = True
-            st.session_state.evaluation_input = {
-                "result": 0  # Agree with the prediction
-            }
-            st.session_state.placeholder = "Income is sufficient, given applicant's background"
-        if no_button:
-            st.session_state.eval_selected = True
-            desired_output = not predictions[0]
-            st.session_state.evaluation_input = {
-                "result": 1,  # Disagree with the prediction
-                "value": {"predictions": [desired_output]},
-            }
-            st.session_state.placeholder = "Income is too low, given applicant's background"
-
-        if st.session_state.eval_selected:
-            comment = st.text_input("Would you like to add a comment?", placeholder=st.session_state.placeholder)
-            if comment:
-                st.session_state.evaluation_input["explanation"] = comment
-
-            st.button("Submit", key="submit_button", on_click=submit_and_clear)                    
+            cols = st.columns(4)
+            col_yes, col_no = cols[:2]
+            with col_yes:
+                yes_button = st.button(
+                    "Yes, I agree",
+                    key="yes_button",
+                    use_container_width=True,
+                    help="Click if you agree with the prediction",
+                )
+                ChangeButtonColour("Yes, I agree", "white", "green")
+            with col_no:
+                no_button = st.button(
+                    "No, I disagree",
+                    key="no_button",
+                    use_container_width=True,
+                    help="Click if you disagree with the prediction",
+                    type="primary",
+                )
+                ChangeButtonColour("No, I disagree", "white", "#DD360C")
+                # ChangeButtonColour("No, I disagree", "#DD360C", "#F0F0F0")
+            if "eval_selected" not in st.session_state:
+                st.session_state["eval_selected"] = False
+            if yes_button:
+                st.session_state.eval_selected = True
+                st.session_state.evaluation_input = {
+                    "result": 0  # Agree with the prediction
+                }
+                st.session_state.placeholder = "Income is sufficient, given applicant's background"
+            if no_button:
+                st.session_state.eval_selected = True
+                desired_output = not predictions[0]
+                st.session_state.evaluation_input = {
+                    "result": 1,  # Disagree with the prediction
+                    "value": {"predictions": [desired_output]},
+                }
+                st.session_state.placeholder = "Income is too low, given applicant's background"
+
+            if st.session_state.eval_selected:
+                comment = st.text_input("Would you like to add a comment?", placeholder=st.session_state.placeholder)
+                if comment:
+                    st.session_state.evaluation_input["explanation"] = comment
+
+                st.button("Submit", key="submit_button", on_click=submit_and_clear)
+        else:
+            st.write("Loan application already evaluated - send another loan application to evaluate again.")                    
 
     except Exception as e:
         logging.error(e)