Enhance app.py: Update init_baselines to return both roo_data and sd_roo_data. Adjusted data handling in the UI to support selection between "Full Slate" and "Showdown" types, improving user experience and data display functionality.

app.py

@@ -39,8 +39,15 @@ def init_baselines():
 
     roo_data = player_frame.drop(columns=['_id', 'index', 'Timestamp'])
     roo_data['Salary'] = roo_data['Salary'].astype(int)
+
+    collection = db["PGA_SD_ROO"] 
+    cursor = collection.find()
+    player_frame = pd.DataFrame(cursor)
+
+    sd_roo_data = player_frame.drop(columns=['_id', 'index', 'Timestamp'])
+    sd_roo_data['Salary'] = sd_roo_data['Salary'].astype(int)
     
-    return roo_data, timestamp
+    return roo_data, sd_roo_data, timestamp
 
 @st.cache_data(ttl = 60)
 def init_DK_lineups():  
@@ -90,7 +97,7 @@ def convert_df(array):
     array = pd.DataFrame(array, columns=column_names)
     return array.to_csv().encode('utf-8')
 
-roo_data, timestamp = init_baselines()
+roo_data, sd_roo_data, timestamp = init_baselines()
 dk_lineups = init_DK_lineups()
 fd_lineups = init_FD_lineups()
 hold_display = roo_data
@@ -107,7 +114,7 @@ with tab1:
             # Clear values from *all* all in-memory and on-disk data caches:
             # i.e. clear values from both square and cube
             st.cache_data.clear()
-            roo_data, timestamp = init_baselines()
+            roo_data, sd_roo_data, timestamp = init_baselines()
             dk_lineups = init_DK_lineups()
             fd_lineups = init_FD_lineups()
             hold_display = roo_data
@@ -119,10 +126,17 @@ with tab1:
     hold_container = st.empty()
 
     with options_container:
-        site_var = st.selectbox("Select a Site", ["Draftkings", "FanDuel"])
+        col1, col2 = st.columns([4, 4])
+        with col1:
+            site_var = st.selectbox("Select a Site", ["Draftkings", "FanDuel"])
+        with col2:
+            type_var = st.selectbox("Select a Type", ["Full Slate", "Showdown"])
         
     with hold_container:
-        display = hold_display[hold_display['Site'] == site_var]
+        if type_var == "Full Slate":
+            display = hold_display[hold_display['Site'] == site_var]
+        elif type_var == "Showdown":
+            display = sd_roo_data
         st.dataframe(display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(player_roo_format, precision=2), height=750, use_container_width = True)
     
     st.download_button(
@@ -137,7 +151,7 @@ with tab2:
     with col1:
         if st.button("Load/Reset Data", key='reset2'):
             st.cache_data.clear()
-            roo_data, timestamp = init_baselines()
+            roo_data, sd_roo_data, timestamp = init_baselines()
             hold_display = roo_data
             dk_lineups = init_DK_lineups()
             fd_lineups = init_FD_lineups()