update

app.py

@@ -2,8 +2,8 @@
 import time
 import os
 import gradio as gr
-from pnpxai.core.experiment.auto_explanation import AutoExplanationForImageClassification
-from pnpxai.core.detector.detector import extract_graph_data, symbolic_trace
+from pnpxai.core.experiment import AutoExplanationForImageClassification
+from pnpxai.core.detector import extract_graph_data, symbolic_trace
 import matplotlib.pyplot as plt
 import plotly.graph_objects as go
 import plotly.express as px
@@ -16,6 +16,7 @@ N_FEATURES_TO_SHOW = 5
 OPT_N_TRIALS = 10
 OBJECTIVE_METRIC = "AbPC"
 SAMPLE_METHOD = "tpe"
+DEFAULT_EXPLAINER = ["GradientXInput", "IntegratedGradients", "LRPEpsilonPlus"]
 
 class App:
     def __init__(self):
@@ -388,7 +389,12 @@ class ExplainerCheckboxGroup(Component):
 
         self.info = []
         for exp, exp_id in zip(explainers, exp_ids):
-            self.info.append({'nm': exp.__class__.__name__, 'id': exp_id, 'pp_id' : 0, 'mode': 'default', 'checked': True})
+            exp_nm = exp.__class__.__name__
+            if exp_nm in DEFAULT_EXPLAINER:
+                checked = True
+            else:
+                checked = False
+            self.info.append({'nm': exp_nm, 'id': exp_id, 'pp_id' : 0, 'mode': 'default', 'checked': checked})
 
     def update_check(self, exp_id, val=None):
         for info in self.info:
@@ -407,7 +413,9 @@ class ExplainerCheckboxGroup(Component):
     def update_gallery_change(self):
         checkboxes = []
         bttns = []
-        checkboxes += [gr.Checkbox(label="Default Parameter", value=True, interactive=True)] * len(self.explainer_objs)
+        for exp in self.explainer_objs:
+            val = exp.explainer_name in DEFAULT_EXPLAINER
+            checkboxes.append(gr.Checkbox(label="Default Parameter", value=val, interactive=True))
         checkboxes += [gr.Checkbox(label="Optimized Parameter (Not Optimal)", value=False, interactive=False)] * len(self.explainer_objs)
         bttns += [gr.Button(value="Optimize", size="sm", variant="primary")] * len(self.explainer_objs)
 
@@ -504,7 +512,8 @@ class ExplainerCheckbox(Component):
 
     def show(self):
         with gr.Accordion(self.explainer_name, open=False):
-            self.default_check = gr.Checkbox(label="Default Parameter", value=True, interactive=True)
+            checked = next(filter(lambda x: x['nm'] == self.explainer_name, self.groups.info))['checked']
+            self.default_check = gr.Checkbox(label="Default Parameter", value=checked, interactive=True)
             self.opt_check = gr.Checkbox(label="Optimized Parameter (Not Optimal)", interactive=False)
 
             self.default_check.select(self.default_on_select)
@@ -587,7 +596,9 @@ class ImageClsApp(App):
     def title(self):
         return """
         <div style="text-align: center;">
+        <a href="https://openxaiproject.github.io/pnpxai/">
             <img src="/file=data/static/XAI-Top-PnP.svg" width="100" height="100">
+        </a>
             <h1> Plug and Play XAI Platform for Image Classification </h1>
         </div>
         """