add dropdown for split_name

app.py

@@ -25,7 +25,7 @@ print(f"Initializing data plotter...")
 data_plotter = MatplotlibDataPlotter(single_df, pair_df, num_domains_in_region_df)
 
 
-def update_all_plots(frequency, split_name='stratified'):
+def update_all_plots(frequency, split_name):
     return data_plotter.plot_single_domains(frequency, split_name), data_plotter.plot_pair_domains(frequency, split_name)
 
 
@@ -45,6 +45,12 @@ with gr.Blocks(title="Interactive Wave Plotter") as demo:
         )
     
     with gr.Row():
+        with gr.Column():
+            split_selector = gr.Dropdown(
+                choices=["stratified"] + BIOSYN_CLASS_NAMES,
+                value="stratified",
+                label="Split name"
+            )
         with gr.Column():
             single_domains_plot = gr.Plot(
                 label="Single domains",
@@ -66,7 +72,7 @@ with gr.Blocks(title="Interactive Wave Plotter") as demo:
     
     frequency_slider.release(
         fn=update_all_plots,
-        inputs=[frequency_slider],
+        inputs=[frequency_slider, split_selector],
         outputs=[single_domains_plot, pair_domains_plot]#, cosine_plot]
     )