chore: move code blocks together and leave comment

Author: isekovanic

src/components/MessageInput/MessageInputFlat.tsx

@@ -71,10 +71,6 @@ export const MessageInputFlat = <
     StopAIGenerationButton: StopAIGenerationButtonOverride,
     EmojiPicker,
   } = useComponentContext<StreamChatGenerics>('MessageInputFlat');
-  const StopAIGenerationButton =
-    StopAIGenerationButtonOverride === undefined
-      ? DefaultStopAIGenerationButton
-      : StopAIGenerationButtonOverride;
   const {
     acceptedFiles = [],
     multipleUploads,
@@ -146,6 +142,15 @@ export const MessageInputFlat = <
   const recordingEnabled = !!(recordingController.recorder && navigator.mediaDevices); // account for requirement on iOS as per this bug report: https://bugs.webkit.org/show_bug.cgi?id=252303
   const isRecording = !!recordingController.recordingState;
 
+  /* This bit here is needed to make sure that we can get rid of the default behaviour
+   * if need be. Essentially this allows us to pass StopAIGenerationButton={null} and
+   * completely circumvent the default logic if it's not what we want. We need it as a
+   * prop because there is no other trivial way to override the SendMessage button otherwise.
+   */
+  const StopAIGenerationButton =
+    StopAIGenerationButtonOverride === undefined
+      ? DefaultStopAIGenerationButton
+      : StopAIGenerationButtonOverride;
   const shouldDisplayStopAIGeneration =
     [AIStates.Thinking, AIStates.Generating].includes(aiState) && !!StopAIGenerationButton;