Refactor player data handling in Streamlit app; removed 'version' column from display and improved QB positioning in combinations for stack finder functionality.

src/streamlit_app.py

@@ -87,14 +87,14 @@ def init_baselines():
     cursor = collection.find()
 
     raw_display = pd.DataFrame(list(cursor))
-    raw_display = raw_display[['Team', 'QB', 'WR1_TE', 'WR2_TE', 'Total', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '60+%', '2x%', '3x%', '4x%', 'Own', 'LevX', 'slate', 'version']]
+    raw_display = raw_display[['Team', 'QB', 'WR1_TE', 'WR2_TE', 'Total', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '60+%', '2x%', '3x%', '4x%', 'Own', 'LevX', 'slate']]
     dk_stacks_raw = raw_display.copy()
 
     collection = db["FD_DFS_Stacks"] 
     cursor = collection.find()
 
     raw_display = pd.DataFrame(list(cursor))
-    raw_display = raw_display[['Team', 'QB', 'WR1_TE', 'WR2_TE', 'Total', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '60+%', '2x%', '3x%', '4x%', 'Own', 'LevX', 'slate', 'version']]
+    raw_display = raw_display[['Team', 'QB', 'WR1_TE', 'WR2_TE', 'Total', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '60+%', '2x%', '3x%', '4x%', 'Own', 'LevX', 'slate']]
     fd_stacks_raw = raw_display.copy()
 
     return player_stats, dk_stacks_raw, fd_stacks_raw, dk_roo_raw, fd_roo_raw
@@ -224,6 +224,7 @@ if selected_tab == 'Stack Finder':
 
         # Create a position dictionary mapping players to their eligible positions
         pos_dict = dict(zip(raw_baselines.Player, raw_baselines.Position))
+        pos_reverse_dict = dict(zip(raw_baselines.Position, raw_baselines.Player))
 
         def is_valid_combination(combo):
             # Count positions in this combination
@@ -252,17 +253,22 @@ if selected_tab == 'Stack Finder':
             working_baselines = working_baselines[working_baselines['Position'] != 'DST']
             working_baselines = working_baselines[working_baselines['Position'] != 'K']
             qb_var = qb_dict[cur_team]
-            order_list = working_baselines['Player'].unique()
+            
+            # Create order_list with QBs first to ensure they appear in first column
+            qb_players = working_baselines[working_baselines['Position'] == 'QB']['Player'].unique()
+            other_players = working_baselines[working_baselines['Position'] != 'QB']['Player'].unique()
+            order_list = list(qb_players) + list(other_players)
 
             comb = combinations(order_list, stack_size)
 
             for i in list(comb):
-                if qb_var in i:
-                    comb_list.append(i)
-
-            for i in list(comb):
-                if is_valid_combination(i):
-                    comb_list.append(i)
+                if qb_var in i and is_valid_combination(i):
+                    # Ensure QB is in first position
+                    combo_list = list(i)
+                    if qb_var in combo_list:
+                        qb_index = combo_list.index(qb_var)
+                        combo_list[0], combo_list[qb_index] = combo_list[qb_index], combo_list[0]
+                        comb_list.append(tuple(combo_list))
 
         comb_DF = pd.DataFrame(comb_list)
 
@@ -416,6 +422,7 @@ if selected_tab == 'Stack Finder':
                     cut_var += 1
                 except:
                     cut_var += 1
+        
         st.session_state['display_frame'] = comb_DF
     if 'display_frame' in st.session_state:
         st.dataframe(st.session_state['display_frame'].style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), hide_index=True, use_container_width = True)