Update pages/chapter_index.py

pages/chapter_index.py

@@ -11,11 +11,6 @@ df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows.csv")
 
 startrowindex_list = df_chapters["from_row_index"].tolist()
 endrowindex_list = df_chapters["to_row_index"].tolist()
-#st.write("START num:", startrowindex_list)
-#st.write("END num:", endrowindex_list)
-#df_chapters['range_of_rows'] = df_chapters.apply(lambda row: list(range(row.min(), row.max()+1)), axis=1)
-#df_chapters['range_of_rows'] = range((s for s in startrowindex_list), (e for e in endrowindex_list))
-#df_chapters['range_of_rows'] = df_chapters.apply(lambda row: list(range((s for s in startrowindex_list), (e for e in endrowindex_list)), axis=1))
 allchapters_rows_list = []
 for s, e in zip(startrowindex_list, endrowindex_list):
     eachchapter_rows_list = list(range(s,e))
@@ -40,7 +35,9 @@ def dataframe_with_selections(df_chapters: pd.DataFrame, init_value: bool = Fals
 
 if "selected_chapters" not in st.session_state:
     st.session_state['selected_chapters'] = []
+    st.session_state['selected_rows'] = []
 selected_chapters_list = st.session_state.selected_chapters
+selected_rows_list = st.session_state.selected_rows
 
 selected_chapters = dataframe_with_selections(df_chapters)
 st.write("Your selection:")
@@ -50,11 +47,7 @@ st.write("SELECTED CHAPTERS: ", selected_chapters_list)
 selected_rows_list = selected_chapters.iloc[:,6].tolist() 
 st.write("SELECTED ROWS: ", selected_rows_list) 
 
-#for item in selected_chapters_list[]:
-#    #start_row = item[4], end_row = item[5]
-#    st.write("ZZZZZZZZZZZZZ: ", item)
-#selected_rows_list = [row for rows in selected_chapters_list.iloc[;,;][4]]
-
-
 if selected_chapters is not None:
-    st.session_state.selected_chapters = selected_chapters_list
+    st.session_state.selected_chapters = selected_chapters_list
+if selected_rows is not None:
+    st.session_state.selected_rows = selected_rows_list