Enhance map_mask_parse function to include site-specific variable handling for DraftKings and FanDuel, ensuring accurate projection and ownership calculations across data exports.

src/streamlit_app.py

@@ -517,8 +517,11 @@ def normalize_special_characters(text):
     return ascii_text
 
 @st.cache_data
-def map_mask_parse(df: pd.DataFrame, map: dict, threshold: float):
-    proj_df = df.iloc[:, :-7].replace(map).astype(float)
+def map_mask_parse(df: pd.DataFrame, map: dict, threshold: float, site_var: str):
+    if site_var == 'Draftkings':
+        proj_df = df.iloc[:, :-7].replace(map).astype(float)
+    elif site_var == 'Fanduel':
+        proj_df = df.iloc[:, :-7].replace(map).astype(float)
     print(proj_df.head(10))
     mask = (proj_df >= threshold).all(axis=1)
     df = df[mask]
@@ -1272,22 +1275,22 @@ if selected_tab == 'Optimals':
                 data_export = data_export[data_export['Team_count'] >= min_stacks_var]
                 data_export = data_export[data_export['Team_count'] <= max_stacks_var]
                 if site_var == 'Draftkings':
-                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var)
-                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var)
+                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Median'])), projection_var, site_var)
+                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Own'])), ownership_var, site_var)
                 elif site_var == 'Fanduel':
-                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var)
-                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var)
+                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Median'])), projection_var, site_var)
+                    data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Own'])), ownership_var, site_var)
 
                 name_export = name_export[name_export['salary'] >= salary_min_var]
                 name_export = name_export[name_export['salary'] <= salary_max_var]
                 name_export = name_export[name_export['Team_count'] >= min_stacks_var]
                 name_export = name_export[name_export['Team_count'] <= max_stacks_var]
                 if site_var == 'Draftkings':
-                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var)
-                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var)
+                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Median'])), projection_var, site_var)
+                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Own'])), ownership_var, site_var)
                 elif site_var == 'Fanduel':
-                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var)
-                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var)
+                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Median'])), projection_var, site_var)
+                    name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Own'])), ownership_var, site_var)
                 
                 reg_opt_col, pm_opt_col = st.columns(2)
                 with reg_opt_col:
@@ -1364,11 +1367,11 @@ if selected_tab == 'Optimals':
     st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['salary'] >= salary_min_var]
     st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['salary'] <= salary_max_var]
     if site_var == 'Draftkings':
-        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var)
-        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var)
+        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Median'])), projection_var, site_var)
+        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Own'])), ownership_var, site_var)
     elif site_var == 'Fanduel':
-        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var)
-        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var)
+        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Median'])), projection_var, site_var)
+        st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Own'])), ownership_var, site_var)
     st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['Team_count'] >= min_stacks_var]
     st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['Team_count'] <= max_stacks_var]
     st.session_state.data_export_display = st.session_state.data_export_display.reset_index(drop=True)