Update my_pages/information_loss.py

my_pages/information_loss.py

@@ -1,5 +1,6 @@
 import pandas as pd
 import streamlit as st
+from streamlit_extras.stylable_container import stylable_container
 from utils import go_to
 
 ALL_FEATURES = [
@@ -33,11 +34,12 @@ def render():
         Consider the automation of loan approval using AI models. <br>
         Choose the features below that you would like to collect to train your AI model. <br><br>
         </div>
-        ---
         """,
         unsafe_allow_html=True
     )
 
+    st.markdown("---")
+
     # Session state init
     # if "available_features" not in st.session_state:
     #     st.session_state.available_features = ALL_FEATURES.copy()
@@ -61,10 +63,17 @@ def render():
         for feature in ALL_FEATURES.copy():
             if feature in st.session_state.selected_features:
                 feature_name = feature[1]
-                _ = st.button(feature_name, key=feature_name, type="selected")
+                with stylable_container(
+                    feature_name,
+                    css_styles=f"""
+                    button {{
+                        background: green!important;            
+                    }}""",
+                ):
+                    _ = st.button(feature_name, type="selected")
             else:
                 feature_name = feature[0]
-                if st.button(feature_name, key=feature_name):
+                if st.button(feature_name):
                     st.session_state.selected_features.append(feature)
                     if feature[2] is not None:
                         st.session_state.show_message = (feature[2])