Enhance contest file loading in app.py and load_contest_file.py to support additional data structures

- Updated load_contest_file function to return salary_df, team_df, and pos_df based on the helper variable, improving data handling for contest files.
- Modified app.py to accommodate the new return values, ensuring proper session state management for salary, team, and position data.
- Maintained existing functionality while expanding the data output capabilities during contest file processing.

app.py

@@ -125,9 +125,9 @@ with tab1:
 
     if 'Contest_file' in st.session_state:
         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['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['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'], 1, 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['entry_list'], check_lineups = load_contest_file(st.session_state['Contest_file'], st.session_state['player_info'], 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'], 0, st.session_state['player_info'], 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:
@@ -137,9 +137,14 @@ 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'] = 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']
+        if 'Contest_file_helper' not in st.session_state:
+            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']
+        else:
+            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.write(st.session_state['salary_dict'])
 

global_func/load_contest_file.py

@@ -2,7 +2,7 @@ import streamlit as st
 import pandas as pd
 from rapidfuzz import process, fuzz
 
-def load_contest_file(upload, helper = None, sport = None):
+def load_contest_file(upload, helper_var, helper = None, sport = None):
     if upload is not None:
         try:
             try:
@@ -136,7 +136,10 @@ def load_contest_file(upload, helper = None, sport = None):
             entry_list = list(set(df['BaseName']))
             entry_list.sort()
             
-            return cleaned_df, ownership_df, fpts_df, entry_list, check_lineups
+            if helper_var == 1:
+                return cleaned_df, ownership_df, fpts_df, salary_df, team_df, pos_df, entry_list, check_lineups
+            else:
+                return cleaned_df, ownership_df, fpts_df, entry_list, check_lineups
             
         except Exception as e:
             st.error(f'Error loading file: {str(e)}')