Refactor custom export options in app.py: reorganize layout to separate buttons for adding all and selected rows to the custom export, improving user interface clarity and functionality.

app.py

@@ -1201,22 +1201,20 @@ with tab2:
                         st.session_state['export_file'][col] = st.session_state['export_file'][col].map(st.session_state['export_dict'])
             
             if 'export_file' in st.session_state:
-                download_port, merge_port, clear_export, blank_export_col = st.columns([1, 1, 1, 8])
+                download_port, merge_port, partial_col, clear_export, blank_export_col = st.columns([1, 1, 1, 8])
                 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:
-                    add_all_col, add_select_col = st.columns(2)
-                    with add_all_col:
-                        if st.button("Add all to Custom Export"):
-                            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 add_select_col:
-                        select_custom_index = st.multiselect("Select rows to add to Custom Export", 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'].iloc[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)
+                    if st.button("Add all to Custom Export"):
+                        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:
+                    select_custom_index = st.multiselect("Select rows to add to Custom Export", 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'].iloc[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)