From a93596e28e24fd0ca2c75e65c4887e11abc6507b Mon Sep 17 00:00:00 2001
From: Peter Clement <peter@budibase.com>
Date: Tue, 15 Apr 2025 11:49:02 +0100
Subject: [PATCH] proper handling of ai being enabled

---
 .../src/components/common/CodeEditor/AIGen.svelte    | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/packages/builder/src/components/common/CodeEditor/AIGen.svelte b/packages/builder/src/components/common/CodeEditor/AIGen.svelte
index 3b1e8ffff2..53ae78e76f 100644
--- a/packages/builder/src/components/common/CodeEditor/AIGen.svelte
+++ b/packages/builder/src/components/common/CodeEditor/AIGen.svelte
@@ -24,7 +24,7 @@
     accept: void
     reject: { code: string | null }
   }>()
-
+  $: console.log($auth.user?.llm)
   let promptInput: HTMLInputElement
   let buttonElement: HTMLButtonElement
   let promptLoading = false
@@ -34,13 +34,13 @@
   let containerWidth = "auto"
   let promptText = ""
   let animateBorder = false
-  let aiEnabled = false
   let creditsExceeded = false // TODO: Make this computed when quota is implemented
   let switchOnAIModal: Modal
   let addCreditsModal: Modal
 
   $: accountPortalAccess = $auth?.user?.accountPortalAccess
   $: accountPortal = $admin.accountPortalUrl
+  $: aiEnabled = !!$auth?.user?.llm
 
   async function generateJs(prompt: string) {
     if (!prompt.trim()) return
@@ -144,6 +144,8 @@
         src={BBAI}
         alt="AI"
         class="ai-icon"
+        class:disabled={expanded &&
+          (suggestedCode !== null || !aiEnabled || creditsExceeded)}
         on:click={e => {
           e.stopPropagation()
           toggleExpand()
@@ -378,6 +380,7 @@
     gap: var(--spacing-s);
     z-index: 4;
     flex-shrink: 0;
+    margin-right: var(--spacing-s);
   }
 
   .button-content-wrapper {
@@ -396,4 +399,9 @@
     color: var(--spectrum-global-color-gray-500);
     cursor: not-allowed;
   }
+
+  .ai-icon.disabled {
+    filter: grayscale(1) brightness(1.5);
+    opacity: 0.5;
+  }
 </style>