segmented controllers rather than tabs

src/streamlit_app.py

@@ -25,6 +25,42 @@ freq_format = {'Exposure': '{:.2%}', 'Proj Own': '{:.2%}', 'Edge': '{:.2%}'}
 dk_columns = ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 fd_columns = ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 
+st.markdown("""
+<style>
+    /* Tab styling */
+    .stElementContainer [data-baseweb="button-group"] {
+        gap: 8px;
+        padding: 4px;
+    }
+    .stElementContainer [kind="segmented_control"] {
+        height: 45px;
+        white-space: pre-wrap;
+        background-color: #DAA520;
+        color: white;
+        border-radius: 10px;
+        gap: 1px;
+        padding: 10px 20px;
+        font-weight: bold;
+        transition: all 0.3s ease;
+    }
+    .stElementContainer [kind="segmented_controlActive"] {
+        height: 50px;
+        background-color: #DAA520;
+        border: 3px solid #FFD700;
+        color: white;
+    }
+    .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)
+
 @st.cache_data(ttl = 600)
 def init_DK_seed_frames(sharp_split):
 
@@ -183,146 +219,17 @@ def sim_contest(Sim_size, seed_frame, maps_dict, Contest_Size):
         
     return Sim_Winners
 
-tab1, tab2 = st.tabs(['Contest Sims', 'Data Export'])
-with tab2:
-    col1, col2 = st.columns([1, 7])
-    with col1:
-        if st.button("Load/Reset Data", key='reset1'):
-              st.cache_data.clear()
-              for key in st.session_state.keys():
-                  del st.session_state[key]
-              DK_seed = init_DK_seed_frames(10000)
-              FD_seed = init_FD_seed_frames(10000)
-              dk_raw, fd_raw = init_baselines('Main Slate')
-              dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_ID))
-              fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_ID))
-              
-        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'))
-        site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
-        sharp_split_var = st.number_input("How many lineups do you want?", value=10000, max_value=500000, min_value=10000, step=10000)
-        
-        if site_var1 == 'Draftkings':
+selected_tab = st.segmented_control(
+    "Select Tab",
+    options=["Contest Sims", "Data Export"],
+    selection_mode='single',
+    default='Contest Sims',
+    width='stretch',
+    label_visibility='collapsed',
+    key='tab_selector'
+)
             
-            team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
-            if team_var1 == 'Specific Teams':
-                    dk_raw, fd_raw = init_baselines('Main Slate')
-                    team_var2 = st.multiselect('Which teams do you want?', options = dk_raw['Team'].unique())
-            elif team_var1 == 'Full Slate':
-                    dk_raw, fd_raw = init_baselines('Main Slate')
-                    team_var2 = dk_raw.Team.values.tolist()
-            
-            stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
-            if stack_var1 == 'Specific Stack Sizes':
-                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = [5, 4, 3, 2, 1, 0])
-            elif stack_var1 == 'Full Slate':
-                    stack_var2 = [5, 4, 3, 2, 1, 0]
-                    
-        elif site_var1 == 'Fanduel':
-            
-            team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
-            if team_var1 == 'Specific Teams':
-                    dk_raw, fd_raw = init_baselines('Main Slate')
-                    team_var2 = st.multiselect('Which teams do you want?', options = fd_raw['Team'].unique())
-            elif team_var1 == 'Full Slate':
-                    dk_raw, fd_raw = init_baselines('Main Slate')
-                    team_var2 = fd_raw.Team.values.tolist()
-            
-            stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
-            if stack_var1 == 'Specific Stack Sizes':
-                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = [5, 4, 3, 2, 1, 0])
-            elif stack_var1 == 'Full Slate':
-                    stack_var2 = [5, 4, 3, 2, 1, 0]
-        
-
-        if st.button("Prepare data export", key='data_export'):
-                if 'working_seed' in st.session_state:
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
-                elif 'working_seed' not in st.session_state:
-                    if site_var1 == 'Draftkings':
-                        if slate_var1 == 'Main Slate':
-                            st.session_state.working_seed = init_DK_seed_frames(sharp_split_var)
-                            dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                        elif slate_var1 == 'Secondary Slate':
-                            st.session_state.working_seed = init_DK_Secondary_seed_frames(sharp_split_var)
-                            dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                            
-                        raw_baselines = dk_raw
-                        column_names = dk_columns
-                
-                    elif site_var1 == 'Fanduel':
-                        if slate_var1 == 'Main Slate':
-                            st.session_state.working_seed = init_FD_seed_frames(sharp_split_var)
-                            fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                        elif slate_var1 == 'Secondary Slate':
-                            st.session_state.working_seed = init_FD_Secondary_seed_frames(sharp_split_var)
-                            fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                            
-                        raw_baselines = fd_raw
-                        column_names = fd_columns
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
-                data_export = st.session_state.working_seed.copy()
-                for col in range(9):
-                    data_export[:, col] = np.array([dk_id_dict.get(x, x) for x in data_export[:, col]])
-                st.download_button(
-                    label="Export optimals set",
-                    data=convert_df(data_export),
-                    file_name='NFL_optimals_export.csv',
-                    mime='text/csv',
-                )
-            
-    with col2:
-        if st.button("Load Data", key='load_data'):
-            if site_var1 == 'Draftkings':
-                if 'working_seed' in st.session_state:
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
-                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
-                elif 'working_seed' not in st.session_state:
-                    if slate_var1 == 'Main Slate':
-                        st.session_state.working_seed = init_DK_seed_frames(sharp_split_var)
-                        dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                        dk_raw, fd_raw = init_baselines('Main Slate')
-                        
-                    elif slate_var1 == 'Secondary Slate':
-                        st.session_state.working_seed = init_DK_Secondary_seed_frames(sharp_split_var)
-                        dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                        dk_raw, fd_raw = init_baselines('Secondary Slate')
-                        
-                    raw_baselines = dk_raw
-                    column_names = dk_columns
-                
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
-                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
-                
-            elif site_var1 == 'Fanduel':
-                if 'working_seed' in st.session_state:
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
-                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
-                elif 'working_seed' not in st.session_state:
-                    if slate_var1 == 'Main Slate':
-                        st.session_state.working_seed = init_FD_seed_frames(sharp_split_var)
-                        fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                        dk_raw, fd_raw = init_baselines('Main Slate')
-                    elif slate_var1 == 'Secondary Slate':
-                        st.session_state.working_seed = init_FD_Secondary_seed_frames(sharp_split_var)
-                        fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
-                        dk_raw, fd_raw = init_baselines('Secondary Slate')
-                        
-                    raw_baselines = fd_raw
-                    column_names = fd_columns
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
-                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
-                
-        with st.container():
-            if 'data_export_display' in st.session_state:
-                st.dataframe(st.session_state.data_export_display.style.format(freq_format, precision=2), use_container_width = True)
-            
-with tab1:
+if selected_tab == "Contest Sims":
     col1, col2 = st.columns([1, 7])
     with col1:
         if st.button("Load/Reset Data", key='reset2'):
@@ -799,4 +706,142 @@ with tab1:
                         file_name='team_freq.csv',
                         mime='text/csv',
                         key='team'
-                    )
+                    )
+
+if selected_tab == "Data Export":
+    col1, col2 = st.columns([1, 7])
+    with col1:
+        if st.button("Load/Reset Data", key='reset1'):
+              st.cache_data.clear()
+              for key in st.session_state.keys():
+                  del st.session_state[key]
+              DK_seed = init_DK_seed_frames(10000)
+              FD_seed = init_FD_seed_frames(10000)
+              dk_raw, fd_raw = init_baselines('Main Slate')
+              dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_ID))
+              fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_ID))
+              
+        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'))
+        site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
+        sharp_split_var = st.number_input("How many lineups do you want?", value=10000, max_value=500000, min_value=10000, step=10000)
+        
+        if site_var1 == 'Draftkings':
+            
+            team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
+            if team_var1 == 'Specific Teams':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
+                    team_var2 = st.multiselect('Which teams do you want?', options = dk_raw['Team'].unique())
+            elif team_var1 == 'Full Slate':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
+                    team_var2 = dk_raw.Team.values.tolist()
+            
+            stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
+            if stack_var1 == 'Specific Stack Sizes':
+                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = [5, 4, 3, 2, 1, 0])
+            elif stack_var1 == 'Full Slate':
+                    stack_var2 = [5, 4, 3, 2, 1, 0]
+                    
+        elif site_var1 == 'Fanduel':
+            
+            team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
+            if team_var1 == 'Specific Teams':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
+                    team_var2 = st.multiselect('Which teams do you want?', options = fd_raw['Team'].unique())
+            elif team_var1 == 'Full Slate':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
+                    team_var2 = fd_raw.Team.values.tolist()
+            
+            stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
+            if stack_var1 == 'Specific Stack Sizes':
+                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = [5, 4, 3, 2, 1, 0])
+            elif stack_var1 == 'Full Slate':
+                    stack_var2 = [5, 4, 3, 2, 1, 0]
+        
+
+        if st.button("Prepare data export", key='data_export'):
+                if 'working_seed' in st.session_state:
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
+                elif 'working_seed' not in st.session_state:
+                    if site_var1 == 'Draftkings':
+                        if slate_var1 == 'Main Slate':
+                            st.session_state.working_seed = init_DK_seed_frames(sharp_split_var)
+                            dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                        elif slate_var1 == 'Secondary Slate':
+                            st.session_state.working_seed = init_DK_Secondary_seed_frames(sharp_split_var)
+                            dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                            
+                        raw_baselines = dk_raw
+                        column_names = dk_columns
+                
+                    elif site_var1 == 'Fanduel':
+                        if slate_var1 == 'Main Slate':
+                            st.session_state.working_seed = init_FD_seed_frames(sharp_split_var)
+                            fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                        elif slate_var1 == 'Secondary Slate':
+                            st.session_state.working_seed = init_FD_Secondary_seed_frames(sharp_split_var)
+                            fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                            
+                        raw_baselines = fd_raw
+                        column_names = fd_columns
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
+                data_export = st.session_state.working_seed.copy()
+                for col in range(9):
+                    data_export[:, col] = np.array([dk_id_dict.get(x, x) for x in data_export[:, col]])
+                st.download_button(
+                    label="Export optimals set",
+                    data=convert_df(data_export),
+                    file_name='NFL_optimals_export.csv',
+                    mime='text/csv',
+                )
+            
+    with col2:
+        if st.button("Load Data", key='load_data'):
+            if site_var1 == 'Draftkings':
+                if 'working_seed' in st.session_state:
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
+                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
+                elif 'working_seed' not in st.session_state:
+                    if slate_var1 == 'Main Slate':
+                        st.session_state.working_seed = init_DK_seed_frames(sharp_split_var)
+                        dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                        dk_raw, fd_raw = init_baselines('Main Slate')
+                        
+                    elif slate_var1 == 'Secondary Slate':
+                        st.session_state.working_seed = init_DK_Secondary_seed_frames(sharp_split_var)
+                        dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                        dk_raw, fd_raw = init_baselines('Secondary Slate')
+                        
+                    raw_baselines = dk_raw
+                    column_names = dk_columns
+                
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
+                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
+                
+            elif site_var1 == 'Fanduel':
+                if 'working_seed' in st.session_state:
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
+                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
+                elif 'working_seed' not in st.session_state:
+                    if slate_var1 == 'Main Slate':
+                        st.session_state.working_seed = init_FD_seed_frames(sharp_split_var)
+                        fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                        dk_raw, fd_raw = init_baselines('Main Slate')
+                    elif slate_var1 == 'Secondary Slate':
+                        st.session_state.working_seed = init_FD_Secondary_seed_frames(sharp_split_var)
+                        fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_ID))
+                        dk_raw, fd_raw = init_baselines('Secondary Slate')
+                        
+                    raw_baselines = fd_raw
+                    column_names = fd_columns
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
+                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
+                
+        with st.container():
+            if 'data_export_display' in st.session_state:
+                st.dataframe(st.session_state.data_export_display.style.format(freq_format, precision=2), use_container_width = True)