update

.gitignore

@@ -1,2 +1,3 @@
+__pycache__
 .vscode
 .tmp/res

app.py

@@ -478,6 +478,7 @@ class ExplainerCheckbox(Component):
         self.experiment = experiment
         self.gallery = gallery
         
+        self.optimal_exp_id = gr.State(None)
         self.default_exp_id = self.get_explainer_id_by_name(explainer_name)
         self.obj_metric = self.get_metric_id_by_name(OBJECTIVE_METRIC)
 
@@ -498,9 +499,9 @@ class ExplainerCheckbox(Component):
         checkbox_group_info = self.groups.update_check(checkbox_group_info, self.default_exp_id, evt._data['value'])
         return checkbox_group_info
 
-    def optimal_on_select(self, evt: gr.EventData, checkbox_group_info):
-        if hasattr(self, "optimal_exp_id"):
-            checkbox_group_info = self.groups.update_check(checkbox_group_info, self.optimal_exp_id, evt._data['value'])
+    def optimal_on_select(self, evt: gr.EventData, checkbox_group_info, optimal_exp_id):
+        if optimal_exp_id is not None:
+            checkbox_group_info = self.groups.update_check(checkbox_group_info, optimal_exp_id, evt._data['value'])
         else:
             raise ValueError("Optimal explainer id is not found.")
         return checkbox_group_info
@@ -513,12 +514,12 @@ class ExplainerCheckbox(Component):
             self.opt_check = gr.Checkbox(label="Optimized Parameter (Not Optimal)", interactive=False)
 
             self.default_check.select(self.default_on_select, self.groups.info, self.groups.info)
-            self.opt_check.select(self.optimal_on_select, self.groups.info, self.groups.info)
+            self.opt_check.select(self.optimal_on_select, [self.groups.info, self.optimal_exp_id], self.groups.info)
 
             self.bttn = gr.Button(value="Optimize", size="sm", variant="primary")
 
             @spaces.GPU
-            def optimize(checkbox_group_info):
+            def optimize(optimal_exp_id, checkbox_group_info):
                 data_id = self.gallery.selected_index
         
                 opt_output = self.experiment.optimize(
@@ -541,13 +542,13 @@ class ExplainerCheckbox(Component):
                 self.experiment.manager._explainers.append(opt_output.explainer)
                 self.experiment.manager._explainer_ids.append(opt_explainer_id)
                 self.groups.insert_check(checkbox_group_info, self.explainer_name, opt_explainer_id, opt_postprocessor_id)
-                self.optimal_exp_id = opt_explainer_id
+                optimal_exp_id = opt_explainer_id
                 checkbox = gr.update(label="Optimized Parameter (Optimal)", interactive=True)
                 bttn = gr.update(value="Optimized", variant="secondary")
 
-                return [checkbox_group_info, checkbox, bttn]
+                return [optimal_exp_id, checkbox_group_info, checkbox, bttn]
 
-            self.bttn.click(optimize, inputs=[self.groups.info], outputs=[self.groups.info, self.opt_check, self.bttn], queue=True, concurrency_limit=1)
+            self.bttn.click(optimize, inputs=[self.optimal_exp_id, self.groups.info], outputs=[self.optimal_exp_id, self.groups.info, self.opt_check, self.bttn], queue=True, concurrency_limit=1)
         
 
 class ExpRes(Component):