adjusting stacking sports logic

app.py

@@ -30,7 +30,8 @@ from global_func.reassess_edge import reassess_edge
 
 freq_format = {'Finish_percentile': '{:.2%}', 'Lineup Edge': '{:.2%}', 'Win%': '{:.2%}'}
 stacking_sports = ['MLB', 'NHL', 'NFL', 'LOL', 'NCAAF']
-all_column_stack_sports = ['LOL', 'NCAAF', 'WNBA', 'NBA', 'CS2']
+# all_column_stack_sports = ['LOL', 'NCAAF', 'WNBA', 'NBA', 'CS2']
+exclude_stacks = ['BaseName', 'EntryCount', 'SP', 'SP1', 'SP2', 'P1', 'P2', 'RB1', 'RB2', 'DST', 'G']
 player_wrong_names_mlb = ['Enrique Hernandez', 'Joseph Cantillo', 'Mike Soroka', 'Jakob Bauers', 'Temi Fágbénlé']
 player_right_names_mlb = ['Kike Hernandez', 'Joey Cantillo', 'Michael Soroka', 'Jake Bauers', 'Temi Fagbenle']
 
@@ -441,39 +442,24 @@ if selected_tab == 'Data Load':
             # Update projections with matched names
             projections['player_names'] = projections['player_names'].map(lambda x: projections_match_dict.get(x, x))
             st.session_state['projections_df'] = projections
+            st.session_state['stack_columns'] = [col for col in st.session_state['portfolio'].columns if col not in exclude_stacks]
 
             if sport_var in stacking_sports:
                 team_dict = dict(zip(st.session_state['projections_df']['player_names'], st.session_state['projections_df']['team']))
-                if sport_var in all_column_stack_sports:
-                    st.session_state['portfolio']['Stack'] = st.session_state['portfolio'].apply(
-                        lambda row: Counter(
-                            team_dict.get(player, '') for player in row
-                            if team_dict.get(player, '') != ''
-                        ).most_common(1)[0][0] if any(team_dict.get(player, '') for player in row) else '',
-                        axis=1
-                    )
-                    st.session_state['portfolio']['Size'] = st.session_state['portfolio'].apply(
-                        lambda row: Counter(
-                            team_dict.get(player, '') for player in row
-                            if team_dict.get(player, '') != ''
-                        ).most_common(1)[0][1] if any(team_dict.get(player, '') for player in row) else 0,
-                        axis=1
-                    )
-                else:
-                    st.session_state['portfolio']['Stack'] = st.session_state['portfolio'].apply(
-                        lambda row: Counter(
-                            team_dict.get(player, '') for player in row[2:]
-                            if team_dict.get(player, '') != ''
-                        ).most_common(1)[0][0] if any(team_dict.get(player, '') for player in row[2:]) else '',
-                        axis=1
-                    )
-                    st.session_state['portfolio']['Size'] = st.session_state['portfolio'].apply(
-                        lambda row: Counter(
-                            team_dict.get(player, '') for player in row[2:]
-                            if team_dict.get(player, '') != ''
-                        ).most_common(1)[0][1] if any(team_dict.get(player, '') for player in row[2:]) else 0,
-                        axis=1
-                    )
+                st.session_state['portfolio']['Stack'] = st.session_state['portfolio'].apply(
+                    lambda row: Counter(
+                        team_dict.get(player, '') for player in row[st.session_state['stack_columns']]
+                        if team_dict.get(player, '') != ''
+                    ).most_common(1)[0][0] if any(team_dict.get(player, '') for player in row[st.session_state['stack_columns']]) else '',
+                    axis=1
+                )
+                st.session_state['portfolio']['Size'] = st.session_state['portfolio'].apply(
+                    lambda row: Counter(
+                        team_dict.get(player, '') for player in row[st.session_state['stack_columns']]
+                        if team_dict.get(player, '') != ''
+                    ).most_common(1)[0][1] if any(team_dict.get(player, '') for player in row[st.session_state['stack_columns']]) else 0,
+                    axis=1
+                )
                 st.session_state['stack_dict'] = dict(zip(st.session_state['portfolio'].index, st.session_state['portfolio']['Stack']))
                 st.session_state['size_dict'] = dict(zip(st.session_state['portfolio'].index, st.session_state['portfolio']['Size']))