Update export file assignment logic in app.py: streamline the assignment of export_file by directly referencing display_frame based on user selection, improving clarity and consistency in data handling for export processes.

app.py

@@ -1085,14 +1085,14 @@ with tab2:
             display_frame_source = st.selectbox("Display:", options=['Portfolio', 'Export Base'], key='display_frame_source')
             if display_frame_source == 'Portfolio':
                 display_frame = st.session_state['working_frame']
-                st.session_state['export_file'] = st.session_state['working_frame'].copy()
+                st.session_state['export_file'] = display_frame.copy()
                 
                 for col in st.session_state['export_file'].columns:
                     if col not in excluded_cols:
                         st.session_state['export_file'][col] = st.session_state['export_file'][col].map(st.session_state['export_dict'])
             elif display_frame_source == 'Export Base':
                 display_frame = st.session_state['export_base']
-                st.session_state['export_file'] = st.session_state['working_frame'].copy()
+                st.session_state['export_file'] = display_frame.copy()
                 
                 for col in st.session_state['export_file'].columns:
                     if col not in excluded_cols:
@@ -1110,6 +1110,10 @@ with tab2:
                 with clear_export:
                     if st.button("Clear Custom Export"):
                         st.session_state['export_base'] = pd.DataFrame(columns=st.session_state['working_frame'].columns)
+                        if display_frame_source == 'Portfolio':
+                            display_frame = st.session_state['working_frame']
+                        elif display_frame_source == 'Export Base':
+                            display_frame = st.session_state['export_base']
 
             total_rows = len(display_frame)
             rows_per_page = 500