Enhance lineup building logic in app.py to support both Draftkings and Fanduel platforms

Updated player selection and position limits based on the selected site, ensuring accurate handling of player positions and salary constraints for each platform. This change improves the user experience by providing tailored lineup building options for Draftkings and Fanduel.

app.py

@@ -824,24 +824,48 @@ with tab4:
         st.header("Handbuilder")
     with col2:
         slate_var3 = st.selectbox("Slate Selection", options=['Main', 'Secondary', 'Auxiliary'])
-        if slate_var3 == 'Main':
-            handbuild_roo = dk_roo[dk_roo['Slate'] == 'main_slate']
-        elif slate_var3 == 'Secondary':
-            handbuild_roo = dk_roo[dk_roo['Slate'] == 'secondary_slate']
-        elif slate_var3 == 'Auxiliary':
-            handbuild_roo = dk_roo[dk_roo['Slate'] == 'turbo_slate']
+        if site_var == 'Draftkings':
+            if slate_var3 == 'Main':
+                handbuild_roo = dk_roo[dk_roo['Slate'] == 'main_slate']
+            elif slate_var3 == 'Secondary':
+                handbuild_roo = dk_roo[dk_roo['Slate'] == 'secondary_slate']
+            elif slate_var3 == 'Auxiliary':
+                handbuild_roo = dk_roo[dk_roo['Slate'] == 'turbo_slate']
+        else:
+            if slate_var3 == 'Main':
+                handbuild_roo = fd_roo[fd_roo['Slate'] == 'main_slate']
+            elif slate_var3 == 'Secondary':
+                handbuild_roo = fd_roo[fd_roo['Slate'] == 'secondary_slate']
+            elif slate_var3 == 'Auxiliary':
+                handbuild_roo = fd_roo[fd_roo['Slate'] == 'turbo_slate']
 
     # --- POSITION LIMITS ---
-    position_limits = {
-        'SP': 2,
-        'C': 1,
-        '1B': 1,
-        '2B': 1,
-        '3B': 1,
-        'SS': 1,
-        'OF': 3,
-        # Add more as needed
-    }
+    if site_var == 'Draftkings':
+        position_limits = {
+            'SP': 2,
+            '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 = {
+            'P': 1,
+            'C_1B': 1,
+            '2B': 1,
+            '3B': 1,
+            'SS': 1,
+            'OF': 3,
+            'UTIL': 1,
+            # Add more as needed
+        }
+        max_salary = 35000
+        max_players = 9
 
     # --- LINEUP STATE ---
     if 'handbuilder_lineup' not in st.session_state:
@@ -926,10 +950,29 @@ with tab4:
             for _, player_row in selected_players.iterrows():
                 eligible_positions = re.split(r'[/, ]+', player_row['Position'])
                 slot_to_fill = None
-                for pos in eligible_positions:
-                    if slot_counts.get(pos, 0) < position_limits.get(pos, 0):
-                        slot_to_fill = pos
-                        break
+
+                if site_var == 'Fanduel':
+                    # Logic for handling Fanduel Positions (with C/1B and UTIL)
+                    for slot in ['P', 'C_1B', '2B', '3B', 'SS', 'OF', 'UTIL']:
+                        if slot_counts.get(slot, 0) < position_limits.get(slot, 0):
+                            if slot == 'C_1B':
+                                if any(pos in eligible_positions for pos in ['C', '1B']):
+                                    slot_to_fill = slot
+                                    break
+                            elif slot == 'UTIL':
+                                if 'P' not in eligible_positions:
+                                    slot_to_fill = slot
+                                    break
+                            elif slot in eligible_positions:
+                                slot_to_fill = slot
+                                break
+                else:
+                    # General logic for handling Draftkings Positions
+                    for pos in eligible_positions:
+                        if slot_counts.get(pos, 0) < position_limits.get(pos, 0):
+                            slot_to_fill = pos
+                            break
+
                 if slot_to_fill is not None:
                     # Avoid duplicates
                     if player_row['Player'] not in st.session_state['handbuilder_lineup']['Player'].values:
@@ -974,10 +1017,29 @@ with tab4:
             eligible_positions = re.split(r'[/, ]+', player_row['Position'].iloc[0])
             # Find the first eligible slot that is not full
             slot_to_fill = None
-            for pos in eligible_positions:
-                if slot_counts.get(pos, 0) < position_limits.get(pos, 0):
-                    slot_to_fill = pos
-                    break
+
+            if site_var == 'Fanduel':
+                # Logic for handling Fanduel Positions (with C/1B and UTIL)
+                for slot in ['P', 'C_1B', '2B', '3B', 'SS', 'OF', 'UTIL']:
+                    if slot_counts.get(slot, 0) < position_limits.get(slot, 0):
+                        if slot == 'C_1B':
+                            if any(pos in eligible_positions for pos in ['C', '1B']):
+                                slot_to_fill = slot
+                                break
+                        elif slot == 'UTIL':
+                            if 'P' not in eligible_positions:
+                                slot_to_fill = slot
+                                break
+                        elif slot in eligible_positions:
+                            slot_to_fill = slot
+                            break
+            else:
+                # General logic for handling Draftkings Positions
+                for pos in eligible_positions:
+                    if slot_counts.get(pos, 0) < position_limits.get(pos, 0):
+                        slot_to_fill = pos
+                        break
+                    
             if slot_to_fill is not None:
                 # Avoid duplicates
                 if not player_row['Player'].iloc[0] in st.session_state['handbuilder_lineup']['Player'].values:
@@ -995,7 +1057,10 @@ with tab4:
         st.subheader("Lineup Build")
 
         # --- EXPLICIT LINEUP ORDER ---
-        lineup_slots = ['SP', 'SP', 'C', '1B', '2B', '3B', 'SS', 'OF', 'OF', 'OF']
+        if site_var == 'Draftkings':
+            lineup_slots = ['SP', 'SP', 'C', '1B', '2B', '3B', 'SS', 'OF', 'OF', 'OF']
+        else:
+            lineup_slots = ['P', 'C_1B', '2B', '3B', 'SS', 'OF', 'OF', 'OF', 'UTIL']
         display_columns = ['Slot', 'Player', 'Order', 'Team', 'Salary', 'Median', 'Own%']
 
         filled_lineup = st.session_state['handbuilder_lineup']
@@ -1073,7 +1138,7 @@ with tab4:
                 '2x%': [avg_2x],
                 'Own%': [total_own]
             })
-            summary_row = summary_row[['Salary', 'Median', 'Own%']].head(10)
+            summary_row = summary_row[['Salary', 'Median', 'Own%']].head(max_players)
             
             col1, col3 = st.columns([2, 3])
 
@@ -1081,20 +1146,20 @@ with tab4:
                 if (10 - len(filled_lineup)) > 0:
                     st.markdown(f"""
                             <div style='text-align: left; vertical-align: top; margin-top: 0; padding-top: 0;''>
-                                <b>💰 Per Player:</b> ${round((50000 - total_salary) / (10 - len(filled_lineup)), 0)} &nbsp;
+                                <b>💰 Per Player:</b> ${round((max_salary - total_salary) / (max_players - len(filled_lineup)), 0)} &nbsp;
                             </div>
                             """,
                             unsafe_allow_html=True)
                 else:
                     st.markdown(f"""
                             <div style='text-align: left; vertical-align: top; margin-top: 0; padding-top: 0;''>
-                                <b>💰 Leftover:</b> ${round(50000 - total_salary, 0)} &nbsp;
+                                <b>💰 Leftover:</b> ${round(max_salary - total_salary, 0)} &nbsp;
                             </div>
                             """,
                             unsafe_allow_html=True)
 
             with col3:
-                if total_salary <= 50000:
+                if total_salary <= max_salary:
                     st.markdown(
                         f"""
                         <div style='text-align: right; vertical-align: top; margin-top: 0; padding-top: 0;''>