Add size comparison feature in app.py and create_size_comparison.py

- Introduced a new function, create_size_comparison, to enable comparison of stack sizes based on user selections.
- Updated app.py to include a form for selecting sizes to compare, enhancing user interaction and data analysis capabilities.
- Improved data display for size comparisons, ensuring accurate representation of exposure data in the DataFrame.

app.py

@@ -65,6 +65,7 @@ from global_func.create_general_exposures import create_general_exposures
 from global_func.grab_contest_data import grab_contest_data
 from global_func.create_player_comparison import create_player_comparison
 from global_func.create_stack_comparison import create_stack_comparison
+from global_func.create_size_comparison import create_size_comparison
 
 def is_valid_input(file):
     if isinstance(file, pd.DataFrame):
@@ -518,21 +519,36 @@ with tab2:
                             format(formatter='{:.2%}', subset=st.session_state['stack_frame'].iloc[:, 1:].select_dtypes(include=['number']).columns),
                             hide_index=True)
             with tab3:
-                
-                if st.session_state['entry_parse_var'] == 'All':
-                    st.session_state['stack_size_frame'] = create_stack_size_exposures(st.session_state['display_contest_info'])
-                    st.dataframe(st.session_state['stack_size_frame'].
-                        sort_values(by='Exposure Overall', ascending=False).
-                        style.background_gradient(cmap='RdYlGn').
-                        format(formatter='{:.2%}', subset=st.session_state['stack_size_frame'].iloc[:, 1:].select_dtypes(include=['number']).columns),
-                        hide_index=True)
+                with st.form(key='size_exp_comp_form'):
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        comp_size_var = st.selectbox("Would you like to compare with anyone?",  ['No', 'Yes'], key='comp_size_var')
+                    with col2:
+                        comp_size_select = st.multiselect("Select sizes to compare with:", st.session_state['display_contest_info']['BaseName'].sort_values().unique(), key='comp_size_select')
+                    submitted = st.form_submit_button("Submit")
+                    if submitted:
+                        if comp_size_var == 'No':
+                            comp_size_select = None
+                        else:
+                            comp_size_select = comp_size_select
+                if comp_size_var == 'Yes':
+                    size_exp_comp = create_size_comparison(st.session_state['display_contest_info'], comp_size_select)
+                    st.dataframe(size_exp_comp.style.background_gradient(cmap='RdYlGn', axis=0).format(formatter='{:.2%}', subset=size_exp_comp.select_dtypes(include=['number']).columns), hide_index=True)
                 else:
-                    st.session_state['stack_size_frame'] = create_stack_size_exposures(st.session_state['display_contest_info'], st.session_state['entry_names'])
-                    st.dataframe(st.session_state['stack_size_frame'].
-                        sort_values(by='Exposure Overall', ascending=False).
-                        style.background_gradient(cmap='RdYlGn').
-                        format(formatter='{:.2%}', subset=st.session_state['stack_size_frame'].iloc[:, 1:].select_dtypes(include=['number']).columns),
-                        hide_index=True)
+                    if st.session_state['entry_parse_var'] == 'All':
+                        st.session_state['stack_size_frame'] = create_stack_size_exposures(st.session_state['display_contest_info'])
+                        st.dataframe(st.session_state['stack_size_frame'].
+                            sort_values(by='Exposure Overall', ascending=False).
+                            style.background_gradient(cmap='RdYlGn').
+                            format(formatter='{:.2%}', subset=st.session_state['stack_size_frame'].iloc[:, 1:].select_dtypes(include=['number']).columns),
+                            hide_index=True)
+                    else:
+                        st.session_state['stack_size_frame'] = create_stack_size_exposures(st.session_state['display_contest_info'], st.session_state['entry_names'])
+                        st.dataframe(st.session_state['stack_size_frame'].
+                            sort_values(by='Exposure Overall', ascending=False).
+                            style.background_gradient(cmap='RdYlGn').
+                            format(formatter='{:.2%}', subset=st.session_state['stack_size_frame'].iloc[:, 1:].select_dtypes(include=['number']).columns),
+                            hide_index=True)
             
             with tab4:
                 

global_func/create_size_comparison.py

@@ -0,0 +1,22 @@
+import pandas as pd
+
+def create_stack_comparison(df: pd.DataFrame, entrants: list):
+    overall_players = pd.Series(list(df['stack_size'])).value_counts()
+    contest_len = len(df)
+
+    set_frame = overall_players.to_frame().reset_index().rename(columns={'index': 'Player', 'count': 'Count'})
+    set_frame['Percent'] = set_frame['Count'] / contest_len
+    set_frame = set_frame[['Player', 'Percent']]
+    set_frame = set_frame.rename(columns={'Percent': f'Exposure Overall'})
+    player_frame = set_frame
+    
+    for each_user in entrants:
+        overall_players = pd.Series(list(df[df['BaseName'] == each_user]['stack_size'])).value_counts()
+    
+        set_frame = overall_players.to_frame().reset_index().rename(columns={'index': 'Player', 'count': 'Count'})
+        set_frame['Percent'] = set_frame['Count'] / len(df[df['BaseName'] == each_user])
+        set_frame = set_frame[['Player', 'Percent']]
+        set_frame = set_frame.rename(columns={'Percent': f'Exposure {each_user}'})
+        player_frame = pd.merge(player_frame, set_frame, on='Player', how='outer')
+    
+    return player_frame.sort_values(by='Exposure Overall', ascending=False)