Hide submit button and comment box after submission

app.py

@@ -42,7 +42,7 @@ def send_evaluation(
 ):
     """Send evaluation to Deeploy."""
     try:
-        with st.spinner("Submitting response..."):
+        with st.spinner("Submitting evaluation..."):
             # Call the explain endpoint as it also includes the prediction
             client.evaluate(
                 deployment_id, request_log_id, prediction_log_id, evaluation_input
@@ -52,7 +52,7 @@ def send_evaluation(
         logging.error(e)
         st.error(
             "Failed to submit feedback."
-            + "Check whether you are using the right model URL and Token. "
+            + "Check whether you are using the right model URL and token for evaluations. "
             + "Contact Deeploy if the problem persists."
         )
         st.write(f"Error message: {e}")
@@ -144,10 +144,10 @@ with st.expander("Application form", expanded=False):
             "Working hours per week", min_value=0, max_value=100, value=40
         )
         capital_gain = st.number_input(
-            "Yearly income", min_value=0, max_value=1000000, value=70000
+            "Yearly income [€]", min_value=0, max_value=1000000, value=70000
         )
         capital_loss = st.number_input(
-            "Yearly expenditures", min_value=0, max_value=1000000, value=60000
+            "Yearly expenditures [€]", min_value=0, max_value=1000000, value=60000
         )
     data_df = pd.DataFrame(
         [
@@ -207,8 +207,9 @@ if deployment_token != "my-secret-token":
 
 if st.session_state.selected == "Loan Decision":
     # If no prediction, show "loading..."
+    # FIXME: Cleanup session state so that it's not sending a request to the explain endpoint each time
     try:
-        with st.spinner("Loading..."):
+        with st.spinner("Loading prediction and explanation..."):
             # Call the explain endpoint as it also includes the prediction
             exp = client.explain(
                 request_body=request_body, deployment_id=deployment_id
@@ -361,13 +362,13 @@ if st.session_state.selected == "Loan Decision":
             )
             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
+        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
             }
+            placeholder = "Income is sufficient, given applicant's background"
         if no_button:
             st.session_state.eval_selected = True
             desired_output = not predictions[0]
@@ -375,29 +376,46 @@ if st.session_state.selected == "Loan Decision":
                 "result": 1,  # Disagree with the prediction
                 "value": {"predictions": [desired_output]},
             }
+            placeholder = "Income is too low, given applicant's background"
 
-        success = False
+        if 'success' not in st.session_state:
+            st.session_state['success'] = False
         if st.session_state.eval_selected:
-            comment = st.text_input("Would you like to add a comment?")
+            comment = st.text_input("Would you like to add a comment?", placeholder=placeholder)
             if comment:
                 st.session_state.evaluation_input["explanation"] = comment
             logging.debug(
                 "Selected feedback:" + str(st.session_state.evaluation_input)
             )
-        if st.button("Submit", key="submit_button"):
-            # FIXME: Hide comment box after submission
-            st.session_state.eval_selected = False
-            yes_button = False
-            no_button = False
-            success = send_evaluation(
-                client,
-                deployment_id,
-                request_log_id,
-                prediction_log_id,
-                st.session_state.evaluation_input,
-            )
-        if success:
-            st.session_state.eval_selected = False
+            success = True
+            print(success)
+            if st.button("Submit", key="submit_button"):
+                with st.spinner("Submitting evaluation..."):
+                    try:
+                        # Call the explain endpoint as it also includes the prediction
+                        client.evaluate(
+                            deployment_id, request_log_id, prediction_log_id, evaluation_input
+                        )
+                    except Exception as e:
+                        success = False
+                        logging.error(e)
+                        st.error(
+                            "Failed to submit feedback."
+                            + "Check whether you are using the right model URL and token for evaluations. "
+                            + "Contact Deeploy if the problem persists."
+                        )
+                        st.write(f"Error message: {e}")
+                    print(success)
+                if success:
+                    st.session_state.success = True
+                else:
+                    st.error("Evaluation failed. Check whether your token allows for evaluations.")
+                # st.session_state.eval_selected = False
+                # yes_button = False
+                # no_button = False
+                # print(st.session_state.success)
+        # print(st.session_state.success)
+        if st.session_state.success:
             st.success("Feedback submitted successfully.")
 
     except Exception as e: