handle cb3

app.py

@@ -92,18 +92,30 @@ def display_heroes_from_df(df,display_cols=display_cols, show_df=True):
 def return_costume_list(df0, hero_name):
     assert hero_name in df0.name.values
 
+    if hero_name[-2:] == "C3":
+        return ['None', 'CB1', 'CB2', 'CB3']
     if hero_name[-2:] == "C2":
-        return ['None', 'CB1', 'CB2']
+        hero_name2 = hero_name[:-1] + "3"
+        if hero_name2 in df0.name.values: # if this hero has C3
+            return ['None', 'CB1', 'CB2', 'CB3']
+        else:
+            return ['None', 'CB1', 'CB2']
     elif hero_name[-2:] == " C":
         hero_name2 = hero_name + "2"
-        if hero_name2 in df0.name.values: # if this hero has C2
+        hero_name3 = hero_name + "3"
+        if hero_name3 in df0.name.values: # if this hero has C2
+            return ['None', 'CB1', 'CB2', 'CB3']
+        elif hero_name2 in df0.name.values: # if this hero has C2
             return ['None', 'CB1', 'CB2']
         else:
             return ['None', 'CB1']
     else:
         hero_name1 = hero_name + " C"
         hero_name2 = hero_name + " C2"
-        if hero_name2 in df0.name.values: # if this hero has C2
+        hero_name3 = hero_name + " C3"
+        if hero_name3 in df0.name.values: # if this hero has C2
+            return ['None', 'CB1', 'CB2', 'CB3']
+        elif hero_name2 in df0.name.values: # if this hero has C2
             return ['None', 'CB1', 'CB2']
         elif hero_name1 in df0.name.values: # if this hero has C2
             return ['None', 'CB1']
@@ -158,7 +170,7 @@ st.header(f'HeroPlan Explorer')
 st.write('Powered by Heroplan.io : Thanks E&P community for continually update hero data.')
 
 df = pd.read_csv('heroes_ep.csv')
-st.write(f'### Updated: Oct 23, 2023 -- Total heroes in HeroPlan database = {len(df)}')
+st.write(f'### Updated: Nov 10, 2023 -- Total heroes in HeroPlan database = {len(df)} (including Toon Costume3)')
 
 df_extra = pd.read_csv("heroes_ep_extra.csv")
 all_name_extra = sorted(list(df_extra['name'].values))
@@ -187,7 +199,7 @@ if genre == ':rainbow[Heroes Explorer]':
     col1, col2, col3 = st.columns(3)
     with col1:
         st.header("Standard Filters:")
-        st.write("Tips: filter costume by typing ' C' or 'C2' in the Name box.")
+        st.write("Tips: filter costume by typing ' C' 'C2' or 'C3' in the Name box.")
         with st.expander("Filter Options"):
             name_option = st.text_input(label="Name:", value="")
             star_option = st.selectbox(label='Star:', options=star_values, index=0)
@@ -311,27 +323,6 @@ elif genre == "Team Simulation":
 
     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'======================')
@@ -361,4 +352,4 @@ else:
     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
+    display_heroes_from_df(df_ret,display_cols=df_ret.columns[:-2]) # display all except special-skill text