update check

app.py

@@ -442,7 +442,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"): # TO BE FIXED : exp.optimal_exp_id is gr.State Object
+            if hasattr(exp, "optimal_exp_id"): 
                 checkbox_group_info = self.update_check(checkbox_group_info, exp.optimal_exp_id, False)
         return checkboxes + bttns + [checkbox_group_info]
 
@@ -504,9 +504,8 @@ class ExplainerCheckbox(Component):
         return checkbox_group_info
 
     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'])
+        if hasattr(self, "optimal_exp_id"):
+            checkbox_group_info = self.groups.update_check(checkbox_group_info, self.optimal_exp_id, evt._data['value'])
         else:
             raise ValueError("Optimal result is not found.")
         return checkbox_group_info
@@ -559,6 +558,7 @@ class ExplainerCheckbox(Component):
                 return [opt_res, checkbox_group_info, checkbox, bttn]
 
             def update_exp(exp_res):
+                _id = exp_res['id']
                 try:
                     kwargs = {}
                     has_baseline = False
@@ -604,13 +604,13 @@ class ExplainerCheckbox(Component):
                     
                     kwargs['model'] = self.experiment.model
                     explainer = exp_res['class'](**kwargs)
-                    _id = exp_res['id']
                 except Exception as e:
                     # If the optimization is failed, use the default parameter explainer as optimal
                     explainer = self.experiment.manager._explainers[self.default_exp_id]
 
                 self.experiment.manager._explainers.append(explainer)
                 self.experiment.manager._explainer_ids.append(_id)
+                self.optimal_exp_id = _id
 
             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(update_exp, self.opt_res)