adding ability to recalculate stacks and sizes

app.py

@@ -1109,6 +1109,24 @@ def parse_portfolio_on_mapped(portfolio, map_dict, map_key, filter_keys_pos, fil
     
     return portfolio[mask]
 
+def recalc_stacks_sizes(df, player_columns, map_dict):
+    team_map = map_dict['team_map']  # Use this directly (player_name -> team)
+    df['Stack'] = df.apply(
+        lambda row: Counter(
+            team_map.get(player, '') for player in row[stack_column_dict[site_var][type_var][sport_var]]
+            if team_map.get(player, '') != ''
+        ).most_common(1)[0][0] if any(team_map.get(player, '') for player in row[stack_column_dict[site_var][type_var][sport_var]]) else '',
+        axis=1
+    )
+    df['Size'] = df.apply(
+        lambda row: Counter(
+            team_map.get(player, '') for player in row[stack_column_dict[site_var][type_var][sport_var]]
+            if team_map.get(player, '') != ''
+        ).most_common(1)[0][1] if any(team_map.get(player, '') for player in row[stack_column_dict[site_var][type_var][sport_var]]) else 0,
+        axis=1
+    )
+    return df
+
 with st.container():
     
     col1, col2, col3, col4 = st.columns([1, 4, 4, 4])
@@ -1929,7 +1947,7 @@ if selected_tab == 'Manage Portfolio':
             st.session_state['base_frame_names'] = {}
     if 'origin_portfolio' in st.session_state and 'projections_df' in st.session_state:
         with st.container():
-            reset_port_col, recalc_div_col, set_base_col, blank_reset_col, contest_size_col = st.columns([.15, .15, .15, .35, .30])
+            reset_port_col, recalc_stacks_col, recalc_div_col, set_base_col, blank_reset_col, contest_size_col = st.columns([.15, .15, .15, .15, .35, .30])
             with reset_port_col:
                 with st.popover("Reset Portfolio"):
                     st.markdown("choose a base to reset to:")
@@ -1940,6 +1958,10 @@ if selected_tab == 'Manage Portfolio':
                             st.rerun()
                     else:
                         st.info("No saved base frames available")
+            with recalc_stacks_col:
+                if st.button("Recalculate Stacks and Sizes"):
+                    st.session_state['working_frame'] = recalc_stacks_sizes(st.session_state['working_frame'], st.session_state['player_columns'], st.session_state['map_dict'])
+                    st.rerun()
             with recalc_div_col:
                 if st.button("Recalculate Diversity"):
                         st.session_state['working_frame']['Diversity'] = recalc_diversity(st.session_state['display_frame'], st.session_state['player_columns'])