Update app.py

app.py

@@ -191,8 +191,24 @@ with tab1:
                                            split_portfolio['F'].map(player_own_dict),
                                            split_portfolio['UTIL'].map(player_own_dict)])
                                  
-                            
+                            display_portfolio = split_portfolio[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'UTIL', 'Salary', 'Projection', 'Ownership']]
+                            hold_portfolio = display_portfolio.sort_values(by='Projection', ascending=False)
                             gc.collect() 
                             
 with tab2:
-    st.dataframe(split_portfolio.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+    col1, col2 = st.columns([1, 7])
+    with col1:
+        if st.button("Load/Reset Data", key='reset1'):
+            display_portfolio = hold_portfolio
+        if st.button("Trim Lineups", key='trim1'):
+            max_proj = 10000
+            max_own = 10000
+            x = 0
+            for index, row in display_portfolio.iterrows():
+                if row['Ownership'] > max_own:
+                    max_own = row['Ownership']
+                    display_portfolio.drop(index, inplace=True)
+                    
+    with col2:    
+        with st.container():
+            st.dataframe(display_portfolio.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)