Refactor portfolio copy handling in app.py: simplify the logic for creating a portfolio copy and streamline export file updates, enhancing data management and user experience during portfolio interactions.

app.py

@@ -844,7 +844,6 @@ with tab3:
                         
                         st.session_state['portfolio'] = trim_portfolio(st.session_state['portfolio'], performance_type, own_type)
                         st.session_state['portfolio'] = st.session_state['portfolio'].sort_values(by='median', ascending=False)
-                        st.session_state['download_portfolio'] = st.session_state['portfolio'].copy()
 
         with col2:
             st.session_state['portfolio'] = predict_dupes(st.session_state['portfolio'], map_dict, site_var, type_var, Contest_Size, strength_var)
@@ -914,11 +913,7 @@ with tab3:
                                         key=key
                                     )
                         
-                        # Create a copy of the portfolio
-                        if 'download_portfolio' not in st.session_state:
-                            portfolio_copy = st.session_state['portfolio'].copy()
-                        else:
-                            portfolio_copy = st.session_state['download_portfolio'].copy()
+                        portfolio_copy = st.session_state['portfolio'].copy()
                         
                         submitted = st.form_submit_button("Submit")
                         if submitted:
@@ -939,15 +934,15 @@ with tab3:
                             portfolio_concat = pd.concat(selected_rows)
                             
                             # Update export_file with filtered data
-                            export_file = portfolio_concat.copy()
-                            for col in export_file.columns:
+                            st.session_state['export_file'] = portfolio_concat.copy()
+                            for col in st.session_state['export_file'].columns:
                                 if col not in excluded_cols:
-                                    export_file[col] = export_file[col].map(st.session_state['export_dict'])
+                                    st.session_state['export_file'][col] = st.session_state['export_file'][col].map(st.session_state['export_dict'])
                             st.write('Export portfolio updated!')
-            try:    
-                st.download_button(label="Download Portfolio", data=export_file.to_csv(index=False), file_name="portfolio.csv", mime="text/csv")
-            except:
-                st.error("No portfolio to download")
+                            try:    
+                                st.download_button(label="Download Portfolio", data=st.session_state['export_file'].to_csv(index=False), file_name="portfolio.csv", mime="text/csv")
+                            except:
+                                st.error("No portfolio to download")
             
             # Add pagination controls below the dataframe
             total_rows = len(st.session_state['portfolio'])