feat(gui): highlight tokens that have been used in prompt

ssube · Nov 12, 2023 · fd34b4a · fd34b4a
1 parent a4bf4ac
commit fd34b4a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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'