Update pages/chapter_index.py

pages/chapter_index.py

@@ -1,14 +1,35 @@
 import streamlit as st
-import pandas as pd
+import pandas as pd 
+#import numpy as np
 
 #st.title("📘SBS mapper")
 #st.write("Map internal descriptions to SBS codes in the below chapters* (work in progress)")
 #st.image("images/SBS_Chapter_Index.png", use_container_width=True) 
-st.write("Select specific topic for quicker results")
+st.header("Select specific topic for quicker results")
 df_chapters = pd.read_csv("SBS_V2_Chapter_Index.csv")
 df_chapters.iloc[25] = {"Chapter": "ALL", "Chapter title": "ALL TOPICS (will take more time)", "Blocks": "Total", "No. Codes": 10081}
 st.dataframe(df_chapters, hide_index=True)
 
+def dataframe_with_selections(df: pd.DataFrame, init_value: bool = False) -> pd.DataFrame:
+    df_with_selections = df.copy()
+    df_with_selections.insert(0, "Select", init_value)
+
+    # Get dataframe row-selections from user with st.data_editor
+    edited_df = st.data_editor(
+        df_with_selections,
+        hide_index=True,
+        column_config={"Select": st.column_config.CheckboxColumn(required=True)},
+        disabled=df.columns,
+    )
+
+    # Filter the dataframe using the temporary column, then drop the column
+    selected_rows = edited_df[edited_df.Select]
+    return selected_rows.drop('Select', axis=1)
+
+selection = dataframe_with_selections(df)
+st.write("Your selection:")
+st.write(selection)
+
 st.subheader("* Demo currently limited to Chapter 20: Imaging services")
 
 #'Procedures on nervous system'