team simulation draft

app.py

@@ -66,11 +66,12 @@ def display_image(url, scale=0.5):
     image = Image.open(urlopen(url))
     st.image(image.resize(( int(image.width * scale), int(image.height * scale))))
 
-def display_heroes_from_df(df,display_cols=display_cols):
+def display_heroes_from_df(df,display_cols=display_cols, show_df=True):
     vtob = "is" if len(df)<=1 else "are"
     st.write(f'There {vtob} {len(df)} heroes in the filtered list')
-    
-    st.dataframe(df[display_cols],
+
+    if show_df:
+        st.dataframe(df[display_cols],
                  column_config={
                          "image": st.column_config.ImageColumn("Avatar", help="")},
                  use_container_width=True,
@@ -159,6 +160,9 @@ st.write('Powered by Heroplan.io : Thanks E&P community for continually update h
 df = pd.read_csv('heroes_ep.csv')
 st.write(f'### Updated: Oct 23, 2023 -- Total heroes in HeroPlan database = {len(df)}')
 
+df_extra = pd.read_csv("heroes_ep_extra.csv")
+all_name_extra = sorted(list(df_extra['name'].values))
+
 #########################################
 
 class_values = ['None'] + list(df['class'].unique()) 
@@ -173,7 +177,7 @@ source_values = ['None'] + list(df['source'].unique()) # Contain lot of typo bug
 with st.sidebar:
     genre = st.radio(
     "Choose how to explore heroes",
-    [":rainbow[Heroes Explorer]", "***LB/CB Hero Stat*** :movie_camera:"],
+    [":rainbow[Heroes Explorer]", "Team Simulation","***LB/CB Hero Stat*** :movie_camera:"],
     captions = ["Filter only heroes with certain properties", "Co-powered by Elioty33's DataVault"])
 
 #########################################
@@ -266,9 +270,32 @@ if genre == ':rainbow[Heroes Explorer]':
 
     display_heroes_from_df(df2.sort_values(sort_option, ascending=False))
 #########################################
-## Program 2
+## Program 2 "Team Simulation"
+elif genre == "Team Simulation":
+
+    def choose_hero():
+        name_choice = st.selectbox(label='Hero Name:', options=all_name_extra, index=0)
+        costume_list = return_costume_list(df_extra, name_choice)
+        costume_choice = st.selectbox(label='Costume:', options=costume_list, index=0)
+        lb_list = ['None', 'LB1', 'LB2']
+        lb_choice = st.selectbox(label='Limit Break:', options=lb_list, index=0)
+    
+        df_ret = return_hero_stat(df_extra, name_choice, lb_choice=lb_choice, costume_choice=costume_choice)
+        return df_ret
+    
+    col1, col2, col3, col4, col5 = st.columns(5)
+    with col1:
+        df_hero1 = choose_hero()
+    with col2:
+        df_hero2 = choose_hero()
+
+    df_hero_all5 = pd.concat([df_hero1, df_hero2])
+    display_heroes_from_df(df_hero_all5, display_cols=df_ret.columns[:-2], show_df=False) # display all except special-skill text
+    
+#########################################
+## Program 3 "Individual Stat"
 else:
-    df_extra = pd.read_csv("heroes_ep_extra.csv")
+    
     
     st.header("Analyze Hero LB/CB Stat (without Emblem)")
     st.write("HeroPlan and DataVault are combined here. Thanks ***@Elioty33*** for his DataVault contribution")
@@ -281,11 +308,11 @@ else:
     name_values = sorted(list(df_extra['name'].values))
     name_choice = st.selectbox(label='Hero Name:', options=name_values, index=0)
 
-    lb_list = ['None', 'LB1', 'LB2']
     costume_list = return_costume_list(df_extra, name_choice)
-
-    lb_choice = st.selectbox(label='Limit Break:', options=lb_list, index=0)
     costume_choice = st.selectbox(label='Costume:', options=costume_list, index=0)
+    
+    lb_list = ['None', 'LB1', 'LB2']
+    lb_choice = st.selectbox(label='Limit Break:', options=lb_list, index=0)
 
     df_ret = return_hero_stat(df_extra, name_choice, lb_choice=lb_choice, costume_choice=costume_choice)
     display_heroes_from_df(df_ret,display_cols=df_ret.columns[:-2]) # display all except special-skill text