improve useSubmitHook

src/components/Chat/ChatContent/Message/MessageContent.tsx

@@ -12,10 +12,10 @@ import CrossIcon from '@icon/CrossIcon';
 import RefreshIcon from '@icon/RefreshIcon';
 import DownChevronArrow from '@icon/DownChevronArrow';
 
-import { MessageInterface } from '@type/chat';
-
 import useSubmit from '@hooks/useSubmit';
 
+import { MessageInterface } from '@type/chat';
+
 const MessageContent = ({
   role,
   content,
@@ -91,6 +91,13 @@ const ContentView = ({
     setMessages(updatedMessages);
   };
 
+  const handleRefresh = () => {
+    const updatedMessages = JSON.parse(JSON.stringify(messages));
+    updatedMessages.splice(updatedMessages.length - 1, 1);
+    setMessages(updatedMessages);
+    handleSubmit();
+  };
+
   return (
     <>
       <div className='markdown prose w-full break-words dark:prose-invert dark'>
@@ -161,11 +168,7 @@ const ContentView = ({
         {isDelete || (
           <>
             {role === 'assistant' && messageIndex === messages?.length - 1 && (
-              <RefreshButton
+              <RefreshButton onClick={handleRefresh} />
-                onClick={() => {
-                  handleSubmit(true);
-                }}
-              />
             )}
             {messageIndex !== 0 && (
               <UpButton onClick={() => handleMove('up')} />
@@ -289,6 +292,34 @@ const EditView = ({
     }
   };
 
+  const handleSave = () => {
+    if (_content === '') return;
+    const updatedMessages: MessageInterface[] = JSON.parse(
+      JSON.stringify(messages)
+    );
+    if (sticky) {
+      updatedMessages.push({ role: inputRole, content: _content });
+      _setContent('');
+    } else {
+      updatedMessages[messageIndex].content = _content;
+      setIsEdit(false);
+    }
+    setMessages(updatedMessages);
+  };
+
+  // only for sticky forms
+  const { handleSubmit } = useSubmit();
+  const handleSaveAndSubmit = () => {
+    if (_content == '') return;
+    const updatedMessages: MessageInterface[] = JSON.parse(
+      JSON.stringify(messages)
+    );
+    updatedMessages.push({ role: inputRole, content: _content });
+    _setContent('');
+    setMessages(updatedMessages);
+    handleSubmit();
+  };
+
   useEffect(() => {
     if (textareaRef.current) {
       textareaRef.current.style.height = 'auto';
@@ -308,28 +339,29 @@ const EditView = ({
         }}
         value={_content}
         onInput={handleInput}
-        rows={sticky ? 10 : 1}
+        rows={1}
       ></textarea>
       <div className='text-center mt-2 flex justify-center'>
+        {sticky && (
           <button
             className='btn relative btn-primary mr-2'
-          onClick={() => {
+            onClick={handleSaveAndSubmit}
-            if (_content === '') return;
+          >
-            const updatedMessages: MessageInterface[] = JSON.parse(
+            <div className='flex items-center justify-center gap-2'>
-              JSON.stringify(messages)
+              Save and Submit
-            );
+            </div>
-            if (sticky) {
+          </button>
-              updatedMessages.push({ role: inputRole, content: _content });
+        )}
-              _setContent('');
+
-            } else {
+        <button
-              updatedMessages[messageIndex].content = _content;
+          className={`btn relative mr-2 ${
-              setIsEdit(false);
+            sticky ? 'btn-neutral' : 'btn-primary'
-            }
+          }`}
-            setMessages(updatedMessages);
+          onClick={handleSave}
-          }}
         >
           <div className='flex items-center justify-center gap-2'>Save</div>
         </button>
+
         {sticky || (
           <button
             className='btn relative btn-neutral'

src/hooks/useSubmit.ts

@@ -7,48 +7,35 @@ import { parseEventSource } from '@api/helper';
 
 const useSubmit = () => {
   const [error, setError] = useState<string>('');
-  const [messages, apiFree, apiKey, setMessages, setGenerating, generating] =
+  const [apiFree, apiKey, setMessages, setGenerating, generating] = useStore(
-    useStore((state) => [
+    (state) => [
-      state.messages,
       state.apiFree,
       state.apiKey,
       state.setMessages,
       state.setGenerating,
       state.generating,
-    ]);
+    ]
+  );
 
-  const handleSubmit = async (refresh?: boolean) => {
+  const handleSubmit = async () => {
     if (generating) return;
+    const messages = useStore.getState().messages;
 
     const updatedMessages: MessageInterface[] = JSON.parse(
       JSON.stringify(messages)
     );
-    if (refresh) {
+
-      updatedMessages[updatedMessages.length - 1] = {
-        role: 'assistant',
-        content: '',
-      };
-    } else {
     updatedMessages.push({ role: 'assistant', content: '' });
-    }
+
     setMessages(updatedMessages);
     setGenerating(true);
     let stream;
 
     try {
       if (apiFree) {
-        if (refresh)
+        stream = await getChatCompletionStreamFree(messages);
-          stream = await getChatCompletionStreamFree(
-            updatedMessages.slice(0, updatedMessages.length - 1)
-          );
-        else stream = await getChatCompletionStreamFree(messages);
       } else if (apiKey) {
-        if (refresh)
+        stream = await getChatCompletionStreamFree(messages);
-          stream = await getChatCompletionStreamCustom(
-            apiKey,
-            updatedMessages.slice(0, updatedMessages.length - 1)
-          );
-        else stream = await getChatCompletionStreamFree(messages);
       }
 
       if (stream) {