From 8e30ad41fe43a981bdb658b6f43d8da25ccb0927 Mon Sep 17 00:00:00 2001
From: Lukas Schaefer <lukas@lschaefer.xyz>
Date: Tue, 30 Jun 2026 17:37:22 -0400
Subject: [PATCH] Add improve button to text outputs to allow for iteration

Signed-off-by: Lukas Schaefer <lukas@lschaefer.xyz>
---
 .../AssistantTextProcessingForm.vue           | 33 +++++++++
 src/components/fields/TextInput.vue           | 67 ++++++++++++-------
 2 files changed, 74 insertions(+), 26 deletions(-)

diff --git a/src/components/AssistantTextProcessingForm.vue b/src/components/AssistantTextProcessingForm.vue
index c3006e8b..9103550e 100644
--- a/src/components/AssistantTextProcessingForm.vue
+++ b/src/components/AssistantTextProcessingForm.vue
@@ -215,6 +215,7 @@ import { saveLastSelectedTaskType } from '../assistant.js'
 
 const TEXT2TEXT_TASK_TYPE_ID = 'core:text2text'
 const CHAT_TASK_TYPE_ID = 'chatty-llm'
+const TEXT2TEXT_IMPROVE_TASK_TYPE_ID = 'core:text2text:improve'
 
 export default {
 	name: 'AssistantTextProcessingForm',
@@ -253,6 +254,24 @@ export default {
 		return {
 			providedCurrentTaskId: () => this.selectedTaskId,
 			streaming: () => this.streaming,
+			improveOutput: (content) => {
+				if (!this.showImproveWithNewInstructionsButton) {
+					return
+				}
+				this.$emit('new-task')
+				this.mySelectedTaskTypeId = TEXT2TEXT_IMPROVE_TASK_TYPE_ID
+				this.$nextTick(() => {
+					this.$refs?.assistantFormInputs?.setDefaultValues(true)
+					this.$nextTick(() => {
+						this.myInputs = {
+							...this.myInputs,
+							input: content,
+						}
+						saveLastSelectedTaskType(this.mySelectedTaskTypeId)
+					})
+				})
+			},
+			canImproveOutput: () => this.showImproveWithNewInstructionsButton,
 		}
 	},
 	props: {
@@ -460,6 +479,20 @@ export default {
 		actionButtonsToShow() {
 			return this.hasOutput ? this.actionButtons : []
 		},
+		contextWriteTaskType() {
+			const taskType = this.taskTypes.find(tt => tt.id === TEXT2TEXT_IMPROVE_TASK_TYPE_ID)
+			if (taskType === undefined) {
+				return null
+			}
+			if (this.taskTypeIdList !== null && !this.taskTypeIdList.includes(TEXT2TEXT_IMPROVE_TASK_TYPE_ID)) {
+				return null
+			}
+			return taskType
+		},
+		showImproveWithNewInstructionsButton() {
+			return this.hasOutput
+				&& this.contextWriteTaskType !== null
+		},
 		showRunningEmptyContent() {
 			return this.showSyncTaskRunning && this.myOutputs === null
 		},
diff --git a/src/components/fields/TextInput.vue b/src/components/fields/TextInput.vue
index 8372ce86..88ae7c52 100644
--- a/src/components/fields/TextInput.vue
+++ b/src/components/fields/TextInput.vue
@@ -22,23 +22,33 @@
 			:title="title"
 			@submit="hasValue && $emit('submit', $event)"
 			@update:model-value="$emit('update:value', $event)" />
-		<NcButton v-if="isOutput && hasValue"
-			class="copy-button"
-			variant="secondary"
-			:title="t('assistant', 'Copy output')"
-			@click="onCopy">
-			<template #icon>
-				<NcLoadingIcon v-if="streaming()" />
-				<ClipboardCheckOutlineIcon v-else-if="copied" />
-				<ContentCopyIcon v-else />
-			</template>
-			<span v-if="streaming()">
-				{{ t('assistant', 'Getting results...') }}
-			</span>
-			<span v-else>
-				{{ t('assistant', 'Copy') }}
-			</span>
-		</NcButton>
+		<div v-if="isOutput && hasValue"
+			class="output-buttons">
+			<NcButton v-if="!streaming() && canImproveOutput()"
+				class="improve-button"
+				variant="secondary"
+				:title="t('assistant', 'Improve with new instructions')"
+				@click="improveOutput(formattedValue)">
+				{{ t('assistant', 'Improve') }}
+			</NcButton>
+			<NcButton
+				class="copy-button"
+				variant="secondary"
+				:title="t('assistant', 'Copy output')"
+				@click="onCopy">
+				<template #icon>
+					<NcLoadingIcon v-if="streaming()" />
+					<ClipboardCheckOutlineIcon v-else-if="copied" />
+					<ContentCopyIcon v-else />
+				</template>
+				<span v-if="streaming()">
+					{{ t('assistant', 'Getting results...') }}
+				</span>
+				<span v-else>
+					{{ t('assistant', 'Copy') }}
+				</span>
+			</NcButton>
+		</div>
 		<NcButton v-if="!isOutput && !hasValue && showChooseButton"
 			class="choose-file-button"
 			variant="secondary"
@@ -95,9 +105,7 @@ export default {
 		isMobile,
 	],
 
-	inject: [
-		'streaming',
-	],
+	inject: ['streaming', 'improveOutput', 'canImproveOutput'],
 
 	props: {
 		id: {
@@ -230,21 +238,28 @@ body[dir="rtl"] .choose-file-button {
 	right: unset;
 }
 
+body[dir="rtl"] .output-buttons {
+	left: 4px;
+	right: unset;
+}
+
 .text-input {
 	position: relative;
 
-	.copy-button,
+	.output-buttons,
 	.choose-file-button {
 		position: absolute !important;
 	}
 
-	.choose-file-button {
-		bottom: 2px;
-	}
-
-	.copy-button {
+	.output-buttons {
 		bottom: 4px;
 		right: 4px;
+		display: flex;
+		gap: 4px;
+	}
+
+	.choose-file-button {
+		bottom: 2px;
 	}
 
 	.rich-contenteditable__input {