Store Customization > Fetch product images from the Pexels API

src/BlockPatterns.php

@@ -1,11 +1,11 @@
 <?php
 namespace Automattic\WooCommerce\Blocks;
 
+use Automattic\WooCommerce\Blocks\AI\Connection;
+use Automattic\WooCommerce\Blocks\Images\Pexels;
 use Automattic\WooCommerce\Blocks\Domain\Package;
 use Automattic\WooCommerce\Blocks\Patterns\PatternUpdater;
 use Automattic\WooCommerce\Blocks\Patterns\ProductUpdater;
-use Automattic\WooCommerce\Blocks\Verticals\Client;
-use Automattic\WooCommerce\Blocks\Verticals\VerticalsSelector;
 
 /**
  * Registers patterns under the `./patterns/` directory and updates their content.
@@ -53,10 +53,37 @@ public function __construct( Package $package ) {
 
 		add_action( 'init', array( $this, 'register_block_patterns' ) );
 		add_action( 'update_option_woo_ai_describe_store_description', array( $this, 'schedule_on_option_update' ), 10, 2 );
+		add_action( 'update_option_woo_ai_describe_store_description', array( $this, 'update_ai_connection_allowed_option' ), 10, 2 );
 		add_action( 'upgrader_process_complete', array( $this, 'schedule_on_plugin_update' ), 10, 2 );
 		add_action( 'woocommerce_update_patterns_content', array( $this, 'update_patterns_content' ) );
 	}
 
+	/**
+	 * Make sure the 'woocommerce_blocks_allow_ai_connection' option is set to true if the site is connected to AI.
+	 *
+	 * @param string $option The option name.
+	 * @param string $value The option value.
+	 *
+	 * @return bool
+	 */
+	public function update_ai_connection_allowed_option( $option, $value ): bool {
+		$ai_connection = new Connection();
+
+		$site_id = $ai_connection->get_site_id();
+
+		if ( is_wp_error( $site_id ) ) {
+			return update_option( 'woocommerce_blocks_allow_ai_connection', false );
+		}
+
+		$token = $ai_connection->get_jwt_token( $site_id );
+
+		if ( is_wp_error( $token ) ) {
+			return update_option( 'woocommerce_blocks_allow_ai_connection', false );
+		}
+
+		return update_option( 'woocommerce_blocks_allow_ai_connection', true );
+	}
+
 	/**
 	 * Registers the block patterns and categories under `./patterns/`.
 	 */
@@ -262,7 +289,7 @@ public function schedule_patterns_content_update( $business_description ) {
 	 *
 	 * @param string $value The new value saved for the add_option_woo_ai_describe_store_description option.
 	 *
-	 * @return bool|int|string|\WP_Error
+	 * @return bool|string|\WP_Error
 	 */
 	public function update_patterns_content( $value ) {
 		$allow_ai_connection = get_option( 'woocommerce_blocks_allow_ai_connection' );
@@ -274,16 +301,40 @@ public function update_patterns_content( $value ) {
 			);
 		}
 
-		$vertical_id = ( new VerticalsSelector() )->get_vertical_id( $value );
+		$ai_connection = new Connection();
+
+		$site_id = $ai_connection->get_site_id();
+
+		if ( is_wp_error( $site_id ) ) {
+			return $site_id->get_error_message();
+		}
+
+		$token = $ai_connection->get_jwt_token( $site_id );
+
+		if ( is_wp_error( $token ) ) {
+			return $token->get_error_message();
+		}
+
+		$business_description = get_option( 'woo_ai_describe_store_description' );
+
+		$images = ( new Pexels() )->get_images( $ai_connection, $token, $business_description );
 
-		if ( is_wp_error( $vertical_id ) ) {
-			return $vertical_id;
+		if ( is_wp_error( $images ) ) {
+			return $images->get_error_message();
 		}
 
-		$vertical_images      = ( new Client() )->get_vertical_images( $vertical_id );
-		$business_description = get_option( VerticalsSelector::STORE_DESCRIPTION_OPTION_KEY );
+		$populate_patterns = ( new PatternUpdater() )->generate_content( $ai_connection, $token, $images, $business_description );
+
+		if ( is_wp_error( $populate_patterns ) ) {
+			return $populate_patterns->get_error_message();
+		}
+
+		$populate_products = ( new ProductUpdater() )->generate_content( $ai_connection, $token, $images, $business_description );
+
+		if ( is_wp_error( $populate_products ) ) {
+			return $populate_products->get_error_message();
+		}
 
-		( new PatternUpdater() )->generate_content( $vertical_images, $business_description );
-		( new ProductUpdater() )->generate_content( $vertical_images, $business_description );
+		return true;
 	}
 }

src/Images/Pexels.php

@@ -0,0 +1,97 @@
+<?php
+
+namespace Automattic\WooCommerce\Blocks\Images;
+
+use Automattic\WooCommerce\Blocks\AI\Connection;
+
+/**
+ * Pexels API client.
+ *
+ * @internal
+ */
+class Pexels {
+
+	/**
+	 * The Pexels API endpoint.
+	 */
+	const EXTERNAL_MEDIA_PEXELS_ENDPOINT = '/wpcom/v2/external-media/list/pexels';
+
+	/**
+	 * Returns the list of images for the given search criteria.
+	 *
+	 * @param Connection $ai_connection The AI connection.
+	 * @param string     $token The JWT token.
+	 * @param string     $business_description The business description.
+	 *
+	 * @return array|\WP_Error Array of images, or WP_Error if the request failed.
+	 */
+	public function get_images( $ai_connection, $token, $business_description ) {
+		$search_term = $this->define_search_term( $ai_connection, $token, $business_description );
+
+		if ( is_wp_error( $search_term ) ) {
+			return $search_term;
+		}
+
+		return $this->request( $search_term );
+	}
+
+	/**
+	 * Define the search term to be used on Pexels using the AI endpoint.
+	 *
+	 * The search term is a shorter description of the business.
+	 *
+	 * @param Connection $ai_connection The AI connection.
+	 * @param string     $token The JWT token.
+	 * @param string     $business_description The business description.
+	 *
+	 * @return mixed|\WP_Error
+	 */
+	private function define_search_term( $ai_connection, $token, $business_description ) {
+		$prompt = sprintf( 'Based on the description "%s", provide a one-word product description for the store\'s item. Do not include any adjectives or descriptions of the qualities of the product. The returned word should be simple.', $business_description );
+
+		$response = $ai_connection->fetch_ai_response( $token, $prompt );
+
+		if ( is_wp_error( $response ) || ! isset( $response['completion'] ) ) {
+			return new \WP_Error( 'search_term_definition_failed', __( 'The search term definition failed.', 'woo-gutenberg-products-block' ) );
+		}
+
+		return $response['completion'];
+	}
+
+	/**
+	 * Make a request to the Pexels API.
+	 *
+	 * @param string $search_term The search term to use.
+	 * @param int    $per_page The number of images to return.
+	 *
+	 * @return array|\WP_Error The response body, or WP_Error if the request failed.
+	 */
+	private function request( string $search_term, int $per_page = 90 ) {
+		$request = new \WP_REST_Request( 'GET', self::EXTERNAL_MEDIA_PEXELS_ENDPOINT );
+
+		$request->set_param( 'search', esc_html( $search_term ) );
+		$request->set_param( 'number', $per_page );
+
+		$response      = rest_do_request( $request );
+		$response_data = $response->get_data();
+
+		if ( $response->is_error() ) {
+			$error_msg = [
+				'code' => $response->get_status(),
+				'data' => $response_data,
+			];
+
+			return new \WP_Error( 'pexels_api_error', __( 'Request to the Pexels API failed.', 'woo-gutenberg-products-block' ), $error_msg );
+		}
+
+		$response = $response_data['media'] ?? $response_data;
+
+		if ( is_array( $response ) ) {
+			shuffle( $response );
+
+			return $response;
+		}
+
+		return array();
+	}
+}