Update my_pages/information_loss.py

my_pages/information_loss.py

@@ -65,7 +65,7 @@ def render():
                 _ = st.button(feature[0], type="tertiary")
             else:
                 if st.button(feature[0]):
-                    st.session_state.selected_features.extend(feature)
+                    st.session_state.selected_features.append(feature)
                     st.session_state.show_message = feature[2]
                     st.rerun()
 
@@ -77,16 +77,15 @@ def render():
         del st.session_state.show_message
 
     #### Selected features shown at the bottom
-    st.markdown("---")
+    display_features_list = []
+    for ele in st.session_state.selected_features:
+        display_features_list.extend(ele[1])
     
+    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[1]}")
-            else:
-                col2.markdown(f"- {item[1]}")
-    else:
-        st.info("No features added yet. Click a feature above to add it to the dataset.")
+    col1, col2 = st.columns(2)
+    for i, item in enumerate(display_features_list):
+        if i % 2 == 0:
+            col1.markdown(f"- {item[1]}")
+        else:
+            col2.markdown(f"- {item[1]}")