got rid of error setting session state

app.py

@@ -5,13 +5,13 @@ from deeploy import Client
 from utils import get_request_body, get_fake_certainty, get_model_url, get_random_suspicious_transaction
 from utils import get_explainability_texts, get_explainability_values, send_evaluation, get_comment_explanation
 from utils import COL_NAMES, feature_texts
-from utils import create_data_input_table, create_table, ChangeButtonColour
+from utils import create_data_input_table, create_table, ChangeButtonColour, get_weights
 
 logging.basicConfig(level=logging.INFO)
 
 st.set_page_config(layout="wide")
 
-st.title("Money Laundering System")
+st.title("Smart AML:tm:")
 st.divider()
 
 data = pd.read_pickle("data/preprocessed_data.pkl")
@@ -27,7 +27,6 @@ if "disabled" not in st.session_state:
     st.session_state.disabled = False
 
 def disabled():
-    # st.session_state.predict_button_clicked = True
     st.session_state.disabled = True
 
 def rerun():
@@ -37,9 +36,6 @@ def rerun():
 def submitted_disabled():
     st.session_state.submitted_disabled = True
 
-if 'predict_button' not in st.session_state:
-    st.session_state.predict_button = False
-
 st.markdown("""
     <style>
         [data-testid=stSidebar] {
@@ -60,8 +56,8 @@ with st.sidebar:
             "Please enter Deeploy API token."
         )
     else: 
-        st.button("Get suspicious transaction", key="predict_button", help="Click to get a suspicious transaction", use_container_width=True,  on_click=disabled, disabled=st.session_state.disabled
-    ) #on_click=lambda: st.experimental_rerun()
+        st.button("Get suspicious transaction", key="predict_button", help="Click to get a suspicious transaction", use_container_width=True, on_click=disabled, disabled=st.session_state.disabled
+    ) #on_click=lambda: st.experimental_rerun() 
         ChangeButtonColour("Get suspicious transaction", '#FFFFFF', "#00052D")#'#FFFFFF', "#00052D"
 
 
@@ -73,11 +69,12 @@ with st.sidebar:
     }
     client = Client(**client_options)
 
+if 'predict_button' not in st.session_state:
+    st.session_state.predict_button = False
+
 if st.session_state.predict_button: #  and not st.session_state.predict_button_clicked
     st.session_state.predict_button_clicked = True
 
-# if 'explanation' not in st.session_state:
-#     st.session_state.explanation = ""
 if 'got_explanation' not in st.session_state:
     st.session_state.got_explanation = False
 
@@ -123,14 +120,18 @@ if st.session_state.got_explanation:
         create_data_input_table(datapoint_pd, COL_NAMES)
 
     with col2:
-        st.subheader('Prediction Explanation: ')
-        # st.metric(label='#### Model Certainty', value=certainty)
-        st.markdown('#### Model Certainty')
-        st.success(f'{certainty}')
+        st.subheader('AML Model Hit')
+        # st.success(f'{certainty}')
+        # st.metric(label='Model Certainty', value=certainty)
+        # style_metric_cards(border_left_color='#00052D', box_shadow=False)
+        # # st.markdown('#### Model Certainty')
+        st.metric(label='Model Certainty', value=certainty)
+        
 
         explainability_texts, sorted_indices = get_explainability_texts(shap_values, feature_texts)
+        weights = get_weights(shap_values, sorted_indices)
         explainability_values = get_explainability_values(sorted_indices, datapoint_pd)
-        create_table(explainability_texts, explainability_values, 'Important Suspicious Variables: ')
+        create_table(explainability_texts, explainability_values, weights, 'Important Suspicious Factors')
 
     st.subheader("")
     # st.markdown("<h2 style='text-align: center; white: red;'>Evaluation</h2>", unsafe_allow_html=True)
@@ -143,7 +144,7 @@ if st.session_state.got_explanation:
     with col3:
         eval1 = st.empty()
         eval1.button("Send to FIU", key="yes_button", use_container_width=True, disabled=st.session_state.submitted_disabled)
-        ChangeButtonColour("Send to FIU", '#FFFFFF', "#DD360C")
+        ChangeButtonColour("Send to FIU", '#FFFFFF', '#4C506C') #'#FFFFFF', "#DD360C"
         st.session_state.yes_button_clicked = False
 
     if st.session_state.yes_button:
@@ -155,7 +156,7 @@ if st.session_state.got_explanation:
     with col4:
         eval2 = st.empty()
         eval2.button("Not money laundering", key="no_button", use_container_width=True, disabled=st.session_state.submitted_disabled)
-        ChangeButtonColour("Not money laundering", '#FFFFFF', "#46B071")
+        ChangeButtonColour("Not money laundering", '#FFFFFF', '#4C506C') # '#FFFFFF', "#46B071",   '#FFFFFF', "#666666"
         st.session_state.no_button_clicked = False
 
 
@@ -193,6 +194,6 @@ if st.session_state.got_explanation:
         eval1.empty()
         eval2.empty()
         eval3.empty()
-        st.success("Feedback submitted successfully.")
+        st.warning("Feedback submitted successfully")
         st.button("Next", key='next', use_container_width=True, on_click=rerun)
-        ChangeButtonColour("Next", '#FFFFFF', "#F9B917")
+        ChangeButtonColour("Next", '#FFFFFF', "#00052D") #'#FFFFFF', #F9B917" "#DD360C  #457EA4