deliver optimal result

app.py

@@ -438,7 +438,7 @@ class ExplainerCheckboxGroup(Component):
         for exp in self.explainer_objs:
             val = exp.explainer_name in DEFAULT_EXPLAINER
             checkbox_group_info = self.update_check(checkbox_group_info, exp.default_exp_id, val)
-            if hasattr(exp, "optimal_exp_id"):
+            if hasattr(exp, "optimal_exp_id"): # TO BE FIXED : exp.optimal_exp_id is gr.State Object
                 checkbox_group_info = self.update_check(checkbox_group_info, exp.optimal_exp_id, False)
         return checkboxes + bttns + [checkbox_group_info]
 
@@ -478,7 +478,7 @@ class ExplainerCheckbox(Component):
         self.experiment = experiment
         self.gallery = gallery
         
-        self.optimal_exp_id = gr.State(None)
+        self.opt_res = 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)
 
@@ -499,13 +499,19 @@ 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, optimal_exp_id):
-        if optimal_exp_id is not None:
+    def optimal_on_select(self, evt: gr.EventData, checkbox_group_info, opt_res):
+        if opt_res is not None:
+            optimal_exp_id = opt_res['id']
             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.")
+            raise ValueError("Optimal result is not found.")
         return checkbox_group_info
 
+    def update_exp(self, exp_res):
+        self.experiment.manager._explainers.append(exp_res['explainer'])
+        self.experiment.manager._explainer_ids.append(exp_res['id'])
+
+
     def show(self):
         val = self.explainer_name in DEFAULT_EXPLAINER
         with gr.Accordion(self.explainer_name, open=val):
@@ -514,12 +520,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.optimal_exp_id], self.groups.info)
+            self.opt_check.select(self.optimal_on_select, [self.groups.info, self.opt_res], self.groups.info)
 
             self.bttn = gr.Button(value="Optimize", size="sm", variant="primary")
 
             @spaces.GPU
-            def optimize(optimal_exp_id, checkbox_group_info):
+            def optimize(checkbox_group_info):
                 data_id = self.gallery.selected_index
         
                 opt_output = self.experiment.optimize(
@@ -537,18 +543,23 @@ class ExplainerCheckbox(Component):
                         opt_postprocessor_id = pp_id
                         break
 
-                opt_explainer_id = max([x['id'] for x in checkbox_group_info]) + 1
+                opt_exp_id = max([x['id'] for x in checkbox_group_info]) + 1
                 opt_output.explainer.model = self.experiment.model
-                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)
-                optimal_exp_id = opt_explainer_id
+                # self.experiment.manager._explainers.append(opt_output.explainer)
+                # self.experiment.manager._explainer_ids.append(opt_exp_id)
+                
+                opt_res = {
+                    'id': opt_exp_id,
+                    'explainer': opt_output.explainer,
+                }
+                self.groups.insert_check(checkbox_group_info, self.explainer_name, opt_exp_id, opt_postprocessor_id)
                 checkbox = gr.update(label="Optimized Parameter (Optimal)", interactive=True)
                 bttn = gr.update(value="Optimized", variant="secondary")
 
-                return [optimal_exp_id, checkbox_group_info, checkbox, bttn]
+                return [opt_res, checkbox_group_info, checkbox, bttn]
 
-            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)
+            self.bttn.click(optimize, inputs=[self.groups.info], outputs=[self.opt_res, self.groups.info, self.opt_check, self.bttn], queue=True, concurrency_limit=1)
+            self.opt_res.change(self.update_exp, self.opt_res)
         
 
 class ExpRes(Component):