Enhance position management in Streamlit app by implementing position limits for DraftKings and FanDuel. Update styling for tabs and add functionality to validate player combinations based on eligible positions. Refactor position selection logic for improved user experience.

app.py

@@ -36,7 +36,7 @@ st.markdown("""
     .stTabs [data-baseweb="tab"] {
         height: 50px;
         white-space: pre-wrap;
-        background-color: #FFD700;
+        background-color: #DAA520;
         color: white;
         border-radius: 10px;
         gap: 1px;
@@ -46,12 +46,20 @@ st.markdown("""
     }
     .stTabs [aria-selected="true"] {
         background-color: #DAA520;
+        border: 3px solid #FFD700;
         color: white;
     }
     .stTabs [data-baseweb="tab"]:hover {
-        background-color: #DAA520;
+        background-color: #FFD700;
         cursor: pointer;
     }
+    div[data-baseweb="select"] > div {
+        background-color: #DAA520;
+        color: white;
+    }
+    div{
+        box-sizing: content-box !important;
+    }
 </style>""", unsafe_allow_html=True)
 
 @st.cache_resource(ttl = 60)
@@ -105,7 +113,7 @@ with col1:
             team_var2 = raw_baselines.Team.unique().tolist()
         pos_split2 = st.radio("Are you viewing all positions, specific groups, or specific positions?", ('All Positions', 'Specific Positions'), key='pos_split2')
         if pos_split2 == 'Specific Positions':
-            pos_var2 = st.multiselect('What Positions would you like to view?', options = ['SP', 'P', 'C', '1B', '2B', '3B', 'SS', 'OF'])
+            pos_var2 = st.multiselect('What Positions would you like to view?', options = ['C', '1B', '2B', '3B', 'SS', 'OF'])
         elif pos_split2 == 'All Positions':
             pos_var2 = 'All'
         if site_var1 == 'Draftkings':
@@ -126,6 +134,30 @@ with col1:
         cost_dict = dict(zip(raw_baselines.Player, raw_baselines.Salary))
 
 with col2:
+
+    if site_var1 == 'Draftkings':
+        position_limits = {
+            'C': 1,
+            '1B': 1,
+            '2B': 1,
+            '3B': 1,
+            'SS': 1,
+            'OF': 3,
+            # Add more as needed
+        }
+        max_salary = 50000
+        max_players = 10
+    else:
+        position_limits = {
+            'C_1B': 1,
+            '2B': 1,
+            '3B': 1,
+            'SS': 1,
+            'OF': 3,
+            'UTIL': 1,
+            # Add more as needed
+        }
+
     stack_hold_container = st.empty()
     comb_list = []
     if pos_split2 == 'All Positions':
@@ -133,6 +165,41 @@ with col2:
     elif pos_split2 != 'All Positions':
         raw_baselines = raw_baselines[raw_baselines['Position'].str.contains('|'.join(pos_var2))]
 
+    # Create a position dictionary mapping players to their eligible positions
+    pos_dict = dict(zip(raw_baselines.Player, raw_baselines.Position))
+
+    def is_valid_combination(combo):
+        # Count positions in this combination
+        position_counts = {pos: 0 for pos in position_limits.keys()}
+        
+        # For each player in the combination
+        for player in combo:
+            # Get their eligible positions
+            player_positions = pos_dict[player].split('/')
+            
+            # For each position they can play
+            for pos in player_positions:
+                # Handle special cases for FanDuel
+                if site_var1 == 'Fanduel':
+                    if pos in ['C', '1B']:
+                        position_counts['C_1B'] += 1
+                    elif pos == 'UTIL':
+                        # UTIL can be filled by any position
+                        for p in position_counts:
+                            position_counts[p] += 1
+                else:  # DraftKings
+                    if pos in position_counts:
+                        position_counts[pos] += 1
+        
+        # Check if any position exceeds its limit
+        for pos, limit in position_limits.items():
+            if position_counts[pos] > limit:
+                return False
+        
+        return True
+
+    # Modify the combination generation code
+    comb_list = []
     for cur_team in team_var2:
         working_baselines = raw_baselines
         working_baselines = working_baselines[working_baselines['Team'] == cur_team]
@@ -142,8 +209,10 @@ with col2:
 
         comb = combinations(order_list, stack_size)
 
+        # Only add combinations that satisfy position limits
         for i in list(comb):
-            comb_list.append(i)
+            if is_valid_combination(i):
+                comb_list.append(i)
 
     comb_DF = pd.DataFrame(comb_list)