Enhance player ROO data handling in app.py by introducing separate dataframes for DraftKings and FanDuel, improving clarity and organization in the data processing flow while maintaining existing functionality.

app.py

@@ -68,6 +68,9 @@ def init_baselines():
 
     roo_data = player_frame.drop(columns=['_id'])
     roo_data['Salary'] = roo_data['Salary'].astype(int)
+    
+    dk_roo = roo_data[roo_data['Site'] == 'Draftkings']
+    fd_roo = roo_data[roo_data['Site'] == 'Fanduel']
 
     collection = db["Player_SD_Range_Of_Outcomes"] 
     cursor = collection.find()
@@ -85,7 +88,7 @@ def init_baselines():
     scoring_percentages['8+ runs'] = scoring_percentages['8+ runs'].replace('%', '', regex=True).astype(float) / 100
     scoring_percentages['Win Percentage'] = scoring_percentages['Win Percentage'].replace('%', '', regex=True).astype(float) / 100
     
-    return roo_data, sd_roo_data, scoring_percentages
+    return roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo
 
 @st.cache_data(ttl = 60)
 def init_DK_lineups(type_var, slate_var):  
@@ -241,7 +244,7 @@ col1, col2 = st.columns([1, 9])
 with col1:
     if st.button("Load/Reset Data", key='reset'):
         st.cache_data.clear()
-        roo_data, sd_roo_data, scoring_percentages = init_baselines()
+        roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo = init_baselines()
         hold_display = roo_data
         dk_lineups = init_DK_lineups('Regular', 'Main')
         fd_lineups = init_FD_lineups('Regular', 'Main')
@@ -258,7 +261,7 @@ with col2:
 
 tab1, tab2, tab3 = st.tabs(["Scoring Percentages", "Player ROO", "Optimals"])
 
-roo_data, sd_roo_data, scoring_percentages = init_baselines()
+roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo = init_baselines()
 hold_display = roo_data
 
 with tab1:
@@ -286,25 +289,27 @@ with tab2:
             else:
                 team_select2 = None
     if slate_type_var2 == 'Regular':
-        player_roo_raw = roo_data.copy()
         if site_var == 'Draftkings':
-            player_roo_raw['Site'] = 'Draftkings'
+        
+            player_roo_raw = dk_roo.copy()
 
             if pos_var2 == 'All':
                 pass
             elif pos_var2 == 'Pitchers':
-                player_roo_raw = player_roo_raw[player_roo_raw['Position'] == 'SP']
+                player_roo_raw = player_roo_raw[player_roo_raw['Position'].str.contains('P')]
             elif pos_var2 == 'Hitters':
-                player_roo_raw = player_roo_raw[player_roo_raw['Position'] != 'SP']
+                player_roo_raw = player_roo_raw[player_roo_raw['Position'].str.contains('P')]
+
         elif site_var == 'Fanduel':
-            player_roo_raw['Site'] = 'Fanduel'
+            
+            player_roo_raw = fd_roo.copy()
 
             if pos_var2 == 'All':
                 pass
             elif pos_var2 == 'Pitchers':
-                player_roo_raw = player_roo_raw[player_roo_raw['Position'] == 'P']
+                player_roo_raw = player_roo_raw[player_roo_raw['Position'].str.contains('P')]
             elif pos_var2 == 'Hitters':
-                player_roo_raw = player_roo_raw[player_roo_raw['Position'] != 'P']
+                player_roo_raw = player_roo_raw[player_roo_raw['Position'].str.contains('P')]
         
         if slate_var2 == 'Main':
             player_roo_raw = player_roo_raw[player_roo_raw['Slate'] == 'main_slate']