Update app.py

app.py

@@ -95,6 +95,10 @@ color_values = ['None'] + list(df['color'].unique())
 speed_values = ['None'] + list(df['speed'].unique())
 source_values = ['None'] + list(df['source'].unique())
 
+defense_values = ['None'] + list(df['defense'].unique())
+attack_values = ['None'] + list(df['attack'].unique())
+health_values = ['None'] + list(df['health'].unique())
+
 #########################################
 ## Select options
 ## TODO: family, costume
@@ -107,6 +111,11 @@ with st.sidebar:
     speed_option = st.selectbox(label='Speed:', options=speed_values, index=0)
     class_option = st.selectbox(label='Class:', options=class_values, index=0)
     source_option = st.selectbox(label='Origin:', options=source_values, index=0)
+    
+    defense_option = st.selectbox(label='Defense:', options=defense_values, index=0)
+    attack_option = st.selectbox(label='Attack:', options=attack_values, index=0)
+    health_option = st.selectbox(label='Health:', options=health_values, index=0)
+    
     special_type_option = st.text_input(label="SpecialSkill Category", value="Hit 3")
     special_text_option = st.text_input(label="SpecialSkill Text", value="Dispel")
 
@@ -133,6 +142,15 @@ if class_option != 'None':
 if source_option != 'None':
     idx_all.append(filter_by_1col(df, 'source', source_option, exact_flag=False))    
 
+if defense_option != 'None':
+    idx_all.append(filter_by_1col_num(df, 'defense', defense_option, oper_flag="ge"))   
+
+if attack_option != 'None':
+    idx_all.append(filter_by_1col_num(df, 'attack', attack_option, oper_flag="ge"))   
+
+if health_option != 'None':
+    idx_all.append(filter_by_1col_num(df, 'health', health_option, oper_flag="ge"))   
+
 
 if special_type_option  != '':
     idx_all.append(filter_by_1col(df, 'types', special_type_option, exact_flag=False))    
@@ -141,6 +159,7 @@ if special_text_option != '':
     idx_all.append(filter_by_1col(df, 'effects', special_text_option, exact_flag=False))    
     
 #########################################
+st.title(f'Updated: Oct 23, 2023 -- Total heroes = {len(df)}')
 
 df2 = df[np.all(idx_all,axis=0)]