Enhance lineup optimization in app.py: Modify optimize_lineup function to accept sidebar_site parameter, allowing for different salary cap constraints for DraftKings and FanDuel. Update UI to display optimal players by position, improving clarity and user experience.

app.py

@@ -124,7 +124,7 @@ tab1, tab2, tab3, tab4, tab5, tab6, tab7 = st.tabs(["Team Stacks Range of Outcom
 # ... existing code ...
 
 @st.cache_data
-def optimize_lineup(player_data):
+def optimize_lineup(player_data, sidebar_site):
     """
     Creates optimal lineup based on median projections while respecting position and salary constraints
     """
@@ -140,7 +140,10 @@ def optimize_lineup(player_data):
     prob += pulp.lpSum([players[row['Player']] * row['Median'] for idx, row in player_data.iterrows()])
     
     # Constraint: Salary cap
-    prob += pulp.lpSum([players[row['Player']] * row['Salary'] for idx, row in player_data.iterrows()]) <= 50000
+    if sidebar_site == 'Draftkings':
+        prob += pulp.lpSum([players[row['Player']] * row['Salary'] for idx, row in player_data.iterrows()]) <= 50000
+    elif sidebar_site == 'Fanduel':
+        prob += pulp.lpSum([players[row['Player']] * row['Salary'] for idx, row in player_data.iterrows()]) <= 60000
 
     # Constraint: 9 players
     prob += pulp.lpSum([players[row['Player']] for idx, row in player_data.iterrows()]) == 9
@@ -215,11 +218,43 @@ with st.sidebar:
             roo_data = fd_roo_raw[fd_roo_raw['slate'] == str(sidebar_slate)]
             roo_data = roo_data[roo_data['version'] == 'overall']
             
-        optimal_players, total_salary, total_median = optimize_lineup(roo_data)
-        
+        optimal_players, total_salary, total_median = optimize_lineup(roo_data, sidebar_site)
+       
         st.write("Optimal Lineup:")
-        for player in optimal_players:
-            st.write(f"{player['Position']}: {player['Player']} (${player['Salary']:,})")
+        
+        # Sort players into position groups
+        qb = [p for p in optimal_players if p['Position'] == 'QB'][0]
+        rbs = [p for p in optimal_players if p['Position'] == 'RB']
+        wrs = [p for p in optimal_players if p['Position'] == 'WR']
+        tes = [p for p in optimal_players if p['Position'] == 'TE']
+        dst = [p for p in optimal_players if p['Position'] == 'DST'][0]
+        
+        # Display QB
+        st.write(f"QB: {qb['Player']} (${qb['Salary']:,})")
+        
+        # Display RB1 and RB2
+        st.write(f"RB: {rbs[0]['Player']} (${rbs[0]['Salary']:,})")
+        st.write(f"RB: {rbs[1]['Player']} (${rbs[1]['Salary']:,})")
+        
+        # Display WR1, WR2, WR3
+        st.write(f"WR: {wrs[0]['Player']} (${wrs[0]['Salary']:,})")
+        st.write(f"WR: {wrs[1]['Player']} (${wrs[1]['Salary']:,})")
+        st.write(f"WR: {wrs[2]['Player']} (${wrs[2]['Salary']:,})")
+        
+        # Display TE1
+        st.write(f"TE: {tes[0]['Player']} (${tes[0]['Salary']:,})")
+        
+        # Display FLEX (either RB3, WR4, or TE2)
+        if len(rbs) > 2:
+            st.write(f"FLEX (RB): {rbs[2]['Player']} (${rbs[2]['Salary']:,})")
+        elif len(wrs) > 3:
+            st.write(f"FLEX (WR): {wrs[3]['Player']} (${wrs[3]['Salary']:,})")
+        elif len(tes) > 1:
+            st.write(f"FLEX (TE): {tes[1]['Player']} (${tes[1]['Salary']:,})")
+            
+        # Display DST
+        st.write(f"DST: {dst['Player']} (${dst['Salary']:,})")
+        
         st.write(f"Total Salary: ${total_salary:,}")
         st.write(f"Projected Points: {total_median:.2f}")