Update app.py

app.py

@@ -49,7 +49,7 @@ def display_image(url, scale=0.5):
 
 def display_heroes_from_df(df):
     display_cols = ['image','name', 'color', 'star', 'class', 'speed', 'power', 'attack', 'defense', 'health', 'types', 'source', 'family']
-    df[display_cols]
+
     st.dataframe(df[display_cols],
                  column_config={
                          "image": st.column_config.ImageColumn("Avatar", help="")},
@@ -65,14 +65,43 @@ def display_heroes_from_df(df):
         #     st.write(sp)
 
 #########################################
-
+## Load the main file (TODO: caching)=
 
 df = pd.read_csv('heroes_ep.csv')
+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())
 
+#########################################
+## Select options
+with st.sidebar:
+    # 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)
+    special_type_option = st.text_input(label="SpecialSkill Category", value="Hit 3")
+    special_text_option = st.text_input(label="SpecialSkill Text", value="Dispel")
 idx_all = []
-idx_all.append(filter_by_1col(df, 'types', 'hit 3'))
-idx_all.append(filter_by_1col(df, 'effects', 'dispel'))
-idx_all.append(filter_by_1col(df, 'speed', 'fast', exact_flag=False))
+# if star_option is not None:
+#     idx_all.append(filter_by_1col(df, 'star', star_option, exact_flag=False))    
+
+if speed_option is not None:
+    idx_all.append(filter_by_1col(df, 'speed', speed_option, exact_flag=True))    
+
+if color_option is not None:
+    idx_all.append(filter_by_1col(df, 'color', color_option, exact_flag=False))    
+
+if class_option is not None:
+    idx_all.append(filter_by_1col(df, 'class', class_option, exact_flag=False))    
+
+if special_type_option is not None:
+    idx_all.append(filter_by_1col(df, 'types', special_type_option, exact_flag=False))    
+
+if special_text_option is not None:
+    idx_all.append(filter_by_1col(df, 'effects', special_text_option, exact_flag=False))    
+    
+#########################################
 
 df2 = df[np.all(idx_all,axis=0)]