Add functions to define showdown slates for DraftKings and FanDuel, enhancing slate selection in the app.

src/streamlit_app.py

@@ -69,6 +69,40 @@ st.markdown("""
 
 </style>""", unsafe_allow_html=True)
 
+@st.cache_resource(ttl=60)
+def define_dk_showdown_slates():
+    collection = db["DK_SD_NFL_ROO"]
+    cursor = collection.find()
+    raw_display = pd.DataFrame(list(cursor))
+    unique_slates = raw_display['slate'].unique()
+
+    slate_names = []
+
+    for slate in unique_slates:
+        slate_data = raw_display[raw_display['slate'] == slate]
+        slate_name = slate_data['Team'] + ' vs. ' + slate_data['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
+@st.cache_resource(ttl=60)
+def define_fd_showdown_slates():
+    collection = db["FD_SD_NFL_ROO"]
+    cursor = collection.find()
+    raw_display = pd.DataFrame(list(cursor))
+    unique_slates = raw_display['slate'].unique()
+
+    slate_names = []
+
+    for slate in unique_slates:
+        slate_data = raw_display[raw_display['slate'] == slate]
+        slate_name = slate_data['Team'] + ' vs. ' + slate_data['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
 @st.cache_resource(ttl=60)
 def init_handbuilder_data(site_var):
     """Load only the data needed for handbuilder"""
@@ -435,6 +469,9 @@ with app_view_site_column:
         with app_site_column:
             site_var = st.selectbox("What site do you want to view?", ('Draftkings', 'Fanduel'), key='site_selectbox')
 
+slate_names_dk, slate_name_lookup_dk = define_dk_showdown_slates()
+slate_names_fd, slate_name_lookup_fd = define_fd_showdown_slates()
+
 selected_tab = st.segmented_control(
     "Select Tab",
     options=["Player ROO", "Stacks ROO", "Handbuilder", "Optimals"],
@@ -921,19 +958,19 @@ if selected_tab == 'Player ROO':
         if slate_type_var2 == 'Regular':
             slate_var2 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate', 'Late Slate', 'Thurs-Mon Slate'), key='slate_var2_radio')
         elif slate_type_var2 == 'Showdown':
-            slate_var2 = st.radio("Which slate are you loading?", ("Showdown #1", "Showdown #2", "Showdown #3"), key='slate_var2_radio')
+            slate_var2 = st.radio("Which slate are you loading?", (slate_names_dk if site_var == 'Draftkings' else slate_names_fd), key='slate_var2_radio')
         if site_var == 'Draftkings':
             if slate_type_var2 == 'Regular':
                 raw_baselines = dk_roo_raw[dk_roo_raw['slate'] == str(slate_var2)]
             elif slate_type_var2 == 'Showdown':
-                raw_baselines = dk_sd_roo_raw[dk_sd_roo_raw['slate'] == str(slate_var2)]
+                raw_baselines = dk_sd_roo_raw[dk_sd_roo_raw['slate'] == str(slate_name_lookup_dk[slate_var2])]
             
             raw_baselines = raw_baselines.iloc[:,:-2]
         elif site_var == 'Fanduel':
             if slate_type_var2 == 'Regular':
                 raw_baselines = fd_roo_raw[fd_roo_raw['slate'] == str(slate_var2)]
             elif slate_type_var2 == 'Showdown':
-                raw_baselines = fd_sd_roo_raw[fd_sd_roo_raw['slate'] == str(slate_var2)]
+                raw_baselines = fd_sd_roo_raw[fd_sd_roo_raw['slate'] == str(slate_name_lookup_fd[slate_var2])]
             raw_baselines = raw_baselines.iloc[:,:-2]
 
         split_var2 = st.radio("Would you like to view the whole slate or just specific games?", ('Full Slate Run', 'Specific Games'), key='split_var2_radio')