Update app.py

app.py

@@ -121,13 +121,22 @@ with tab2:
         if site_var1 == 'Draftkings':
               if slate_var1 == 'User':
                   init_baselines = proj_dataframe
+                  
               elif slate_var1 != 'User':
                   init_baselines = dk_raw_proj
+                  if mainvar1 == 'Main Slate':
+                      init_baselines = init_baselines.loc[init_baselines['MainSlateDK'] == 'Main']
+                  if mainvar1 != 'Main Slate':
+                      init_baselines = init_baselines.loc[init_baselines['MainSlateDK'] != 'Main']
         elif site_var1 == 'Fanduel':
               if slate_var1 == 'User':
                   init_baselines = proj_dataframe
               elif slate_var1 != 'User':
                   init_baselines = fd_raw_proj
+                  if mainvar1 == 'Main Slate':
+                      init_baselines = init_baselines.loc[init_baselines['MainSlateFD'] == ' Main ']
+                  if mainvar1 != 'Main Slate':
+                      init_baselines = init_baselines.loc[init_baselines['MainSlateFD'] != ' Main ']
         contest_var1 = st.selectbox("What contest type are you optimizing for?", ('Cash', 'Small Field GPP', 'Large Field GPP', 'Round Robin'), key='contest_var1')
         if contest_var1 != 'Cash':
             stack_var1 = st.selectbox('Which team are you stacking?', options = init_baselines['Team'].unique(), key='stack_var1')