Refactor 'Manage Portfolio' section in 'app.py' to improve layout and functionality by adding buttons for recalculating diversity, adding/removing rows from custom export, and adjusting column configurations for better user interaction.

app.py

@@ -1084,13 +1084,16 @@ if selected_tab == 'Data Load':
 if selected_tab == 'Manage Portfolio':
     if 'origin_portfolio' in st.session_state and 'projections_df' in st.session_state:
         with st.container():
-            col1, col2 = st.columns(2)
+            reset_port_col, recalc_div_col, blank_reset_col, contest_size_col = st.columns(2, 2, 4, 2)
             with col1:
                 if st.button('Reset Portfolio', key='reset_port'):
                     st.session_state['settings_base'] = True
                     st.session_state['working_frame'] = st.session_state['base_frame'].copy()
-
             with col2:
+                if st.button("Recalculate Diversity"):
+                        st.session_state['display_frame']['Diversity'] = recalc_diversity(st.session_state['display_frame'], st.session_state['player_columns'])
+
+            with col4:
                 with st.form(key='contest_size_form'):
                     size_col, strength_col, submit_col = st.columns(3)
                     with size_col:
@@ -1974,7 +1977,7 @@ if selected_tab == 'Manage Portfolio':
                         st.session_state['export_file'][col] = st.session_state['export_file'][col].map(position_dict)
             
             if 'export_file' in st.session_state:
-                download_port, merge_port, partial_col, clear_export, recalc_div_col, blank_export_col = st.columns([1, 1, 1, 1, 1, 8])
+                download_port, merge_port, clear_export, add_rows_col, remove_rows_col, blank_export_col = st.columns([1, 1, 1, 2, 2, 6])
                 with download_port:
                     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:
@@ -1982,13 +1985,7 @@ if selected_tab == 'Manage Portfolio':
                         st.session_state['export_base'] = pd.concat([st.session_state['export_base'], st.session_state['export_merge']])
                         st.session_state['export_base'] = st.session_state['export_base'].drop_duplicates()
                         st.session_state['export_base'] = st.session_state['export_base'].reset_index(drop=True)
-                with partial_col:
-                    if 'export_merge' in st.session_state:
-                        select_custom_index = st.multiselect("Select rows to add (based on first column):", options=st.session_state['export_merge'].index, default=[])
-                        if st.button("Add selected to Custom Export"):
-                            st.session_state['export_base'] = pd.concat([st.session_state['export_base'], st.session_state['export_merge'].loc[select_custom_index]])
-                        st.session_state['export_base'] = st.session_state['export_base'].drop_duplicates()
-                        st.session_state['export_base'] = st.session_state['export_base'].reset_index(drop=True)
+                    
                 with clear_export:
                     if st.button("Clear Custom Export"):
                         st.session_state['export_base'] = pd.DataFrame(columns=st.session_state['working_frame'].columns)
@@ -1996,9 +1993,21 @@ if selected_tab == 'Manage Portfolio':
                             st.session_state['display_frame'] = st.session_state['working_frame']
                         elif display_frame_source == 'Export Base':
                             st.session_state['display_frame'] = st.session_state['export_base']
-                with recalc_div_col:
-                    if st.button("Recalculate Diversity"):
-                        st.session_state['display_frame']['Diversity'] = recalc_diversity(st.session_state['display_frame'], st.session_state['player_columns'])
+                    
+                with add_rows_col:
+                    select_custom_index = st.multiselect("Select rows to add (based on first column):", options=st.session_state['export_merge'].index, default=[])
+                    if st.button("Add selected to Custom Export"):
+                        st.session_state['export_base'] = pd.concat([st.session_state['export_base'], st.session_state['export_merge'].loc[select_custom_index]])
+                    st.session_state['export_base'] = st.session_state['export_base'].drop_duplicates()
+                    st.session_state['export_base'] = st.session_state['export_base'].reset_index(drop=True)
+                    
+                with remove_rows_col:
+                    remove_custom_index = st.multiselect("Remove rows (based on first column):", options=st.session_state['display_frame'].index, default=[])
+                    if st.button("Remove selected from Display"):
+                        st.session_state['display_frame'] = st.session_state['display_frame'].drop(remove_custom_index)
+                    st.session_state['display_frame'] = st.session_state['display_frame'].drop_duplicates()
+                    st.session_state['display_frame'] = st.session_state['display_frame'].reset_index(drop=True)
+                    
 
             total_rows = len(st.session_state['display_frame'])
             rows_per_page = 100