Update app.py

app.py

@@ -90,15 +90,23 @@ with tab1:
             raw_baselines = dk_raw
         elif site_var1 == 'Fanduel':
             raw_baselines = fd_raw
+            
         team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
         if team_var1 == 'Specific Teams':
                 team_var2 = st.multiselect('Which teams do you want?', options = DK_seed['Team'].unique())
         elif team_var1 == 'Full Slate':
                 team_var2 = DK_seed.Team.values.tolist()
+
+        stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
+        if stack_var1 == 'Specific Stack Sizes':
+                stack_var2 = st.multiselect('Which stack sizes do you want?', options = DK_seed['Team_count'].unique())
+        elif stack_var1 == 'Full Slate':
+                stack_var2 = DK_seed.Team_count.values.tolist()
     
     with col2:
         if site_var1 == 'Draftkings':
             DK_seed_parse = DK_seed[DK_seed['Team'].isin(team_var2)]
+            DK_seed_parse = DK_seed[DK_seed['Team_count'].isin(stack_var2)]
             st.session_state.data_export_display = DK_seed_parse.head(1000)
             st.session_state.data_export = DK_seed_parse
             st.session_state.data_export_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.data_export.iloc[:,0:9].values, return_counts=True)),
@@ -117,6 +125,7 @@ with tab1:
             st.dataframe(st.session_state.data_export_freq.style.format(percentages_format, precision=2), height=500, use_container_width=True)
         elif site_var1 == 'Fanduel':
             FD_seed_parse = FD_seed[FD_seed['Team'].isin(team_var2)]
+            FD_seed_parse = FD_seed[FD_seed['Team_count'].isin(stack_var2)]
             st.session_state.data_export_display = FD_seed_parse.head(1000)
             st.session_state.data_export = FD_seed_parse
             st.session_state.data_export_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.data_export.iloc[:,0:8].values, return_counts=True)),