minor fix team simulation avatar

app.py

@@ -78,8 +78,11 @@ def display_heroes_from_df(df,display_cols=display_cols, show_df=True):
     if show_df:
         st.dataframe(df[display_cols],
                  column_config={
-                         "image": st.column_config.ImageColumn("Avatar", help="")},
+                         "image": st.column_config.ImageColumn("Avatar", help="",
+                                                              # width="medium", # width alone doesnt matter and there is no height option
+                                                              )},
                  use_container_width=True,
+                 # height=128, # this is height of the whold df, not on each line
                  hide_index=True)
 
     for i in range(len(df)):
@@ -313,8 +316,8 @@ if genre == ':rainbow[Heroes Explorer]':
 ## Program 2 "Team Simulation"
 elif genre == "Team Simulation":
     
-    def choose_hero(key="Hero1"):
-        name_choice = st.selectbox(label='Hero Name:', options=all_name_extra, index=0, key=key+"_name")
+    def choose_hero(key="Hero1", default_index=0):
+        name_choice = st.selectbox(label='Hero Name:', options=all_name_extra, index=default_index, key=key+"_name")
         costume_list = return_costume_list(df_extra, name_choice)
         costume_choice = st.selectbox(label='Costume:', options=costume_list, index=0, key=key+"_costume")
         lb_list = ['None', 'LB1', 'LB2']
@@ -348,7 +351,7 @@ elif genre == "Team Simulation":
     total_power = 0
     for ii in range(nheroes_choice):
         with col_list[ii]:
-            df_hero_list.append(choose_hero(key=f"Hero{ii+1}")) # 'key' in st.selectbox to differentiate widgets
+            df_hero_list.append(choose_hero(key=f"Hero{ii+1}", default_index=ii)) # 'key' in st.selectbox to differentiate widgets
             write_short_description(df_hero_list[-1])
         total_power += df_hero_list[ii]['power'].values[0]
 
@@ -362,7 +365,7 @@ elif genre == "Team Simulation":
     st.write(f'### Total power = {total_power}')
     st.write(f'======================')
     
-    display_heroes_from_df(df_hero_all5, display_cols=df_hero_all5.columns[:-2], show_df=False) # display all except special-skill text
+    display_heroes_from_df(df_hero_all5, display_cols=df_hero_all5.columns[:-2], show_df=True) # display all except special-skill text
     
 #########################################
 ## Program 3 "Individual Stat"