Add size selection functionality in app.py: introduce options to include player sizes in lineups, enhancing user control over player configurations and improving data representation.

app.py

@@ -272,7 +272,15 @@ with tab1:
                     ).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
+                )
                 stack_dict = dict(zip(st.session_state['portfolio'].index, st.session_state['portfolio']['Stack']))
+                size_dict = dict(zip(st.session_state['portfolio'].index, st.session_state['portfolio']['Size']))
 
             working_frame = st.session_state['portfolio'].copy()
             try:
@@ -909,6 +917,7 @@ with tab2:
                     st.session_state['working_frame']['Own'] = st.session_state['working_frame'].apply(lambda row: sum(st.session_state['map_dict']['own_map'].get(player, 0) for player in row), axis=1)
                     if stack_dict is not None:
                         st.session_state['working_frame']['Stack'] = st.session_state['working_frame'].index.map(stack_dict)
+                        st.session_state['working_frame']['Size'] = st.session_state['working_frame'].index.map(size_dict)
             elif type_var == 'Showdown':
                 # Calculate salary (CPT uses cpt_salary_map, others use salary_map)
                 st.session_state['working_frame']['salary'] = st.session_state['working_frame'].apply(
@@ -1004,6 +1013,7 @@ with tab2:
                     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=[])
                     team_remove = st.multiselect("Remove teams?", options=sorted(list(set(st.session_state['projections_df']['team'].unique()))), default=[])
+                    size_include = st.multiselect("Include sizes?", options=sorted(list(set(st.session_state['working_frame']['Size'].unique()))), default=[])
                     
                     submitted = st.form_submit_button("Submit")
                     if submitted:
@@ -1048,6 +1058,9 @@ with tab2:
                                 lambda row: not any(team in list(row) for team in team_remove), axis=1
                             )
                             parsed_frame = parsed_frame[remove_mask]
+                        
+                        if size_include:
+                            parsed_frame = parsed_frame[parsed_frame['Size'].isin(size_include)]
                             
                         st.session_state['working_frame'] = parsed_frame.sort_values(by='median', ascending=False)
                         st.session_state['export_merge'] = st.session_state['working_frame'].copy()