Update app.py to switch database connections for DK and FD lineups, modify displayed columns, and enhance player data filtering with a radio button for site selection, improving user experience and data management.

app.py

@@ -77,17 +77,18 @@ def init_baselines():
     cursor = collection.find()
     team_frame = pd.DataFrame(cursor)
     scoring_percentages = team_frame.drop(columns=['_id'])
+    scoring_percentages = scoring_percentages[['Names', 'Avg First Inning', 'First Inning Lead Percentage', 'Avg Fifth Inning', 'Fifth Inning Lead Percentage', 'Avg Score', '8+ runs', 'Win Percentage']]
     
     return roo_data, sd_roo_data, scoring_percentages
 
 @st.cache_data(ttl = 60)
 def init_DK_lineups():  
         
-        collection = db['DK_MLB_SD1_seed_frame']
+        collection = db2['DK_MLB_SD1_seed_frame']
         cursor = collection.find().limit(10000)
     
         raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'FLEX6', 'salary', 'proj', 'Own']]
+        raw_display = raw_display[['CPT', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'FLEX6', 'salary', 'proj', 'Own']]
         DK_seed = raw_display.to_numpy()
 
         return DK_seed
@@ -95,11 +96,11 @@ def init_DK_lineups():
 @st.cache_data(ttl = 60)
 def init_FD_lineups():  
         
-        collection = db['FD_MLB_SD1_seed_frame']
+        collection = db2['FD_MLB_SD1_seed_frame']
         cursor = collection.find().limit(10000)
     
         raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'FLEX6', 'salary', 'proj', 'Own']]
+        raw_display = raw_display[['CPT', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Own']]
         FD_seed = raw_display.to_numpy()
 
         return FD_seed
@@ -125,7 +126,15 @@ with tab1:
 
 with tab2:
     st.title("Player ROO")
-    st.dataframe(sd_roo_data)
+    with st.expander("Info and Filters"):
+        site_var = st.radio("Select site", ["Draftkings", "Fanduel"])
+        if site_var == "Draftkings":
+            display_data = sd_roo_data[sd_roo_data['site'] == 'Draftkings']
+            display_data = display_data[display_data['slate'] == 'DK SD1']
+        elif site_var == "Fanduel":
+            display_data = sd_roo_data[sd_roo_data['site'] == 'Fanduel']
+            display_data = display_data[display_data['slate'] == 'FD SD1']
+        st.dataframe(display_data)
 
 with tab3:
     with st.expander("Info and Filters"):