From a99d99bbd8bca47fc33db348ef2f3c0ecc1d6d83 Mon Sep 17 00:00:00 2001
From: Andrew Michael McNutt <mcnutt.andrew@gmail.com>
Date: Tue, 15 Oct 2024 06:19:41 -0600
Subject: [PATCH] Allow AI Mute

---
 apps/color-buddy/src/controls/AddColor.svelte         |  3 +++
 apps/color-buddy/src/controls/Config.svelte           |  1 +
 apps/color-buddy/src/controls/DesignPit.svelte        |  9 +++++++--
 apps/color-buddy/src/lib/api-calls.ts                 |  3 ++-
 .../src/linting/LintToolTipContents.svelte            | 11 +++++++----
 5 files changed, 20 insertions(+), 7 deletions(-)

diff --git a/apps/color-buddy/src/controls/AddColor.svelte b/apps/color-buddy/src/controls/AddColor.svelte
index 1c4b79e..b5bf979 100644
--- a/apps/color-buddy/src/controls/AddColor.svelte
+++ b/apps/color-buddy/src/controls/AddColor.svelte
@@ -15,6 +15,9 @@
   let requestState: "idle" | "loading" | "loaded" | "failed" = "idle";
 
   function getColorForSearch(searchedString: string) {
+    if ($configStore.engine === "none") {
+      return;
+    }
     const allowedStates = new Set(["idle", "loaded", "failed"]);
     if (!allowedStates.has(requestState)) {
       return;
diff --git a/apps/color-buddy/src/controls/Config.svelte b/apps/color-buddy/src/controls/Config.svelte
index 563dc11..fabae1e 100644
--- a/apps/color-buddy/src/controls/Config.svelte
+++ b/apps/color-buddy/src/controls/Config.svelte
@@ -12,6 +12,7 @@
     // "google",
     "openai",
     "anthropic",
+    "none",
   ] as string[];
 
   const isMac = navigator.userAgent.indexOf("Mac OS X") !== -1;
diff --git a/apps/color-buddy/src/controls/DesignPit.svelte b/apps/color-buddy/src/controls/DesignPit.svelte
index 5a42a1d..c935f51 100644
--- a/apps/color-buddy/src/controls/DesignPit.svelte
+++ b/apps/color-buddy/src/controls/DesignPit.svelte
@@ -1,6 +1,7 @@
 <script lang="ts">
   import focusStore from "../stores/focus-store";
   import colorStore from "../stores/color-store";
+  import configStore from "../stores/config-store";
 
   import Tooltip from "../components/Tooltip.svelte";
   import AdjustColor from "./AdjustColor.svelte";
@@ -28,12 +29,16 @@
   style={`max-height: calc(100% - (450px + 65px + 48px + 20px))`}
 >
   <div class="flex w-full flex-wrap">
-    <SuggestionModificationToSelection />
+    {#if $configStore.engine !== "none"}
+      <SuggestionModificationToSelection />
+    {/if}
     {#if numFocused > 0}
       <DupAndDelete />
     {/if}
   </div>
-  <div class={breaker} />
+  {#if !($configStore.engine === "none" && numFocused === 0)}
+    <div class={breaker} />
+  {/if}
   {#if numFocused >= 1}
     <!-- <div class="font-bold">Adjust</div> -->
     <AdjustColor />
diff --git a/apps/color-buddy/src/lib/api-calls.ts b/apps/color-buddy/src/lib/api-calls.ts
index a659557..6452735 100644
--- a/apps/color-buddy/src/lib/api-calls.ts
+++ b/apps/color-buddy/src/lib/api-calls.ts
@@ -4,7 +4,7 @@ import LintWorker from "./lint-worker.worker?worker";
 import { summarizePal } from "./utils";
 import type { WorkerCommand } from "./worker-types";
 
-export type Engine = "openai" | "google" | "anthropic";
+export type Engine = "openai" | "google" | "anthropic" | "none";
 type SimplePal = { background: string; colors: string[] };
 const palToString = (pal: Palette) => ({
   background: pal.background.toHex(),
@@ -85,6 +85,7 @@ const engineToScaffold = {
   openai: openAIScaffold,
   google: googleScaffold,
   anthropic: anthropicScaffold,
+  none: openAIScaffold,
 };
 
 // supports the add color search function
diff --git a/apps/color-buddy/src/linting/LintToolTipContents.svelte b/apps/color-buddy/src/linting/LintToolTipContents.svelte
index 3cd337e..2c72c52 100644
--- a/apps/color-buddy/src/linting/LintToolTipContents.svelte
+++ b/apps/color-buddy/src/linting/LintToolTipContents.svelte
@@ -1,9 +1,9 @@
 <script lang="ts">
   import type { LintResult } from "color-buddy-palette-lint";
   import type { Palette } from "color-buddy-palette";
-  import { suggestLintFix } from "color-buddy-palette-lint";
+  import { linter, suggestLintFix } from "color-buddy-palette-lint";
+
   import { suggestLintAIFix, suggestLintMonteFix } from "../lib/lint-fixer";
-  import { linter } from "color-buddy-palette-lint";
   import Equal from "virtual:icons/fa6-solid/equals";
 
   import colorStore from "../stores/color-store";
@@ -81,8 +81,11 @@
   $: waitingOnFixes = 0;
   function generateFixes() {
     let numAdd = 0;
-    proposeFix("ai", "LLM Suggestion");
-    numAdd += 1;
+    if ($configStore.engine !== "none") {
+      proposeFix("ai", "LLM Suggestion");
+      numAdd += 1;
+    }
+
     if (lintProgram && lintProgram.program.length) {
       proposeFix("monte", "Monte Carlo Suggestion");
       numAdd += 1;