Update export logic in app.py: introduce a new session state variable to store a copy of the portfolio for export, ensuring accurate data handling when adding to the export base and improving display options for paginated data.

app.py

@@ -1022,6 +1022,7 @@ with tab2:
                 )
                 st.session_state['portfolio'] = st.session_state['portfolio'][lock_mask]
             st.session_state['portfolio'] = st.session_state['portfolio'].sort_values(by='median', ascending=False)
+            st.session_state['export_merge'] = st.session_state['portfolio'].copy()
             with st.expander("Download options"):
                 if stack_dict is not None:
                     download_type = st.selectbox("Simple or Advanced Download?", options=['Simple', 'Advanced'], key='download_choice')
@@ -1105,60 +1106,53 @@ with tab2:
                     st.download_button(label="Download Portfolio", data=st.session_state['export_file'].to_csv(index=False), file_name="portfolio.csv", mime="text/csv")
                 with merge_port:
                     if st.button("Add to export"):
-                        st.session_state['export_base'] = pd.concat([st.session_state['export_base'], st.session_state['portfolio']])
+                        st.session_state['export_base'] = pd.concat([st.session_state['export_base'], st.session_state['export_merge']])
             else:
                 st.error("No portfolio to download")
             
             display_frame_source = st.selectbox("Display:", options=['Portfolio', 'Export Base'], key='display_frame_source')
             if display_frame_source == 'Portfolio':
-                total_rows = len(st.session_state['portfolio'])
-                rows_per_page = 500
-                total_pages = (total_rows + rows_per_page - 1) // rows_per_page  # Ceiling division
+                display_frame = st.session_state['portfolio']
+            elif display_frame_source == 'Export Base':
+                display_frame = st.session_state['export_base']
+            total_rows = len(display_frame)
+            rows_per_page = 500
+            total_pages = (total_rows + rows_per_page - 1) // rows_per_page  # Ceiling division
 
-                # Initialize page number in session state if not exists
-                if 'current_page' not in st.session_state:
-                    st.session_state.current_page = 1
+            # Initialize page number in session state if not exists
+            if 'current_page' not in st.session_state:
+                st.session_state.current_page = 1
 
-                # Display current page range info and pagination control in a single line
-                st.write(
-                    f"Showing rows {(st.session_state.current_page - 1) * rows_per_page + 1} "
-                    f"to {min(st.session_state.current_page * rows_per_page, total_rows)} of {total_rows}"
-                )
-                
-                # Add page number input
-                st.session_state.current_page = st.number_input(
-                    f"Page (1-{total_pages})", 
-                    min_value=1, 
-                    max_value=total_pages,
-                    value=st.session_state.current_page
-                )
+            # Display current page range info and pagination control in a single line
+            st.write(
+                f"Showing rows {(st.session_state.current_page - 1) * rows_per_page + 1} "
+                f"to {min(st.session_state.current_page * rows_per_page, total_rows)} of {total_rows}"
+            )
+            
+            # Add page number input
+            st.session_state.current_page = st.number_input(
+                f"Page (1-{total_pages})", 
+                min_value=1, 
+                max_value=total_pages,
+                value=st.session_state.current_page
+            )
 
-                # Calculate start and end indices for current page
-                start_idx = (st.session_state.current_page - 1) * rows_per_page
-                end_idx = min(start_idx + rows_per_page, total_rows)
+            # Calculate start and end indices for current page
+            start_idx = (st.session_state.current_page - 1) * rows_per_page
+            end_idx = min(start_idx + rows_per_page, total_rows)
 
-                # Get the subset of data for the current page
-                current_page_data = st.session_state['portfolio'].iloc[start_idx:end_idx]
-                # Display the paginated dataframe first
-                st.dataframe(
-                    current_page_data.style
-                    .background_gradient(axis=0)
-                    .background_gradient(cmap='RdYlGn')
-                    .background_gradient(cmap='RdYlGn_r', subset=['Finish_percentile', 'Own', 'Dupes'])
-                    .format(freq_format, precision=2), 
-                    height=1000,
-                    use_container_width=True
-                )
-            elif display_frame_source == 'Export Base':
-                st.dataframe(
-                    st.session_state['export_base'].style
-                    .background_gradient(axis=0)
-                    .background_gradient(cmap='RdYlGn')
-                    .background_gradient(cmap='RdYlGn_r', subset=['Finish_percentile', 'Own', 'Dupes'])
-                    .format(freq_format, precision=2), 
-                    height=1000,
-                    use_container_width=True
-                )
+            # Get the subset of data for the current page
+            current_page_data = display_frame.iloc[start_idx:end_idx]
+            # Display the paginated dataframe first
+            st.dataframe(
+                current_page_data.style
+                .background_gradient(axis=0)
+                .background_gradient(cmap='RdYlGn')
+                .background_gradient(cmap='RdYlGn_r', subset=['Finish_percentile', 'Own', 'Dupes'])
+                .format(freq_format, precision=2), 
+                height=1000,
+                use_container_width=True
+            )
             
         # Create player summary dataframe
         player_stats = []