diff --git a/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue b/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue
index 1da4109da..1e910624d 100644
--- a/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue
+++ b/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue
@@ -53,17 +53,29 @@
             </option>
           </select>
         </div>
+        <div v-if="uiFlags.isContentGenerated">
+          <label>
+            {{ $t('INTEGRATION_SETTINGS.OPEN_AI.GENERATE_TITLE') }}
+          </label>
+          <p>
+            {{ generatedMessage }}
+          </p>
+        </div>
         <div class="modal-footer flex-container align-right">
           <woot-button variant="clear" size="small" @click="closeDropdown">
             {{ $t('INTEGRATION_SETTINGS.OPEN_AI.BUTTONS.CANCEL') }}
           </woot-button>
           <woot-button
+            v-if="!uiFlags.isContentGenerated"
             :is-loading="uiFlags.rephrase"
             size="small"
             @click="processEvent('rephrase')"
           >
             {{ buttonText }}
           </woot-button>
+          <woot-button v-else size="small" @click="replaceText('rephrase')">
+            {{ $t('INTEGRATION_SETTINGS.OPEN_AI.BUTTONS.REPLACE') }}
+          </woot-button>
         </div>
       </div>
     </div>
@@ -98,7 +110,9 @@ export default {
         rephrase: false,
         reply_suggestion: false,
         summarize: false,
+        isisContentGenerated: false,
       },
+      generatedMessage: '',
       showDropdown: false,
       activeTone: 'professional',
       tones: [
@@ -140,6 +154,8 @@ export default {
   },
   methods: {
     toggleDropdown() {
+      this.uiFlags.isContentGenerated = false;
+      this.generatedMessage = '';
       this.showDropdown = !this.showDropdown;
     },
     closeDropdown() {
@@ -154,6 +170,11 @@ export default {
         });
       }
     },
+    replaceText(type) {
+      this.$emit('replace-text', this.generatedMessage || this.message);
+      this.recordAnalytics({ type, tone: this.activeTone });
+      this.closeDropdown();
+    },
     async processEvent(type = 'rephrase') {
       this.uiFlags[type] = true;
       try {
@@ -167,9 +188,12 @@ export default {
         const {
           data: { message: generatedMessage },
         } = result;
-        this.$emit('replace-text', generatedMessage || this.message);
-        this.closeDropdown();
-        this.recordAnalytics({ type, tone: this.activeTone });
+        this.uiFlags.isContentGenerated = true;
+        this.generatedMessage = generatedMessage;
+        // It the type is not rephrase, replace the text
+        if (type !== 'rephrase') {
+          this.replaceText(type);
+        }
       } catch (error) {
         this.showAlert(this.$t('INTEGRATION_SETTINGS.OPEN_AI.GENERATE_ERROR'));
       } finally {
diff --git a/app/javascript/dashboard/i18n/locale/en/integrations.json b/app/javascript/dashboard/i18n/locale/en/integrations.json
index 3821fe6e8..a4e528045 100644
--- a/app/javascript/dashboard/i18n/locale/en/integrations.json
+++ b/app/javascript/dashboard/i18n/locale/en/integrations.json
@@ -97,8 +97,10 @@
       "BUTTONS": {
         "GENERATE": "Generate",
         "GENERATING": "Generating...",
+        "REPLACE": "Replace",
         "CANCEL": "Cancel"
       },
+      "GENERATE_TITLE": "Generated Text",
       "GENERATE_ERROR": "There was an error processing the content, please try again"
     },
     "DELETE": {