Update app.py

app.py

@@ -475,6 +475,10 @@ def show_tabular_data(df, co):
     start_idx = (st.session_state.current_page - 1) * rows_per_page
     end_idx = start_idx + rows_per_page
     
+    # Initialize or update selected_rows in session state
+    if 'selected_rows' not in st.session_state or len(st.session_state.selected_rows) != len(df):
+        st.session_state.selected_rows = [False] * len(df)
+    
     # Add a "Calculate Relevancy" button at the top
     if st.button("Calculate Relevancy for Selected"):
         selected_indices = [i for i, selected in enumerate(st.session_state.selected_rows) if selected]
@@ -491,12 +495,8 @@ def show_tabular_data(df, co):
     for col, header in zip(cols, headers):
         col.write(f"**{header}**")
 
-    # Initialize selected_rows in session state if not present
-    if 'selected_rows' not in st.session_state:
-        st.session_state.selected_rows = [False] * len(df)
-
     # Display each row
-    for index, row in df.iloc[start_idx:end_idx].iterrows():
+    for i, (index, row) in enumerate(df.iloc[start_idx:end_idx].iterrows()):
         cols = st.columns([0.5, 3, 2, 1, 1, 1, 1, 1, 1])
         
         # Checkbox for row selection