diff --git a/src/client/pages/InventorsAssistant/FourthStep.tsx b/src/client/pages/InventorsAssistant/FourthStep.tsx
index 4cade574..d078dfa9 100644
--- a/src/client/pages/InventorsAssistant/FourthStep.tsx
+++ b/src/client/pages/InventorsAssistant/FourthStep.tsx
@@ -9,12 +9,14 @@ import LlmResponse from './LlmResponse'
 type FourthStepProps = {
   setAiResponse3: Dispatch<SetStateAction<string>>
   aiResponse: string
+  aiResponseReady: boolean
   setEditModeGlobal: Dispatch<SetStateAction<boolean>>
 }
 
 const FourthStep = ({
   setAiResponse3,
   aiResponse,
+  aiResponseReady,
   setEditModeGlobal,
 }: FourthStepProps) => (
   <>
@@ -45,6 +47,7 @@ const FourthStep = ({
 
     <LlmResponse
       aiResponse={aiResponse}
+      aiResponseReady={aiResponseReady}
       setEditedResponse={setAiResponse3}
       setEditModeGlobal={setEditModeGlobal}
     />
diff --git a/src/client/pages/InventorsAssistant/InventionReport.tsx b/src/client/pages/InventorsAssistant/InventionReport.tsx
index 09d0e119..b975a40a 100644
--- a/src/client/pages/InventorsAssistant/InventionReport.tsx
+++ b/src/client/pages/InventorsAssistant/InventionReport.tsx
@@ -9,12 +9,14 @@ import LlmResponse from './LlmResponse'
 
 type InventionReportProps = {
   aiResponse: string
+  aiResponseReady: boolean
   setAiResponse: Dispatch<SetStateAction<string>>
   headingLevel: 'h2' | 'h3'
 }
 
 const InventionReport = ({
   aiResponse,
+  aiResponseReady,
   setAiResponse,
   headingLevel,
 }: InventionReportProps) => {
@@ -26,7 +28,11 @@ const InventionReport = ({
         {t('inventorsAssistant:finalStepSummary')}
       </SectionHeading>
 
-      <LlmResponse aiResponse={aiResponse} setEditedResponse={setAiResponse} />
+      <LlmResponse
+        aiResponse={aiResponse}
+        aiResponseReady={aiResponseReady}
+        setEditedResponse={setAiResponse}
+      />
     </Box>
   )
 }
diff --git a/src/client/pages/InventorsAssistant/InventorPhase2.tsx b/src/client/pages/InventorsAssistant/InventorPhase2.tsx
index 1f5c5d99..a989f7d7 100644
--- a/src/client/pages/InventorsAssistant/InventorPhase2.tsx
+++ b/src/client/pages/InventorsAssistant/InventorPhase2.tsx
@@ -82,6 +82,7 @@ const InventorPhase2 = () => {
           <SecondStep
             setAiResponse1={setAiResponse1}
             aiResponse={aiResponse1}
+            aiResponseReady={aiResponse1Ready}
             setEditModeGlobal={setEditModeGlobal}
           />
         )}
@@ -112,6 +113,7 @@ const InventorPhase2 = () => {
           <ThirdStep
             setAiResponse2={setAiResponse2}
             aiResponse={aiResponse2}
+            aiResponseReady={aiResponse2Ready}
             setEditModeGlobal={setEditModeGlobal}
           />
         )}
@@ -142,6 +144,7 @@ const InventorPhase2 = () => {
           <FourthStep
             setAiResponse3={setAiResponse3}
             aiResponse={aiResponse3}
+            aiResponseReady={aiResponse3Ready}
             setEditModeGlobal={setEditModeGlobal}
           />
         )}
diff --git a/src/client/pages/InventorsAssistant/InventorPhase3.tsx b/src/client/pages/InventorsAssistant/InventorPhase3.tsx
index 081c3749..595926cb 100644
--- a/src/client/pages/InventorsAssistant/InventorPhase3.tsx
+++ b/src/client/pages/InventorsAssistant/InventorPhase3.tsx
@@ -41,6 +41,7 @@ const InventorPhase3 = () => {
         {currentStep > 7 && (
           <FinalStep
             aiResponse={aiResponse4}
+            aiResponseReady={aiResponse4Ready}
             originalIdea={inventiveMessage}
             ideaRefinement={aiResponse1}
             industrialApplicability={aiResponse2}
diff --git a/src/client/pages/InventorsAssistant/LlmResponse.tsx b/src/client/pages/InventorsAssistant/LlmResponse.tsx
index f2869fd6..80f26926 100644
--- a/src/client/pages/InventorsAssistant/LlmResponse.tsx
+++ b/src/client/pages/InventorsAssistant/LlmResponse.tsx
@@ -5,12 +5,14 @@ import Markdown from '../../components/Common/Markdown'
 
 type LlmResponseProps = {
   aiResponse: string
+  aiResponseReady: boolean
   setEditedResponse: Dispatch<SetStateAction<string>>
   setEditModeGlobal?: Dispatch<SetStateAction<boolean>>
 }
 
 const LlmResponse = ({
   aiResponse,
+  aiResponseReady,
   setEditedResponse,
   setEditModeGlobal,
 }: LlmResponseProps) => {
@@ -73,7 +75,7 @@ const LlmResponse = ({
           <Alert severity='success' sx={{ py: 4, px: 2, width: '100%' }}>
             <Markdown>{aiResponse}</Markdown>
           </Alert>
-          {aiResponse && aiResponse.length > 0 && (
+          {aiResponse && aiResponseReady && (
             <Button
               data-cy='edit-ai-response-button'
               variant='contained'
diff --git a/src/client/pages/InventorsAssistant/SecondStep.tsx b/src/client/pages/InventorsAssistant/SecondStep.tsx
index 8263b95d..4fbf96ab 100644
--- a/src/client/pages/InventorsAssistant/SecondStep.tsx
+++ b/src/client/pages/InventorsAssistant/SecondStep.tsx
@@ -9,12 +9,14 @@ import LlmResponse from './LlmResponse'
 type SecondStepProps = {
   setAiResponse1: Dispatch<SetStateAction<string>>
   aiResponse: string
+  aiResponseReady: boolean
   setEditModeGlobal: Dispatch<SetStateAction<boolean>>
 }
 
 const SecondStep = ({
   setAiResponse1,
   aiResponse,
+  aiResponseReady,
   setEditModeGlobal,
 }: SecondStepProps) => (
   <>
@@ -32,6 +34,7 @@ const SecondStep = ({
 
     <LlmResponse
       aiResponse={aiResponse}
+      aiResponseReady={aiResponseReady}
       setEditedResponse={setAiResponse1}
       setEditModeGlobal={setEditModeGlobal}
     />
diff --git a/src/client/pages/InventorsAssistant/ThirdStep.tsx b/src/client/pages/InventorsAssistant/ThirdStep.tsx
index 5c633a44..4a07f496 100644
--- a/src/client/pages/InventorsAssistant/ThirdStep.tsx
+++ b/src/client/pages/InventorsAssistant/ThirdStep.tsx
@@ -9,12 +9,14 @@ import LlmResponse from './LlmResponse'
 type ThirdStepProps = {
   setAiResponse2: React.Dispatch<React.SetStateAction<string>>
   aiResponse: string
+  aiResponseReady: boolean
   setEditModeGlobal: Dispatch<SetStateAction<boolean>>
 }
 
 const ThirdStep: React.FC<ThirdStepProps> = ({
   setAiResponse2,
   aiResponse,
+  aiResponseReady,
   setEditModeGlobal,
 }) => (
   <>
@@ -30,6 +32,7 @@ const ThirdStep: React.FC<ThirdStepProps> = ({
     <Box component='section' sx={{ mt: 4 }}>
       <LlmResponse
         aiResponse={aiResponse}
+        aiResponseReady={aiResponseReady}
         setEditedResponse={setAiResponse2}
         setEditModeGlobal={setEditModeGlobal}
       />