Update app.py

app.py

@@ -29,13 +29,18 @@ narrator_bios.loc[49845, 'Number of Narrations'] = 327512
 narrator_bios['Generation'] = narrator_bios['Generation'].replace([None], [-1])
 narrator_bios['Generation'] = narrator_bios['Generation'].astype(int)
 
-
+cols = narrator_bios.columns.tolist()
+def filter_df(df, col, val):
+    return df[df[col] == val]
 
 edge_info = dataset.to_pandas()
 taraf_info = dataset2.to_pandas()
 min_year = int(taraf_info['Year'].min())
 max_year = int(taraf_info['Year'].max())
 cmap = plt.colormaps['cool']
+
+
+
 def value_to_hex(value):
     rgba_color = cmap(value)
     return "#{:02X}{:02X}{:02X}".format(int(rgba_color[0] * 255), int(rgba_color[1] * 255), int(rgba_color[2] * 255))
@@ -94,6 +99,12 @@ with gr.Blocks() as demo:
         text_output = gr.DataFrame()
         text_button = gr.Button("Search")
         text_button.click(narrator_retriever, inputs=text_input, outputs=text_output)
+
+        col = gr.Dropdown(choices = cols)
+        df_filter = gr.Textbox()
+        filter_button = gr.Button('Filter')
+        output2 = gr.DataFrame(filter_df, inputs = [text_output, col, df_filter], outputs= [gr.DataFrame()])
+        filter_button.click()
     with gr.Tab("Visualize Network"):
         with gr.Row():
             image_input = gr.Number()