Enhance exposure_spread function to include an option for removing specified teams, improving flexibility in player comparison and exposure calculations.

app.py

@@ -1172,10 +1172,11 @@ with tab2:
                         exposure_stack_bool = st.selectbox("Maintain Stacks?", options=['Yes', 'No'], index=0)
                     else:
                         exposure_stack_bool = 'No'
+                    remove_teams_exposure = st.multiselect("Removed/Locked teams?", options=sorted(list(set(st.session_state['projections_df']['team'].unique()))), default=[])
                     submitted = st.form_submit_button("Submit")
                     if submitted:
                         st.session_state['settings_base'] = False
-                        parsed_frame = exposure_spread(st.session_state['working_frame'], exposure_player, exposure_target, exposure_stack_bool, st.session_state['projections_df'], sport_var, type_var)
+                        parsed_frame = exposure_spread(st.session_state['working_frame'], exposure_player, exposure_target, exposure_stack_bool, remove_teams_exposure, st.session_state['projections_df'], sport_var, type_var)
                         st.session_state['working_frame'] = parsed_frame.reset_index(drop=True)
                         st.session_state['export_merge'] = st.session_state['working_frame'].copy()
         with st.container():

global_func/exposure_spread.py

@@ -209,7 +209,7 @@ def check_position_eligibility(sport, column_name, player_positions):
         # Default fallback - assume exact position match
         return column_name in player_positions
 
-def exposure_spread(working_frame, exposure_player, exposure_target, exposure_stack_bool, projections_df, sport_var, type_var):
+def exposure_spread(working_frame, exposure_player, exposure_target, exposure_stack_bool, remove_teams, projections_df, sport_var, type_var):
     # Find comparable players in the projections
     comparable_players = projections_df[projections_df['player_names'] == exposure_player]
     
@@ -240,7 +240,16 @@ def exposure_spread(working_frame, exposure_player, exposure_target, exposure_st
         (projections_df['median'] <= comp_projection_high) &
         (projections_df['position'].apply(lambda x: has_position_overlap(x, comp_player_position)))
     ]
+
+    if exposure_target == 0:
+        comparable_players = comparable_players[comparable_players['player_names'] != exposure_player]
     
+    if remove_teams is not None:
+        remove_mask = player_rows.apply(
+                lambda row: not any(team in list(row) for team in remove_teams), axis=1
+            )
+        player_rows = player_rows[remove_mask]
+        
     # Create a list of comparable players
     comparable_player_list = comparable_players['player_names'].tolist()
 
@@ -255,11 +264,9 @@ def exposure_spread(working_frame, exposure_player, exposure_target, exposure_st
 
     # isolate the rows that contain the player
     player_rows = working_frame[player_mask]
-    print(player_rows.head(10))
     if comparable_stack != 0:
         player_rows = player_rows[player_rows['Stack'] != comparable_stack]
-    print(player_rows.head(10))
-    print(working_frame.head(10))
+    
 
     change_counter = 0