Update app.py

app.py

@@ -125,20 +125,24 @@ def explainability(_):
         else:
             sv = shap_values
 
-        # Align number of columns
-        if sv.shape[1] != X_test.shape[1]:
-            X_test_trimmed = pd.DataFrame(X_test.values[:, :sv.shape[1]], columns=[f"Feature_{i}" for i in range(sv.shape[1])])
+        # Use safe feature names if mismatch, fallback to dummy
+        num_features = sv.shape[1]
+        if num_features <= X_test.shape[1]:
+        feature_names = X_test.columns[:num_features]
         else:
-            X_test_trimmed = X_test
+        feature_names = [f"Feature_{i}" for i in range(num_features)]
 
-        shap.summary_plot(sv, X_test_trimmed, show=False)
+        X_shap_safe = pd.DataFrame(np.zeros_like(sv), columns=feature_names)
+
+        shap.summary_plot(sv, X_shap_safe, show=False)
         shap_path = "./shap_plot.png"
         plt.title("SHAP Summary")
         plt.savefig(shap_path)
         if wandb.run:
-            wandb.log({"shap_summary": wandb.Image(shap_path)})
+        wandb.log({"shap_summary": wandb.Image(shap_path)})
         plt.clf()
 
+
     except Exception as e:
         shap_path = "./shap_error.png"
         print("SHAP plotting failed:", e)