Update app.py

app.py

@@ -74,19 +74,31 @@ class_values = ['None'] + list(df['class'].unique())
 star_values = ['None'] + list(df['star'].unique())
 color_values = ['None'] + list(df['color'].unique())
 speed_values = ['None'] + list(df['speed'].unique())
+source_values = ['None'] + list(df['source'].unique())
 
 #########################################
 ## Select options
-## Source, family, costume, sorting
+## TODO: family, costume
 
 with st.sidebar:
+    st.title('Filter Options')
+    name_option = st.text_input(label="Name:", value="")
     # star_option = st.selectbox(label='Speed:', options=star_values, index=5)
     color_option = st.selectbox(label='Color:', options=color_values, index=0)
     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)
     special_type_option = st.text_input(label="SpecialSkill Category", value="Hit 3")
     special_text_option = st.text_input(label="SpecialSkill Text", value="Dispel")
+
+    st.title('Sorted By')
+    sort_option = st.selectbox(label='Sort by', options=display_cols[1:], index=0)
+    
 idx_all = []
+
+if name_option != '':
+    idx_all.append(filter_by_1col(df, 'name', name_option, exact_flag=False)) 
+
 # if star_option is not None:
 #     idx_all.append(filter_by_1col(df, 'star', star_option, exact_flag=False))    
 
@@ -99,6 +111,10 @@ if color_option != 'None':
 if class_option != 'None':
     idx_all.append(filter_by_1col(df, 'class', class_option, exact_flag=False))    
 
+if source_option != 'None':
+    idx_all.append(filter_by_1col(df, 'source', source_option, exact_flag=False))    
+
+
 if special_type_option  != '':
     idx_all.append(filter_by_1col(df, 'types', special_type_option, exact_flag=False))    
 
@@ -109,4 +125,4 @@ if special_text_option != '':
 
 df2 = df[np.all(idx_all,axis=0)]
 
-display_heroes_from_df(df2.sort_values("power", ascending=False))
+display_heroes_from_df(df2.sort_values(sort_option, ascending=False))