Update app.py

app.py

@@ -117,16 +117,20 @@ with tab2:
               
         slate_var1 = st.radio("Which data are you loading?", ('Paydirt', 'User'), key='slate_var1')
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='site_var1')
+        if slate_var1 != 'User':
+            mainvar1 = st.radio("Main slate or Secondary?", ('Main Slate', 'Secondary'), key='mainvar1')
         if site_var1 == 'Draftkings':
               if slate_var1 == 'User':
                   init_baselines = proj_dataframe
               elif slate_var1 != 'User':
                   init_baselines = dk_raw_proj
+                  init_baselines = init_baselines[init_baselines['Slate'] == mainvar1]
         elif site_var1 == 'Fanduel':
               if slate_var1 == 'User':
                   init_baselines = proj_dataframe
               elif slate_var1 != 'User':
                   init_baselines = fd_raw_proj
+                  init_baselines = init_baselines[init_baselines['Slate'] == mainvar1]
         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')