Use current dataset settings for setting suggested settings

backend/src/logic.py

@@ -281,6 +281,6 @@ def compute_suggested_settings(
 
     opt_norm = float(np.linalg.norm(moore_penrose, ord=2))
 
-    reg_levels = [i / 4 * opt_norm for i in range(3)]
+    reg_levels = [i / 4 * opt_norm for i in range(1, 4)]
 
     return w1_range, w2_range, reg_levels

backend/src/manager.py

@@ -37,6 +37,36 @@ class Manager:
         x2_col: int,
         y_col: int,
     ) -> None:
+        self.dataset = self._compute_dataset(
+            dataset_type,
+            function,
+            x1_range_input,
+            x2_range_input,
+            x_selection_method,
+            sigma,
+            nsample,
+            csv_file,
+            has_header,
+            x1_col,
+            x2_col,
+            y_col,
+        )
+
+    def _compute_dataset(
+        self,
+        dataset_type: str,
+        function: str,
+        x1_range_input: str,
+        x2_range_input: str,
+        x_selection_method: str,
+        sigma: float,
+        nsample: int,
+        csv_file: str,
+        has_header: bool,
+        x1_col: int,
+        x2_col: int,
+        y_col: int,
+    ) -> Dataset:
         if dataset_type == "Generate":
             try:
                 parsed_function = sympify(function)
@@ -50,7 +80,7 @@ class Manager:
             if method not in ("grid", "random"):
                 raise ValueError(f"Invalid x_selection_method: {x_selection_method}")
 
-            self.dataset = generate_dataset(
+            return generate_dataset(
                 parsed_function,
                 x1_range,
                 x2_range,
@@ -60,7 +90,7 @@ class Manager:
         elif dataset_type == "CSV":
             csv_path = self._resolve_csv_path(csv_file)
             try:
-                self.dataset = load_dataset_from_csv(
+                return load_dataset_from_csv(
                     csv_path,
                     bool(has_header),
                     int(x1_col),
@@ -157,11 +187,37 @@ class Manager:
         strength_plot = self._generate_strength_plot(self.plots_data.path)
         return self, contour_plot, data_plot, strength_plot
 
-    def handle_use_suggested_settings(self) -> tuple[Manager, str, str, str]:
-        if self.dataset is None:
-            raise ValueError("Dataset is not initialized")
+    def handle_use_suggested_settings(
+        self,
+        dataset_type: str,
+        function: str,
+        x1_range_input: str,
+        x2_range_input: str,
+        x_selection_method: str,
+        sigma: float,
+        nsample: int,
+        csv_file: str,
+        has_header: bool,
+        x1_col: int,
+        x2_col: int,
+        y_col: int,
+    ) -> tuple[Manager, str, str, str]:
+        dataset = self._compute_dataset(
+            dataset_type,
+            function,
+            x1_range_input,
+            x2_range_input,
+            x_selection_method,
+            sigma,
+            nsample,
+            csv_file,
+            has_header,
+            x1_col,
+            x2_col,
+            y_col,
+        )
 
-        w1_range, w2_range, reg_levels = compute_suggested_settings(self.dataset)
+        w1_range, w2_range, reg_levels = compute_suggested_settings(dataset)
 
         w1_range_input = f"{w1_range[0]:.2f}, {w1_range[1]:.2f}"
         w2_range_input = f"{w2_range[0]:.2f}, {w2_range[1]:.2f}"

frontends/gradio/main.py

@@ -94,8 +94,33 @@ def handle_generate_plots(
 
 def handle_use_suggested_settings(
     manager: Manager,
+    dataset_type: str,
+    function: str,
+    x1_range_input: str,
+    x2_range_input: str,
+    x_selection_method: str,
+    sigma: float,
+    nsample: int,
+    csv_file: str,
+    has_header: bool,
+    x1_col: int,
+    x2_col: int,
+    y_col: int,
 ) -> tuple[Manager, str, str, str]:
-    return manager.handle_use_suggested_settings()
+    return manager.handle_use_suggested_settings(
+        dataset_type,
+        function,
+        x1_range_input,
+        x2_range_input,
+        x_selection_method,
+        sigma,
+        nsample,
+        csv_file,
+        has_header,
+        x1_col,
+        x2_col,
+        y_col,
+    )
 
 
 def launch():
@@ -307,7 +332,21 @@ def launch():
                     
                     gr.Button("Use suggested settings").click(
                         fn=handle_use_suggested_settings,
-                        inputs=[manager_state],
+                        inputs=[
+                            manager_state,
+                            dataset_type,
+                            function,
+                            x1_textbox,
+                            x2_textbox,
+                            x_selection_method,
+                            sigma,
+                            nsample,
+                            csv_file,
+                            has_header,
+                            x1_col,
+                            x2_col,
+                            y_col,
+                        ],
                         outputs=[
                             manager_state,
                             w1_range_textbox,