Add a button to update setting changes

regularization.py

@@ -263,7 +263,8 @@ class Regularization:
             l1_ratio = 1 if reg_type == "l1" else 0
             alphas, coefs, *_ = ElasticNet.path(X, y, l1_ratio=l1_ratio, alphas=alphas)
         else:
-            coefs = np.random.randn(2, len(alphas))  # temporary
+            return Image.new("RGB", (800, 800), color="white")
+
         coefs = coefs.T
 
         fig, ax = plt.subplots(figsize=(8, 8))
@@ -287,6 +288,9 @@ class Regularization:
             raise ValueError(f"loss_type must be one of {self.LOSS_TYPES}")
         return loss_type
 
+    def update_reg_path_visibility(self, loss_type: str):
+        return gr.update(visible=(loss_type == "l2"))
+
     def update_regularizer(self, reg_type: str):
         if reg_type not in self.REGULARIZER_TYPES:
             raise ValueError(f"reg_type must be one of {self.REGULARIZER_TYPES}")
@@ -408,19 +412,20 @@ class Regularization:
                                 visible=True,
                             )
 
-                        with gr.Row():
-                            regularizer_type_selection = gr.Dropdown(
-                                choices=['l1', 'l2'],
-                                label='Regularizer type',
-                                value='l2',
-                                visible=True,
-                            )
-
-                            reg_textbox = gr.Textbox(
-                                label="Regularizer levels",
-                                value="10, 20, 30",
-                                interactive=True,
-                            )
+                        with gr.Group():
+                            with gr.Row():
+                                regularizer_type_selection = gr.Dropdown(
+                                    choices=['l1', 'l2'],
+                                    label='Regularizer type',
+                                    value='l2',
+                                    visible=True,
+                                )
+
+                                reg_textbox = gr.Textbox(
+                                    label="Regularizer levels",
+                                    value="10, 20, 30",
+                                    interactive=True,
+                                )
 
                         with gr.Row():
                             w1_textbox = gr.Textbox(
@@ -444,8 +449,11 @@ class Regularization:
                                 label="Resolution (#points)",
                             )
 
+                        submit_button = gr.Button("Submit changes")
+
                         with gr.Row():
                             path_checkbox = gr.Checkbox(label="Show regularization path", value=False)
+
                     with gr.Tab("Data"):
                         dataset_view = DatasetView()
                         dataset_view.build(state=dataset)
@@ -496,6 +504,10 @@ class Regularization:
                         fn=self.update_loss_type, 
                         inputs=[loss_type_selection],
                         outputs=[loss_type],
+                    ).then(
+                        fn=self.update_reg_path_visibility,
+                        inputs=[loss_type_selection],
+                        outputs=[path_checkbox],
                     ).then(
                         fn=self.compute_and_plot_loss_and_reg,
                         inputs=[
@@ -611,6 +623,33 @@ class Regularization:
                         outputs=self.loss_and_regularization_plot,
                     )
 
+                    submit_button.click(
+                        self.update_w1_range, 
+                        inputs=[w1_textbox], 
+                        outputs=[w1_range],
+                    ).then(
+                        self.update_w2_range, 
+                        inputs=[w2_textbox], 
+                        outputs=[w2_range],
+                    ).then(
+                        self.update_reg_levels, 
+                        inputs=[reg_textbox], 
+                        outputs=[reg_levels],
+                    ).then(
+                        fn=self.compute_and_plot_loss_and_reg,
+                        inputs=[
+                            dataset,
+                            loss_type,
+                            reg_type,
+                            reg_levels,
+                            w1_range,
+                            w2_range,
+                            num_dots,
+                            plot_regularization_path,
+                        ],
+                        outputs=self.loss_and_regularization_plot,
+                    )
+
                     resolution_slider.change(
                         self.update_resolution, 
                         inputs=[resolution_slider],