From 530d871fd33df82fadd40af7a4d6f2048979ef16 Mon Sep 17 00:00:00 2001
From: Spicy_Marinara <mgrabower97@gmail.com>
Date: Mon, 29 Dec 2025 18:07:33 +0100
Subject: [PATCH] Update layout.js

---
 src/systems/ui/layout.js | 32 ++++++++++++++++++++++++--------
 1 file changed, 24 insertions(+), 8 deletions(-)

diff --git a/src/systems/ui/layout.js b/src/systems/ui/layout.js
index bd77dd0..f3130a2 100644
--- a/src/systems/ui/layout.js
+++ b/src/systems/ui/layout.js
@@ -18,15 +18,31 @@ import { i18n } from '../../core/i18n.js';
  * Toggles the visibility of plot buttons based on settings.
  */
 export function togglePlotButtons() {
-    if (extensionSettings.enablePlotButtons && extensionSettings.enabled) {
-        $('#rpg-plot-buttons').show();
+    if (!extensionSettings.enabled) {
+        $('#rpg-plot-buttons').hide();
+        return;
+    }
 
-        // Show/hide encounter button based on encounter settings
-        if (extensionSettings.encounterSettings?.enabled) {
-            $('#rpg-encounter-button').show();
-        } else {
-            $('#rpg-encounter-button').hide();
-        }
+    // Show/hide plot progression buttons based on enablePlotButtons setting
+    if (extensionSettings.enablePlotButtons) {
+        $('#rpg-plot-random').show();
+        $('#rpg-plot-natural').show();
+    } else {
+        $('#rpg-plot-random').hide();
+        $('#rpg-plot-natural').hide();
+    }
+
+    // Show/hide encounter button independently based on encounter settings
+    if (extensionSettings.encounterSettings?.enabled) {
+        $('#rpg-encounter-button').show();
+    } else {
+        $('#rpg-encounter-button').hide();
+    }
+
+    // Show the container if at least one button is visible
+    const shouldShowContainer = extensionSettings.enablePlotButtons || extensionSettings.encounterSettings?.enabled;
+    if (shouldShowContainer) {
+        $('#rpg-plot-buttons').show();
     } else {
         $('#rpg-plot-buttons').hide();
     }