Update my_pages/information_loss.py

my_pages/information_loss.py

@@ -3,15 +3,15 @@ import streamlit as st
 from utils import add_navigation, add_instruction_text
 
 ALL_FEATURES = [
-    ("Liquid Assets", "Liquid Assets", "Added"), 
-    ("Illiquid Assets", "Property appraisals, Insurance valuations, etc.", 
+    ("Liquid Assets", ["Liquid Assets"], "Added"), 
+    ("Illiquid Assets", ["Property appraisals, Insurance valuations, etc."], 
      "Sorry, it is not possible to get the precise value of illiquid assets. We will use some approximate alternatives instead."),
-    ("Current Debts", "Current Debts", "Added"),
-    ("Income Stability", "Past Income Stability",
+    ("Current Debts", ["Current Debts"], "Added"),
+    ("Income Stability", ["Past Income Stability"],
      "Sorry, it is not possible to know precisely how stable someone's income will be in future. Until someone invents a time machine (we're hopeful!), we will instead use past income stability."),
-    ("Health Trajectory", "Current Health Indicators", 
+    ("Health Trajectory", ["Current Health Indicators"], 
      "Sorry, it is not possible to know precisely how will someone's health change in the future. Until someone invents a time machine (we're hopeful!), we will instead use current health indicators."),
-    ("Financial Discipline", "Credit Score", 
+    ("Financial Discipline", ["Credit Score"], 
      "Sorry, financial discipline is not a directly measurable quality. We will use credit score as an approximation instead.")
 ]
 
@@ -63,7 +63,7 @@ def render():
                 _ = st.button(feature[0], type="tertiary")
             else:
                 if st.button(feature[0]):
-                    st.session_state.selected_features.append(feature)
+                    st.session_state.selected_features.extend(feature[1])
                     st.session_state.show_message = feature[2]
                     st.rerun()
 
@@ -74,13 +74,17 @@ def render():
         )
         del st.session_state.show_message
 
-    # st.markdown("---")
-
-    # # Bottom half: dataset table
-    # st.markdown("### Current Dataset")
-    # if st.session_state.selected_features:
-    #     cols = st.session_state.selected_features + ["Target Label"]
-    #     df = pd.DataFrame(columns=cols)
-    #     st.dataframe(df, use_container_width=True)
-    # else:
-    #     st.info("No features added yet. Click a feature above to add it to the dataset.")
+    #### Selected features shown at the bottom
+    st.markdown("---")
+    
+    st.markdown("### Set of Selected Features:")
+    if st.session_state.selected_features:
+        col1, col2 = st.columns(2)
+        
+        for i, item in enumerate(st.session_state.selected_features):
+            if i % 2 == 0:
+                col1.markdown(f"- {item}")
+            else:
+                col2.markdown(f"- {item}")
+    else:
+        st.info("No features added yet. Click a feature above to add it to the dataset.")