Update app.py

app.py

@@ -628,12 +628,16 @@ def show_tabular_data(df, co, country_code):
     )
     if st.button("Click here to calculate relevancy for selected pages"):
         selected_indices = [i for i, selected in enumerate(st.session_state.selected_rows) if selected]
-        with st.spinner('Calculating relevancy scores...'):
-            for index in selected_indices:
-                if pd.isna(df.iloc[index]['relevancy_score']) or df.iloc[index]['relevancy_score'] == 0:
-                    df.iloc[index, df.columns.get_loc('relevancy_score')] = calculate_single_relevancy(df.iloc[index])
-        st.success(f"Calculated relevancy scores for {len(selected_indices)} selected rows.")
-        st.rerun()
+        if selected_indices:
+            progress_bar = st.progress(0)
+            for i, index in enumerate(selected_indices):
+                if pd.isna(df.at[index, 'relevancy_score']) or df.at[index, 'relevancy_score'] == 0:
+                    df.at[index, 'relevancy_score'] = calculate_single_relevancy(df.iloc[index])
+                progress_bar.progress((i + 1) / len(selected_indices))
+            st.success(f"Calculated relevancy scores for {len(selected_indices)} selected rows.")
+            st.session_state.report_data = df  # Update the report_data in session state
+        else:
+            st.warning("No rows selected. Please select at least one row to calculate relevancy.")
     
     # Display column headers
     cols = st.columns([0.5, 3, 2, 1, 1, 1, 1, 1, 1])
@@ -645,10 +649,9 @@ def show_tabular_data(df, co, country_code):
     for i, row in enumerate(df.iloc[start_idx:end_idx].itertuples(), start=start_idx):
         cols = st.columns([0.5, 3, 2, 1, 1, 1, 1, 1, 1])
         
-        # Checkbox for row selection with a label
+        # Checkbox for row selection
         cols[0].checkbox("Select", key=f"select_{i}", value=st.session_state.selected_rows[i],
-                         on_change=lambda idx=i: setattr(st.session_state, 'selected_rows', 
-                                                       [True if j == idx else x for j, x in enumerate(st.session_state.selected_rows)]))
+                         on_change=lambda idx=i: update_selected_rows(idx))
         
         # Truncate and make the URL clickable
         truncated_url = row.page[:30] + '...' if len(row.page) > 30 else row.page
@@ -731,6 +734,9 @@ def show_tabular_data(df, co, country_code):
 
     return df  # Return the updated dataframe
 
+def update_selected_rows(idx):
+    st.session_state.selected_rows[idx] = not st.session_state.selected_rows[idx]
+
 def show_date_range_selector():
   #  logging.info("Showing date range selector")
     return st.selectbox(
@@ -883,7 +889,8 @@ def main():
             if st.session_state.report_data is not None and not st.session_state.report_data.empty:
                 st.write("Data fetched successfully.")
                 
-                st.session_state.report_data = show_tabular_data(st.session_state.report_data, co, country_code)
+                updated_df = show_tabular_data(st.session_state.report_data, co, country_code)
+                st.session_state.report_data = updated_df  # Update the report_data with the potentially modified dataframe
                 
                 download_csv_link(st.session_state.report_data)
             elif st.session_state.report_data is not None: