generalize team simulation

app.py

@@ -273,6 +273,8 @@ if genre == ':rainbow[Heroes Explorer]':
 ## Program 2 "Team Simulation"
 elif genre == "Team Simulation":
 
+    nheroes_choice = st.selectbox(label='Number of Heroes:', options=[2,3,4,5], index=-1)
+    
     def choose_hero(key="Hero1"):
         name_choice = st.selectbox(label='Hero Name:', options=all_name_extra, index=0, key=key+"_name")
         costume_list = return_costume_list(df_extra, name_choice)
@@ -294,27 +296,42 @@ elif genre == "Team Simulation":
         st.write(f'Class: {df_hero["class"].values[0]}')
         st.write(f'Types: {df_hero["types"].values[0]}')
     
-    col1, col2, col3, col4, col5, col6 = st.columns(6)
-    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")
-        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)
-    with col6:
-        txt = st.text_area("Write your note about team synergy", max_chars=1000)
-
-    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]
+    
+    col_list = st.columns(nheroes_choice+1)
+    df_hero_list = []
+    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
+            write_short_description(df_hero_list[-1])
+        total_power += df_hero_list[ii]['power'].values[0]
+
+    with col_list[-1]:
+        txt = st.text_area("Write your note about team synergy", max_chars=1000, height = 480)
+
+    df_hero_all5 = pd.concat(df_hero_list)
+        
+    # col1, col2, col3, col4, col5, col6 = st.columns(6)    
+    # 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")
+    #     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)
+    # with col6:
+    #     txt = st.text_area("Write your note about team synergy", max_chars=1000)
+
+    # 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'======================')
     st.write(f'### Total power = {total_power}')
     st.write(f'======================')