From 469359807d197d71bba84c621232dca6601dc181 Mon Sep 17 00:00:00 2001
From: Pabloader <pabloader@pabloader.ru>
Date: Fri, 1 Nov 2024 21:27:20 +0000
Subject: [PATCH] AIStory: add blocking ui on generation

---
 src/games/ai/assets/style.css                 |  6 ++++
 src/games/ai/components/input.tsx             |  4 ++-
 src/games/ai/components/minichat/minichat.tsx | 28 +++++++++++++------
 src/games/ai/contexts/llm.tsx                 |  4 +--
 4 files changed, 31 insertions(+), 11 deletions(-)

diff --git a/src/games/ai/assets/style.css b/src/games/ai/assets/style.css
index 890c9ff..da4c5be 100644
--- a/src/games/ai/assets/style.css
+++ b/src/games/ai/assets/style.css
@@ -53,6 +53,12 @@ button {
     background-color: var(--backgroundColor);
     color: var(--color);
     cursor: pointer;
+    user-select: none;
+
+    &.disabled {
+        pointer-events: none;
+        opacity: 0.5;
+    }
 }
 
 body {
diff --git a/src/games/ai/components/input.tsx b/src/games/ai/components/input.tsx
index 9094c38..5ccbfb6 100644
--- a/src/games/ai/components/input.tsx
+++ b/src/games/ai/components/input.tsx
@@ -7,6 +7,8 @@ export const Input = () => {
     const { input, setInput, addMessage, continueMessage } = useContext(StateContext);
     const { generating } = useContext(LLMContext);
 
+    console.log({generating});
+
     const handleChange = useCallback((e: Event) => {
         if (e.target instanceof HTMLTextAreaElement) {
             setInput(e.target.value);
@@ -36,7 +38,7 @@ export const Input = () => {
     return (
         <div class="chat-input">
             <textarea onInput={handleChange} onKeyDown={handleKeyDown} value={input} />
-            <button onClick={handleSend} disabled={generating}>{input ? 'Send' : 'Continue'}</button>
+            <button onClick={handleSend} class={`${generating ? 'disabled': ''}`}>{input ? 'Send' : 'Continue'}</button>
         </div>
     );
 }
\ No newline at end of file
diff --git a/src/games/ai/components/minichat/minichat.tsx b/src/games/ai/components/minichat/minichat.tsx
index 5acefb0..da8680d 100644
--- a/src/games/ai/components/minichat/minichat.tsx
+++ b/src/games/ai/components/minichat/minichat.tsx
@@ -27,17 +27,22 @@ export const MiniChat = ({ history = [], buttons = {}, open, onClose }: IProps)
         ref.current?.querySelectorAll('textarea').forEach(height ? DOMTools.fixHeight : DOMTools.scrollDown);
     }, []);
 
+    const handleInit = useCallback((force = false) => {
+        if (force || confirm('Clear chat?')) {
+            setMessages([MessageTools.create('', 'user', true)]);
+        }
+    }, []);
+
     useEffect(() => {
         fitTextareas();
     }, [messages]);
 
     useEffect(() => {
+        if (messages.length === 0) {
+            handleInit(true);
+        }
         DOMTools.scrollDown(ref.current);
-    }, [messages.length]);
-
-    useEffect(() => {
-        setMessages([MessageTools.create('', 'user', true)]);
-    }, [open])
+    }, [messages.length, handleInit]);
 
     const handleGenerate = useCallback(async () => {
         if (messages.length > 0 && !generating) {
@@ -97,11 +102,18 @@ export const MiniChat = ({ history = [], buttons = {}, open, onClose }: IProps)
                 </div>
             </div>
             <div class={styles.buttons}>
-                <button onClick={handleGenerate} disabled={generating}>
+                <button onClick={handleGenerate} class={`${generating ? 'disabled' : ''}`}>
                     Generate
                 </button>
-                {answer && Object.entries(buttons).map(([label, onClick], i) => (
-                    <button key={i} onClick={() => onClick(answer)}>
+                <button onClick={() => handleInit()} class={`${generating ? 'disabled' : ''}`}>
+                    Clear
+                </button>
+                {Object.entries(buttons).map(([label, onClick], i) => (
+                    <button
+                        key={i}
+                        onClick={() => onClick(answer ?? '')}
+                        class={`${(generating || !answer) ? 'disabled' : ''}`}
+                    >
                         {label}
                     </button>
                 ))}
diff --git a/src/games/ai/contexts/llm.tsx b/src/games/ai/contexts/llm.tsx
index ee59373..9a7cd46 100644
--- a/src/games/ai/contexts/llm.tsx
+++ b/src/games/ai/contexts/llm.tsx
@@ -129,7 +129,7 @@ export const LLMContextProvider = ({ children }: { children?: any }) => {
     }), [connectionUrl]);
 
     useEffect(() => void (async () => {
-        if (triggerNext && !generating) {
+        if (triggerNext && !generating.value) {
             setTriggerNext(false);
 
             let messageId = messages.length - 1;
@@ -152,7 +152,7 @@ export const LLMContextProvider = ({ children }: { children?: any }) => {
 
             MessageTools.playReady();
         }
-    })(), [triggerNext, messages, generating]);
+    })(), [triggerNext, messages, generating.value]);
 
     const rawContext: IContext = {
        generating: generating.value,