← All changes | includes/Core/WriteWithAI.php +641 -1232 4.5.6 → 4.9.4 View file →

		@@ -12,9 +12,13 @@
12	12	use WPDeveloper\BetterDocs\Core\PostType;
13	13
14	14	use WPDeveloper\BetterDocs\Utils\Helper;
15	15	use WPDeveloper\BetterDocs\Utils\AIHelper;
	16	+ use WPDeveloper\BetterDocs\AI\ProviderFactory;
	17	+ use WPDeveloper\BetterDocs\AI\ModelRegistry;
16	18	use WPDeveloper\BetterDocs\Utils\AIUsage;
	19	+ use WPDeveloper\BetterDocs\REST\AIEdit;
	20	+ use WPDeveloper\BetterDocs\REST\WriteWithAI as RESTWriteWithAI;
17	21
18	22	class WriteWithAI extends Base {
19	23
20	24	public $settings;
		@@ -32,11 +36,11 @@
32	36	$post_type = '';
33	37	}
34	38
35	39	if ( ! empty( $this->isEnabledWriteWithAI() ) && 'docs' == $post_type ) {
36		- add_action( 'admin_footer', array( $this, 'ai_autowrite_button' ) );
37	40	add_action( 'admin_enqueue_scripts', array( $this, 'enqueue_ai_edit_assets' ) );
38	41	}
	42	+ // Legacy AJAX handler kept for back-compat; the redesigned modal uses the REST route.
39	43	add_action( 'wp_ajax_generate_openai_content', array( $this, 'generate_openai_content_callback' ) );
40	44	}
41	45
42	46	public function enqueue_ai_edit_assets( $hook ) {
		@@ -48,9 +52,85 @@
48	52	if ( 'docs' !== $post_type ) {
49	53	return;
50	54	}
51	55
52		- if ( empty( $this->get_api_key() ) ) {
	56	+ $factory = new ProviderFactory( $this->settings );
	57	+ $api_key = $factory->api_key_for( $factory->active_platform() );
	58	+ $has_key = ! empty( $api_key );
	59	+
	60	+ // Resolve the *active* AI platform + model (multi-platform aware) so the
	61	+ // modal reflects the current Settings → AI selection instead of the legacy
	62	+ // OpenAI-only `write_with_ai_model` key.
	63	+ $active_platform = $factory->active_platform();
	64	+ $active_model = $factory->active_model();
	65	+ $platform_labels = ModelRegistry::platforms();
	66	+ $model_labels = ModelRegistry::models( $active_platform );
	67	+
	68	+ // Glossary suggestions are existing-terms-only, and the glossaries taxonomy is only
	69	+ // registered by Pro (see BetterDocsPro\Core\GlossaryTaxonomy::register_glossaries_taxonomy).
	70	+ // So the "Suggest glossaries" control must require, on top of the two settings, that Pro is
	71	+ // active AND at least one glossary term exists — otherwise the modal advertises an offer that can never
	72	+ // return anything. Mirrors the Docs-AI-suite availability check in Core\DocsAISuite.
	73	+ $glossary_count = wp_count_terms( array( 'taxonomy' => 'glossaries', 'hide_empty' => false ) );
	74	+ $has_glossary_terms = ! is_wp_error( $glossary_count ) && (int) $glossary_count > 0;
	75	+ $glossary_suggestions_enabled = (bool) $this->settings->get( 'enable_glossaries', false )
	76	+ && (bool) $this->settings->get( 'show_glossary_suggestions', true )
	77	+ && betterdocs()->is_pro_active()
	78	+ && $has_glossary_terms;
	79	+
	80	+ // Write with AI — loads even without a key so the modal can show its
	81	+ // "add an API key" banner (matches the legacy inline form behavior).
	82	+ betterdocs()->assets->enqueue( 'betterdocs-write-with-ai', 'blocks/write-with-ai.js' );
	83	+ betterdocs()->assets->enqueue( 'betterdocs-write-with-ai-style', 'blocks/write-with-ai-style.css' );
	84	+
	85	+ wp_localize_script(
	86	+ 'betterdocs-write-with-ai',
	87	+ 'betterdocsWriteWithAI',
	88	+ array(
	89	+ 'rest_url' => esc_url_raw( rest_url( 'betterdocs/v1/write-with-ai' ) ),
	90	+ 'rest_nonce' => wp_create_nonce( 'wp_rest' ),
	91	+ 'post_id' => get_the_ID(),
	92	+ 'has_key' => $has_key,
	93	+ // Term-suggestion (Docs AI Suite) wiring reused by the Write-with-AI
	94	+ // preview step. Endpoint + gate mirror REST\DocsAISuite / Core\DocsAISuite.
	95	+ 'rest_suggest_url' => esc_url_raw( rest_url( 'betterdocs/v1/ai-suggest-terms' ) ),
	96	+ 'suggest_terms_enabled' => (bool) $this->settings->get( 'enable_docs_ai_suite', true ),
	97	+ // Glossary suggestions follow the glossary feature AND real availability
	98	+ // (Pro active + at least one glossary term); see the computation above.
	99	+ 'glossary_suggestions_enabled' => $glossary_suggestions_enabled,
	100	+ 'platform' => $active_platform,
	101	+ 'platform_label' => isset( $platform_labels[ $active_platform ] ) ? $platform_labels[ $active_platform ] : ucfirst( (string) $active_platform ),
	102	+ 'model' => $active_model,
	103	+ 'model_label' => isset( $model_labels[ $active_model ] ) ? $model_labels[ $active_model ] : $active_model,
	104	+ 'max_token' => (int) $this->settings->get( 'ai_autowrite_max_token', 2500 ),
	105	+ // Attachment limits, resolved server-side so the modal can refuse
	106	+ // an oversize file before uploading it — and so it advertises the
	107	+ // host's real ceiling rather than ours when the host is smaller.
	108	+ 'max_upload_bytes' => RESTWriteWithAI::upload_cap( 'file' ),
	109	+ 'max_media_bytes' => RESTWriteWithAI::upload_cap( 'media' ),
	110	+ 'media_exts' => RESTWriteWithAI::media_exts(),
	111	+ 'supports_transcription' => $this->platform_supports( 'transcription', $active_platform ),
	112	+ 'settings_url' => esc_url( admin_url( 'admin.php?page=betterdocs-settings#betterdocs-ai' ) ),
	113	+ 'woo_active' => class_exists( 'WooCommerce' ),
	114	+ 'is_multilingual_active' => Helper::is_multilingual_active(),
	115	+ 'language_options' => Helper::get_active_languages(),
	116	+ 'instructions' => $this->get_instruction_choices(),
	117	+ // From Git is a Pro feature. Pro is detected here; Git enabled/auth
	118	+ // status is filled in by Pro via the filter below (default: off).
	119	+ 'is_pro_active' => betterdocs()->is_pro_active(),
	120	+ 'git' => apply_filters(
	121	+ 'betterdocs_write_with_ai_git',
	122	+ array(
	123	+ 'enabled' => false,
	124	+ 'connected' => false,
	125	+ 'settings_url' => admin_url( 'admin.php?page=betterdocs-settings#git-sync' ),
	126	+ )
	127	+ ),
	128	+ )
	129	+ );
	130	+
	131	+ // AI Edit — only meaningful with a key configured.
	132	+ if ( ! $has_key ) {
53	133	return;
54	134	}
55	135
56	136	betterdocs()->assets->enqueue( 'betterdocs-ai-edit', 'blocks/ai-edit.js' );
		@@ -61,9 +141,11 @@
61	141	'betterdocsAIEdit',
62	142	array(
63	143	'rest_url' => esc_url_raw( rest_url( 'betterdocs/v1/ai-edit' ) ),
64	144	'rest_nonce' => wp_create_nonce( 'wp_rest' ),
65		- 'post_id' => get_the_ID()
	145	+ 'post_id' => get_the_ID(),
	146	+ 'instructions' => $this->get_instruction_choices(),
	147	+ 'actions' => AIEdit::get_localized_actions()
66	148	)
67	149	);
68	150	}
69	151
		@@ -73,58 +155,34 @@
73	155	}
74	156
75	157	public function isValidAPIKey( $apiKey ) {
76	158	if ( empty( $apiKey ) ) {
77		- $api_response[ 'valid' ] = false;
78		- $api_response[ 'message' ] = 'Please Insert your <a href="/admin.php?page=betterdocs-settings#betterdocs-ai">OpenAI API Key</a> to use this Write with AI feature.';
79		-
80		- return $api_response;
	159	+ return array(
	160	+ 'valid' => false,
	161	+ 'message' => 'Please Insert your <a href="/admin.php?page=betterdocs-settings#betterdocs-ai">API Key</a> to use this Write with AI feature.'
	162	+ );
81	163	}
82	164
83		- $api_response = array();
84		-
85		- $response = wp_safe_remote_get(
86		- 'https://api.openai.com/v1/engines',
87		- array(
88		- 'headers' => array(
89		- 'Content-Type' => 'application/json',
90		- 'Authorization' => 'Bearer ' . $apiKey,
91		- ),
92		- 'timeout' => 15,
93		- )
94		- );
95		-
96		- if ( is_wp_error( $response ) ) {
97		- $api_response[ 'valid' ] = false;
98		- $api_response[ 'message' ] = $response->get_error_message();
99		- return $api_response;
100		- }
101		-
102		- $httpCode = (int) wp_remote_retrieve_response_code( $response );
103		- $body = wp_remote_retrieve_body( $response );
104		-
105		- if ( 200 === $httpCode ) {
106		- $api_response[ 'valid' ] = true;
107		- $api_response[ 'message' ] = 'Valid API Key';
108		- } else {
109		- $responseData = json_decode( $body, true );
110		- $messageData = ! empty( $responseData[ 'error' ] ) ? $responseData[ 'error' ] : array();
111		- $api_response[ 'valid' ] = false;
112		- $api_response[ 'message' ] = ! empty( $messageData[ 'message' ] ) ? $messageData[ 'message' ] : 'Invalid API Key';
113		- }
114		-
115		- // print_r($response);
116		-
117		- return $api_response;
	165	+ $factory = new ProviderFactory( $this->settings );
	166	+ return $factory->validate( $factory->active_platform(), $apiKey );
118	167	}
119	168
120	169	public function get_api_key() {
121		- $api_key = $this->settings->get( 'ai_autowrite_api_key', '' );
122		- return $api_key;
	170	+ $factory = new ProviderFactory( $this->settings );
	171	+ return $factory->api_key_for( $factory->active_platform() );
123	172	}
124	173
125		- public function get_system_prompt() {
126		- $prompt = <<<'PROMPT'
	174	+ /**
	175	+ * The built-in "Default" instruction body.
	176	+ *
	177	+ * Single source of truth shared by {@see Settings::get_default()} (which seeds
	178	+ * the editable "Default" instruction set) and {@see self::get_system_prompt()}
	179	+ * (the fallback when no saved Default content exists).
	180	+ *
	181	+ * @return string
	182	+ */
	183	+ public static function default_instruction_content() {
	184	+ return <<<'PROMPT'
127	185	You are a Senior Technical Writer specializing in comprehensive, high-quality documentation. Your goal is to produce documentation that scores 100/100 on clarity, completeness, and structure.
128	186
129	187	## Output format
130	188
		@@ -146,1299 +204,650 @@
146	204	Do not emit `<style>`, `<script>`, `<iframe>`, `<form>`, or inline `style=""` / `class=""` attributes (other than the `highlight` class above).
147	205
148	206	If — and only if — the user's prompt explicitly asks for raw/custom HTML, embed code, or a non-standard structure, follow that request instead of these defaults.
149	207	PROMPT;
150		-
151		- return apply_filters( 'betterdocs_write_with_ai_system_prompt', $prompt );
152	208	}
153	209
154		- public function generate_openai_response( $prompt, $keywords ) {
155		- try {
156		- $api_key = $this->settings->get( 'ai_autowrite_api_key', '' );
157		- $max_tokens = $this->settings->get( 'ai_autowrite_max_token', 2500 );
158		- $model = $this->settings->get( 'write_with_ai_model', 'gpt-4o-mini' );
	210	+ /**
	211	+ * The saved instruction sets, normalized to a list of { id, title, content }.
	212	+ *
	213	+ * Stored in the `write_with_ai_instructions` setting. Always returns at least
	214	+ * the built-in "Default" set so callers never have to special-case an empty
	215	+ * option (e.g. a site whose stored value predates this feature).
	216	+ *
	217	+ * @return array<int,array{id:string,title:string,content:string}>
	218	+ */
	219	+ public function get_instructions() {
	220	+ $stored = $this->settings->get( 'write_with_ai_instructions', array() );
159	221
160		- $api_endpoint = 'https://api.openai.com/v1/chat/completions'; // Update the endpoint based on OpenAI API version
161		-
162		- $request_body = AIHelper::build_openai_payload(
163		- $model,
164		- array(
165		- array(
166		- 'role' => 'system',
167		- 'content' => $this->get_system_prompt()
168		- ),
169		- array(
170		- 'role' => 'user',
171		- 'content' => $prompt
172		- )
173		- ),
174		- $max_tokens,
175		- null,
176		- 'write_with_ai'
177		- );
178		-
179		- $request_options = array(
180		- 'headers' => array(
181		- 'Content-Type' => 'application/json',
182		- 'Authorization' => 'Bearer ' . $api_key
183		- ),
184		- 'body' => json_encode( $request_body ),
185		- 'timeout' => 300
186		- );
187		-
188		- // GPT-5.5 reasoning can run well past the default limits; give PHP and
189		- // the HTTP call room to finish (still subject to server php-fpm/nginx limits).
190		- if ( function_exists( 'set_time_limit' ) ) {
191		- set_time_limit( 300 ); // phpcs:ignore Squiz.PHP.DiscouragedFunctions.Discouraged -- long-running AI generation needs an extended limit; still bounded by server fpm/nginx timeouts.
	222	+ $instructions = array();
	223	+ if ( is_array( $stored ) ) {
	224	+ foreach ( $stored as $item ) {
	225	+ if ( ! is_array( $item ) || empty( $item['id'] ) ) {
	226	+ continue;
	227	+ }
	228	+ $instructions[] = array(
	229	+ 'id' => sanitize_key( (string) $item['id'] ),
	230	+ 'title' => isset( $item['title'] ) ? (string) $item['title'] : '',
	231	+ 'content' => isset( $item['content'] ) ? (string) $item['content'] : '',
	232	+ );
192	233	}
	234	+ }
193	235
194		- $response = wp_remote_post( $api_endpoint, $request_options );
195		-
196		- if ( is_wp_error( $response ) ) {
197		- return 'Error: ' . $response->get_error_message();
198		- } else {
199		- $body = wp_remote_retrieve_body( $response );
200		-
201		- $data = json_decode( $body, true );
202		-
203		- if ( ! empty( $data[ 'error' ] ) ) {
204		- return $data[ 'error' ][ 'message' ];
205		- }
206		-
207		- return $data[ 'choices' ][ 0 ][ 'message' ][ 'content' ]; // Update this line to get the assistant's message
	236	+ // Guarantee a Default set is always present.
	237	+ $has_default = false;
	238	+ foreach ( $instructions as $item ) {
	239	+ if ( 'default' === $item['id'] ) {
	240	+ $has_default = true;
	241	+ break;
208	242	}
209		- } catch ( Exception $error ) {
210		- return 'Error: ' . $error->getMessage();
211	243	}
212		- }
213		-
214		- public function generate_openai_response_ai_edit( $prompt ) {
215		- $api_key = $this->settings->get( 'ai_autowrite_api_key', '' );
216		- $max_tokens = $this->settings->get( 'ai_autowrite_max_token', 2500 );
217		- $model = $this->settings->get( 'write_with_ai_model', 'gpt-4o-mini' );
218		-
219		- $api_endpoint = 'https://api.openai.com/v1/chat/completions';
220		-
221		- $payload = AIHelper::build_openai_payload(
222		- $model,
223		- array(
	244	+ if ( ! $has_default ) {
	245	+ array_unshift(
	246	+ $instructions,
224	247	array(
225		- 'role' => 'system',
226		- 'content' => $this->get_system_prompt()
227		- ),
228		- array(
229		- 'role' => 'user',
230		- 'content' => $prompt
	248	+ 'id' => 'default',
	249	+ 'title' => __( 'Default/Core', 'betterdocs' ),
	250	+ 'content' => self::default_instruction_content(),
231	251	)
232		- ),
233		- $max_tokens,
234		- null,
235		- 'write_with_ai'
236		- );
237		-
238		- $request_options = array(
239		- 'headers' => array(
240		- 'Content-Type' => 'application/json',
241		- 'Authorization' => 'Bearer ' . $api_key
242		- ),
243		- 'body' => wp_json_encode( $payload ),
244		- 'timeout' => 300
245		- );
246		-
247		- // GPT-5.5 reasoning can run well past the default limits; give PHP and
248		- // the HTTP call room to finish (still subject to server php-fpm/nginx limits).
249		- if ( function_exists( 'set_time_limit' ) ) {
250		- set_time_limit( 300 ); // phpcs:ignore Squiz.PHP.DiscouragedFunctions.Discouraged -- long-running AI generation needs an extended limit; still bounded by server fpm/nginx timeouts.
251		- }
252		-
253		- $response = wp_remote_post( $api_endpoint, $request_options );
254		-
255		- if ( is_wp_error( $response ) ) {
256		- return array(
257		- 'success' => false,
258		- 'error' => $response->get_error_message(),
259		- 'model' => $model
260	252	);
261	253	}
262	254
263		- $body = wp_remote_retrieve_body( $response );
264		- $data = json_decode( $body, true );
	255	+ return $instructions;
	256	+ }
265	257
266		- if ( ! empty( $data[ 'error' ] ) ) {
267		- return array(
268		- 'success' => false,
269		- 'error' => isset( $data[ 'error' ][ 'message' ] ) ? $data[ 'error' ][ 'message' ] : 'OpenAI error',
270		- 'model' => $model,
271		- 'raw' => $data
	258	+ /**
	259	+ * The selectable (non-default) instruction sets exposed to the editor UI.
	260	+ *
	261	+ * Only id + title are sent to the browser; the prompt body stays server-side
	262	+ * and is resolved at request time by {@see self::get_instruction_messages()}.
	263	+ *
	264	+ * @return array<int,array{id:string,title:string}>
	265	+ */
	266	+ public function get_instruction_choices() {
	267	+ $choices = array();
	268	+ foreach ( $this->get_instructions() as $item ) {
	269	+ if ( 'default' === $item['id'] ) {
	270	+ continue;
	271	+ }
	272	+ if ( '' === trim( $item['content'] ) ) {
	273	+ continue; // skip empty sets — they'd inject nothing.
	274	+ }
	275	+ $choices[] = array(
	276	+ 'id' => $item['id'],
	277	+ 'title' => '' !== trim( $item['title'] ) ? $item['title'] : $item['id'],
272	278	);
273	279	}
274		-
275		- $content = isset( $data[ 'choices' ][ 0 ][ 'message' ][ 'content' ] ) ? $data[ 'choices' ][ 0 ][ 'message' ][ 'content' ] : '';
276		- $usage = isset( $data[ 'usage' ] ) && is_array( $data[ 'usage' ] ) ? $data[ 'usage' ] : array();
277		-
278		- return array(
279		- 'success' => true,
280		- 'content' => $content,
281		- 'model' => $model,
282		- 'prompt_tokens' => isset( $usage[ 'prompt_tokens' ] ) ? (int) $usage[ 'prompt_tokens' ] : null,
283		- 'completion_tokens' => isset( $usage[ 'completion_tokens' ] ) ? (int) $usage[ 'completion_tokens' ] : null,
284		- 'total_tokens' => isset( $usage[ 'total_tokens' ] ) ? (int) $usage[ 'total_tokens' ] : null,
285		- 'finish_reason' => isset( $data[ 'choices' ][ 0 ][ 'finish_reason' ] ) ? $data[ 'choices' ][ 0 ][ 'finish_reason' ] : null
286		- );
	280	+ return $choices;
287	281	}
288	282
289	283	/**
290		- * Generic chat completion using the Write-with-AI model/token/key settings.
	284	+ * Resolve selected instruction ids into extra `system` messages.
291	285	*
292		- * Shared by lightweight, plain-text generators (glossary definitions, FAQ answers)
293		- * that need the same dynamic model as Write with AI / AI Edit but a caller-supplied
294		- * system prompt instead of the doc-authoring HTML prompt. Returns the same structured
295		- * array shape as {@see self::generate_openai_response_ai_edit()}.
	286	+ * The "Default" set is intentionally excluded — it is always sent as the base
	287	+ * system prompt by {@see self::get_system_prompt()}. Unknown or empty ids are
	288	+ * silently ignored. Output order follows the requested $ids order.
296	289	*
297		- * @param string $user_prompt The user message.
298		- * @param string $system_prompt Optional system message (omitted when empty).
299		- * @return array { success:bool, content?:string, error?:string, model:string, *_tokens?:int }
	290	+ * @param array $ids
	291	+ * @return array<int,array{role:string,content:string}>
300	292	*/
301		- public function generate_text( $user_prompt, $system_prompt = '' ) {
302		- $api_key = $this->settings->get( 'ai_autowrite_api_key', '' );
303		- $max_tokens = $this->settings->get( 'ai_autowrite_max_token', 2500 );
304		- $model = $this->settings->get( 'write_with_ai_model', 'gpt-4o-mini' );
	293	+ public function get_instruction_messages( $ids ) {
	294	+ if ( empty( $ids ) || ! is_array( $ids ) ) {
	295	+ return array();
	296	+ }
305	297
	298	+ $by_id = array();
	299	+ foreach ( $this->get_instructions() as $item ) {
	300	+ $by_id[ $item['id'] ] = $item;
	301	+ }
	302	+
306	303	$messages = array();
307		- if ( $system_prompt !== '' ) {
308		- $messages[] = array(
	304	+ $seen = array();
	305	+ foreach ( $ids as $id ) {
	306	+ $id = sanitize_key( (string) $id );
	307	+ if ( 'default' === $id || isset( $seen[ $id ] ) || ! isset( $by_id[ $id ] ) ) {
	308	+ continue;
	309	+ }
	310	+ $content = trim( $by_id[ $id ]['content'] );
	311	+ if ( '' === $content ) {
	312	+ continue;
	313	+ }
	314	+ $seen[ $id ] = true;
	315	+ $messages[] = array(
309	316	'role' => 'system',
310		- 'content' => $system_prompt
	317	+ 'content' => $content,
311	318	);
312	319	}
313		- $messages[] = array(
314		- 'role' => 'user',
315		- 'content' => $user_prompt
316		- );
317	320
318		- $api_endpoint = 'https://api.openai.com/v1/chat/completions';
	321	+ return $messages;
	322	+ }
319	323
320		- $payload = AIHelper::build_openai_payload( $model, $messages, $max_tokens, null, 'write_with_ai' );
	324	+ /**
	325	+ * Coerce a list of extra messages into well-formed `system` chat messages.
	326	+ *
	327	+ * Accepts either pre-built [ 'role'=>'system', 'content'=>… ] entries (as
	328	+ * returned by {@see self::get_instruction_messages()}) or bare strings, and
	329	+ * drops anything empty. Keeps the OpenAI payload builders tolerant of either
	330	+ * shape so callers can pass instruction ids or ready-made messages.
	331	+ *
	332	+ * @param array $extra_system
	333	+ * @return array<int,array{role:string,content:string}>
	334	+ */
	335	+ protected function normalize_extra_system( $extra_system ) {
	336	+ if ( empty( $extra_system ) || ! is_array( $extra_system ) ) {
	337	+ return array();
	338	+ }
321	339
322		- $request_options = array(
323		- 'headers' => array(
324		- 'Content-Type' => 'application/json',
325		- 'Authorization' => 'Bearer ' . $api_key
326		- ),
327		- 'body' => wp_json_encode( $payload ),
328		- 'timeout' => 300
329		- );
	340	+ $messages = array();
	341	+ foreach ( $extra_system as $entry ) {
	342	+ if ( is_string( $entry ) ) {
	343	+ $content = trim( $entry );
	344	+ $role = 'system';
	345	+ } elseif ( is_array( $entry ) && isset( $entry['content'] ) ) {
	346	+ $content = trim( (string) $entry['content'] );
	347	+ $role = isset( $entry['role'] ) ? (string) $entry['role'] : 'system';
	348	+ } else {
	349	+ continue;
	350	+ }
330	351
331		- // GPT-5.5 reasoning can run well past the default limits; give PHP and
332		- // the HTTP call room to finish (still subject to server php-fpm/nginx limits).
333		- if ( function_exists( 'set_time_limit' ) ) {
334		- set_time_limit( 300 );
335		- }
	352	+ if ( '' === $content ) {
	353	+ continue;
	354	+ }
336	355
337		- $response = wp_remote_post( $api_endpoint, $request_options );
338		-
339		- if ( is_wp_error( $response ) ) {
340		- return array(
341		- 'success' => false,
342		- 'error' => $response->get_error_message(),
343		- 'model' => $model
	356	+ $messages[] = array(
	357	+ 'role' => $role,
	358	+ 'content' => $content,
344	359	);
345	360	}
346	361
347		- $body = wp_remote_retrieve_body( $response );
348		- $data = json_decode( $body, true );
	362	+ return $messages;
	363	+ }
349	364
350		- if ( ! empty( $data[ 'error' ] ) ) {
351		- return array(
352		- 'success' => false,
353		- 'error' => isset( $data[ 'error' ][ 'message' ] ) ? $data[ 'error' ][ 'message' ] : 'OpenAI error',
354		- 'model' => $model,
355		- 'raw' => $data
356		- );
	365	+ public function get_system_prompt() {
	366	+ // Prefer the saved (editable) "Default" instruction set; fall back to the
	367	+ // built-in body when it's missing or has been cleared.
	368	+ $prompt = self::default_instruction_content();
	369	+ foreach ( $this->get_instructions() as $item ) {
	370	+ if ( 'default' === $item['id'] && '' !== trim( $item['content'] ) ) {
	371	+ $prompt = $item['content'];
	372	+ break;
	373	+ }
357	374	}
358	375
359		- $content = isset( $data[ 'choices' ][ 0 ][ 'message' ][ 'content' ] ) ? $data[ 'choices' ][ 0 ][ 'message' ][ 'content' ] : '';
360		- $usage = isset( $data[ 'usage' ] ) && is_array( $data[ 'usage' ] ) ? $data[ 'usage' ] : array();
	376	+ return apply_filters( 'betterdocs_write_with_ai_system_prompt', $prompt );
	377	+ }
361	378
362		- return array(
363		- 'success' => true,
364		- 'content' => $content,
365		- 'model' => $model,
366		- 'prompt_tokens' => isset( $usage[ 'prompt_tokens' ] ) ? (int) $usage[ 'prompt_tokens' ] : null,
367		- 'completion_tokens' => isset( $usage[ 'completion_tokens' ] ) ? (int) $usage[ 'completion_tokens' ] : null,
368		- 'total_tokens' => isset( $usage[ 'total_tokens' ] ) ? (int) $usage[ 'total_tokens' ] : null,
369		- 'finish_reason' => isset( $data[ 'choices' ][ 0 ][ 'finish_reason' ] ) ? $data[ 'choices' ][ 0 ][ 'finish_reason' ] : null
	379	+ /**
	380	+ * Build the system + user messages and chat options shared by the
	381	+ * Write-with-AI generator and the in-editor AI-Edit endpoint. Routes through
	382	+ * the active AI platform (ProviderFactory), so it works for every provider.
	383	+ *
	384	+ * @param string $prompt
	385	+ * @param int|null $max_tokens Optional cap override (e.g. a "large" doc).
	386	+ * @param array $extra_system Optional extra system messages.
	387	+ * @return array array( array $messages, array $options )
	388	+ */
	389	+ private function ai_request_args( $prompt, $max_tokens = null, $extra_system = array() ) {
	390	+ $messages = array_merge(
	391	+ array( array( 'role' => 'system', 'content' => $this->get_system_prompt() ) ),
	392	+ $this->normalize_extra_system( $extra_system ),
	393	+ array( array( 'role' => 'user', 'content' => $prompt ) )
370	394	);
	395	+
	396	+ return array( $messages, $this->ai_chat_options( $max_tokens ) );
371	397	}
372	398
373		- public function generate_openai_content_callback() {
374		- // Verify the nonce
375		- $ai_nonce = isset( $_POST['ai_nonce'] ) ? sanitize_text_field( wp_unslash( $_POST['ai_nonce'] ) ) : '';
376		- if ( ! wp_verify_nonce( $ai_nonce, 'generate_openai_content_nonce' ) ) {
377		- wp_send_json_error( 'Invalid nonce' );
378		- wp_die();
	399	+ /**
	400	+ * Assemble the chat options passed to the active provider. Document
	401	+ * generation is long-running on every platform: reasoning models (gpt-5*)
	402	+ * and larger non-OpenAI models (e.g. Claude Opus) routinely need well over
	403	+ * the old 50s default to return a full doc, which timed them out mid-
	404	+ * generation. Give every request a generous ceiling; fast models simply
	405	+ * finish early and are unaffected.
	406	+ *
	407	+ * @param int|null $max_tokens Optional token cap override.
	408	+ * @param float|null $temperature Optional sampling temperature.
	409	+ * @return array
	410	+ */
	411	+ private function ai_chat_options( $max_tokens = null, $temperature = null ) {
	412	+ $timeout = 300;
	413	+ if ( function_exists( 'set_time_limit' ) ) {
	414	+ set_time_limit( 300 );
379	415	}
380	416
381		- if ( ! current_user_can( 'edit_posts' ) ) {
382		- wp_send_json_error( 'Insufficient permissions' );
383		- wp_die();
384		- }
	417	+ $options = array(
	418	+ 'max_tokens' => null !== $max_tokens ? (int) $max_tokens : (int) $this->settings->get( 'ai_autowrite_max_token', 2500 ),
	419	+ 'context' => 'write_with_ai',
	420	+ 'timeout' => $timeout,
	421	+ );
385	422
386		- $prompt = isset( $_POST['prompt'] ) ? sanitize_text_field( wp_unslash( $_POST['prompt'] ) ) : '';
387		- $keywords = isset( $_POST['keywords'] ) ? sanitize_text_field( wp_unslash( $_POST['keywords'] ) ) : '';
388		-
389		- $ai_instance = new WriteWithAI( $this->settings );
390		-
391		- $generated_content = $ai_instance->generate_openai_response( $prompt, $keywords );
392		-
393		- // Count a successful generation (skip obvious upstream errors).
394		- if ( is_string( $generated_content ) && $generated_content !== '' && strpos( $generated_content, 'Error:' ) !== 0 ) {
395		- $post_id = isset( $_POST[ 'post_id' ] ) ? intval( $_POST[ 'post_id' ] ) : 0; //phpcs:ignore
396		- AIUsage::record( 'write_with_ai', $post_id );
	423	+ if ( null !== $temperature ) {
	424	+ $options['temperature'] = $temperature;
397	425	}
398	426
399		- // Send the generated content as the AJAX response
400		- wp_send_json_success( $generated_content );
401		- wp_die();
	427	+ return $options;
402	428	}
403	429
404		- public function ai_autowrite_button() {
	430	+ public function generate_openai_response( $prompt, $keywords, $max_tokens = null, $extra_system = array() ) {
	431	+ try {
	432	+ list( $messages, $options ) = $this->ai_request_args( $prompt, $max_tokens, $extra_system );
405	433
406		- ?>
407		- <script>
408		- var docsTitle;
	434	+ $result = ( new ProviderFactory( $this->settings ) )->make()->chat( $messages, $options );
409	435
410		- function responseMessage(message) {
411		- return `<div class="warning-message">
412		- <span class="dashicons dashicons-warning"></span>
413		- <div class="footer-message">
414		- ${message}
415		- </div>
416		- </div>`;
417		- }
	436	+ if ( is_wp_error( $result ) ) {
	437	+ return $result->get_error_message();
	438	+ }
418	439
419		- function generateArticle(e) {
	440	+ return $result['content'];
	441	+ } catch ( \Exception $error ) {
	442	+ return 'Error: ' . $error->getMessage();
	443	+ }
	444	+ }
420	445
	446	+ /**
	447	+ * Generate documentation from an uploaded image using a vision-capable model.
	448	+ *
	449	+ * Mirrors generate_openai_response() but sends the picture alongside the text
	450	+ * prompt as an OpenAI-format multimodal user message
	451	+ * (`content: [ {type:text}, {type:image_url} ]`). The OpenAI-compatible
	452	+ * provider forwards that message array to the wire verbatim, so no provider
	453	+ * change is needed. Only OpenAI vision models are wired — Claude and Gemini
	454	+ * use a different image envelope, so they are refused with a clear error
	455	+ * instead of being sent a payload they would reject.
	456	+ *
	457	+ * @param string $prompt Composed instruction prompt.
	458	+ * @param array $image { data_uri:string, mime:string }.
	459	+ * @param int|null $max_tokens Optional token cap.
	460	+ * @param array $extra_system Extra system messages (instruction sets).
	461	+ * @return string|\WP_Error Generated content, or WP_Error on guard/failure.
	462	+ */
	463	+ public function generate_vision_response( $prompt, $image, $max_tokens = null, $extra_system = array() ) {
	464	+ if ( empty( $image['data_uri'] ) ) {
	465	+ return new \WP_Error( 'ai_vision_no_image', __( 'No image data to send to the AI.', 'betterdocs' ) );
	466	+ }
421	467
422		- const title = document.getElementById('betterdocs-ai-title').value;
423		- const keywords = document.getElementById('betterdocs-ai-keyword').value;
424		- // const sectionOptions = document.getElementById('bd-autowrtite-content-section');
425		- // const numOfSections = sectionOptions.options[sectionOptions.selectedIndex].value;
	468	+ $factory = new ProviderFactory( $this->settings );
	469	+ $platform = $factory->active_platform();
	470	+ $model = $factory->active_model( $platform );
426	471
427		- // const paragraphOptions = document.getElementById('bd-autowrtite-content-paragraph');
428		- // const numOfParagraphs = paragraphOptions.options[paragraphOptions.selectedIndex].value;
	472	+ if ( ! $this->platform_supports( 'vision', $platform, $model ) ) {
	473	+ return new \WP_Error(
	474	+ 'ai_no_vision',
	475	+ sprintf(
	476	+ /* translators: 1: AI platform id, 2: model name. */
	477	+ __( 'The configured AI model (%1$s / %2$s) can\'t read images. Switch to an OpenAI vision model such as GPT-4o or GPT-4o mini in BetterDocs → Settings → AI Content Suite, or upload a PDF/DOCX/TXT instead.', 'betterdocs' ),
	478	+ $platform,
	479	+ '' !== (string) $model ? $model : 'default'
	480	+ )
	481	+ );
	482	+ }
429	483
430		- const contentTextArea = document.getElementById('betterdocs-ai-content');
431		- const docGenerateBtnTxt = document.querySelector('.generate-btn span');
	484	+ try {
	485	+ $messages = array_merge(
	486	+ array( array( 'role' => 'system', 'content' => $this->get_system_prompt() ) ),
	487	+ $this->normalize_extra_system( $extra_system ),
	488	+ array(
	489	+ array(
	490	+ 'role' => 'user',
	491	+ 'content' => array(
	492	+ array( 'type' => 'text', 'text' => (string) $prompt ),
	493	+ array( 'type' => 'image_url', 'image_url' => array( 'url' => (string) $image['data_uri'] ) ),
	494	+ ),
	495	+ ),
	496	+ )
	497	+ );
432	498
433		- // Add nonce value to the data being sent
434		- const nonce = document.querySelector('input[name="ai_nonce"]').value;
435		- const isOverwrite = document.getElementById('betterdocs-ai-checkbox').checked;
436		- const generateDocLabel = "<?php echo esc_html__( 'Generate Doc', 'betterdocs' ); ?>";
437		- const reGenerateDocLabel = "<?php echo esc_html__( 'Regenerate Doc', 'betterdocs' ); ?>";
	499	+ $result = $factory->make()->chat( $messages, $this->ai_chat_options( $max_tokens ) );
438	500
	501	+ if ( is_wp_error( $result ) ) {
	502	+ return $result;
	503	+ }
439	504
440		- // contentTextArea.value = `Write an article about ${title} in English. The article is organized by the following exact ${numOfSections} headings. Write exact ${numOfParagraphs} number of paragraphs per heading.${keywords} Use Markdown for formatting. Add an introduction and a conclusion. Style: creative. Tone: cheerful.`;
	505	+ return $result['content'];
	506	+ } catch ( \Exception $error ) {
	507	+ return new \WP_Error( 'ai_vision_failed', 'Error: ' . $error->getMessage() );
	508	+ }
	509	+ }
441	510
	511	+ /**
	512	+ * Transcribe an uploaded recording to plain text.
	513	+ *
	514	+ * Deliberately not a doc-generation call: this returns the raw transcript so
	515	+ * the author can correct it, and generation then runs through the ordinary
	516	+ * grounded from-source path. That keeps one generation path in the plugin
	517	+ * instead of a second, media-shaped one.
	518	+ *
	519	+ * @since 4.9.4
	520	+ *
	521	+ * @param array $file `[ 'path', 'filename', 'mime' ]` — a PHP temp upload.
	522	+ * @return string|\WP_Error Transcript text.
	523	+ */
	524	+ public function transcribe( $file ) {
	525	+ $factory = new ProviderFactory( $this->settings );
	526	+ $platform = $factory->active_platform();
442	527
443		- // Check if the title is not empty
444		- if (title.trim() !== '' && keywords.trim() !== '') {
445		- e.preventDefault();
	528	+ if ( ! $this->platform_supports( 'transcription', $platform ) ) {
	529	+ return new \WP_Error(
	530	+ 'ai_no_transcription',
	531	+ sprintf(
	532	+ /* translators: %s: AI platform id, e.g. "claude". */
	533	+ __( 'The configured AI platform (%s) can\'t read audio or video. Switch to OpenAI or Google Gemini in BetterDocs → Settings → AI Content Suite, or upload a text file instead.', 'betterdocs' ),
	534	+ $platform
	535	+ )
	536	+ );
	537	+ }
446	538
447		- if (!jQuery('#betterdocs-ai-error-message').parent().hasClass('hidden')) {
448		- jQuery('#betterdocs-ai-error-message').parent().addClass('hidden');
449		- }
	539	+ // 300s, matching every other AI call here: a 25 MB recording can take a
	540	+ // minute or more to come back, well past the 50s provider default.
	541	+ $options = $this->ai_chat_options();
	542	+ $options['model'] = ModelRegistry::transcription_model( $platform );
450	543
451		- docGenerateBtnTxt.innerHTML = "<?php echo esc_html__( 'Generating...', 'betterdocs' ); ?>";
452		- jQuery('.generate-btn').prop('disabled', true);
	544	+ try {
	545	+ return $factory->make()->transcribe( $file, $options );
	546	+ } catch ( \Exception $error ) {
	547	+ return new \WP_Error( 'ai_transcribe_failed', 'Error: ' . $error->getMessage() );
	548	+ }
	549	+ }
453	550
454		- jQuery.post({
455		- url: ajaxurl, // Make sure ajaxurl is defined in your script
456		- type: 'POST',
457		- data: {
458		- action: 'generate_openai_content',
459		- prompt: document.querySelector('#betterdocs-ai-content').value,
460		- // numOfSections: numOfSections,
461		- // numOfParagraphs: numOfParagraphs,
462		- keywords: keywords,
463		- post_id: (document.getElementById('post_ID') ? document.getElementById('post_ID').value : 0),
464		- ai_nonce: nonce, // Pass the nonce value
465		- },
466		- success: function(response) {
467		- // Update the content in the textarea
	551	+ /**
	552	+ * Whether a platform + model can handle a given input capability.
	553	+ *
	554	+ * One entry point for every "can this provider read that file?" question, so
	555	+ * a new capability is a case here rather than another bespoke check beside
	556	+ * the call site.
	557	+ *
	558	+ * @since 4.9.4
	559	+ *
	560	+ * @param string $capability `vision` | `transcription`
	561	+ * @param string $platform
	562	+ * @param string $model Chat model. Ignored for transcription, which uses
	563	+ * its own model (see ModelRegistry).
	564	+ * @return bool
	565	+ */
	566	+ public function platform_supports( $capability, $platform, $model = '' ) {
	567	+ switch ( $capability ) {
	568	+ case 'vision':
	569	+ return $this->platform_supports_vision( $platform, $model );
468	570
469		- if (response.success && (typeof response.data === 'string')) {
470		- docGenerateBtnTxt.innerHTML = "<?php echo esc_html__( 'Generated', 'betterdocs' ); ?>";
471		- setTimeout(() => {
472		- insertContentToEditor(title, response.data, isOverwrite, keywords);
473		- docGenerateBtnTxt.innerHTML = reGenerateDocLabel;
474		- jQuery('.generate-btn').removeAttr('disabled');
475		- }, 2000);
476		- } else {
477		- // alert(response.data.message);
478		- jQuery('#betterdocs-ai-error-message').parent().removeClass('hidden');
479		- jQuery('#betterdocs-ai-error-message').html(responseMessage(response.data.message));
480		- docGenerateBtnTxt.innerHTML = generateDocLabel;
481		- jQuery('.generate-btn').removeAttr('disabled');
	571	+ case 'transcription':
	572	+ // Presence in the transcription map is the whole test — the chat
	573	+ // model is irrelevant, since transcription runs as its own call
	574	+ // with its own model before any doc is written.
	575	+ return '' !== ModelRegistry::transcription_model( $platform );
	576	+ }
482	577
483		- }
484		- },
485		- error: function(jqXHR, textStatus, errorThrown) {
486		- console.error(jqXHR, textStatus, errorThrown);
	578	+ return false;
	579	+ }
487	580
488		- // Reset the UI so a slow/failed request never leaves a permanent "Generating...".
489		- var timedOut = ( textStatus === 'timeout' ) || ( jqXHR && jqXHR.status === 504 ) || ( jqXHR && jqXHR.status === 0 );
490		- var errMessage = timedOut
491		- ? "<?php echo esc_html__( 'The request timed out before a response came back. The AI may still have generated content on the server — try again, or pick a faster model / lower the token limit.', 'betterdocs' ); ?>"
492		- : "<?php echo esc_html__( 'Something went wrong while generating. Please try again.', 'betterdocs' ); ?>";
	581	+ /**
	582	+ * Whether the active platform + model can accept image input in the OpenAI
	583	+ * multimodal format. Deliberately conservative: only OpenAI vision model
	584	+ * families qualify, because Claude and Gemini require a different image
	585	+ * envelope this path does not build. gpt-3.5 (text-only) is excluded.
	586	+ *
	587	+ * @param string $platform
	588	+ * @param string $model
	589	+ * @return bool
	590	+ */
	591	+ protected function platform_supports_vision( $platform, $model ) {
	592	+ if ( 'openai' !== $platform ) {
	593	+ return false;
	594	+ }
493	595
494		- jQuery('#betterdocs-ai-error-message').parent().removeClass('hidden');
495		- jQuery('#betterdocs-ai-error-message').html(responseMessage(errMessage));
496		- docGenerateBtnTxt.innerHTML = generateDocLabel;
497		- jQuery('.generate-btn').removeAttr('disabled');
498		- },
499		- });
	596	+ $model = strtolower( (string) $model );
500	597
501		- }
502		- }
	598	+ if ( '' === $model || false !== strpos( $model, 'gpt-3.5' ) ) {
	599	+ return false;
	600	+ }
503	601
504		- // Function to update the textarea
505		- function updateTextarea() {
506		- const title = document.getElementById('betterdocs-ai-title').value;
507		- const keywords = document.getElementById('betterdocs-ai-keyword').value;
508		- const sectionOptions = document.getElementById('bd-autowrtite-content-section');
509		- let language = 'English';
	602	+ foreach ( array( 'gpt-4o', 'gpt-4.1', 'gpt-4-turbo', 'gpt-4-vision', 'chatgpt-4o', 'gpt-5', 'o1', 'o3', 'o4' ) as $family ) {
	603	+ if ( false !== strpos( $model, $family ) ) {
	604	+ return true;
	605	+ }
	606	+ }
510	607
511		- if (language === '') {
512		- language = 'English';
513		- }
	608	+ return false;
	609	+ }
514	610
515		- let keywordsText = keywords !== '' ? ` and incorporate the keywords: ${keywords}` : '';
	611	+ public function get_outline_system_prompt() {
	612	+ $prompt = <<<'PROMPT'
	613	+You are a Senior Technical Writer. Produce a documentation OUTLINE only — not the full article.
516	614
517		- const contentTextArea = document.getElementById('betterdocs-ai-content');
	615	+Return ONLY a JSON array (no prose, no markdown, no code fences). Each element is an object:
	616	+ { "level": "h2" | "h3", "text": "Section heading" }
518	617
519		- if (!jQuery('#betterdocs-ai-error-message').parent().hasClass('hidden')) {
520		- jQuery('#betterdocs-ai-error-message').parent().addClass('hidden');
521		- }
	618	+Rules:
	619	+- 5 to 9 top-level "h2" sections that comprehensively cover the topic.
	620	+- Add "h3" sub-sections only where they genuinely help; keep each one directly after its parent h2.
	621	+- Headings are concise, descriptive, and free of numbering.
	622	+- Do not include an h1/title and do not include any text outside the JSON array.
	623	+PROMPT;
522	624
523		- contentTextArea.value = `Write documentation for '${title}'. Cover these topics in depth: ${keywords}. Use headings, paragraphs, lists, links, code blocks, and tables wherever they help the reader. Follow the output-format rules in the system instructions.`;
524		- }
	625	+ return apply_filters( 'betterdocs_write_with_ai_outline_system_prompt', $prompt );
	626	+ }
525	627
526		- function convertMarkdownToHTML(markdownContent) {
527		- // Simple Markdown to HTML conversion (you may need to enhance this based on your needs)
528		- const htmlContent = markdownContent
529		- .replace(/^# (.+)$/gm, '<h1>$1</h1>')
530		- .replace(/^## (.+)$/gm, '<h2>$1</h2>')
531		- .replace(/^### (.+)$/gm, '<h3>$1</h3>')
532		- .replace(/\*\*(.+?)\*\*/g, '<strong>$1</strong>')
533		- .replace(/\*(.+?)\*/g, '<em>$1</em>');
	628	+ /**
	629	+ * Generate a documentation outline as a structured array of sections.
	630	+ *
	631	+ * @param string $user_prompt The composed prompt (title/keywords/instructions).
	632	+ * @return array { success:bool, outline?:array<int,array{level:string,text:string}>, error?:string }
	633	+ */
	634	+ public function generate_outline_response( $user_prompt, $extra_system = array() ) {
	635	+ $result = $this->generate_text( $user_prompt, $this->get_outline_system_prompt(), $extra_system );
534	636
535		- return htmlContent;
536		- }
	637	+ if ( empty( $result['success'] ) ) {
	638	+ return array(
	639	+ 'success' => false,
	640	+ 'error' => isset( $result['error'] ) ? $result['error'] : 'AI error',
	641	+ );
	642	+ }
537	643
538		- function replaceHeadingTitle(content, title, overviewTitle) {
539		- let regex = new RegExp(title, 'g');
540		- let updateContent = content.replace(regex, overviewTitle);
541		- return updateContent;
542		- }
	644	+ $outline = $this->parse_outline( (string) $result['content'] );
543	645
544		- function removeFirstHeading(htmlString) {
545		- var newDiv = document.createElement('div');
546		- newDiv.innerHTML = htmlString;
547		- var firstHeading = newDiv.querySelector('h1');
548		- if (firstHeading) {
549		- firstHeading.parentNode.removeChild(firstHeading);
550		- }
551		- return newDiv.innerHTML;
552		- }
	646	+ if ( empty( $outline ) ) {
	647	+ return array(
	648	+ 'success' => false,
	649	+ 'error' => 'The AI returned an outline we could not read. Please try again.',
	650	+ );
	651	+ }
553	652
	653	+ return array(
	654	+ 'success' => true,
	655	+ 'outline' => $outline,
	656	+ );
	657	+ }
554	658
555		- function escapeHtml(str) {
556		- return String(str).replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;');
557		- }
	659	+ /**
	660	+ * Parse a model outline response (ideally a JSON array) into a clean list of
	661	+ * { level, text } sections. Tolerates code fences and stray prose.
	662	+ *
	663	+ * @param string $content
	664	+ * @return array<int,array{level:string,text:string}>
	665	+ */
	666	+ protected function parse_outline( $content ) {
	667	+ $content = trim( $content );
	668	+ $content = preg_replace( '/^```(?:json)?\s*/i', '', $content );
	669	+ $content = preg_replace( '/```\s*$/', '', $content );
558	670
559		- function escapeRegex(str) {
560		- return String(str).replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
561		- }
	671	+ // Grab the first JSON array if the model wrapped it in prose.
	672	+ if ( preg_match( '/\[[\s\S]*\]/', $content, $m ) ) {
	673	+ $content = $m[0];
	674	+ }
562	675
563		- function stripCodeFences(htmlString) {
564		- if (!htmlString) {
565		- return '';
566		- }
567		- return htmlString
568		- .replace(/^\s*```(?:html|HTML)?\s*\n?/, '')
569		- .replace(/\n?\s*```\s*$/, '');
570		- }
	676	+ $decoded = json_decode( $content, true );
	677	+ if ( ! is_array( $decoded ) ) {
	678	+ return array();
	679	+ }
571	680
572		- function wrapwithHeighlight(inputString, keywords) {
573		- if (!inputString) {
574		- return '';
575		- }
	681	+ $outline = array();
	682	+ foreach ( $decoded as $item ) {
	683	+ if ( ! is_array( $item ) || empty( $item['text'] ) ) {
	684	+ continue;
	685	+ }
	686	+ $level = isset( $item['level'] ) && 'h3' === strtolower( (string) $item['level'] ) ? 'h3' : 'h2';
	687	+ $text = sanitize_text_field( (string) $item['text'] );
	688	+ if ( '' === $text ) {
	689	+ continue;
	690	+ }
	691	+ $outline[] = array( 'level' => $level, 'text' => $text );
	692	+ }
576	693
577		- const container = document.createElement('div');
578		- container.innerHTML = inputString;
	694	+ return $outline;
	695	+ }
579	696
580		- // Wrap each heading's visible text with the highlight span (skip if already present)
581		- container.querySelectorAll('h1, h2, h3, h4, h5, h6').forEach(function (heading) {
582		- if (heading.querySelector('.highlight')) {
583		- return;
584		- }
585		- const text = heading.textContent;
586		- if (!text || !text.trim()) {
587		- return;
588		- }
589		- heading.innerHTML = '<span class="highlight">' + escapeHtml(text) + '</span>';
590		- });
	697	+ public function generate_openai_response_ai_edit( $prompt, $extra_system = array() ) {
	698	+ $factory = new ProviderFactory( $this->settings );
	699	+ $model = $factory->active_model();
591	700
592		- // Wrap keywords only inside text nodes — never inside attributes, code, or existing highlights
593		- const keywordList = (keywords || '')
594		- .split(',')
595		- .map(function (k) { return k.trim(); })
596		- .filter(Boolean);
	701	+ $messages = array_merge(
	702	+ array( array( 'role' => 'system', 'content' => $this->get_system_prompt() ) ),
	703	+ $this->normalize_extra_system( $extra_system ),
	704	+ array( array( 'role' => 'user', 'content' => $prompt ) )
	705	+ );
597	706
598		- if (keywordList.length) {
599		- const pattern = new RegExp('\\b(' + keywordList.map(escapeRegex).join('|') + ')\\b', 'gi');
600		- const walker = document.createTreeWalker(container, NodeFilter.SHOW_TEXT, {
601		- acceptNode: function (node) {
602		- let parent = node.parentNode;
603		- while (parent && parent !== container) {
604		- if (parent.nodeType === 1) {
605		- const tag = parent.tagName.toLowerCase();
606		- if (tag === 'code' || tag === 'pre') {
607		- return NodeFilter.FILTER_REJECT;
608		- }
609		- if (parent.classList && parent.classList.contains('highlight')) {
610		- return NodeFilter.FILTER_REJECT;
611		- }
612		- }
613		- parent = parent.parentNode;
614		- }
615		- return NodeFilter.FILTER_ACCEPT;
616		- }
617		- });
	707	+ $result = $factory->make()->chat( $messages, $this->ai_chat_options() );
618	708
619		- const textNodes = [];
620		- let current;
621		- while ((current = walker.nextNode())) {
622		- textNodes.push(current);
623		- }
	709	+ if ( is_wp_error( $result ) ) {
	710	+ return array(
	711	+ 'success' => false,
	712	+ 'error' => $result->get_error_message(),
	713	+ 'model' => $model
	714	+ );
	715	+ }
624	716
625		- textNodes.forEach(function (textNode) {
626		- pattern.lastIndex = 0;
627		- if (!pattern.test(textNode.nodeValue)) {
628		- return;
629		- }
630		- pattern.lastIndex = 0;
631		- const fragmentHost = document.createElement('span');
632		- fragmentHost.innerHTML = escapeHtml(textNode.nodeValue).replace(pattern, '<span class="highlight">$1</span>');
633		- const parent = textNode.parentNode;
634		- while (fragmentHost.firstChild) {
635		- parent.insertBefore(fragmentHost.firstChild, textNode);
636		- }
637		- parent.removeChild(textNode);
638		- });
639		- }
	717	+ $usage = isset( $result['usage'] ) && is_array( $result['usage'] ) ? $result['usage'] : array();
640	718
641		- return container.innerHTML;
642		- }
	719	+ return array(
	720	+ 'success' => true,
	721	+ 'content' => $result['content'],
	722	+ 'model' => isset( $result['model'] ) ? $result['model'] : $model,
	723	+ 'prompt_tokens' => isset( $usage['prompt_tokens'] ) ? $usage['prompt_tokens'] : null,
	724	+ 'completion_tokens' => isset( $usage['completion_tokens'] ) ? $usage['completion_tokens'] : null,
	725	+ 'total_tokens' => isset( $usage['total_tokens'] ) ? $usage['total_tokens'] : null,
	726	+ 'finish_reason' => isset( $result['finish_reason'] ) ? $result['finish_reason'] : null
	727	+ );
	728	+ }
643	729
644		- function getBodyContent(htmlString) {
645		- if (!htmlString) {
646		- return '';
647		- }
648		- const cleaned = stripCodeFences(htmlString);
649		- const match = cleaned.match(/<body[^>]*>([\s\S]*?)<\/body>/i);
650		- if (match) {
651		- return match[1].replace(/<p>\s+/g, '<p>');
652		- }
653		- return cleaned;
654		- }
	730	+ /**
	731	+ * Generic chat completion using the active Write-with-AI platform/model/token
	732	+ * settings. Shared by lightweight, plain-text generators (glossary definitions,
	733	+ * FAQ answers, outlines) that need the same dynamic model as Write with AI /
	734	+ * AI Edit but a caller-supplied system prompt instead of the doc-authoring HTML
	735	+ * prompt. Routes through ProviderFactory so every provider is supported.
	736	+ *
	737	+ * @param string $user_prompt The user message.
	738	+ * @param string $system_prompt Optional system message (omitted when empty).
	739	+ * @param array $extra_system Optional extra system messages.
	740	+ * @return array { success:bool, content?:string, error?:string, model:string, *_tokens?:int }
	741	+ */
	742	+ public function generate_text( $user_prompt, $system_prompt = '', $extra_system = array() ) {
	743	+ $factory = new ProviderFactory( $this->settings );
	744	+ $model = $factory->active_model();
655	745
	746	+ $messages = array();
	747	+ if ( $system_prompt !== '' ) {
	748	+ $messages[] = array( 'role' => 'system', 'content' => $system_prompt );
	749	+ }
	750	+ foreach ( $this->normalize_extra_system( $extra_system ) as $extra ) {
	751	+ $messages[] = $extra;
	752	+ }
	753	+ $messages[] = array( 'role' => 'user', 'content' => $user_prompt );
656	754
	755	+ $result = $factory->make()->chat( $messages, $this->ai_chat_options() );
657	756
658		- function insertContentToEditor(title, htmlContent, isOverwrite, keywords) {
	757	+ if ( is_wp_error( $result ) ) {
	758	+ return array(
	759	+ 'success' => false,
	760	+ 'error' => $result->get_error_message(),
	761	+ 'model' => $model
	762	+ );
	763	+ }
659	764
660		- const {
661		- dispatch,
662		- select
663		- } = wp.data;
	765	+ $usage = isset( $result['usage'] ) && is_array( $result['usage'] ) ? $result['usage'] : array();
664	766
665		- htmlContent = getBodyContent(htmlContent);
666		- htmlContent = removeFirstHeading(htmlContent);
667		- htmlContent = wrapwithHeighlight(htmlContent, keywords);
668		- const isPostContent = `<?php echo isset( $_GET['post'] ) ? esc_html( get_the_content( absint( wp_unslash( $_GET['post'] ) ) ) ) : ''; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- read-only post id from URL. ?>`;
	767	+ return array(
	768	+ 'success' => true,
	769	+ 'content' => $result['content'],
	770	+ 'model' => isset( $result['model'] ) ? $result['model'] : $model,
	771	+ 'prompt_tokens' => isset( $usage['prompt_tokens'] ) ? $usage['prompt_tokens'] : null,
	772	+ 'completion_tokens' => isset( $usage['completion_tokens'] ) ? $usage['completion_tokens'] : null,
	773	+ 'total_tokens' => isset( $usage['total_tokens'] ) ? $usage['total_tokens'] : null,
	774	+ 'finish_reason' => isset( $result['finish_reason'] ) ? $result['finish_reason'] : null
	775	+ );
	776	+ }
669	777
670		- const blocks = wp.blocks.rawHandler({
671		- HTML: htmlContent
672		- });
	778	+ /**
	779	+ * Generate a chat completion with a caller-supplied system + user prompt.
	780	+ * Mirrors generate_openai_response_ai_edit() (same model/token floor/timeout
	781	+ * handling and return shape) but does NOT force the documentation-writer
	782	+ * system prompt, so callers such as the Docs AI Suite (taxonomy suggestions,
	783	+ * excerpts) can supply task-appropriate instructions. Routes through the active
	784	+ * provider via ProviderFactory.
	785	+ *
	786	+ * @param string $system_prompt System instruction for the model.
	787	+ * @param string $user_prompt User message / content payload.
	788	+ * @param float|null $temperature Optional sampling temperature (ignored for gpt-5*).
	789	+ * @return array{success:bool,content?:string,error?:string,model:string,...}
	790	+ */
	791	+ public function generate_openai_response_raw( $system_prompt, $user_prompt, $temperature = null ) {
	792	+ $factory = new ProviderFactory( $this->settings );
	793	+ $model = $factory->active_model();
673	794
674		- dispatch('core/editor').editPost({
675		- title: title,
676		- });
	795	+ $messages = array(
	796	+ array( 'role' => 'system', 'content' => $system_prompt ),
	797	+ array( 'role' => 'user', 'content' => $user_prompt )
	798	+ );
677	799
678		- const selectedBlockClientId = select('core/block-editor').getSelectedBlockClientId();
	800	+ $result = $factory->make()->chat( $messages, $this->ai_chat_options( null, $temperature ) );
679	801
680		- const allBlocks = select('core/block-editor').getBlocks();
	802	+ if ( is_wp_error( $result ) ) {
	803	+ return array(
	804	+ 'success' => false,
	805	+ 'error' => $result->get_error_message(),
	806	+ 'model' => $model
	807	+ );
	808	+ }
681	809
682		- if (isOverwrite || isPostContent === '') {
	810	+ $usage = isset( $result['usage'] ) && is_array( $result['usage'] ) ? $result['usage'] : array();
683	811
684		- allBlocks.forEach((block) => {
685		- dispatch('core/block-editor').removeBlock(block.clientId);
686		- });
687		- dispatch('core/block-editor').insertBlocks(blocks);
	812	+ return array(
	813	+ 'success' => true,
	814	+ 'content' => $result['content'],
	815	+ 'model' => isset( $result['model'] ) ? $result['model'] : $model,
	816	+ 'prompt_tokens' => isset( $usage['prompt_tokens'] ) ? $usage['prompt_tokens'] : null,
	817	+ 'completion_tokens' => isset( $usage['completion_tokens'] ) ? $usage['completion_tokens'] : null,
	818	+ 'total_tokens' => isset( $usage['total_tokens'] ) ? $usage['total_tokens'] : null,
	819	+ 'finish_reason' => isset( $result['finish_reason'] ) ? $result['finish_reason'] : null
	820	+ );
	821	+ }
688	822
689		- } else if (selectedBlockClientId) {
690		- const selectedIndex = select('core/block-editor').getBlockIndex(selectedBlockClientId);
691		- dispatch('core/block-editor').insertBlocks(blocks, selectedIndex + 1);
692		- } else {
693		- dispatch('core/block-editor').insertBlocks(blocks);
694		- }
	823	+ public function generate_openai_content_callback() {
	824	+ // Verify the nonce
	825	+ $ai_nonce = isset( $_POST['ai_nonce'] ) ? sanitize_text_field( wp_unslash( $_POST['ai_nonce'] ) ) : '';
	826	+ if ( ! wp_verify_nonce( $ai_nonce, 'generate_openai_content_nonce' ) ) {
	827	+ wp_send_json_error( 'Invalid nonce' );
	828	+ wp_die();
	829	+ }
695	830
696		- closeWriteWithAIForm('afterInsertContent');
697		- }
	831	+ if ( ! current_user_can( 'edit_posts' ) ) {
	832	+ wp_send_json_error( 'Insufficient permissions' );
	833	+ wp_die();
	834	+ }
698	835
699		- function closeWriteWithAIForm(after) {
700		- jQuery('.betterdocs-ai-autowrite-form-container').addClass('hidden');
701		- jQuery('#betterdocs-ai-autowrite-form-container-overlay').addClass('hidden');
702		- jQuery('.betterdocs-ai-button').addClass('regenerate-btn');
703		- }
	836	+ $prompt = isset( $_POST['prompt'] ) ? sanitize_text_field( wp_unslash( $_POST['prompt'] ) ) : '';
	837	+ $keywords = isset( $_POST['keywords'] ) ? sanitize_text_field( wp_unslash( $_POST['keywords'] ) ) : '';
704	838
	839	+ $ai_instance = new WriteWithAI( $this->settings );
705	840
706		- document.addEventListener('DOMContentLoaded', function() {
	841	+ $generated_content = $ai_instance->generate_openai_response( $prompt, $keywords );
707	842
708		- // Subscribe to changes in the editor state
709		- let previousDocsTitle = '';
	843	+ // Count a successful generation (skip obvious upstream errors).
	844	+ if ( is_string( $generated_content ) && $generated_content !== '' && strpos( $generated_content, 'Error:' ) !== 0 ) {
	845	+ $post_id = isset( $_POST[ 'post_id' ] ) ? intval( $_POST[ 'post_id' ] ) : 0; //phpcs:ignore
	846	+ AIUsage::record( 'write_with_ai', $post_id );
	847	+ }
710	848
711		- wp.data.subscribe(() => {
712		- // Get the updated post title
713		- // const docsTitle = wp.data.select('core/editor').getEditedPostAttribute('title');
714		-
715		- const docsTitle = wp.data.select('core/editor').getEditedPostAttribute('title');
716		-
717		- // Check if the title has changed
718		- if (docsTitle !== previousDocsTitle) {
719		- let currentKeywords = jQuery('#betterdocs-ai-keyword').val();
720		- if (!currentKeywords) {
721		- currentKeywords = '{Documentation Keywords}';
722		- }
723		- const currentPrompt = `Write documentation for '${docsTitle?docsTitle:'{Documentation title}'}'. Cover these topics in depth: ${currentKeywords}. Use headings, paragraphs, lists, links, code blocks, and tables wherever they help the reader. Follow the output-format rules in the system instructions.`;
724		- jQuery('#betterdocs-ai-content').val(currentPrompt);
725		-
726		- jQuery('#betterdocs-ai-title').val(docsTitle);
727		-
728		- previousDocsTitle = docsTitle;
729		- }
730		-
731		-
732		- });
733		- });
734		-
735		-
736		- // This example assumes that this code is executed when your script is loaded.
737		- // You may want to place it within the appropriate context in your application.
738		-
739		-
740		- function writeWithAIForm() {
741		-
742		- let title = `<?php echo isset( $_GET['post'] ) ? esc_html( get_the_title( absint( wp_unslash( $_GET['post'] ) ) ) ) : ''; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- read-only post id from URL. ?>`;
743		- if (docsTitle) {
744		- title = docsTitle;
745		- }
746		-
747		- const promtTitle = `<?php echo isset( $_GET['post'] ) ? esc_html( get_the_title( absint( wp_unslash( $_GET['post'] ) ) ) ) : '{Documentation Title}'; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- read-only post id from URL. ?>`;
748		- const titlePlaceholder = "<?php echo esc_attr__( 'Enter a descriptive title for your documentation.', 'betterdocs' ); ?>";
749		- const keywords = "<?php echo esc_attr( '{Documentation Keywords}' ); ?>";
750		- const keywordsPlaceholder = "<?php echo esc_attr__( 'Add keywords to generate precise & relevant documentation (comma-separated).', 'betterdocs' ); ?>";
751		- const language = "<?php echo esc_attr( 'English' ); ?>";
752		- const titleLabel = "<?php echo esc_html__( 'Documentation Title:', 'betterdocs' ); ?>";
753		- const keywordLabel = "<?php echo esc_html__( 'Keywords:', 'betterdocs' ); ?>";
754		- const secLabel = "<?php echo esc_html__( '# of Sections:', 'betterdocs' ); ?>";
755		- const paraLabel = "<?php echo esc_html__( '# of Paragraph Per Section: ', 'betterdocs' ); ?>";
756		- const langLabel = "<?php echo esc_html__( 'Language:', 'betterdocs' ); ?>";
757		- const promtLabel = "<?php echo esc_html__( 'Prompt:', 'betterdocs' ); ?>";
758		- const promtBtnLabel = "<?php echo esc_html__( 'Generate Prompt', 'betterdocs' ); ?>";
759		- let promtArticleLabel = "<?php echo esc_html__( 'Generate Doc', 'betterdocs' ); ?>";
760		- const checkboxLabel = "<?php echo esc_html__( 'Overwrite your existing Doc:', 'betterdocs' ); ?>";
761		- const nonce = "<?php echo esc_attr( wp_create_nonce( 'generate_openai_content_nonce' ) ); ?>";
762		-
763		- <?php
764		- $is_valid = $this->get_api_key();
765		- $disable_field = 'disabled-input-field';
766		- if ( $this->get_api_key() ) {
767		- $disable_field = '';
768		- }
769		- ?>
770		-
771		- let hiddenClass = 'hidden';
772		- let newDocPageAtt = 'data-new-doc-page="true"';
773		- const isPostContent = `<?php echo isset( $_GET['post'] ) ? esc_html( get_the_content( absint( wp_unslash( $_GET['post'] ) ) ) ) : ''; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- read-only post id from URL. ?>`;
774		-
775		- if (isPostContent !== '') {
776		- hiddenClass = '';
777		- newDocPageAtt = '';
778		- }
779		-
780		-
781		- // const prompt = `Make a knowledge base article with html heading tags about [${title}] in [${language}]. Here is some topic to describe the article: [${keywords}]. and wrap the content with p tag also add a span tag with a class named 'highlight' to all the heading and topic tags in the article.`;
782		-
783		- // const prompt = `Create a details knowledge base article in [${language}] about [${title}] with html heading tags. Here is some topic to describe the article: [${keywords}]. And wrap the content with p tag also add a span tag with a class named 'highlight' to all the heading and topic tags in the article.`;
784		-
785		- const prompt = `Write documentation for '${promtTitle}'. Cover these topics in depth: ${keywords}. Use headings, paragraphs, lists, links, code blocks, and tables wherever they help the reader. Follow the output-format rules in the system instructions.`;
786		-
787		- const aiIcon = `<svg xmlns="http://www.w3.org/2000/svg" width="21" height="20" viewBox="0 0 21 20" fill="none">
788		- <g clip-path="url(#clip0_2460_1729)">
789		- <path d="M10.3956 18.8608L6.10991 19.2322L6.48134 14.9465L15.3956 6.08936C15.5287 5.95329 15.6876 5.84518 15.863 5.77137C16.0384 5.69756 16.2268 5.65954 16.4171 5.65954C16.6074 5.65954 16.7957 5.69756 16.9711 5.77137C17.1466 5.84518 17.3054 5.95329 17.4385 6.08936L19.2528 7.91793C19.3865 8.05083 19.4926 8.20886 19.565 8.38293C19.6375 8.55701 19.6747 8.74368 19.6747 8.93221C19.6747 9.12075 19.6375 9.30742 19.565 9.48149C19.4926 9.65557 19.3865 9.8136 19.2528 9.9465L10.3956 18.8608ZM1.7042 5.67507C1.20277 5.58793 1.20277 4.86793 1.7042 4.78079C2.59203 4.62626 3.41374 4.21085 4.06456 3.58751C4.71538 2.96417 5.16583 2.16113 5.35848 1.28079L5.38848 1.14221C5.49705 0.647929 6.20277 0.643643 6.31705 1.13936L6.35277 1.29936C6.55248 2.17579 7.0067 2.97367 7.65837 3.59281C8.31005 4.21195 9.13013 4.62475 10.0156 4.77936C10.5199 4.86507 10.5199 5.58936 10.0156 5.67793C9.13005 5.83218 8.3098 6.24465 7.65787 6.86354C7.00594 7.48243 6.5514 8.28014 6.35134 9.1565L6.3142 9.31793C6.20134 9.81221 5.49563 9.80936 5.38705 9.31364L5.35848 9.17507C5.16564 8.29433 4.71476 7.49102 4.06339 6.86764C3.41202 6.24426 2.58969 5.82908 1.70134 5.67507H1.7042Z" stroke="white" stroke-width="1.42857" stroke-linecap="round" stroke-linejoin="round"/>
790		- </g>
791		- <defs>
792		- <clipPath id="clip0_2460_1729">
793		- <rect width="20" height="20" fill="white" transform="translate(0.5)"/>
794		- </clipPath>
795		- </defs>
796		- </svg>`;
797		-
798		-
799		-
800		- return `
801		- <div class="betterdocs-ai-autowrite-form-container hidden" ${newDocPageAtt}>
802		- <div class="betterdocs-ai-autowrite-form-content">
803		- <div class="betterdocs-ai-autowrite-top-part">
804		- <div class="autowrite-heading">
805		- <span class="autowrite-icon">
806		- <svg xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32" fill="none">
807		- <g clip-path="url(#clip0_2453_20882)">
808		- <path d="M15.8322 30.1765L8.97508 30.7708L9.56936 23.9136L23.8322 9.74219C24.0451 9.52448 24.2993 9.3515 24.58 9.23341C24.8606 9.11531 25.162 9.05447 25.4665 9.05447C25.771 9.05447 26.0724 9.11531 26.3531 9.23341C26.6337 9.3515 26.8879 9.52448 27.1008 9.74219L30.0037 12.6679C30.2176 12.8805 30.3874 13.1334 30.5033 13.4119C30.6192 13.6904 30.6788 13.9891 30.6788 14.2908C30.6788 14.5924 30.6192 14.8911 30.5033 15.1696C30.3874 15.4481 30.2176 15.701 30.0037 15.9136L15.8322 30.1765ZM1.92593 9.07933C1.12365 8.9399 1.12365 7.7879 1.92593 7.64848C3.34647 7.40124 4.6612 6.73658 5.70251 5.73923C6.74382 4.74188 7.46455 3.45703 7.77279 2.04848L7.82079 1.82676C7.99451 1.03591 9.12365 1.02905 9.30651 1.82219L9.36365 2.07819C9.68319 3.48048 10.4099 4.75709 11.4526 5.74772C12.4953 6.73834 13.8074 7.39881 15.2242 7.64619C16.0311 7.78333 16.0311 8.94219 15.2242 9.0839C13.8073 9.33071 12.4949 9.99066 11.4518 10.9809C10.4087 11.9711 9.68146 13.2474 9.36136 14.6496L9.30193 14.9079C9.12136 15.6988 7.99222 15.6942 7.81851 14.901L7.77279 14.6793C7.46424 13.2702 6.74284 11.9849 5.70064 10.9874C4.65845 9.99003 3.34273 9.32574 1.92136 9.07933H1.92593Z" stroke="#E31B54" stroke-width="2.28571" stroke-linecap="round" stroke-linejoin="round"/>
809		- </g>
810		- <defs>
811		- <clipPath id="clip0_2453_20882">
812		- <rect width="32" height="32" fill="white"/>
813		- </clipPath>
814		- </defs>
815		- </svg>
816		- </span>
817		- <h1><?php echo esc_html__( 'Write Documentation with BetterDocs AI', 'betterdocs' ); ?></h1>
818		-
819		- </div>
820		- <div class="autowrite-subheadding">
821		- <p><?php echo esc_html__( 'Generate documentation effortlessly with BetterDocs AI. Simply input your doc title, keywords, prompt and let the system automatically generate comprehensive documentation tailored to your needs.', 'betterdocs' ); ?></p>
822		- </div>
823		-
824		- <?php if ( empty( $this->get_api_key() ) ): ?>
825		- <div id="betterdocs-ai-message">
826		- <div class="warning-message">
827		- <span class="dashicons dashicons-warning"></span>
828		- <div>
829		- <?php echo wp_kses_post( 'Please Insert your <a target="_blank" href="' . esc_url( admin_url( 'admin.php?page=betterdocs-settings#betterdocs-ai' ) ) . '">OpenAI API Key</a> to use this Write with AI feature.', 'betterdocs' ); ?>
830		- </div>
831		- </div>
832		-
833		- </div>
834		- <?php endif; ?>
835		-
836		- <div class="form-group hidden">
837		- <div id="betterdocs-ai-error-message"></div>
838		- </div>
839		-
840		- </div>
841		- <form id="betterdocs-ai-form" class="<?php echo esc_attr( $disable_field ); ?>">
842		- <div class="form-inner-content">
843		- <div class="form-group">
844		- <label for="betterdocs-ai-title">${titleLabel}</label>
845		- <input type="text" placeholder="${titlePlaceholder}" id="betterdocs-ai-title" name="betterdocs-ai-title" required value="${title}">
846		- </div>
847		-
848		- <div class="form-group">
849		- <label for="betterdocs-ai-keyword">${keywordLabel}</label>
850		- <input type="text" placeholder="${keywordsPlaceholder}" id="betterdocs-ai-keyword" name="betterdocs-ai-keyword" required>
851		- </div>
852		-
853		- <div class="form-group prompt-field">
854		- <label for="betterdocs-ai-content">${promtLabel}</label>
855		- <textarea id="betterdocs-ai-content" name="betterdocs-ai-content" rows="6" required>${prompt}</textarea>
856		- <p class="input-description"><?php echo esc_html__( 'Ensure you include a clear and detailed prompt to receive the desired output. Follow the guidelines provided for the best results.', 'betterdocs' ); ?></p>
857		- </div>
858		- <div class="betterdocs-ai-checkbox-field ${hiddenClass}">
859		- <div class="form-group">
860		- <div class="checkbox-field-label">
861		- <label for="betterdocs-ai-checkbox">${checkboxLabel}</label>
862		- <p class="input-description"><?php echo esc_html__( 'Overwrite the existing doc with the AI Generated Content. If Disabled, new AI Generated content will be added in the section you are currently editing.', 'betterdocs' ); ?></p>
863		- </div>
864		-
865		- <div class="checkbox-input-field">
866		- <input type="checkbox" id="betterdocs-ai-checkbox" name="betterdocs-ai-checkbox" data-overwrite="false">
867		- <label for="betterdocs-ai-checkbox" class="toggle-label"></label>
868		- </div>
869		- </div>
870		- </div>
871		-
872		- <input type="hidden" name="ai_nonce" value="${nonce}" />
873		- </div>
874		- <div class="generate-button-container">
875		- <button type="submit" class="generate-btn" onclick="generateArticle(event)">
876		- ${aiIcon} <span>${promtArticleLabel}</span>
877		- </button>
878		- </div>
879		- </form>
880		- </div>
881		-
882		- <div class="bd-close-button" onclick="closeWriteWithAIForm('afterClosedClicked')">✕</div>
883		- </div>
884		-
885		- <div id="betterdocs-ai-autowrite-form-container-overlay" class="hidden"></div>
886		- `;
887		- }
888		-
889		- jQuery(document).on('click', '.regenerate-btn', function() {
890		- jQuery('.betterdocs-ai-autowrite-form-container').removeClass('hidden');
891		- jQuery('#betterdocs-ai-autowrite-form-container-overlay').removeClass('hidden');
892		- });
893		-
894		- jQuery(document).ready(function($) {
895		- // Check if we are in the Edit Post screen
896		-
897		- if ($('.block-editor-page').length > 0) {
898		-
899		- const bdAIButton = $(`<div class="betterdocs-ai-button-container">
900		- <button class="betterdocs-ai-button">
901		- <img src="<?php echo esc_url( BETTERDOCS_ABSURL . 'assets/admin/images/betterdocs-icon-white.png' ); ?>" alt="<?php echo esc_attr__( 'betterdocs icon', 'betterdocs' ); ?>" />
902		- <span><?php echo esc_html__( 'Write with AI', 'betterdocs' ); ?></span>
903		- </button>
904		- </div>`);
905		-
906		-
907		- const closeBtn = $('bd-close-button');
908		-
909		- const formHtml = writeWithAIForm();
910		- $(formHtml).addClass('hidden');
911		-
912		- $(document).on('click', '.betterdocs-ai-button', function() {
913		- $('.betterdocs-ai-autowrite-form-container').removeClass('hidden');
914		- $('#betterdocs-ai-autowrite-form-container-overlay').removeClass('hidden');
915		- });
916		-
917		- // $(document).on('click', '.betterdocs-ai-button', function() {
918		- if ($('.betterdocs-ai-autowrite-form-container').length < 1) {
919		- $('body').append(formHtml);
920		-
921		- // Add event listeners to input elements
922		- document.getElementById('betterdocs-ai-title').addEventListener('input', updateTextarea);
923		- document.getElementById('betterdocs-ai-keyword').addEventListener('input', updateTextarea);
924		- // document.getElementById('bd-autowrtite-content-section').addEventListener('change', updateTextarea);
925		- // document.getElementById('bd-autowrtite-content-paragraph').addEventListener('change', updateTextarea);
926		- // document.getElementById('betterdocs-ai-language').addEventListener('input', updateTextarea);
927		- }
928		- // });
929		- const btn = document.createElement('button');
930		- wp.data.subscribe(function() {
931		- setTimeout(() => {
932		- if ($('.edit-post-header__settings').length > 0) {
933		- $('.edit-post-header__settings').prepend(bdAIButton);
934		- } else if ($('.editor-header__settings').length > 0) {
935		- $('.editor-header__settings').prepend(bdAIButton);
936		- }
937		-
938		- }, 1);
939		- });
940		- }
941		-
942		- $(document).on('click', '#betterdocs-ai-autowrite-form-container-overlay', function() {
943		- closeWriteWithAIForm();
944		- });
945		- });
946		- </script>
947		-
948		- <style>
949		- .hidden {
950		- display: none !important;
951		- }
952		-
953		- .disabled-input-field input,
954		- .disabled-input-field textarea,
955		- .disabled-input-field button,
956		- .disabled-input-field label {
957		- pointer-events: none;
958		- opacity: 0.6;
959		- }
960		-
961		- .disabled-input-field label {
962		- opacity: 1;
963		- }
964		-
965		- .betterdocs-ai-button-container {
966		- margin-left: 10px;
967		- white-space: nowrap;
968		- max-width: 145px;
969		- }
970		-
971		- .autowrite-icon {
972		- display: flex;
973		- align-items: center;
974		- width: 55px;
975		- height: 55px;
976		- background: #FFE4E8;
977		- justify-content: center;
978		- border-radius: 50px;
979		- }
980		-
981		- .autowrite-icon svg {
982		- width: 28px;
983		- height: 28px;
984		- }
985		-
986		- .autowrite-heading {
987		- display: flex;
988		- gap: 10px;
989		- align-items: center;
990		- }
991		-
992		- .autowrite-heading h1 {
993		- color: #1D2939;
994		- font-family: 'IBM Plex Sans', sans-serif;
995		- font-size: 24px;
996		- font-style: normal;
997		- font-weight: 500;
998		- line-height: normal;
999		- margin: 0;
1000		- }
1001		-
1002		- .autowrite-heading p,
1003		- form#betterdocs-ai-form p {
1004		- margin: 0;
1005		- margin-bottom: 24px;
1006		- }
1007		-
1008		- .autowrite-subheadding p {
1009		- font-family: 'IBM Plex Sans', sans-serif;
1010		- font-size: 14px;
1011		- }
1012		-
1013		- .prompt-field .input-description {
1014		- margin: 0 !important;
1015		- }
1016		-
1017		- .autowrite-heading p {
1018		- color: #475467;
1019		- font-family: 'IBM Plex Sans', sans-serif;
1020		- font-size: 16px;
1021		- font-style: normal;
1022		- font-weight: 400;
1023		- }
1024		-
1025		- .betterdocs-ai-button {
1026		- background: #00B884;
1027		- border: 1px solid transparent;
1028		- color: #fff;
1029		- box-shadow: none;
1030		- font-size: 12px;
1031		- margin-right: 8px;
1032		- padding: 0px 15px;
1033		- cursor: pointer;
1034		- border-radius: 3px;
1035		- height: 38px;
1036		- display: flex;
1037		- align-items: center;
1038		- justify-content: space-between;
1039		- gap: 10px;
1040		- width: 135px;
1041		- }
1042		-
1043		- .betterdocs-ai-button img {
1044		- width: 20px;
1045		- height: 20px;
1046		- }
1047		-
1048		- .betterdocs-ai-autowrite-form-container {
1049		- position: fixed;
1050		- top: 50%;
1051		- left: 50%;
1052		- transform: translate(-50%, -50%);
1053		- z-index: 100001;
1054		- width: 650px;
1055		- margin: 0 auto;
1056		- background-color: #fff;
1057		- border-radius: 5px;
1058		- font-family: 'IBM Plex Sans', sans-serif;
1059		- /* max-height: 600px; */
1060		- max-height: 750px;
1061		- max-width: 100%;
1062		- }
1063		-
1064		- .betterdocs-ai-autowrite-form-content {
1065		- background-color: #fff;
1066		- padding: 20px 0px;
1067		- border-radius: 5px;
1068		- padding-bottom: 0;
1069		- overflow: auto;
1070		- /* max-height: 700px; */
1071		- height: 100%;
1072		-
1073		- }
1074		-
1075		- .betterdocs-ai-autowrite-top-part {
1076		- /* padding-bottom: 20px; */
1077		- border-bottom: 1px solid #ddd;
1078		- /* margin-bottom: 20px; */
1079		- padding: 0 40px;
1080		- }
1081		-
1082		- #betterdocs-ai-form {
1083		- display: flex;
1084		- flex-direction: column;
1085		- /* height: 100%; */
1086		- overflow: hidden;
1087		- }
1088		-
1089		- .form-inner-content {
1090		- overflow: auto;
1091		- height: 100%;
1092		- max-height: 435px;
1093		- /* max-height: 330px; */
1094		- padding: 0 40px;
1095		- }
1096		-
1097		- .form-inner-content>.form-group {
1098		- margin-top: 20px;
1099		- }
1100		-
1101		-
1102		- #betterdocs-ai-autowrite-form-container-overlay {
1103		- position: fixed;
1104		- top: 0;
1105		- right: 0;
1106		- bottom: 0;
1107		- left: 0;
1108		- background-color: rgba(0, 0, 0, .7);
1109		- z-index: 100000;
1110		- }
1111		-
1112		-
1113		-
1114		- #betterdocs-ai-form {
1115		- display: flex;
1116		- flex-direction: column;
1117		- }
1118		-
1119		- #betterdocs-ai-form .form-group {
1120		- margin-bottom: 24px;
1121		- }
1122		-
1123		- #betterdocs-ai-form .bd-aiform-flex {
1124		- display: flex;
1125		- justify-content: space-between;
1126		- }
1127		-
1128		- #betterdocs-ai-form label {
1129		- font-weight: 600;
1130		- margin-bottom: 5px;
1131		- display: block;
1132		- font-size: 14px;
1133		- line-height: 20px;
1134		- }
1135		-
1136		- select#bd-autowrtite-content-section,
1137		- #bd-autowrtite-content-paragraph,
1138		- #betterdocs-ai-language {
1139		- width: 150px !important;
1140		- height: 40px;
1141		- }
1142		-
1143		- #betterdocs-ai-form input[type="text"],
1144		- #betterdocs-ai-form textarea {
1145		- width: 100%;
1146		- padding: 8px;
1147		- box-sizing: border-box;
1148		- border: 1px solid #D0D5DD;
1149		- border-radius: 4px;
1150		- color: #667085;
1151		- font-weight: 400;
1152		- }
1153		-
1154		- /* Override autofill text color */
1155		- #betterdocs-ai-form input:-webkit-autofill {
1156		- -webkit-text-fill-color: #667085 !important;
1157		- }
1158		-
1159		- #betterdocs-ai-form input::placeholder {
1160		- color: #acb2bf;
1161		- }
1162		-
1163		- #betterdocs-ai-form textarea {
1164		- color: #667085;
1165		- }
1166		-
1167		- #betterdocs-ai-form input:focus,
1168		- #betterdocs-ai-form textarea:focus {
1169		- box-shadow: none;
1170		- }
1171		-
1172		- #betterdocs-ai-form textarea:focus {
1173		- color: inherit;
1174		- box-shadow: none;
1175		-
1176		- }
1177		-
1178		-
1179		- .generate-button-container {
1180		- position: sticky;
1181		- bottom: 0px;
1182		- width: 100%;
1183		- margin-left: 0;
1184		- z-index: 999999 !important;
1185		- padding: 20px 0;
1186		- display: flex;
1187		- justify-content: end;
1188		- border-top: 1px solid #ddd;
1189		- background: white;
1190		- border-bottom-right-radius: 5px;
1191		- border-bottom-left-radius: 5px;
1192		- }
1193		-
1194		- .generate-button-container button {
1195		- display: flex;
1196		- gap: 8px;
1197		- align-items: center;
1198		- justify-content: center;
1199		- opacity: 1 !important;
1200		- margin-right: 40px;
1201		- }
1202		-
1203		- .input-description {
1204		- font-size: 14px;
1205		- color: #667085;
1206		- }
1207		-
1208		- .betterdocs-ai-checkbox-field .form-group {
1209		- display: flex;
1210		- align-items: start;
1211		- /* gap: 200px; */
1212		- justify-content: space-between;
1213		- }
1214		-
1215		- .betterdocs-ai-checkbox-field input {
1216		- width: 20px;
1217		- height: 20px;
1218		- }
1219		-
1220		- /* Add this CSS to your existing stylesheet or create a new one */
1221		-
1222		- .betterdocs-ai-checkbox-field {
1223		- position: relative;
1224		- font-size: 16px;
1225		- /* Adjust font size as needed */
1226		- }
1227		-
1228		- .betterdocs-ai-checkbox-field input[type=checkbox]:checked::before {
1229		- padding: 2px;
1230		- }
1231		-
1232		- .betterdocs-ai-checkbox-field input[type=checkbox] {
1233		- border: 1px solid #00b884;
1234		- }
1235		-
1236		- .betterdocs-ai-checkbox-field input[type=checkbox]:checked::before {
1237		- content: '\2713';
1238		- color: #00b884;
1239		- }
1240		-
1241		- /* Change the default checkbox color */
1242		- .betterdocs-ai-checkbox-field input[type="checkbox"]:hover {
1243		- -webkit-appearance: none;
1244		- /* WebKit/Blink Browsers */
1245		- -moz-appearance: none;
1246		- /* Firefox */
1247		- appearance: none;
1248		- border: 1px solid #00b884;
1249		- /* Set the height of the checkbox */
1250		- /* Optional: Round the corners */
1251		- outline: none;
1252		- /* Remove the default outline */
1253		- }
1254		-
1255		- /* Style the checked state */
1256		- .betterdocs-ai-checkbox-field input[type="checkbox"]:checked {
1257		- /* Set the background color when checked */
1258		- border: 1px solid #00b884;
1259		- /* Set the border color when checked */
1260		- }
1261		-
1262		- .checkbox-field-label {
1263		- width: calc(100% - 150px);
1264		- }
1265		-
1266		- .checkbox-field-label p {
1267		- margin: 0 !important;
1268		- }
1269		-
1270		- .checkbox-input-field {
1271		- width: 300px;
1272		- text-align: right;
1273		- }
1274		-
1275		- .checkbox-input-field {
1276		- position: relative;
1277		- display: inline-block;
1278		- width: 60px;
1279		- height: 34px;
1280		- }
1281		-
1282		- .checkbox-input-field input {
1283		- display: none;
1284		- }
1285		-
1286		- .toggle-label {
1287		- position: absolute;
1288		- cursor: pointer;
1289		- top: 0;
1290		- left: 0;
1291		- right: 0;
1292		- bottom: 0;
1293		- background-color: #ccc;
1294		- border-radius: 34px;
1295		- transition: background-color 0.3s;
1296		- scale: .9;
1297		- width: 52px;
1298		- height: 26px;
1299		- }
1300		-
1301		-
1302		- .checkbox-input-field input:checked+.toggle-label {
1303		- background-color: #00b884;
1304		- }
1305		-
1306		- .toggle-label:after {
1307		- content: "";
1308		- position: absolute;
1309		- height: 22px;
1310		- width: 22px;
1311		- left: 2px;
1312		- bottom: 2px;
1313		- background-color: white;
1314		- border-radius: 50%;
1315		- transition: transform 0.3s;
1316		- }
1317		-
1318		- .checkbox-input-field input:checked+.toggle-label:after {
1319		- transform: translateX(26px);
1320		- }
1321		-
1322		- .generate-btn,
1323		- .prompt-btn,
1324		- .keep-btn {
1325		- background-color: #00b884;
1326		- color: #fff;
1327		- border: none;
1328		- padding: 10px 15px;
1329		- text-align: center;
1330		- text-decoration: none;
1331		- display: inline-block;
1332		- font-size: 16px;
1333		- cursor: pointer;
1334		- border-radius: 4px;
1335		- }
1336		-
1337		- .generate-btn[disabled] {
1338		- cursor: unset;
1339		- }
1340		-
1341		- .bd-close-button {
1342		- cursor: pointer;
1343		- font-size: 18px;
1344		- font-weight: bold;
1345		- float: right;
1346		- position: absolute;
1347		- top: -16px;
1348		- border-radius: 50%;
1349		- background: #ffffff;
1350		- width: 40px;
1351		- height: 40px;
1352		- display: flex;
1353		- align-items: center;
1354		- justify-content: center;
1355		- color: #281617;
1356		- right: -42px;
1357		- }
1358		-
1359		- div#betterdocs-ai-error-message,
1360		- #betterdocs-ai-message {
1361		- font-size: 14px;
1362		- font-family: 'IBM Plex Sans', sans-serif;
1363		- color: #667085;
1364		- margin-bottom: 20px;
1365		- }
1366		-
1367		- div#betterdocs-ai-error-message a,
1368		- #betterdocs-ai-message a {
1369		- text-decoration: none;
1370		- font-weight: 600;
1371		- }
1372		-
1373		- div#betterdocs-ai-error-message a:focus,
1374		- #betterdocs-ai-message a:focus {
1375		- outline: none;
1376		- box-shadow: none;
1377		- color: #2271b1;
1378		- }
1379		-
1380		- #betterdocs-ai-message span {
1381		- color: #d63638;
1382		- }
1383		-
1384		- .warning-message {
1385		- display: flex;
1386		- align-items: center;
1387		- gap: 10px;
1388		- background: #fbebed;
1389		- padding: 12px;
1390		- border-radius: 5px;
1391		- font-size: 14px;
1392		- line-height: 1.4em;
1393		- border-left: 4px solid #d63638;
1394		- }
1395		-
1396		- .warning-message svg {
1397		- width: 20px;
1398		- height: 20px;
1399		- }
1400		-
1401		- .warning-message span {
1402		- color: #d63638;
1403		- }
1404		-
1405		- .warning-message .footer-message {
1406		- width: calc(100% - 20px);
1407		- }
1408		-
1409		- @media only screen and (max-width: 991px) {
1410		- .betterdocs-ai-autowrite-form-container {
1411		- left: 0;
1412		- right: 0;
1413		- transform: translate(0%, -50%);
1414		- }
1415		-
1416		- .betterdocs-ai-autowrite-form-content {
1417		- overflow-x: auto;
1418		- }
1419		- }
1420		-
1421		- @media only screen and (max-width: 740px) {
1422		-
1423		- /* .bd-close-button {
1424		- right: px;
1425		- top: 1px;
1426		- border-radius: 5px;
1427		- width: 12px;
1428		- height: 12px;
1429		- color: #ffffff;
1430		- background: #00b884;
1431		- } */
1432		- .bd-close-button {
1433		- right: 0px;
1434		- top: 0px;
1435		- width: 12px;
1436		- height: 12px;
1437		- font-size: 14px;
1438		- }
1439		-
1440		- }
1441		- </style>
1442		- <?php
1443		- }
1444		- }
	849	+ // Send the generated content as the AJAX response
	850	+ wp_send_json_success( $generated_content );
	851	+ wp_die();
	852	+ }
	853	+}