Add user-configurable options for lineup generation in Data Load tab

Introduced an "Info and Filters" expander that allows users to set priority variables, lineup counts, and salary constraints for both Draftkings and Fanduel. Updated the lineup loading functions to utilize these new parameters, enhancing customization and user experience in lineup generation.

app.py

@@ -480,6 +480,38 @@ with st.container():
         elif type_var == 'Showdown':
             salary_max = 60000
 
+    with st.expander("Info and Filters"):
+        prio_col, optimals_site_col, optimals_macro_col, optimals_salary_col, optimals_stacks_col = st.columns(5)
+
+        with prio_col:
+            prio_var = st.radio("Which priority variable do you want to use?", ('proj', 'Own', 'Mix'), key='prio_var_radio')
+            prio_mix = st.number_input("If Mix, what split of Projection/Ownership to dedicate to Projection?", min_value=0, max_value=100, value=50, step=1)
+            lineup_num = st.number_input("How many lineups do you want to work with?", min_value=1000, max_value=50000, value=25000, step=100, key='lineup_download_var_input')
+
+        with optimals_site_col:
+            if site_var == 'Draftkings':
+                slate_var3 = st.radio("Which slate data are you loading?", (nfl_slate_names_dk if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
+            elif site_var == 'Fanduel':
+                slate_var3 = st.radio("Which slate data are you loading?", (nfl_slate_names_fd if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
+            
+        with optimals_macro_col:
+            lineup_num_var = st.number_input("How many lineups do you want to display?", min_value=1, max_value=1000, value=150, step=1, key='lineup_num_var_input')
+
+        with optimals_salary_col:
+            if site_var == 'Draftkings':
+                salary_min_var = st.number_input("Minimum salary used", min_value = 0, max_value = 50000, value = 49000, step = 100, key = 'salary_min_var_dk')
+                salary_max_var = st.number_input("Maximum salary used", min_value = 0, max_value = 50000, value = 50000, step = 100, key = 'salary_max_var_dk')
+            elif site_var == 'Fanduel':
+                salary_min_var = st.number_input("Minimum salary used", min_value = 0, max_value = 60000, value = 59000, step = 100, key = 'salary_min_var_fd')
+                salary_max_var = st.number_input("Maximum salary used", min_value = 0, max_value = 60000, value = 60000, step = 100, key = 'salary_max_var_fd')
+        with optimals_stacks_col:
+            if site_var == 'Draftkings':
+                min_stacks_var = st.number_input("Minimum stacks used", min_value = 0, max_value = 5, value = 1, step = 1, key = 'min_stacks_var_dk')
+                max_stacks_var = st.number_input("Maximum stacks used", min_value = 0, max_value = 5, value = 5, step = 1, key = 'max_stacks_var_dk')
+            elif site_var == 'Fanduel':
+                min_stacks_var = st.number_input("Minimum stacks used", min_value = 0, max_value = 4, value = 1, step = 1, key = 'min_stacks_var_fd')
+                max_stacks_var = st.number_input("Maximum stacks used", min_value = 0, max_value = 4, value = 4, step = 1, key = 'max_stacks_var_fd')
+
 try:
     selected_tab = st.segmented_control(
     "Select Tab",
@@ -565,7 +597,7 @@ if selected_tab == 'Data Load':
         if upload_toggle == 'Paydirt DB':
             if st.button("Load from Database after inserting site CSV"):
                 if site_var == 'Draftkings':
-                    portfolio_load = init_DK_NFL_lineups(type_var, 'Main', 'proj', 50, dk_nfl_showdown_db_translation, 1000, [])
+                    portfolio_load = init_DK_NFL_lineups(type_var, 'Main', prio_var, 50, dk_nfl_showdown_db_translation, lineup_num_var, [])
                     st.session_state['db_portfolio_file'] = pd.DataFrame(portfolio_load)
                     st.session_state['portfolio_loaded'] = True
                     if 'portfolio' in st.session_state:
@@ -573,7 +605,7 @@ if selected_tab == 'Data Load':
                     if 'export_portfolio' in st.session_state:
                         del st.session_state['export_portfolio']
                 else:
-                    portfolio_load = init_FD_NFL_lineups(type_var, 'Main', 'proj', 50, fd_nfl_showdown_db_translation, 1000, [])
+                    portfolio_load = init_FD_NFL_lineups(type_var, 'Main', prio_var, 50, fd_nfl_showdown_db_translation, lineup_num_var, [])
                     st.session_state['db_portfolio_file'] = pd.DataFrame(portfolio_load)
                     st.session_state['portfolio_loaded'] = True
                     if 'portfolio' in st.session_state:

database_queries.py

@@ -13,6 +13,50 @@ right_nfl_name_teams = ['Broncos', 'Commanders', 'Bengals', 'Cardinals', 'Rams',
 'Jets', 'Colts', 'Dolphins', 'Lions', 'Raiders', 'Falcons', 'Seahawks', 'Texans', 'Saints', 'Panthers', 'Giants', 'Browns', 'Titans', 'Eagles', 'Cowboys',
 'Chiefs', 'Chargers', 'Ravens', 'Bills', 'Vikings', 'Bears']
 
+def define_dk_nfl_showdown_slates():
+    collection = nfl_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.iloc[0]['Team'] + ' vs. ' + slate_data.iloc[0]['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
+def define_fd_nfl_showdown_slates():
+    collection = nfl_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.iloc[0]['Team'] + ' vs. ' + slate_data.iloc[0]['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
+try:
+    nfl_slate_names_dk, nfl_slate_name_lookup_dk = define_dk_nfl_showdown_slates()
+except:
+    nfl_slate_names_dk = []
+    nfl_slate_name_lookup_dk = {}
+    
+try:
+    nfl_slate_names_fd, nfl_slate_name_lookup_fd = define_fd_nfl_showdown_slates()
+except:
+    nfl_slate_names_fd = []
+    nfl_slate_name_lookup_fd = {}
+
 def init_nfl_baselines(type_var: str, site_var: str, slate_var: str):
 
     if type_var == 'Classic':