Add CPT/FLEX/Overall focus selection for player locking and removal in Showdown mode in 'app.py', enhancing lineup customization options.

app.py

@@ -1256,6 +1256,8 @@ if selected_tab == 'Manage Portfolio':
                     for col in st.session_state['working_frame'].columns:
                         if col not in excluded_cols:
                             player_names.update(st.session_state['working_frame'][col].unique())
+                    if type_var == 'Showdown':
+                        cpt_flex_focus = st.selectbox("Focus on Overall, CPT, or FLEX?", options=['Overall', 'CPT', 'FLEX'], index=0)
                     player_lock = st.multiselect("Lock players?", options=sorted(list(player_names)), default=[])
                     player_remove = st.multiselect("Remove players?", options=sorted(list(player_names)), default=[])
                     team_include = st.multiselect("Include teams?", options=sorted(list(set(st.session_state['projections_df']['team'].unique()))), default=[])
@@ -1275,41 +1277,96 @@ if selected_tab == 'Manage Portfolio':
                         st.session_state['settings_base'] = False
                         parsed_frame = st.session_state['working_frame'].copy()
                         if player_remove:
-                            # Create mask for lineups that contain any of the removed players
-                            remove_mask = parsed_frame[st.session_state['player_columns']].apply(
-                                lambda row: not any(player in list(row) for player in player_remove), axis=1
-                            )
+                            if type_var == 'Showdown':
+                                if cpt_flex_focus == 'CPT':
+                                    remove_mask = parsed_frame[st.session_state['player_columns'][0]].apply(
+                                        lambda row: not any(player in list(row) for player in player_remove), axis=1
+                                    )
+                                elif cpt_flex_focus == 'FLEX':
+                                    remove_mask = parsed_frame[st.session_state['player_columns'][1:]].apply(
+                                        lambda row: not any(player in list(row) for player in player_remove), axis=1
+                                    )
+                                elif cpt_flex_focus == 'Overall':
+                                    remove_mask = parsed_frame[st.session_state['player_columns']].apply(
+                                        lambda row: not any(player in list(row) for player in player_remove), axis=1
+                                    )
+                            else:
+                                # Create mask for lineups that contain any of the removed players
+                                remove_mask = parsed_frame[st.session_state['player_columns']].apply(
+                                    lambda row: not any(player in list(row) for player in player_remove), axis=1
+                                )
                             parsed_frame = parsed_frame[remove_mask]
                         
                         if player_lock:
-                            
-                            lock_mask = parsed_frame[st.session_state['player_columns']].apply(
-                                lambda row: all(player in list(row) for player in player_lock), axis=1
-                            )
+                            if type_var == 'Showdown':
+                                if cpt_flex_focus == 'CPT':
+                                    lock_mask = parsed_frame[st.session_state['player_columns'][0]].apply(
+                                        lambda row: all(player in list(row) for player in player_lock), axis=1
+                                    )
+                                elif cpt_flex_focus == 'FLEX':
+                                    lock_mask = parsed_frame[st.session_state['player_columns'][1:]].apply(
+                                        lambda row: all(player in list(row) for player in player_lock), axis=1
+                                    )
+                                elif cpt_flex_focus == 'Overall':
+                                    lock_mask = parsed_frame[st.session_state['player_columns']].apply(
+                                        lambda row: all(player in list(row) for player in player_lock), axis=1
+                                    )
+                            else:
+                                lock_mask = parsed_frame[st.session_state['player_columns']].apply(
+                                        lambda row: all(player in list(row) for player in player_lock), axis=1
+                                    )
                             parsed_frame = parsed_frame[lock_mask]
                         
                         if team_include:
-                            # Create a copy of the frame with player names replaced by teams, excluding SP1 and SP2
-                            filtered_player_columns = [col for col in st.session_state['player_columns'] if col not in ['SP1', 'SP2']]
-                            team_frame = parsed_frame[filtered_player_columns].apply(
-                                lambda x: x.map(st.session_state['map_dict']['team_map'])
-                            )
-                            # Create mask for lineups that contain any of the included teams
-                            include_mask = team_frame.apply(
-                                lambda row: any(team in list(row) for team in team_include), axis=1
-                            )
+                            if type_var == 'Showdown':
+                                if cpt_flex_focus == 'CPT':
+                                    include_mask = parsed_frame[st.session_state['player_columns'][0]].apply(
+                                        lambda row: any(team in list(row) for team in team_include), axis=1
+                                    )
+                                elif cpt_flex_focus == 'FLEX':
+                                    include_mask = parsed_frame[st.session_state['player_columns'][1:]].apply(
+                                        lambda row: any(team in list(row) for team in team_include), axis=1
+                                    )
+                                elif cpt_flex_focus == 'Overall':
+                                    include_mask = parsed_frame[st.session_state['player_columns']].apply(
+                                        lambda row: any(team in list(row) for team in team_include), axis=1
+                                    )
+                            else:
+                                # Create a copy of the frame with player names replaced by teams, excluding SP1 and SP2
+                                filtered_player_columns = [col for col in st.session_state['player_columns'] if col not in ['SP1', 'SP2']]
+                                team_frame = parsed_frame[filtered_player_columns].apply(
+                                    lambda x: x.map(st.session_state['map_dict']['team_map'])
+                                )
+                                # Create mask for lineups that contain any of the included teams
+                                include_mask = team_frame.apply(
+                                    lambda row: any(team in list(row) for team in team_include), axis=1
+                                )
                             parsed_frame = parsed_frame[include_mask]
                         
                         if team_remove:
-                            # Create a copy of the frame with player names replaced by teams, excluding SP1 and SP2
-                            filtered_player_columns = [col for col in st.session_state['player_columns'] if col not in ['SP1', 'SP2']]
-                            team_frame = parsed_frame[filtered_player_columns].apply(
-                                lambda x: x.map(st.session_state['map_dict']['team_map'])
-                            )
-                            # Create mask for lineups that don't contain any of the removed teams
-                            remove_mask = team_frame.apply(
-                                lambda row: not any(team in list(row) for team in team_remove), axis=1
-                            )
+                            if type_var == 'Showdown':
+                                if cpt_flex_focus == 'CPT':
+                                    remove_mask = parsed_frame[st.session_state['player_columns'][0]].apply(
+                                        lambda row: not any(team in list(row) for team in team_remove), axis=1
+                                    )
+                                elif cpt_flex_focus == 'FLEX':
+                                    remove_mask = parsed_frame[st.session_state['player_columns'][1:]].apply(
+                                        lambda row: not any(team in list(row) for team in team_remove), axis=1
+                                    )
+                                elif cpt_flex_focus == 'Overall':
+                                    remove_mask = parsed_frame[st.session_state['player_columns']].apply(
+                                        lambda row: not any(team in list(row) for team in team_remove), axis=1
+                                    )
+                            else:
+                                # Create a copy of the frame with player names replaced by teams, excluding SP1 and SP2
+                                filtered_player_columns = [col for col in st.session_state['player_columns'] if col not in ['SP1', 'SP2']]
+                                team_frame = parsed_frame[filtered_player_columns].apply(
+                                    lambda x: x.map(st.session_state['map_dict']['team_map'])
+                                )
+                                # Create mask for lineups that don't contain any of the removed teams
+                                remove_mask = team_frame.apply(
+                                    lambda row: not any(team in list(row) for team in team_remove), axis=1
+                                )
                             parsed_frame = parsed_frame[remove_mask]
                         
                         if size_include: