From 0e1b61075778826b5e590864d977df15ed28cbb4 Mon Sep 17 00:00:00 2001
From: Tahier Hussain <tahier@zipstack.com>
Date: Fri, 17 Apr 2026 01:26:55 +0530
Subject: [PATCH 1/3] fix: hide Stop button once response streaming begins
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

When a prompt is running, the Stop button was shown throughout the entire
process (thought chain + response streaming). Now the Stop button is only
shown during the thought chain phase. Once the response starts streaming,
the Send button is displayed instead.

Changes:
- Add isResponseStreaming computed state in ExistingChat.jsx
- Thread isResponseStreaming prop through InputPrompt, PromptInput
- Update button condition in MonacoPromptInput and DefaultPromptInput

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 frontend/src/ide/chat-ai/DefaultPromptInput.jsx | 4 +++-
 frontend/src/ide/chat-ai/ExistingChat.jsx       | 8 ++++++++
 frontend/src/ide/chat-ai/InputPrompt.jsx        | 3 +++
 frontend/src/ide/chat-ai/MonacoPromptInput.jsx  | 5 +++--
 frontend/src/ide/chat-ai/PromptInput.jsx        | 4 ++++
 5 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/frontend/src/ide/chat-ai/DefaultPromptInput.jsx b/frontend/src/ide/chat-ai/DefaultPromptInput.jsx
index ceefb64..82d106e 100644
--- a/frontend/src/ide/chat-ai/DefaultPromptInput.jsx
+++ b/frontend/src/ide/chat-ai/DefaultPromptInput.jsx
@@ -6,6 +6,7 @@ import { Sender } from "@ant-design/x";
 function DefaultPromptInput({
   value,
   isPromptRunning,
+  isResponseStreaming,
   onSenderChange,
   onSubmit,
   onCancel,
@@ -25,7 +26,7 @@ function DefaultPromptInput({
       actions={(_, info) => {
         const { SendButton, LoadingButton } = info.components;
 
-        if (isPromptRunning) {
+        if (isPromptRunning && !isResponseStreaming) {
           return (
             <Tooltip title="Stop">
               <LoadingButton variant="text" color="secondary" shape="default" />
@@ -64,6 +65,7 @@ function DefaultPromptInput({
 DefaultPromptInput.propTypes = {
   value: PropTypes.string.isRequired,
   isPromptRunning: PropTypes.bool.isRequired,
+  isResponseStreaming: PropTypes.bool,
   onSenderChange: PropTypes.func.isRequired,
   onSubmit: PropTypes.func.isRequired,
   onCancel: PropTypes.func.isRequired,
diff --git a/frontend/src/ide/chat-ai/ExistingChat.jsx b/frontend/src/ide/chat-ai/ExistingChat.jsx
index edc2b51..eafeaf2 100644
--- a/frontend/src/ide/chat-ai/ExistingChat.jsx
+++ b/frontend/src/ide/chat-ai/ExistingChat.jsx
@@ -246,6 +246,13 @@ const ExistingChat = memo(function ExistingChat({
     return -1;
   }, [chatMessages, chatIntents]);
 
+  // Check if response is actively streaming (thought chain done, response started)
+  const isResponseStreaming = useMemo(() => {
+    if (!chatMessages?.length) return false;
+    const lastMessage = chatMessages?.[chatMessages.length - 1];
+    return isPromptRunning && lastMessage?.response?.length > 0;
+  }, [chatMessages, isPromptRunning]);
+
   useEffect(() => {
     if (chatMessages.length) {
       setLastChatMessageId(
@@ -454,6 +461,7 @@ const ExistingChat = memo(function ExistingChat({
         <InputPrompt
           savePrompt={handleSavePrompt}
           isPromptRunning={isPromptRunning}
+          isResponseStreaming={isResponseStreaming}
           chatIntents={chatIntents}
           selectedChatIntent={selectedChatIntent}
           setSelectedChatIntent={setSelectedChatIntent}
diff --git a/frontend/src/ide/chat-ai/InputPrompt.jsx b/frontend/src/ide/chat-ai/InputPrompt.jsx
index d09e489..6951781 100644
--- a/frontend/src/ide/chat-ai/InputPrompt.jsx
+++ b/frontend/src/ide/chat-ai/InputPrompt.jsx
@@ -10,6 +10,7 @@ const InputPrompt = memo(function InputPrompt({
   savePrompt,
   isNewChat = false,
   isPromptRunning,
+  isResponseStreaming = false,
   chatIntents,
   selectedChatIntent,
   setSelectedChatIntent,
@@ -209,6 +210,7 @@ const InputPrompt = memo(function InputPrompt({
           value={value}
           editorHeight={editorHeight}
           isPromptRunning={isPromptRunning}
+          isResponseStreaming={isResponseStreaming}
           onSenderChange={handleSenderChange}
           onSubmit={handleSubmit}
           onEditorMount={handleEditorMount}
@@ -252,6 +254,7 @@ InputPrompt.propTypes = {
   savePrompt: PropTypes.func.isRequired,
   isNewChat: PropTypes.bool,
   isPromptRunning: PropTypes.bool.isRequired,
+  isResponseStreaming: PropTypes.bool,
   chatIntents: PropTypes.array.isRequired,
   selectedChatIntent: PropTypes.string,
   setSelectedChatIntent: PropTypes.func.isRequired,
diff --git a/frontend/src/ide/chat-ai/MonacoPromptInput.jsx b/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
index 0c50e6a..f2f4938 100644
--- a/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
+++ b/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
@@ -12,6 +12,7 @@ const MonacoPromptInput = memo(function MonacoPromptInput({
   value,
   editorHeight,
   isPromptRunning,
+  isResponseStreaming,
   onEditorMount,
   onMonacoChange,
   onSubmit,
@@ -159,9 +160,8 @@ const MonacoPromptInput = memo(function MonacoPromptInput({
   return (
     <div className="monaco-editor-wrapper">
       {editorContent}
-
       <div className="monaco-editor-send">
-        {isPromptRunning ? (
+        {isPromptRunning && !isResponseStreaming ? (
           <Tooltip title="Stop">
             <Button type="text" icon={<CloseOutlined />} onClick={onCancel} />
           </Tooltip>
@@ -204,6 +204,7 @@ MonacoPromptInput.propTypes = {
   value: PropTypes.string.isRequired,
   editorHeight: PropTypes.number.isRequired,
   isPromptRunning: PropTypes.bool.isRequired,
+  isResponseStreaming: PropTypes.bool,
   onEditorMount: PropTypes.func,
   onMonacoChange: PropTypes.func.isRequired,
   onSubmit: PropTypes.func.isRequired,
diff --git a/frontend/src/ide/chat-ai/PromptInput.jsx b/frontend/src/ide/chat-ai/PromptInput.jsx
index ae0802e..4a16903 100644
--- a/frontend/src/ide/chat-ai/PromptInput.jsx
+++ b/frontend/src/ide/chat-ai/PromptInput.jsx
@@ -10,6 +10,7 @@ const PromptInput = memo(function PromptInput({
   value,
   editorHeight,
   isPromptRunning,
+  isResponseStreaming,
   onSenderChange,
   onSubmit,
   onEditorMount,
@@ -34,6 +35,7 @@ const PromptInput = memo(function PromptInput({
       <DefaultPromptInput
         value={value}
         isPromptRunning={isPromptRunning}
+        isResponseStreaming={isResponseStreaming}
         onSenderChange={onSenderChange}
         onSubmit={onSubmit}
         onCancel={onCancel}
@@ -49,6 +51,7 @@ const PromptInput = memo(function PromptInput({
       value={value}
       editorHeight={editorHeight}
       isPromptRunning={isPromptRunning}
+      isResponseStreaming={isResponseStreaming}
       onEditorMount={onEditorMount}
       onMonacoChange={onMonacoChange}
       onSubmit={onSubmit}
@@ -70,6 +73,7 @@ PromptInput.propTypes = {
   value: PropTypes.string.isRequired,
   editorHeight: PropTypes.number.isRequired,
   isPromptRunning: PropTypes.bool.isRequired,
+  isResponseStreaming: PropTypes.bool,
   onSenderChange: PropTypes.func.isRequired,
   onSubmit: PropTypes.func.isRequired,
   onEditorMount: PropTypes.func.isRequired,

From f960bffef9d447dc448397cf73276dddc00f60c0 Mon Sep 17 00:00:00 2001
From: Tahier Hussain <tahier@zipstack.com>
Date: Fri, 17 Apr 2026 01:38:06 +0530
Subject: [PATCH 2/3] fix: address code review feedback
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Add explicit default `= false` for isResponseStreaming in MonacoPromptInput and DefaultPromptInput
- Disable SendButton when isPromptRunning to prevent concurrent prompt submission

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 frontend/src/ide/chat-ai/DefaultPromptInput.jsx | 4 ++--
 frontend/src/ide/chat-ai/MonacoPromptInput.jsx  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/frontend/src/ide/chat-ai/DefaultPromptInput.jsx b/frontend/src/ide/chat-ai/DefaultPromptInput.jsx
index 82d106e..e941935 100644
--- a/frontend/src/ide/chat-ai/DefaultPromptInput.jsx
+++ b/frontend/src/ide/chat-ai/DefaultPromptInput.jsx
@@ -6,7 +6,7 @@ import { Sender } from "@ant-design/x";
 function DefaultPromptInput({
   value,
   isPromptRunning,
-  isResponseStreaming,
+  isResponseStreaming = false,
   onSenderChange,
   onSubmit,
   onCancel,
@@ -52,7 +52,7 @@ function DefaultPromptInput({
                 />
               }
               shape="default"
-              disabled={!value}
+              disabled={!value || isPromptRunning}
               className={shouldHighlightSend ? "onboarding-send-highlight" : ""}
             />
           </Tooltip>
diff --git a/frontend/src/ide/chat-ai/MonacoPromptInput.jsx b/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
index f2f4938..204ac01 100644
--- a/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
+++ b/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
@@ -12,7 +12,7 @@ const MonacoPromptInput = memo(function MonacoPromptInput({
   value,
   editorHeight,
   isPromptRunning,
-  isResponseStreaming,
+  isResponseStreaming = false,
   onEditorMount,
   onMonacoChange,
   onSubmit,

From 08cd3b3c1429c6100ade0e8cb40b878f1d2dcfc1 Mon Sep 17 00:00:00 2001
From: Tahier Hussain <tahier@zipstack.com>
Date: Fri, 17 Apr 2026 01:45:04 +0530
Subject: [PATCH 3/3] fix: prevent concurrent submissions in MonacoPromptInput
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Add isPromptRunning to disabled condition to match DefaultPromptInput behavior.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 frontend/src/ide/chat-ai/MonacoPromptInput.jsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/frontend/src/ide/chat-ai/MonacoPromptInput.jsx b/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
index 204ac01..fa64e62 100644
--- a/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
+++ b/frontend/src/ide/chat-ai/MonacoPromptInput.jsx
@@ -189,7 +189,7 @@ const MonacoPromptInput = memo(function MonacoPromptInput({
                   }
                 />
               }
-              disabled={!value}
+              disabled={!value || isPromptRunning}
               onClick={handleSubmit}
               className={shouldHighlightSend ? "onboarding-send-highlight" : ""}
             />