making more options

app.py

@@ -41,18 +41,38 @@ def gradio_interface() -> gr.Blocks:
         ds = load_dataset(results_repo, split='train')
         full_df = pd.DataFrame(ds)
         filenames = full_df['result_filename'].to_list()
+
+        mode_selector = gr.Radio(choices=["Leaderboard", "Upload", "Generate"],
+                                 label="Select input method:",
+                                 value="Leaderboard")
+
+
         with gr.Row():
-            with gr.Column():
+            with gr.Column(visible=True) as leaderboard_ui:
                 dropdown = gr.Dropdown(choices=filenames, label="Choose a leaderboard entry", value=filenames[0])
                 rld_btn = gr.Button(value="Reload")
+            
+            with gr.Column(visible=False) as generate_ui:
+                aspect_ratio = gr.Number(label="Aspect Ratio", value=3)
+                elongation = gr.Number(label="Elongation", value=0.5)
+                rotational_transform = gr.Number(label="Rotational Transform", value=0.4)
+                n_field_periods = gr.Number(label="Number of Period Fields", value=3)
+                generate_btn = gr.Button(value="Generate")
 
-            with gr.Column():
-                plot = gr.Plot()
+            
+            plot = gr.Plot()
 
-        def get_boundary_vis(selected_file):
+        def update_ui(mode):
+            return (
+                gr.update(visible=(mode == "Leaderboard")),
+                gr.update(visible=(mode == "Upload")),
+                gr.update(visible=(mode == "Generate")),
+            )
 
-            row = full_df[full_df['result_filename'] == selected_file].iloc[0]
+        mode_selector.change(update_ui, inputs=[mode_selector], outputs=[leaderboard_ui, generate_ui])
 
+        def get_boundary_from_leaderboard(selected_file):
+            row = full_df[full_df['result_filename'] == selected_file].iloc[0]
             if row['problem_type'] == 'mhd_stable':
                 raise gr.Error("Sorry this isn't implemented for mhd_stable submissions yet!")
             else:
@@ -61,9 +81,17 @@ def gradio_interface() -> gr.Blocks:
             vis = make_visual(boundary)
             return vis
 
-        demo.load(get_boundary_vis, dropdown, plot)
-        rld_btn.click(get_boundary_vis, dropdown, plot)
+        dropdown.change(get_boundary_from_leaderboard, dropdown, plot)
+        rld_btn.click(get_boundary_from_leaderboard, dropdown, plot)
+        
+        def generate_random_boundary(aspect_ratio, elongation, rotational_transform, n_field_periods):
+            boundary = initial_guess.generate_rotating_ellipse(
+                aspect_ratio=aspect_ratio, elongation=elongation, rotational_transform=rotational_transform, n_field_periods=n_field_periods
+            )
+            vis = make_visual(boundary)
+            return vis
         
+        generate_btn.click(generate_random_boundary, [aspect_ratio, elongation, rotational_transform, n_field_periods], plot)
 
     return demo