AI Assistant: Add option to generate title (#30646)

* add option to generate a post title * changelog * renaming lastPromptType attribute * fix text case * replace Done with Accept title
Automattic · May 11, 2023 · 8924786 · 8924786
1 parent 3c7910a
commit 8924786
Show file tree

Hide file tree

Showing 6 changed files with 51 additions and 5 deletions.
diff --git a/projects/plugins/jetpack/changelog/update-ai-assistant-generate-title b/projects/plugins/jetpack/changelog/update-ai-assistant-generate-title
@@ -0,0 +1,5 @@
+Significance: patch
+Type: other
+Comment: Unreleased block
+
+
diff --git a/projects/plugins/jetpack/extensions/blocks/ai-assistant/ai-control.js b/projects/plugins/jetpack/extensions/blocks/ai-assistant/ai-control.js
@@ -20,6 +20,7 @@ const AIControl = ( {
 	getSuggestionFromOpenAI,
 	retryRequest,
 	handleAcceptContent,
+	handleAcceptTitle,
 	handleTryAgain,
 	handleGetSuggestion,
 	isWaitingState,
@@ -32,6 +33,7 @@ const AIControl = ( {
 	postTitle,
 	wholeContent,
 	content,
+	promptType,
 } ) => {
 	const handleInputEnter = event => {
 		if ( event.key === 'Enter' && ! event.shiftKey ) {
@@ -80,13 +82,15 @@ const AIControl = ( {
 					getSuggestionFromOpenAI={ getSuggestionFromOpenAI }
 					retryRequest={ retryRequest }
 					handleAcceptContent={ handleAcceptContent }
+					handleAcceptTitle={ handleAcceptTitle }
 					handleGetSuggestion={ handleGetSuggestion }
 					handleTryAgain={ handleTryAgain }
 					showRetry={ showRetry }
 					toggleAIType={ toggleAIType }
 					contentBefore={ contentBefore }
 					hasPostTitle={ !! postTitle?.length }
 					wholeContent={ wholeContent }
+					promptType={ promptType }
 				/>
 			) }
 			<div className="jetpack-ai-assistant__input-wrapper">
@@ -124,12 +128,14 @@ const ToolbarControls = ( {
 	getSuggestionFromOpenAI,
 	retryRequest,
 	handleAcceptContent,
+	handleAcceptTitle,
 	handleTryAgain,
 	showRetry,
 	toggleAIType,
 	contentBefore,
 	hasPostTitle,
 	wholeContent,
+	promptType,
 } ) => {
 	return (
 		<BlockControls>
@@ -138,11 +144,17 @@ const ToolbarControls = ( {
 				<ToolbarGroup>
 					{ ! showRetry && contentIsLoaded && animationDone && (
 						<>
-							<ToolbarButton onClick={ handleAcceptContent }>
-								{ __( 'Done', 'jetpack' ) }
-							</ToolbarButton>
+							{ promptType === 'generateTitle' ? (
+								<ToolbarButton onClick={ handleAcceptTitle }>
+									{ __( 'Accept title', 'jetpack' ) }
+								</ToolbarButton>
+							) : (
+								<ToolbarButton onClick={ handleAcceptContent }>
+									{ __( 'Done', 'jetpack' ) }
+								</ToolbarButton>
+							) }
 							<ToolbarButton onClick={ handleTryAgain }>
-								{ __( 'Try Again', 'jetpack' ) }
+								{ __( 'Try again', 'jetpack' ) }
 							</ToolbarButton>
 						</>
 					) }
@@ -192,6 +204,11 @@ const ToolbarControls = ( {
 									onClick: () => getSuggestionFromOpenAI( 'simplify' ),
 									isDisabled: ! contentBefore?.length,
 								},
+								{
+									title: __( 'Generate a post title', 'jetpack' ),
+									onClick: () => getSuggestionFromOpenAI( 'generateTitle' ),
+									isDisabled: ! wholeContent?.length,
+								},
 							] }
 						/>
 					) }

diff --git a/projects/plugins/jetpack/extensions/blocks/ai-assistant/attributes.js b/projects/plugins/jetpack/extensions/blocks/ai-assistant/attributes.js
@@ -2,4 +2,7 @@ export default {
 	content: {
 		type: 'string',
 	},
+	promptType: {
+		type: 'string',
+	},
 };
diff --git a/projects/plugins/jetpack/extensions/blocks/ai-assistant/create-prompt.js b/projects/plugins/jetpack/extensions/blocks/ai-assistant/create-prompt.js
@@ -192,6 +192,18 @@ export const createPrompt = (
 		return expandPrompt;
 	}
 
+	if ( type === 'generateTitle' ) {
+		if ( ! content.length ) {
+			return '';
+		}
+
+		return buildPromptTemplate( {
+			request: 'Generate a title for this blog post',
+			rules: [ 'Only output the raw title, without any prefix or quotes' ],
+			content,
+		} );
+	}
+
 	// TODO: add some error handling if user supplied prompts or existing content is too short.
 
 	// We prevent a prompt if everything is empty.

diff --git a/projects/plugins/jetpack/extensions/blocks/ai-assistant/edit.js b/projects/plugins/jetpack/extensions/blocks/ai-assistant/edit.js
@@ -29,7 +29,8 @@ export default function AIAssistantEdit( { attributes, setAttributes, clientId }
 	const { tracks } = useAnalytics();
 	const postId = useSelect( select => select( 'core/editor' ).getCurrentPostId() );
 
-	const { replaceBlocks, replaceBlock } = useDispatch( blockEditorStore );
+	const { replaceBlocks, replaceBlock, removeBlock } = useDispatch( blockEditorStore );
+	const { editPost } = useDispatch( 'core/editor' );
 	const { mediaUpload } = useSelect( select => {
 		const { getSettings } = select( blockEditorStore );
 		const settings = getSettings();
@@ -107,6 +108,11 @@ export default function AIAssistantEdit( { attributes, setAttributes, clientId }
 		replaceBlocks( clientId, rawHandler( { HTML: attributes.content } ) );
 	};
 
+	const handleAcceptTitle = () => {
+		editPost( { title: attributes.content.trim() } );
+		removeBlock( clientId );
+	};
+
 	const handleTryAgain = () => {
 		setAttributes( { content: undefined } );
 	};
@@ -155,6 +161,7 @@ export default function AIAssistantEdit( { attributes, setAttributes, clientId }
 				getSuggestionFromOpenAI={ getSuggestionFromOpenAI }
 				retryRequest={ retryRequest }
 				handleAcceptContent={ handleAcceptContent }
+				handleAcceptTitle={ handleAcceptTitle }
 				handleGetSuggestion={ handleGetSuggestion }
 				handleTryAgain={ handleTryAgain }
 				isWaitingState={ isWaitingState }
@@ -166,6 +173,7 @@ export default function AIAssistantEdit( { attributes, setAttributes, clientId }
 				postTitle={ postTitle }
 				userPrompt={ userPrompt }
 				wholeContent={ wholeContent }
+				promptType={ attributes.promptType }
 			/>
 			{ ! loadingImages && resultImages.length > 0 && (
 				<Flex direction="column" style={ { width: '100%' } }>

diff --git a/projects/plugins/jetpack/extensions/blocks/ai-assistant/use-suggestions-from-openai.js b/projects/plugins/jetpack/extensions/blocks/ai-assistant/use-suggestions-from-openai.js
@@ -177,6 +177,7 @@ const useSuggestionsFromOpenAI = ( {
 
 		if ( ! retryRequest ) {
 			setLastPrompt( prompt );
+			setAttributes( { promptType: type } );
 		}
 
 		apiFetch( {