(.*?)<\/h[12]>\s*/i', '', $content, 1);
}
}
return $content;
}
// 7. CORE GENERATION FUNCTION
// MTAW-MODIFICATION: Added $gemini_model parameter and the entire Gemini API logic branch.
function mtaw_generate_post($topic, $post_status, $schedule_date, $schedule_time, $selected_categories, $content_options, $ai_provider, $openai_model, $gemini_model, $user_id) {
$options = get_option('mtaw_settings');
$pexels_key = $options['mtaw_pexels_api_key'] ?? '';
$youtube_key = $options['mtaw_youtube_api_key'] ?? '';
$ai_key = ($ai_provider === 'gemini') ? ($options['mtaw_gemini_api_key'] ?? '') : ($options['mtaw_openai_api_key'] ?? '');
if (empty($ai_key)) return new WP_Error('api_key_missing', 'AI API key is missing. Please save it in the settings.');
if ($content_options['include_images'] > 0 && empty($pexels_key)) return new WP_Error('pexels_key_missing', 'Pexels API key is required for images.');
if ($content_options['include_video'] && empty($youtube_key)) return new WP_Error('youtube_key_missing', 'YouTube API key is required for video.');
// --- Build Universal Prompt ---
$length_map = ['short' => '3-4', 'medium' => '4-6', 'long' => '6-8', 'very_long' => '15-20'];
$article_structure = 'an introduction, '.($length_map[$content_options['article_length']] ?? '4-6').' main sections each under an heading, and a conclusion';
$json_keys = ['"post_content"'];
// MTAW-MODIFICATION: Removed "categories" from required keys as Gemini can be less reliable with it. We handle it gracefully.
$prompt = "Generate a blog post package for the topic: \"{$topic}\". Return only a single, valid JSON object and nothing else. The JSON should be clean and ready for parsing.\n";
$prompt .= "For the \"post_content\" key: Write a well-structured and engaging blog post. DO NOT include a main H1 title in the content. The post structure must be: {$article_structure}.\n";
if ($content_options['include_images'] > 0) $prompt .= "Inside the post_content HTML, insert placeholders like `[IMAGE_1_HERE]`, `[IMAGE_2_HERE]`, etc., where images would be contextually appropriate.\n";
if ($content_options['include_video']) $prompt .= "Inside the post_content HTML, insert a single `[YOUTUBE_VIDEO_HERE]` placeholder where a video would be most relevant.\n";
if ($content_options['include_qa']) $prompt .= "In the post_content HTML, after the conclusion, add an titled 'Frequently Asked Questions'. Follow it with 3-5 question-and-answer pairs. Each pair must be inside its own `
` tag, with the question wrapped in a `` tag. Example: `Is this a question? This is the answer.
`\n";
if ($content_options['include_internal_links']) $prompt .= "In the post_content HTML, naturally weave in 3-5 internal link placeholders formatted like `relevant anchor text`.\n";
if ($content_options['include_external_links']) $prompt .= "In the post_content HTML, naturally weave in 3 external link placeholders formatted like `relevant anchor text`.\n";
if ($content_options['generate_excerpt']) { $json_keys[] = '"excerpt"'; $prompt .= "For the \"excerpt\" key: Create a concise and compelling meta description or summary (around 150 characters).\n"; }
if ($content_options['generate_tags']) { $json_keys[] = '"tags"'; $prompt .= "For the \"tags\" key: Provide a JSON array of 5 relevant string keywords.\n"; }
$prompt .= "The final JSON object MUST have these keys: " . implode(', ', $json_keys) . ".";
$ai_response_body = '';
// --- API Call Logic ---
if ($ai_provider === 'gemini') {
// --- GEMINI API LOGIC ---
$api_url = "https://generativelanguage.googleapis.com/v1beta/models/{$gemini_model}:generateContent?key={$ai_key}";
$request_body = json_encode([
'contents' => [['parts' => [['text' => $prompt]]]],
'generationConfig' => [
'response_mime_type' => 'application/json',
]
]);
$api_response = wp_remote_post($api_url, [
'body' => $request_body,
'headers' => ['Content-Type' => 'application/json'],
'timeout' => 300,
]);
if (is_wp_error($api_response)) return new WP_Error('gemini_error', 'Gemini API Error: ' . $api_response->get_error_message());
$body = json_decode(wp_remote_retrieve_body($api_response), true);
if (isset($body['error'])) return new WP_Error('gemini_error', 'Gemini API Error: ' . $body['error']['message']);
// Extract the raw text, which should be our JSON string
$ai_response_body = $body['candidates'][0]['content']['parts'][0]['text'] ?? '';
} else {
// --- OPENAI API LOGIC (Original) ---
$api_url = 'https://api.openai.com/v1/chat/completions';
$request_body = json_encode([
'model' => $openai_model,
'messages' => [['role' => 'user', 'content' => $prompt]],
'response_format' => ['type' => 'json_object']
]);
$api_response = wp_remote_post($api_url, [
'body' => $request_body,
'headers' => ['Authorization' => 'Bearer ' . $ai_key, 'Content-Type' => 'application/json'],
'timeout' => 300,
]);
if (is_wp_error($api_response)) return new WP_Error('openai_error', 'OpenAI API Error: ' . $api_response->get_error_message());
$body = json_decode(wp_remote_retrieve_body($api_response), true);
if (isset($body['error'])) return new WP_Error('openai_error', 'OpenAI API Error: ' . $body['error']['message']);
$ai_response_body = $body['choices'][0]['message']['content'] ?? '';
}
// --- Process Response (Universal) ---
if (empty($ai_response_body)) return new WP_Error('ai_no_content', 'The AI provider returned an empty response. Check API key and model compatibility.');
// Sometimes the AI wraps the JSON in ```json ... ```, so we strip that.
$ai_response_body = preg_replace('/^```json\s*|\s*```$/', '', trim($ai_response_body));
$ai_data = json_decode($ai_response_body, true);
if (json_last_error() !== JSON_ERROR_NONE) {
return new WP_Error('json_decode_error', 'Failed to decode AI JSON response. Error: ' . json_last_error_msg());
}
$post_content = $ai_data['post_content'] ?? '';
if (empty($post_content)) return new WP_Error('ai_no_content', 'The AI response did not contain the expected "post_content" field.');
$post_content = mtaw_remove_duplicate_title($post_content, $topic);
$new_post = [
'post_title' => ucwords($topic),
'post_content' => $post_content,
'post_author' => $user_id,
'post_status' => $post_status,
'post_excerpt' => sanitize_text_field($ai_data['excerpt'] ?? ''),
];
if ($post_status === 'future') {
$new_post['post_date'] = $schedule_date . ' ' . $schedule_time . ':00';
$new_post['post_date_gmt'] = get_gmt_from_date($new_post['post_date']);
}
if (!empty($selected_categories)) {
$new_post['post_category'] = $selected_categories;
}
$post_id = wp_insert_post($new_post, true);
if (is_wp_error($post_id)) return $post_id;
// --- Sideload Images and Embed Video ---
$final_content = get_post($post_id)->post_content;
$image_attachments = [];
if ($content_options['include_images'] > 0 && !empty($pexels_key)) {
require_once(ABSPATH . 'wp-admin/includes/media.php');
require_once(ABSPATH . 'wp-admin/includes/file.php');
require_once(ABSPATH . 'wp-admin/includes/image.php');
$num_images_to_fetch = $content_options['include_images'];
$pexels_api_url = "https://api.pexels.com/v1/search?query=" . urlencode($topic) . "&per_page=" . $num_images_to_fetch . "&orientation=landscape";
$pexels_response = wp_remote_get($pexels_api_url, ['headers' => ['Authorization' => $pexels_key]]);
$pexels_photos = [];
if (!is_wp_error($pexels_response) && ($pexels_body = json_decode(wp_remote_retrieve_body($pexels_response), true))) {
$pexels_photos = $pexels_body['photos'] ?? [];
}
for ($i = 1; $i <= $num_images_to_fetch; $i++) {
$placeholder = "[IMAGE_{$i}_HERE]";
if (strpos($final_content, $placeholder) !== false) {
if (isset($pexels_photos[$i - 1])) {
$image_url = $pexels_photos[$i - 1]['src']['large2x'];
$attachment_id = media_sideload_image($image_url, $post_id, $topic, 'id');
if (!is_wp_error($attachment_id)) {
$image_html = wp_get_attachment_image($attachment_id, 'large', false, ['class' => 'aligncenter size-large']);
$final_content = str_replace($placeholder, $image_html, $final_content);
$image_attachments[] = $attachment_id;
}
} else {
$final_content = str_replace($placeholder, "", $final_content);
}
}
}
}
if ($content_options['include_video'] && !empty($youtube_key) && strpos($final_content, '[YOUTUBE_VIDEO_HERE]') !== false) {
$video_embed_code = mtaw_get_youtube_video_embed_code($topic, $youtube_key);
$final_content = str_replace('[YOUTUBE_VIDEO_HERE]', $video_embed_code, $final_content);
}
wp_update_post(['ID' => $post_id, 'post_content' => wp_kses_post($final_content)]);
if (!empty($ai_data['tags'])) wp_set_post_tags($post_id, $ai_data['tags']);
if (!empty($image_attachments)) set_post_thumbnail($post_id, $image_attachments[0]);
return $post_id;
}