Enhance stack selection functionality in app.py: update stack inclusion and removal options for improved user control over lineup configurations.

app.py

@@ -967,9 +967,11 @@ with tab2:
                     max_salary = st.number_input("Max acceptable salary?", value=100000, min_value=1000, step=100)
                     max_finish_percentile = st.number_input("Max acceptable finish percentile?", value=.50, min_value=0.005, step=.001)
                     min_lineup_edge = st.number_input("Min acceptable Lineup Edge?", value=-.5, min_value=-1.00, step=.001)
-                    if stack_dict is not None:
-                        stack_toggle = st.selectbox("Include specific stacks?", options=['All Stacks', 'Specific Stacks'], index=0)
+                    if 'Stack' in st.session_state['working_frame'].columns:
+                        stack_include_toggle = st.selectbox("Include specific stacks?", options=['All Stacks', 'Specific Stacks'], index=0)
                         stack_selections = st.multiselect("If Specific Stacks, Which to include?", options=sorted(list(set(stack_dict.values()))), default=[])
+
+                        stack_remove_toggle = st.selectbox("Remove specific stacks?", options=['No', 'Yes'], index=0)
                         stack_remove = st.multiselect("If Specific Stacks, Which to remove?", options=sorted(list(set(stack_dict.values()))), default=[])
                     
                     submitted = st.form_submit_button("Submit")
@@ -980,13 +982,15 @@ with tab2:
                         parsed_frame = parsed_frame[parsed_frame['salary'] <= max_salary]
                         parsed_frame = parsed_frame[parsed_frame['Finish_percentile'] <= max_finish_percentile]
                         parsed_frame = parsed_frame[parsed_frame['Lineup Edge'] >= min_lineup_edge]
-                        if stack_dict is not None:
-                            if stack_toggle == 'All Stacks':
+                        if 'Stack' in parsed_frame.columns:
+                            if stack_include_toggle == 'All Stacks':
                                 parsed_frame = parsed_frame
-                                parsed_frame = parsed_frame[~parsed_frame['Stack'].isin(stack_remove)]
                             else:
                                 parsed_frame = parsed_frame[parsed_frame['Stack'].isin(stack_selections)]
+                            if stack_remove_toggle == 'Yes':
                                 parsed_frame = parsed_frame[~parsed_frame['Stack'].isin(stack_remove)]
+                            else:
+                                parsed_frame = parsed_frame
                         st.session_state['working_frame'] = parsed_frame.sort_values(by='median', ascending=False)
                         st.session_state['export_merge'] = st.session_state['working_frame'].copy()