adding ROI info tab

app.py

@@ -68,8 +68,10 @@ def grab_contest_payout_info(db, sport, type, contest_date, contest_name, contes
     payout_info = pd.DataFrame(list(cursor)).drop('_id', axis=1)
     payout_info = payout_info[payout_info['Contest Date'] == contest_date]
     payout_info = payout_info[payout_info['Contest ID'] == contest_id_map[contest_name]]
+
+    entry_fee = payout_info['Entry Fee'].iloc[0]
     
-    return payout_info
+    return payout_info, entry_fee
 
 def export_contest_file(db, sport, type, contest_date, contest_id, contest_data):
     if type == 'Classic':
@@ -240,7 +242,7 @@ if selected_tab == 'Data Load':
                     st.session_state['player_info'], st.session_state['info_maps'] = grab_contest_player_info(db, sport_init, type_init, date_select, contest_name_var, contest_id_map)
                     st.session_state['Contest_file'] = grab_contest_data(sport_init, contest_name_var, contest_id_map, date_select, date_select2)
                     try:
-                        st.session_state['payout_info'] = grab_contest_payout_info(db, sport_init, type_init, date_select, contest_name_var, contest_id_map, contest_id_map[contest_name_var])
+                        st.session_state['payout_info'], st.session_state['entry_fee'] = grab_contest_payout_info(db, sport_init, type_init, date_select, contest_name_var, contest_id_map, contest_id_map[contest_name_var])
                     except:
                         st.session_state['payout_info'] = None
             else:
@@ -260,7 +262,7 @@ if selected_tab == 'Data Load':
                 except:
                     st.warning('Please upload a Contest CSV')
                 try:
-                    st.session_state['payout_info'] = grab_contest_payout_info(db, sport_init, type_init, date_select, contest_name_var, contest_id_map, contest_id_map[contest_name_var])
+                    st.session_state['payout_info'], st.session_state['entry_fee'] = grab_contest_payout_info(db, sport_init, type_init, date_select, contest_name_var, contest_id_map, contest_id_map[contest_name_var])
                 except:
                     st.session_state['payout_info'] = None
             else:
@@ -635,7 +637,7 @@ if selected_tab == 'Contest Analysis':
     st.download_button(label="Download Contest Info", data=st.session_state['display_contest_info'].to_csv(index=False), file_name="contest_info.csv", mime="text/csv", key='download_contest')
     if 'Contest' in st.session_state:
         with st.container():
-            tab1, tab2, tab3, tab4, tab5 = st.tabs(['Player Used Info', 'Stack Used Info', 'Stack Size Info', 'General Info', 'Duplication Info'])
+            tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs(['Player Used Info', 'Stack Used Info', 'Stack Size Info', 'General Info', 'Duplication Info', 'ROI Info'])
             with tab1:
                 player_pos_form_col, player_comp_form_col = st.columns(2)
                 with player_pos_form_col:
@@ -887,4 +889,58 @@ if selected_tab == 'Contest Analysis':
                              background_gradient(cmap='RdYlGn', subset=['uniques%', 'under_5%', 'under_10%'], axis=0).
                              background_gradient(cmap='RdYlGn', subset=['uniques', 'under_5', 'under_10'], axis=0).
                              format(dupe_format, precision=2), hide_index=True)
-                st.download_button(label="Download Duplication Info", data=st.session_state['duplication_frame'].to_csv(index=False), file_name="duplication_info.csv", mime="text/csv", key='dupe_exp_comp_download')
+                st.download_button(label="Download Duplication Info", data=st.session_state['duplication_frame'].to_csv(index=False), file_name="duplication_info.csv", mime="text/csv", key='dupe_exp_comp_download')
+            
+            with tab6:
+                if st.session_state['payout_info'] is not None:
+                    with st.form(key='ROI_form'):
+                        col1, col2 = st.columns(2)
+                        with col1:
+                            user_ROI_var = st.selectbox("Which user(s) would you like to view?",  ['All', 'Specific'], key='user_ROI_var')
+                        with col2:
+                            user_ROI_select = st.multiselect("Select your user(s)", st.session_state['display_contest_info']['BaseName'].sort_values().unique(), key='user_ROI_select')
+                        submitted = st.form_submit_button("Submit")
+                        if submitted:
+                            if user_ROI_var == 'Specific':
+                                user_ROI_select = user_ROI_select
+                            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['payout'].sum()
+                        dupe_frame['ROI'] = (dupe_frame['Total Payout'] / dupe_frame['Total Fees'])
+                        dupe_frame = dupe_frame[['BaseName', 'EntryCount', 'finish', 'payout', 'ROI']].drop_duplicates(subset='BaseName', keep='first')
+                        st.session_state['ROI_frame'] = dupe_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)]
+                    
+                    # Initialize pagination in session state if not exists
+                    if 'ROI_page' not in st.session_state:
+                        st.session_state.ROI_page = 1
+
+                    # Calculate total pages
+                    rows_per_page = 50
+                    total_rows = len(st.session_state['ROI_frame'])
+                    total_pages = (total_rows + rows_per_page - 1) // rows_per_page
+
+                    # Create pagination controls in a single row
+                    pagination_cols = st.columns([4, 1, 1, 1, 4])
+                    with pagination_cols[1]:
+                        if st.button(f"Previous ROI Page"):
+                            if st.session_state['ROI_page'] > 1:
+                                st.session_state.ROI_page -= 1
+
+                    with pagination_cols[3]:
+                        if st.button(f"Next ROI Page"):
+                            st.session_state.ROI_page += 1
+
+                    # Calculate start and end indices for current page
+                    start_ROI_idx = (st.session_state.ROI_page - 1) * rows_per_page
+                    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)
+                    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')