James McCool commited on
Commit
83df1df
·
1 Parent(s): be8e70e

Enhance export file processing in app.py: add mapping for non-excluded columns in the export file using the export dictionary, improving data accuracy during portfolio downloads.

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -968,6 +968,9 @@ with tab2:
968
  st.write('Export portfolio updated!')
969
  else:
970
  st.session_state['export_file'] = st.session_state['portfolio'].copy()
 
 
 
971
  if 'export_file' in st.session_state:
972
  st.download_button(label="Download Portfolio", data=st.session_state['export_file'].to_csv(index=False), file_name="portfolio.csv", mime="text/csv")
973
  else:
 
968
  st.write('Export portfolio updated!')
969
  else:
970
  st.session_state['export_file'] = st.session_state['portfolio'].copy()
971
+ for col in st.session_state['export_file'].columns:
972
+ if col not in excluded_cols:
973
+ st.session_state['export_file'][col] = st.session_state['export_file'][col].map(st.session_state['export_dict'])
974
  if 'export_file' in st.session_state:
975
  st.download_button(label="Download Portfolio", data=st.session_state['export_file'].to_csv(index=False), file_name="portfolio.csv", mime="text/csv")
976
  else: