Updated GUI.

app.py

@@ -241,8 +241,7 @@ def run_inference_fast(file):
         save_output_csv(agg, "ranked"),
         save_output_csv(topk, "topk"),
         save_output_csv(df, "per_benchmark"),
-        topk.to_dict("records"),
-        "Inference complete. Generating SHAP plots..."
+        topk.to_dict("records")
     )
 
 
@@ -251,11 +250,8 @@ def run_inference_fast(file):
 # =========================================================
 def generate_shap(topk_records):
     topk_df = pd.DataFrame(topk_records)
-
     if len(topk_df) < 2:
-        return [None] * (len(WORKLOADS) * len(TARGETS) * 2) + [
-            "Need at least 2 configs for SHAP comparison."
-        ]
+        return [None] * (len(WORKLOADS) * len(TARGETS) * 2)
 
     rank1 = topk_df.iloc[0]
     rank2 = topk_df.iloc[1]
@@ -275,7 +271,7 @@ def generate_shap(topk_records):
                     f"model_{workload}_{target}"
                 ).shap_values(X)[0]
 
-                plt.figure(figsize=(7, 4))
+                plt.figure(figsize=(7, 4), dpi=180)
 
                 shap.plots.bar(
                     shap.Explanation(
@@ -290,12 +286,16 @@ def generate_shap(topk_records):
                 path = OUTPUT_DIR / f"shap_{uuid.uuid4().hex[:8]}.png"
 
                 plt.tight_layout()
-                plt.savefig(path, dpi=120, bbox_inches="tight")
+                plt.savefig(
+                    path,
+                    dpi=180,
+                    bbox_inches="tight",
+                    pad_inches=0.05
+                )
                 plt.close()
 
                 outputs.append(str(path))
 
-    outputs.append("SHAP generation complete.")
     return outputs
 
 
@@ -304,20 +304,9 @@ def generate_shap(topk_records):
 # =========================================================
 CUSTOM_CSS = """
 .gradio-container {
-    max-width: 1600px !important;
-    margin: auto !important;
-}
-
-.gr-image,
-.gr-image img,
-.gr-dataframe {
-    max-width: 100% !important;
+    max-width: 1450px !important;
+    margin: auto;
 }
-
-.gr-row {
-    flex-wrap: wrap !important;
-}
-
 footer {
     display: none !important;
 }
@@ -333,8 +322,6 @@ Upload cache configurations once — automatically evaluated across all workload
     file_input = gr.File(label="Upload Cache Config CSV")
     run_btn = gr.Button("Run Inference", variant="primary")
 
-    status_md = gr.Markdown()
-
     with gr.Tabs():
         with gr.Tab("📊 Summary"):
             summary_df = gr.Dataframe(interactive=False)
@@ -379,15 +366,14 @@ Upload cache configurations once — automatically evaluated across all workload
             ranked_csv,
             topk_csv,
             per_benchmark_csv,
-            topk_state,
-            status_md
+            topk_state
         ]
     )
 
     inference_event.then(
         fn=generate_shap,
         inputs=topk_state,
-        outputs=shap_image_components + [status_md]
+        outputs=shap_image_components
     )
 
 demo.launch(