trying to visualize

app.py

@@ -106,10 +106,13 @@ def gradio_interface() -> gr.Blocks:
                 ds = load_dataset(results_repo, split='train')
                 full_df = pd.DataFrame(ds)
                 filenames = full_df['result_filename'].to_list()
-                dropdown = gr.Dropdown(choices=filenames, label="Choose a file", value=filenames[0])
+                with gr.Row():
+                    dropdown = gr.Dropdown(choices=filenames, label="Choose a file", value=filenames[0])
+                    rld_btn = gr.Button(value="Reload")
+
                 plot = gr.Plot()
 
-                def update_plot(selected_file):
+                def get_boundary_vis(selected_file):
                     local_path = read_file_from_hub(selected_file)
                     with Path(local_path).open("r") as f:
                         raw = f.read()
@@ -121,16 +124,11 @@ def gradio_interface() -> gr.Blocks:
                     else:
                         boundary = load_boundary(boundary_json)
 
-                    vis = make_visual(boundary)
+                    vis = make_visual(boundary, inputs=[dropdown])
                     return vis
 
-                dropdown.change(fn=update_plot, inputs=dropdown, outputs=plot)
-                gr.on(
-                    triggers="load",  # when app loads
-                    fn=update_plot,
-                    inputs=[dropdown],
-                    outputs=[plot]
-                )
+                demo.load(get_boundary_vis, dropdown, plot)
+                rld_btn.click(get_boundary_vis, dropdown, plot)
 
             with gr.TabItem("✉️ Submit", elem_id="boundary-benchmark-tab-table"):
                 gr.Markdown(