Add team selection functionality in app.py: introduce options to include and remove teams from lineups for enhanced user control over player configurations.

app.py

@@ -1002,6 +1002,8 @@ with tab2:
                             player_names.update(st.session_state['working_frame'][col].unique())
                     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['working_frame']['team'].unique()))), default=[])
+                    team_remove = st.multiselect("Remove teams?", options=sorted(list(set(st.session_state['working_frame']['team'].unique()))), default=[])
                     
                     submitted = st.form_submit_button("Submit")
                     if submitted:
@@ -1022,6 +1024,31 @@ with tab2:
                                 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 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 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]
+                            
                         st.session_state['working_frame'] = parsed_frame.sort_values(by='median', ascending=False)
                         st.session_state['export_merge'] = st.session_state['working_frame'].copy()