James McCool commited on
Commit
610b496
·
1 Parent(s): dd63c0f

Refactor portfolio export logic in app.py: streamline the concatenation of selected rows and update the export file handling, improving data management and user experience during portfolio updates.

Browse files
Files changed (1) hide show
  1. app.py +2 -8
app.py CHANGED
@@ -919,7 +919,6 @@ with tab3:
919
  portfolio_copy = st.session_state['portfolio'].copy()
920
  else:
921
  portfolio_copy = st.session_state['download_portfolio'].copy()
922
- st.table(portfolio_copy)
923
 
924
  submitted = st.form_submit_button("Submit")
925
  if submitted:
@@ -935,21 +934,16 @@ with tab3:
935
  # Sort by median and take top N rows
936
  top_rows = stack_rows.nlargest(count, sort_var)
937
  selected_rows.append(top_rows)
938
-
939
- st.table(selected_rows)
940
 
941
  # Combine all selected rows
942
- portfolio_copy = pd.concat(selected_rows)
943
-
944
- st.table(portfolio_copy)
945
 
946
  # Update export_file with filtered data
947
- export_file = portfolio_copy.copy()
948
  for col in export_file.columns:
949
  if col not in excluded_cols:
950
  export_file[col] = export_file[col].map(st.session_state['export_dict'])
951
  st.write('Export portfolio updated!')
952
- st.table(export_file)
953
  try:
954
  st.download_button(label="Download Portfolio", data=export_file.to_csv(index=False), file_name="portfolio.csv", mime="text/csv")
955
  except:
 
919
  portfolio_copy = st.session_state['portfolio'].copy()
920
  else:
921
  portfolio_copy = st.session_state['download_portfolio'].copy()
 
922
 
923
  submitted = st.form_submit_button("Submit")
924
  if submitted:
 
934
  # Sort by median and take top N rows
935
  top_rows = stack_rows.nlargest(count, sort_var)
936
  selected_rows.append(top_rows)
 
 
937
 
938
  # Combine all selected rows
939
+ portfolio_concat = pd.concat(selected_rows)
 
 
940
 
941
  # Update export_file with filtered data
942
+ export_file = portfolio_concat.copy()
943
  for col in export_file.columns:
944
  if col not in excluded_cols:
945
  export_file[col] = export_file[col].map(st.session_state['export_dict'])
946
  st.write('Export portfolio updated!')
 
947
  try:
948
  st.download_button(label="Download Portfolio", data=export_file.to_csv(index=False), file_name="portfolio.csv", mime="text/csv")
949
  except: