Add player information retrieval in app.py for enhanced contest data processing

- Introduced a new function, grab_contest_player_info, to fetch player details based on contest selection, improving data accessibility.
- Updated session state management to store player information and related mappings, enhancing user experience during contest analysis.
- Maintained existing functionality while expanding the capabilities of contest data handling.

app.py

@@ -32,6 +32,27 @@ def grab_contest_names(db, sport, type):
     
     return contest_names, contest_id_map, curr_info
 
+def grab_contest_player_info(db, sport, type, contest_name, contest_id_map):
+    if type == 'Classic':
+        db_type = 'reg'
+    elif type == 'Showdown':
+        db_type = 'sd'
+    collection = db[f'{sport}_{db_type}_player_info']
+    cursor = collection.find()
+
+    player_info = pd.DataFrame(list(cursor)).drop('_id', axis=1)
+    player_info = player_info[player_info['Contest ID'] == contest_id_map[contest_name]]
+
+    info_maps = {
+        'position_dict': dict(zip(player_info['Display Name'], player_info['Position'])),
+        'salary_dict': dict(zip(player_info['Display Name'], player_info['Salary'])),
+        'team_dict': dict(zip(player_info['Display Name'], player_info['Team'])),
+        'opp_dict': dict(zip(player_info['Display Name'], player_info['Opp'])),
+        'fpts_avg_dict': dict(zip(player_info['Display Name'], player_info['Avg FPTS']))
+    }
+    
+    return player_info, info_maps
+
 db = init_conn()
 
 ## import global functions
@@ -84,6 +105,7 @@ with tab1:
             if 'Contest_file' not in st.session_state:
                 if st.button('Load Contest Data', key='load_contest_data'):
                     st.session_state['Contest_file'] = grab_contest_data(sport_select, contest_name_var, contest_id_map, date_select, date_select2)
+                    st.session_state['player_info'], st.session_state['info_maps'] = grab_contest_player_info(db, sport_select, type_var, contest_name_var, contest_id_map)
             else:
                 pass
     with col2:
@@ -101,12 +123,10 @@ with tab1:
                 pass
 
     if 'Contest_file' in st.session_state:
-        st.table(st.session_state['Contest_file'].head(100))
-        st.write(st.session_state['Contest_file_helper'])
         if 'Contest_file_helper' in st.session_state:
-            st.session_state['Contest'], st.session_state['ownership_df'], st.session_state['actual_df'], st.session_state['salary_df'], st.session_state['team_df'], st.session_state['pos_df'], st.session_state['entry_list'], check_lineups = load_contest_file(st.session_state['Contest_file'], st.session_state['Contest_file_helper'], sport_select)
+            st.session_state['Contest'], st.session_state['ownership_df'], st.session_state['actual_df'], st.session_state['entry_list'], check_lineups = load_contest_file(st.session_state['Contest_file'], st.session_state['Contest_file_helper'], sport_select)
         else:
-            st.session_state['Contest'], st.session_state['ownership_df'], st.session_state['actual_df'], st.session_state['salary_df'], st.session_state['team_df'], st.session_state['pos_df'], st.session_state['entry_list'], check_lineups = load_contest_file(st.session_state['Contest_file'], None, sport_select)
+            st.session_state['Contest'], st.session_state['ownership_df'], st.session_state['actual_df'], st.session_state['entry_list'], check_lineups = load_contest_file(st.session_state['Contest_file'], None, sport_select)
         st.session_state['Contest'] = st.session_state['Contest'].dropna(how='all')
         st.session_state['Contest'] = st.session_state['Contest'].reset_index(drop=True)
         if st.session_state['Contest'] is not None:
@@ -116,9 +136,9 @@ with tab1:
     if 'Contest_file' in st.session_state:
         st.session_state['ownership_dict'] = dict(zip(st.session_state['ownership_df']['Player'], st.session_state['ownership_df']['Own']))
         st.session_state['actual_dict'] = dict(zip(st.session_state['actual_df']['Player'], st.session_state['actual_df']['FPTS']))
-        st.session_state['salary_dict'] = dict(zip(st.session_state['salary_df']['Player'], st.session_state['salary_df']['Salary']))
-        st.session_state['team_dict'] = dict(zip(st.session_state['team_df']['Player'], st.session_state['team_df']['Team']))
-        st.session_state['pos_dict'] = dict(zip(st.session_state['pos_df']['Player'], st.session_state['pos_df']['Pos']))
+        st.session_state['salary_dict'] = st.session_state['info_maps']['salary_dict']
+        st.session_state['team_dict'] = st.session_state['info_maps']['team_dict']
+        st.session_state['pos_dict'] = st.session_state['info_maps']['position_dict']
 
         st.write(st.session_state['salary_dict'])