working through ROI tab

app.py

@@ -906,12 +906,12 @@ if selected_tab == 'Contest Analysis':
                             else:
                                 user_ROI_select = None
                     if 'ROI_frame' not in st.session_state:
-                        dupe_frame = st.session_state['display_contest_info'][['BaseName', 'EntryCount', 'finish', 'payout']]
-                        dupe_frame['Total Fees'] = dupe_frame['EntryCount'] * st.session_state['entry_fee']
-                        dupe_frame['Total Payout'] = dupe_frame.groupby('BaseName')['payout'].sum()
-                        dupe_frame['ROI'] = (dupe_frame['Total Payout'] / dupe_frame['Total Fees'])
-                        dupe_frame = dupe_frame[['BaseName', 'EntryCount', 'Total Fees', 'Total Payout', 'ROI']].drop_duplicates(subset='BaseName', keep='first')
-                        st.session_state['ROI_frame'] = dupe_frame.sort_values(by='ROI', ascending=False)
+                        roi_frame = st.session_state['display_contest_info'][['BaseName', 'EntryCount', 'finish', 'payout']]
+                        roi_frame['Total Fees'] = roi_frame['EntryCount'] * st.session_state['entry_fee']
+                        roi_frame['Total Payout'] = roi_frame.groupby('BaseName')['payout'].transform('sum')
+                        roi_frame['ROI'] = (roi_frame['Total Payout'] / roi_frame['Total Fees'])
+                        roi_frame = roi_frame[['BaseName', 'EntryCount', 'Total Fees', 'Total Payout', 'ROI']].drop_duplicates(subset='BaseName', keep='first')
+                        st.session_state['ROI_frame'] = roi_frame.sort_values(by='ROI', ascending=False)
                     if user_ROI_var == 'Specific':
                         st.session_state['ROI_frame'] = st.session_state['ROI_frame'][st.session_state['ROI_frame']['BaseName'].isin(user_ROI_select)]
                     
@@ -940,7 +940,7 @@ if selected_tab == 'Contest Analysis':
                     end_ROI_idx = min((st.session_state.ROI_page) * rows_per_page, total_rows)
 
                     st.dataframe(st.session_state['ROI_frame'].iloc[start_ROI_idx:end_ROI_idx].style.
-                                background_gradient(cmap='RdYlGn', subset=['ROI'], axis=0), hide_index=True)
+                                background_gradient(cmap='RdYlGn', subset=['ROI', 'Total Fees', 'Total Payout'], axis=0), hide_index=True)
                     st.download_button(label="Download ROI Info", data=st.session_state['ROI_frame'].to_csv(index=False), file_name="ROI_info.csv", mime="text/csv", key='ROI_exp_comp_download')
                 else:
                     st.write('No ROI info available')