added coloured evaluation buttons

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 import logging
 from deeploy import Client
+from streamlit_extras.stylable_container import stylable_container
 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
 from utils import COL_NAMES, feature_texts
@@ -11,6 +12,9 @@ logging.basicConfig(level=logging.INFO)
 
 st.set_page_config(layout="wide")
 
+st.title("Money Laundering System")
+st.divider()
+
 data = pd.read_pickle("data/preprocessed_data.pkl")
 # data = data.drop('isFraud', axis=1)
 
@@ -60,8 +64,6 @@ with st.sidebar:
 
 if "predict_button" not in st.session_state:
     st.session_state.predict_button = False
-    st.title("Money Laundering System")
-    st.divider()
     st.info(
     "Fill in left hand side and click on button to observe a potential fraudulent transaction"
     )
@@ -81,6 +83,7 @@ if st.session_state.predict_button:
 
             create_data_input_table(datapoint_pd, COL_NAMES)
         with col2:
+            st.subheader('Prediction Explanation: ')
             certainty = get_fake_certainty()
             st.metric(label='#### Model Certainty', value=certainty)
 
@@ -88,30 +91,32 @@ if st.session_state.predict_button:
             explainability_values = get_explainability_values(sorted_indices, datapoint_pd)
             create_table(explainability_texts, explainability_values, 'Important Suspicious Variables: ')
 
-        
-        st.divider()
+        st.subheader("")
+        # st.markdown("<h2 style='text-align: center; white: red;'>Evaluation</h2>", unsafe_allow_html=True)
 
-    
-                # Add prediction evaluation
-        st.subheader("Prediction Evaluation")
-        st.write("Do you agree with the prediction?")
-
-        yes_button = st.button("Yes :thumbsup:", key="yes_button")
-        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
-            }
-        no_button = st.button("No :thumbsdown:", key="no_button")
-        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]},
-            # }
+        col3, col4 = st.columns(2)
+        
+        with col3:
+            st.button("Send to FIU", key="yes_button", use_container_width=True)
+            ChangeButtonColour("Send to FIU", '#FFFFFF', "#DD360C")
+            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
+            #     }
+        
+        with col4:
+            st.button("Not money laundering", key="no_button", use_container_width=True)
+            ChangeButtonColour("Not money laundering", '#FFFFFF', "#46B071")
+            # 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]},
+                # }
 
         success = False
         if st.session_state.eval_selected: