Update app.py

app.py

@@ -198,6 +198,11 @@ with tab2:
                   Max_Rank = fd_Max_Rank
                   Max_Upside = fd_Max_Upside
         contest_var1 = st.selectbox("What contest type are you optimizing for?", ('Cash', 'Small Field GPP', 'Large Field GPP', 'Round Robin'), key='contest_var1')
+        trim_choice1 = st.selectbox("Allow overowned lineups?", options = ['Yes', 'No'])
+        if trim_choice1 == 'Yes':
+            trim_var1 = 0
+        elif trim_choice1 == 'No':
+            trim_var1 = 1
         if contest_var1 == 'Small Field GPP':
             st.info('The Pivot optimal uses backend functions to create a stack and lock in certain pieces, if you want control over QB pairing use the Manual model instead.')
             opto_var1 = st.selectbox("Pivot optimal or Manual?", ('Pivot Optimal', 'Manual'), key='opto_var1')
@@ -585,6 +590,8 @@ with tab2:
                                 elif contest_var1 != 'Cash':
                                     obj_points = {idx: (flex_file['Median'][idx]) for idx in flex_file.index}
                                     total_score += sum([player_vars[idx]*obj_points[idx] for idx in flex_file.index])
+                                if trim_var1 == 1:
+                                    total_score += pulp.lpSum([player_vars[idx]*obj_own_max[idx] for idx in flex_file.index]) <= max_own - .001
                                 
         
                                 total_score.solve()