Implement segmented control for tab selection and enhance session state management in app.py

- Introduced a segmented control for selecting between "Data Load" and "Contest Analysis" tabs, improving user navigation.
- Updated session state handling to ensure consistent access to selected sport and game type across various components, enhancing data integrity and user experience.
- Added custom styling for tabs to improve visual appeal and usability.

app.py

@@ -119,8 +119,64 @@ def highlight_row_condition(row):
 player_exposure_format = {'Exposure Overall': '{:.2%}', 'Exposure Top 1%': '{:.2%}', 'Exposure Top 5%': '{:.2%}', 'Exposure Top 10%': '{:.2%}', 'Exposure Top 20%': '{:.2%}'}
 dupe_format = {'uniques%': '{:.2%}', 'under_5%': '{:.2%}', 'under_10%': '{:.2%}'}
 
-tab1, tab2 = st.tabs(["Data Load", "Contest Analysis"])
-with tab1:
+st.markdown("""
+<style>
+    /* Tab styling */
+    .stElementContainer [data-baseweb="button-group"] {
+        gap: 2.000rem;
+        padding: 4px;
+    }
+    .stElementContainer [kind="segmented_control"] {
+        height: 2.000rem;
+        white-space: pre-wrap;
+        background-color: #DAA520;
+        color: white;
+        border-radius: 20px;
+        gap: 1px;
+        padding: 10px 20px;
+        font-weight: bold;
+        transition: all 0.3s ease;
+    }
+    .stElementContainer [kind="segmented_controlActive"] {
+        height: 3.000rem;
+        background-color: #DAA520;
+        border: 3px solid #FFD700;
+        border-radius: 10px;
+        color: black;
+    }
+    .stElementContainer [kind="segmented_control"]:hover {
+        background-color: #FFD700;
+        cursor: pointer;
+    }
+
+    div[data-baseweb="select"] > div {
+        background-color: #DAA520;
+        color: white;
+    }
+
+</style>""", unsafe_allow_html=True)
+
+try:
+    selected_tab = st.segmented_control(
+        "Select Tab",
+        options=["Data Load", "Contest Analysis"],
+        selection_mode='single',
+        default='Data Load',
+        width='stretch',
+        label_visibility='collapsed',
+        key='tab_selector'
+    )
+except:
+    selected_tab = st.segmented_control(
+        "Select Tab",
+        options=["Data Load", "Contest Analysis"],
+        selection_mode='single',
+        default='Data Load',
+        label_visibility='collapsed',
+        key='tab_selector'
+    )
+
+if selected_tab == 'Data Load':
     data_select, contest_upload, portfolio_upload = st.columns(3)
 
     with data_select:
@@ -129,10 +185,10 @@ with tab1:
         sport_options, date_options = st.columns(2)
         parse_type = 'Manual'
         with sport_options:
-            sport_select = st.selectbox("Select Sport", ['MLB', 'MMA', 'GOLF', 'NBA', 'NHL', 'WNBA', 'NAS'], key='sport_select')
-            type_var = st.selectbox("Select Game Type", ['Classic', 'Showdown'], key='type_var')
+            st.session_state['sport_select'] = st.selectbox("Select Sport", ['MLB', 'MMA', 'GOLF', 'NBA', 'NHL', 'WNBA', 'NAS'], key='sport_select')
+            st.session_state['type_var'] = st.selectbox("Select Game Type", ['Classic', 'Showdown'], key='type_var')
             try:
-                contest_names, curr_info = grab_contest_names(db, sport_select, type_var)
+                contest_names, curr_info = grab_contest_names(db, st.session_state['sport_select'], st.session_state['type_var'])
             except:
                 st.error("No contests found for this sport and/or game type")
                 st.stop()
@@ -156,8 +212,8 @@ with tab1:
                 del st.session_state['Contest_file']
             if 'Contest_file' not in st.session_state:
                 if st.button('Load Contest Data', key='load_contest_data'):
-                    st.session_state['player_info'], st.session_state['info_maps'] = grab_contest_player_info(db, sport_select, type_var, date_select, contest_name_var, contest_id_map)
-                    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, st.session_state['sport_select'], st.session_state['type_var'], date_select, contest_name_var, contest_id_map)
+                    st.session_state['Contest_file'] = grab_contest_data(st.session_state['sport_select'], contest_name_var, contest_id_map, date_select, date_select2)
             else:
                 pass
     with contest_upload:
@@ -169,7 +225,7 @@ with tab1:
                 del st.session_state['Contest_file']
             if 'Contest_file' not in st.session_state:
                 st.session_state['Contest_upload'] = st.file_uploader("Upload Contest File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
-                st.session_state['player_info'], st.session_state['info_maps'] = grab_contest_player_info(db, sport_select, type_var, date_select, contest_name_var, contest_id_map)
+                st.session_state['player_info'], st.session_state['info_maps'] = grab_contest_player_info(db, st.session_state['sport_select'], st.session_state['type_var'], date_select, contest_name_var, contest_id_map)
                 try:
                     st.session_state['Contest_file'] = pd.read_csv(st.session_state['Contest_upload'])
                 except:
@@ -205,7 +261,7 @@ with tab1:
             
 
     if 'Contest_file' 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'], type_var, st.session_state['player_info'], sport_select, st.session_state['portfolio_df'])
+        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['type_var'], st.session_state['player_info'], st.session_state['sport_select'], st.session_state['portfolio_df'])
         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:
@@ -216,7 +272,7 @@ with tab1:
                 st.warning("If you have confirmed that the data is correct, you can send the CSV to the database to enrich Paydirt's sources and help us create actionable tools and algorithms >>")
             with upload_col:
                 if st.button('Send file to Database?', key='export_contest_file'):
-                    return_message = export_contest_file(db, sport_select, type_var, date_select, contest_id_map[contest_name_var], st.session_state['Contest_file'])
+                    return_message = export_contest_file(db, st.session_state['sport_select'], st.session_state['type_var'], date_select, contest_id_map[contest_name_var], st.session_state['Contest_file'])
             with message_col:
                 try:
                     st.info(return_message)
@@ -237,7 +293,7 @@ with tab1:
         st.session_state['team_dict'] = st.session_state['info_maps']['team_dict']
         st.session_state['pos_dict'] = st.session_state['info_maps']['position_dict']
 
-with tab2:
+elif selected_tab == 'Contest Analysis':
     excluded_cols = ['BaseName', 'EntryCount']
     exclude_stacks = ['BaseName', 'EntryCount', 'SP', 'SP1', 'SP2', 'P1', 'P2']
     if 'Contest' in st.session_state and 'display_contest_info' not in st.session_state:
@@ -266,7 +322,7 @@ with tab2:
         actual_map = st.session_state['actual_dict']
         ownership_map = st.session_state['ownership_dict']
         
-        if type_var == 'Classic':
+        if st.session_state['type_var'] == 'Classic':
             # Vectorized stack calculation
             player_teams = working_df[st.session_state['stack_columns']].apply(
                 lambda x: x.map(team_map).fillna('')
@@ -328,7 +384,7 @@ with tab2:
             working_df['finish'] = working_df['index']
             working_df = working_df.drop(['sorted', 'index'], axis=1)
         
-        elif type_var == 'Showdown':
+        elif st.session_state['type_var'] == 'Showdown':
             # Vectorized stack calculation for Showdown
             player_teams = working_df.iloc[:, 2:].apply(
                 lambda x: x.map(team_map).fillna('')
@@ -348,7 +404,7 @@ with tab2:
             working_df['stack'] = [result[0] for result in stack_results]
             working_df['stack_size'] = [result[1] for result in stack_results]
             
-            if sport_select == 'GOLF':
+            if st.session_state['sport_select'] == 'GOLF':
                 # Vectorized calculations for GOLF
                 player_salaries = working_df.apply(
                     lambda x: x.map(salary_map).fillna(0)
@@ -539,21 +595,21 @@ with tab2:
                         with col1:
                             pos_var = st.selectbox("Which position(s) would you like to view?",  ['All', 'Specific'], key='pos_var')
                         with col2:
-                            if sport_select == 'MLB':
+                            if st.session_state['sport_select'] == 'MLB':
                                 pos_select = st.multiselect("Select your position(s)", ['P', 'C', '1B', '2B', '3B', 'SS', 'OF'], key='pos_select')
-                            elif sport_select == 'NBA':
+                            elif st.session_state['sport_select'] == 'NBA':
                                 pos_select = st.multiselect("Select your position(s)", ['PG', 'SG', 'SF', 'PF', 'C'], key='pos_select')
-                            elif sport_select == 'WNBA':
+                            elif st.session_state['sport_select'] == 'WNBA':
                                 pos_select = st.multiselect("Select your position(s)", ['PG', 'SG', 'SF', 'PF'], key='pos_select')
-                            elif sport_select == 'NFL':
+                            elif st.session_state['sport_select'] == 'NFL':
                                 pos_select = st.multiselect("Select your position(s)", ['QB', 'RB', 'WR', 'TE', 'DST'], key='pos_select')
-                            elif sport_select == 'NHL':
+                            elif st.session_state['sport_select'] == 'NHL':
                                 pos_select = st.multiselect("Select your position(s)", ['W', 'C', 'D', 'G'], key='pos_select')
-                            elif sport_select == 'MMA':
+                            elif st.session_state['sport_select'] == 'MMA':
                                 pos_select = st.multiselect("Select your position(s)", ['All the same position', 'So', 'Yeah', 'Idk'], key='pos_select')
-                            elif sport_select == 'GOLF':
+                            elif st.session_state['sport_select'] == 'GOLF':
                                 pos_select = st.multiselect("Select your position(s)", ['All the same position', 'So', 'Yeah', 'Idk'], key='pos_select')
-                            elif sport_select == 'NAS':
+                            elif st.session_state['sport_select'] == 'NAS':
                                 pos_select = st.multiselect("Select your position(s)", ['All the same position', 'So', 'Yeah', 'Idk'], key='pos_select')
                         submitted = st.form_submit_button("Submit")
                         if submitted:
@@ -577,9 +633,9 @@ with tab2:
                 if comp_player_var == 'Yes':
                     player_exp_comp = create_player_comparison(st.session_state['display_contest_info'], st.session_state['player_columns'], comp_player_select)
                     hold_frame = player_exp_comp.copy()
-                    if sport_select == 'GOLF':
+                    if st.session_state['sport_select'] == 'GOLF':
                         hold_frame['Pos'] = 'G'
-                    elif sport_select == 'NAS':
+                    elif st.session_state['sport_select'] == 'NAS':
                         hold_frame['Pos'] = 'C'
                     else:
                         hold_frame['Pos'] = hold_frame['Player'].map(st.session_state['map_dict']['pos_map'])
@@ -595,9 +651,9 @@ with tab2:
 
                         st.session_state['player_frame'] = create_player_exposures(st.session_state['display_contest_info'], st.session_state['player_columns'])
                         hold_frame = st.session_state['player_frame'].copy()
-                        if sport_select == 'GOLF':
+                        if st.session_state['sport_select'] == 'GOLF':
                             hold_frame['Pos'] = 'G'
-                        elif sport_select == 'NAS':
+                        elif st.session_state['sport_select'] == 'NAS':
                             hold_frame['Pos'] = 'NAS'
                         else:
                             hold_frame['Pos'] = hold_frame['Player'].map(st.session_state['map_dict']['pos_map'])
@@ -615,9 +671,9 @@ with tab2:
                     else:
                         st.session_state['player_frame'] = create_player_exposures(st.session_state['display_contest_info'], st.session_state['player_columns'], st.session_state['entry_names'])
                         hold_frame = st.session_state['player_frame'].copy()
-                        if sport_select == 'GOLF':
+                        if st.session_state['sport_select'] == 'GOLF':
                             hold_frame['Pos'] = 'G'
-                        elif sport_select == 'NAS':
+                        elif st.session_state['sport_select'] == 'NAS':
                             hold_frame['Pos'] = 'NAS'
                         else:
                             hold_frame['Pos'] = hold_frame['Player'].map(st.session_state['map_dict']['pos_map'])