Making the initialMessage 'sticky' by saving it in local storage and clearing after the first submit.

Author: Brian Genisio

client/src/components/Chat/ChatView.tsx

@@ -39,7 +39,7 @@ function ChatView({ index = 0 }: { index?: number }) {
   const {presetsQuery, onSelectPreset} = usePresets();
 
   const fileMap = useFileMapContext();
-  const initialMessage = searchParams.get('initialMessage');
+  const initialMessageFromUrl = searchParams.get('initialMessage');
   const usePresetId = searchParams.get('usePresetId');
 
   useEffect(() => {
@@ -50,7 +50,13 @@ function ChatView({ index = 0 }: { index?: number }) {
         searchParams.delete('usePresetId');
       }
     }
-  }, [usePresetId, presetsQuery.data, onSelectPreset])
+  }, [usePresetId, presetsQuery.data, onSelectPreset]);
+
+  useEffect(() => {
+    if (initialMessageFromUrl) {
+      localStorage.setItem('initialMessage', initialMessageFromUrl);
+    }
+  }, [initialMessageFromUrl]);
 
   const { data: messagesTree = null, isLoading } = useGetMessagesByConvoId(conversationId ?? '', {
     select: useCallback(
@@ -65,6 +71,11 @@ function ChatView({ index = 0 }: { index?: number }) {
 
   const chatHelpers = useChatHelpers(index, conversationId);
   const addedChatHelpers = useAddedResponse({ rootIndex: index });
+  const initialMessage = localStorage.getItem('initialMessage');
+
+  if (initialMessage && chatHelpers.isSubmitting) {
+    localStorage.removeItem('initialMessage');
+  }
 
   useSSE(rootSubmission, chatHelpers, false);
   useSSE(addedSubmission, addedChatHelpers, true);

client/src/components/Chat/Input/ChatForm.tsx

@@ -129,11 +129,9 @@ const ChatForm = memo(({ index = 0, initialMessage = ''  }: { index?: number; in
     isSubmitting: isSubmitting || isSubmittingAdded,
   });
 
-  useEffect(() => {
-    if (initialMessage) {
-      methods.setValue('text', initialMessage);
-    }
-  }, [initialMessage, methods]);
+  if (initialMessage && methods.getValues('text') === '') {
+    methods.setValue('text', initialMessage);
+  }
 
   const { submitMessage, submitPrompt } = useSubmitMessage();