Title Optimization: improve button behavior on error screen (#39554)

* Properly close the modal when Cancel is clicked, cleaning up the state

* Skip new generation on try again button when there are options already

* Changelog

projects/plugins/jetpack/changelog/update-jetpack-ai-reset-error-on-title-optimization

@@ -0,0 +1,4 @@
+Significance: patch
+Type: other
+
+Title Optimization: improve button behaviors when handling errors.

projects/plugins/jetpack/extensions/plugins/ai-assistant-plugin/components/title-optimization/index.tsx

@@ -177,8 +177,16 @@ export default function TitleOptimization( {
 
 	const handleTryAgain = useCallback( () => {
 		setError( null );
-		handleRequest( true ); // retry the generation
-	}, [ handleRequest ] );
+
+		/**
+		 * Only try to generate again if there are no options available.
+		 * If there are options, show them so the user can choose one
+		 * or ask for new suggestions.
+		 */
+		if ( options.length === 0 ) {
+			handleRequest( true ); // retry the generation
+		}
+	}, [ handleRequest, options ] );
 
 	const handleTitleOptimizationWithKeywords = useCallback( () => {
 		handleRequest();
@@ -204,6 +212,7 @@ export default function TitleOptimization( {
 	);
 
 	const handleClose = useCallback( () => {
+		setError( null );
 		toggleTitleOptimizationModal();
 		setOptimizationKeywords( '' );
 		stopSuggestion();
@@ -275,7 +284,7 @@ export default function TitleOptimization( {
 								</>
 							) }
 							<div className="jetpack-ai-title-optimization__cta">
-								<Button variant="secondary" onClick={ toggleTitleOptimizationModal }>
+								<Button variant="secondary" onClick={ handleClose }>
 									{ __( 'Cancel', 'jetpack' ) }
 								</Button>
 								{ showTryAgainButton && (