Adding base choice functionality? Or at least trying to

app.py

@@ -1745,15 +1745,23 @@ if selected_tab == 'Manage Portfolio':
         with st.container():
             reset_port_col, recalc_div_col, set_base_col, blank_reset_col, contest_size_col = st.columns([1, 1, 1, 6, 4])
             with reset_port_col:
-                if st.button('Reset Portfolio', key='reset_port'):
-                    st.session_state['settings_base'] = True
-                    st.session_state['working_frame'] = st.session_state['base_frame']
+                with st.popover("Reset Portfolio"):
+                    st.markdown("choose a base to reset to:")
+                    base_choice = st.selectbox("Base Choice", options=st.session_state['base_frame_names'], index=0)
+                    st.session_state['working_frame'] = st.session_state[f'{base_choice}']
             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 set_base_col:
-                if st.button("Set New Base"):
-                    st.session_state['base_frame'] = st.session_state['working_frame']
+                with st.popover("New Base Setting"):
+                    st.markdown("Name of new base:")
+                    new_base_name = st.text_input("New Base Name", value='New Base')
+                    if f'{new_base_name}' not in st.session_state['base_frame_names']:
+                        st.session_state['base_frame_names'].append(f'{new_base_name}')
+                        st.session_state['base_frame_names'] = sorted(st.session_state['base_frame_names'])
+                        st.session_state[f'{new_base_name}'] = st.session_state['working_frame']
+                    else:
+                        st.error("Base name already exists")
 
             with contest_size_col:
                 with st.form(key='contest_size_form'):
@@ -1792,7 +1800,8 @@ if selected_tab == 'Manage Portfolio':
                 st.session_state['working_frame']['Size'] = st.session_state['working_frame'].index.map(st.session_state['size_dict'])
             
             st.session_state['base_frame'] = predict_dupes(st.session_state['working_frame'], st.session_state['map_dict'], site_var, type_var, Contest_Size, strength_var, sport_var, salary_max)
-            st.session_state['working_frame'] = st.session_state['base_frame'].copy()
+            st.session_state[f'{base_choice}'] = st.session_state['base_frame']
+            st.session_state['working_frame'] = st.session_state['base_frame']
             # st.session_state['highest_owned_teams'] = st.session_state['projections_df'][~st.session_state['projections_df']['position'].isin(['P', 'SP'])].groupby('team')['ownership'].sum().sort_values(ascending=False).head(3).index.tolist()
             # st.session_state['highest_owned_pitchers'] = st.session_state['projections_df'][st.session_state['projections_df']['position'].isin(['P', 'SP'])]['player_names'].sort_values(by='ownership', ascending=False).head(3).tolist()