[Assigned] Change Search URL Slug for different language

function custom_search_url_rewrite() {
    if ( is_search() && ! empty( $_GET['s'] ) ) {
        // Check if WPML is active and a language code is defined
        $current_lang = defined('ICL_LANGUAGE_CODE') ? ICL_LANGUAGE_CODE : '';
        
        // If you want to handle default language differently (omitting its code), 
        // uncomment and adjust this section:
        $default_lang = apply_filters( 'wpml_default_language', null );
        if ( $current_lang && $current_lang === $default_lang ) {
        //     // Default language: no language code in the URL
             $lang_prefix = '';
        } else {
        //     // Secondary language: include the language code
             $lang_prefix = $current_lang ? $current_lang . '/' : '';
         }

        // If you want to always show the language code regardless of default language:
       // $lang_prefix = $current_lang ? $current_lang . '/' : '';

        // Redirect to the language-specific search URL
        wp_redirect( home_url( "/" . $lang_prefix . "search/" ) . urlencode( get_query_var( 's' ) ) );
        exit();
    }
}
add_action( 'template_redirect', 'custom_search_url_rewrite' );