You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
412 lines
9.8 KiB
PHP
412 lines
9.8 KiB
PHP
<?php
|
|
namespace Elementor\Modules\Ai\Connect;
|
|
|
|
use Elementor\Core\Common\Modules\Connect\Apps\Library;
|
|
|
|
if ( ! defined( 'ABSPATH' ) ) {
|
|
exit; // Exit if accessed directly
|
|
}
|
|
|
|
class Ai extends Library {
|
|
const API_URL = 'https://my.elementor.com/api/v2/ai/';
|
|
|
|
const STYLE_PRESET = 'style_preset';
|
|
const IMAGE_TYPE = 'image_type';
|
|
const IMAGE_STRENGTH = 'image_strength';
|
|
const ASPECT_RATIO = 'ratio';
|
|
const IMAGE_RESOLUTION = 'image_resolution';
|
|
|
|
const PROMPT = 'prompt';
|
|
|
|
public function get_title() {
|
|
return esc_html__( 'AI', 'elementor' );
|
|
}
|
|
|
|
protected function get_api_url() {
|
|
return static::API_URL . '/';
|
|
}
|
|
|
|
public function get_usage() {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'status/check',
|
|
[
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
/**
|
|
* get_file_payload
|
|
* @param $filename
|
|
* @param $file_type
|
|
* @param $file_path
|
|
* @param $boundary
|
|
*
|
|
* @return string
|
|
*/
|
|
private function get_file_payload( $filename, $file_type, $file_path, $boundary ) {
|
|
$name = $filename ?? basename( $file_path );
|
|
$mine_type = 'image' === $file_type ? image_type_to_mime_type( exif_imagetype( $file_path ) ) : $file_type;
|
|
$payload = '';
|
|
// Upload the file
|
|
$payload .= '--' . $boundary;
|
|
$payload .= "\r\n";
|
|
$payload .= 'Content-Disposition: form-data; name="' . esc_attr( $name ) . '"; filename="' . esc_attr( $name ) . '"' . "\r\n";
|
|
$payload .= 'Content-Type: ' . $mine_type . "\r\n";
|
|
$payload .= "\r\n";
|
|
$payload .= file_get_contents( $file_path );
|
|
$payload .= "\r\n";
|
|
|
|
return $payload;
|
|
}
|
|
|
|
private function get_upload_request_body( $body, $file, $boundary, $file_name = '' ) {
|
|
$payload = '';
|
|
// add all body fields as standard POST fields:
|
|
foreach ( $body as $name => $value ) {
|
|
$payload .= '--' . $boundary;
|
|
$payload .= "\r\n";
|
|
$payload .= 'Content-Disposition: form-data; name="' . esc_attr( $name ) . '"' . "\r\n\r\n";
|
|
$payload .= $value;
|
|
$payload .= "\r\n";
|
|
}
|
|
|
|
if ( is_array( $file ) ) {
|
|
foreach ( $file as $key => $file_data ) {
|
|
$payload .= $this->get_file_payload( $file_data['name'], $file_data['type'], $file_data['path'], $boundary );
|
|
}
|
|
} else {
|
|
$image_mime = image_type_to_mime_type( exif_imagetype( $file ) );
|
|
// @todo: add validation for supported image types
|
|
|
|
if ( empty( $file_name ) ) {
|
|
$file_name = basename( $file );
|
|
}
|
|
$payload .= $this->get_file_payload( $file_name, $image_mime, $file, $boundary );
|
|
}
|
|
|
|
$payload .= '--' . $boundary . '--';
|
|
|
|
return $payload;
|
|
}
|
|
|
|
private function ai_request( $method, $endpoint, $body, $file = false, $file_name = '' ) {
|
|
$headers = [
|
|
'x-elementor-ai-version' => '2',
|
|
];
|
|
|
|
if ( $file ) {
|
|
$boundary = wp_generate_password( 24, false );
|
|
$body = $this->get_upload_request_body( $body, $file, $boundary, $file_name );
|
|
// add content type header
|
|
$headers['Content-Type'] = 'multipart/form-data; boundary=' . $boundary;
|
|
}
|
|
|
|
return $this->http_request(
|
|
$method,
|
|
$endpoint,
|
|
[
|
|
'timeout' => 50,
|
|
'headers' => $headers,
|
|
'body' => $body,
|
|
],
|
|
[
|
|
'return_type' => static::HTTP_RETURN_TYPE_ARRAY,
|
|
]
|
|
);
|
|
}
|
|
|
|
public function set_get_started() {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'status/get-started',
|
|
[
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
public function set_status_feedback( $response_id ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'status/feedback/' . $response_id,
|
|
[
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
public function set_used_gallery_image( $image_id ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'status/used-gallery-image/' . $image_id,
|
|
[
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
public function get_completion_text( $prompt ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'text/completion',
|
|
[
|
|
'prompt' => $prompt,
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
/**
|
|
* get_image_prompt_enhanced
|
|
* @param $prompt
|
|
*
|
|
* @return mixed|\WP_Error
|
|
*/
|
|
public function get_image_prompt_enhanced( $prompt ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'text/enhance-image-prompt',
|
|
[
|
|
'prompt' => $prompt,
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
public function get_edit_text( $input, $instruction ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'text/edit',
|
|
[
|
|
'input' => $input,
|
|
'instruction' => $instruction,
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
public function get_custom_code( $prompt, $language ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'text/custom-code',
|
|
[
|
|
'prompt' => $prompt,
|
|
'language' => $language,
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
public function get_custom_css( $prompt, $html_markup, $element_id ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'text/custom-css',
|
|
[
|
|
'prompt' => $prompt,
|
|
'html_markup' => $html_markup,
|
|
'element_id' => $element_id,
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
/**
|
|
* get_text_to_image
|
|
* @param $prompt
|
|
* @param $prompt_settings
|
|
*
|
|
* @return mixed|\WP_Error
|
|
*/
|
|
public function get_text_to_image( $prompt, $prompt_settings ) {
|
|
return $this->ai_request(
|
|
'POST',
|
|
'image/text-to-image',
|
|
[
|
|
self::PROMPT => $prompt,
|
|
self::IMAGE_TYPE => $prompt_settings[ self::IMAGE_TYPE ] . '/' . $prompt_settings[ self::STYLE_PRESET ],
|
|
self::ASPECT_RATIO => $prompt_settings[ self::ASPECT_RATIO ],
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
]
|
|
);
|
|
}
|
|
|
|
/**
|
|
* get_image_to_image
|
|
* @param $image_data
|
|
*
|
|
* @return mixed|\WP_Error
|
|
* @throws \Exception
|
|
*/
|
|
public function get_image_to_image( $image_data ) {
|
|
$image_file = get_attached_file( $image_data['attachment_id'] );
|
|
|
|
if ( ! $image_file ) {
|
|
throw new \Exception( 'Image file not found' );
|
|
}
|
|
|
|
$result = $this->ai_request(
|
|
'POST',
|
|
'image/image-to-image',
|
|
[
|
|
self::PROMPT => $image_data[ self::PROMPT ],
|
|
self::IMAGE_TYPE => $image_data['promptSettings'][ self::IMAGE_TYPE ] . '/' . $image_data['promptSettings'][ self::STYLE_PRESET ],
|
|
self::IMAGE_STRENGTH => $image_data['promptSettings'][ self::IMAGE_STRENGTH ],
|
|
self::ASPECT_RATIO => $image_data['promptSettings'][ self::ASPECT_RATIO ],
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
],
|
|
$image_file,
|
|
'image'
|
|
);
|
|
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* get_image_to_image_upscale
|
|
* @param $image_data
|
|
*
|
|
* @return mixed|\WP_Error
|
|
* @throws \Exception
|
|
*/
|
|
public function get_image_to_image_upscale( $image_data ) {
|
|
$image_file = get_attached_file( $image_data['attachment_id'] );
|
|
|
|
if ( ! $image_file ) {
|
|
throw new \Exception( 'Image file not found' );
|
|
}
|
|
|
|
$result = $this->ai_request(
|
|
'POST',
|
|
'image/image-to-image/upscale',
|
|
[
|
|
self::IMAGE_RESOLUTION => $image_data['promptSettings']['upscale_to'],
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
],
|
|
$image_file,
|
|
'image'
|
|
);
|
|
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* store_temp_file
|
|
* used to store a temp file for the AI request and deletes it once the request is done
|
|
* @param $file_content
|
|
* @param $file_ext
|
|
*
|
|
* @return string
|
|
*/
|
|
private function store_temp_file( $file_content, $file_ext = '' ) {
|
|
$temp_file = str_replace( '.tmp', '', wp_tempnam() . $file_ext );
|
|
file_put_contents( $temp_file, $file_content );
|
|
// make sure the temp file is deleted on shutdown
|
|
register_shutdown_function( function () use ( $temp_file ) {
|
|
if ( file_exists( $temp_file ) ) {
|
|
unlink( $temp_file );
|
|
}
|
|
} );
|
|
return $temp_file;
|
|
}
|
|
|
|
/**
|
|
* get_image_to_image_out_painting
|
|
* @param $image_data
|
|
*
|
|
* @return mixed|\WP_Error
|
|
* @throws \Exception
|
|
*/
|
|
public function get_image_to_image_out_painting( $image_data ) {
|
|
$img_content = str_replace( ' ', '+', $image_data['mask'] );
|
|
$img_content = substr( $img_content, strpos( $img_content, ',' ) + 1 );
|
|
$img_content = base64_decode( $img_content );
|
|
$mask_file = $this->store_temp_file( $img_content, '.png' );
|
|
|
|
if ( ! $mask_file ) {
|
|
throw new \Exception( 'Expended Image file not found' );
|
|
}
|
|
|
|
$result = $this->ai_request(
|
|
'POST',
|
|
'image/image-to-image/outpainting',
|
|
[
|
|
self::PROMPT => $image_data[ self::PROMPT ],
|
|
self::IMAGE_TYPE => '',
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
],
|
|
[
|
|
[
|
|
'name' => 'image',
|
|
'type' => 'image',
|
|
'path' => $mask_file,
|
|
],
|
|
]
|
|
);
|
|
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* get_image_to_image_mask
|
|
* @param $image_data
|
|
*
|
|
* @return mixed|\WP_Error
|
|
* @throws \Exception
|
|
*/
|
|
public function get_image_to_image_mask( $image_data ) {
|
|
$image_file = get_attached_file( $image_data['attachment_id'] );
|
|
$mask_file = $this->store_temp_file( $image_data['mask'], '.svg' );
|
|
|
|
if ( ! $image_file ) {
|
|
throw new \Exception( 'Image file not found' );
|
|
}
|
|
|
|
if ( ! $mask_file ) {
|
|
throw new \Exception( 'Mask file not found' );
|
|
}
|
|
|
|
$result = $this->ai_request(
|
|
'POST',
|
|
'image/image-to-image/inpainting',
|
|
[
|
|
self::PROMPT => $image_data[ self::PROMPT ],
|
|
self::IMAGE_TYPE => $image_data['promptSettings'][ self::IMAGE_TYPE ] . '/' . $image_data['promptSettings'][ self::STYLE_PRESET ],
|
|
self::IMAGE_STRENGTH => $image_data['promptSettings'][ self::IMAGE_STRENGTH ],
|
|
'api_version' => ELEMENTOR_VERSION,
|
|
'site_lang' => get_bloginfo( 'language' ),
|
|
],
|
|
[
|
|
[
|
|
'name' => 'image',
|
|
'type' => 'image',
|
|
'path' => $image_file,
|
|
],
|
|
[
|
|
'name' => 'mask_image',
|
|
'type' => 'image/svg+xml',
|
|
'path' => $mask_file,
|
|
],
|
|
]
|
|
);
|
|
|
|
return $result;
|
|
}
|
|
|
|
protected function init() {}
|
|
}
|