Update app.py

app.py

@@ -2071,7 +2071,8 @@ with tab_predict:
         with cA:
             do_batch = st.button(f"Compute batch SHAP for first {batch_n} rows", key="batch_shap_btn")
         with cB:
-            max_display = st.slider("Top features to display", 5, 40, 20, 1, key="batch_max_display")
+            max_display_batch = st.slider("Top features to display (batch)",5, 40, 20, 1,key="batch_max_display")
+
         with cC:
             show_beeswarm = st.checkbox("Show beeswarm (slower)", value=True, key="batch_beeswarm")
         
@@ -2186,7 +2187,7 @@ with tab_predict:
                 features=X_dense,
                 feature_names=names,
                 plot_type="bar",
-                max_display=max_display,
+                max_display=max_display_batch,
                 show=False
             )
             fig_bar = plt.gcf()
@@ -2201,7 +2202,7 @@ with tab_predict:
                     shap_vals_batch,
                     features=X_dense,
                     feature_names=names,
-                    max_display=max_display,
+                    max_display=max_display_batch,
                     show=False
                 )
                 fig_swarm = plt.gcf()
@@ -2217,7 +2218,8 @@ with tab_predict:
                 default=[0],
                 key="batch_rows_to_plot"
             )
-        
+            max_display_single = st.slider("Top features to display (single-row SHAP)",5, 40, 20, 1,key="single_max_display")
+
             max_waterfalls = st.slider("Max waterfall plots to render", 1, 10, 3, 1, key="max_waterfalls")
             rows_to_plot = rows_to_plot[:max_waterfalls]
         
@@ -2240,8 +2242,10 @@ with tab_predict:
                 fig_w, ax = plt.subplots(figsize=FIGSIZE, dpi=plot_dpi_screen)
                 plt.sca(ax)  # important: set current axis for SHAP
             
-                shap.plots.waterfall(exp, show=False, max_display=max_display)
-            
+                
+                shap.plots.waterfall(exp, show=False, max_display=max_display_single)
+                shap.plots.bar(exp, show=False, max_display=max_display_single)
+
                 render_plot_with_download(
                     fig_w,
                     title=f"Batch SHAP waterfall (row {r})",