Compatibility layer between the ClassifAI plugin and Polylang

felipeelia-classifai.php

<?php
/**
 * Plugin Name: ClassifAI and Polylang
 * Description: Compatibility layer between the ClassifAI plugin and Polylang
 * Version:     1.0.0
 * Author:      Felipe Elia
 * Author URI:  https://felipeelia.dev/
 * Text Domain: felipeelia-classifai
 * Domain Path: /languages
 *
 * @package FelipeEliaClassifAi
 */

namespace FelipeEliaClassifAi;

defined( 'ABSPATH' ) || exit;

/**
 * Add the language to ClassifAI excerpt prompt
 *
 * @param string $prompt  Prompt sent to ChatGPT
 * @param int    $post_id Post ID
 * @return string
 */
function add_language_to_prompt( $prompt, $post_id ) {
	if ( ! function_exists( 'pll_get_post_language' ) ) {
		return $prompt;
	}

	$language = pll_get_post_language( $post_id, 'locale' );

	if ( in_array( $language, [ 'en', 'en_US', 'en_UK' ], true ) ) {
		return $prompt;
	}

	$prompt .= " in {$language}";

	return $prompt;
}
add_filter( 'classifai_chatgpt_excerpt_prompt', __NAMESPACE__ . '\add_language_to_prompt', 10, 2 );
add_filter( 'classifai_chatgpt_title_prompt', __NAMESPACE__ . '\add_language_to_prompt', 10, 2 );