cont draft

app.py

@@ -282,15 +282,34 @@ elif genre == "Team Simulation":
     
         df_ret = return_hero_stat(df_extra, name_choice, lb_choice=lb_choice, costume_choice=costume_choice)
         return df_ret
+
+    def write_short_description(df_hero):
+        st.write(f'Power: {df_hero['power'].values[0]}')
+        st.write(f'Attack: {df_hero['attack'].values[0]}')
+        st.write(f'Defense: {df_hero['defense'].values[0]}')
+        st.write(f'Health: {df_hero['health'].values[0]}')
     
     col1, col2, col3, col4, col5 = st.columns(5)
     with col1:
         df_hero1 = choose_hero(key="Hero1") # 'key' in st.selectbox to differentiate widgets
+        write_short_description(df_hero1)
     with col2:
         df_hero2 = choose_hero(key="Hero2")
-
-    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
+        write_short_description(df_hero2)
+    with col3:
+        df_hero3 = choose_hero(key="Hero3")
+        write_short_description(df_hero3)
+    with col4:
+        df_hero4 = choose_hero(key="Hero4")
+        write_short_description(df_hero4)
+    with col5:
+        df_hero5 = choose_hero(key="Hero5")
+        write_short_description(df_hero5)
+
+    df_hero_all5 = pd.concat([df_hero1, df_hero2, df_hero3, df_hero4, df_hero5])
+    total_power = df_hero1['power'].values[0]+ df_hero2['power'].values[0]+ df_hero3['power'].values[0]+ df_hero4['power'].values[0]+ df_hero5['power'].values[0]
+    st.write(f'Total power = {total_power}')
+    display_heroes_from_df(df_hero_all5, display_cols=df_hero_all5.columns[:-2], show_df=False) # display all except special-skill text
     
 #########################################
 ## Program 3 "Individual Stat"