From fd34b4a76418a7034d620756d0811963411bc30a Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 12 Nov 2023 15:50:04 -0600
Subject: [PATCH] feat(gui): highlight tokens that have been used in prompt

---
 gui/src/components/input/PromptInput.tsx | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/gui/src/components/input/PromptInput.tsx b/gui/src/components/input/PromptInput.tsx
index fa6a7aba..6fbc7f09 100644
--- a/gui/src/components/input/PromptInput.tsx
+++ b/gui/src/components/input/PromptInput.tsx
@@ -79,7 +79,13 @@ export function PromptInput(props: PromptInputProps) {
         });
       }}
     />
-    {tokens.map(([token, _weight]) => <Chip label={token} onClick={() => addToken(token)} />)}
+    <Stack direction='row' spacing={2}>
+      {tokens.map(([token, _weight]) => <Chip
+        color={prompt.includes(token) ? 'primary' : 'default'}
+        label={token}
+        onClick={() => addToken(token)}
+      />)}
+    </Stack>
     <TextField
       label={t('parameter.negativePrompt')}
       variant='outlined'