Merge pull request #21 from rjmacarthy/fix/disabled-generating

Disable send/chat when generating
twinnydotdev · Jan 4, 2024 · dd0317b · dd0317b
2 parents 8c6a7db + 3c02613
commit dd0317b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/webview/chat.tsx b/src/webview/chat.tsx
@@ -17,6 +17,7 @@ import styles from './index.module.css'
 const global = globalThis as any
 export const Chat = () => {
   const [inputText, setInputText] = useState('')
+  const [isGenerating, setIsGenerating] = useState(false)
   const [loading, setLoading] = useState(false)
   const [messages, setMessages] = useState<Message[]>([])
   const [completion, setCompletion] = useState<Message | null>()
@@ -70,6 +71,7 @@ export const Chat = () => {
       const message: PostMessage = event.data
       switch (message.type) {
         case 'onCompletion': {
+          setIsGenerating(true)
           setLoading(false)
           setCompletion({
             role: BOT_NAME,
@@ -95,6 +97,7 @@ export const Chat = () => {
             ]
           })
           setCompletion(null)
+          setIsGenerating(false)
         }
       }
     })
@@ -130,7 +133,8 @@ export const Chat = () => {
           <Selection onSelect={scrollBottom} />
           <div className={styles.chatbox}>
             <VSCodeTextArea
-              placeholder='Message twinny'
+              disabled={isGenerating}
+              placeholder="Message twinny"
               rows={5}
               value={inputText}
               onChange={(e) => {
@@ -141,7 +145,11 @@ export const Chat = () => {
             />
           </div>
           <div className={styles.send}>
-            <VSCodeButton type="submit" appearance="primary">
+            <VSCodeButton
+              disabled={isGenerating}
+              type="submit"
+              appearance="primary"
+            >
               Send message
             </VSCodeButton>
           </div>