-- katakana_conversion.lua

-- Table mapping half-width katakana to full-width
local half_to_full = {
    ['ｱ'] = 'ア', ['ｲ'] = 'イ', ['ｳ'] = 'ウ', ['ｴ'] = 'エ', ['ｵ'] = 'オ',
    ['ｶ'] = 'カ', ['ｷ'] = 'キ', ['ｸ'] = 'ク', ['ｹ'] = 'ケ', ['ｺ'] = 'コ',
    ['ｻ'] = 'サ', ['ｼ'] = 'シ', ['ｽ'] = 'ス', ['ｾ'] = 'セ', ['ｿ'] = 'ソ',
    ['ﾀ'] = 'タ', ['ﾁ'] = 'チ', ['ﾂ'] = 'ツ', ['ﾃ'] = 'テ', ['ﾄ'] = 'ト',
    ['ﾅ'] = 'ナ', ['ﾆ'] = 'ニ', ['ﾇ'] = 'ヌ', ['ﾈ'] = 'ネ', ['ﾉ'] = 'ノ',
    ['ﾊ'] = 'ハ', ['ﾋ'] = 'ヒ', ['ﾌ'] = 'フ', ['ﾍ'] = 'ヘ', ['ﾎ'] = 'ホ',
    ['ﾏ'] = 'マ', ['ﾐ'] = 'ミ', ['ﾑ'] = 'ム', ['ﾒ'] = 'メ', ['ﾓ'] = 'モ',
    ['ﾔ'] = 'ヤ', ['ﾕ'] = 'ユ', ['ﾖ'] = 'ヨ',
    ['ﾗ'] = 'ラ', ['ﾘ'] = 'リ', ['ﾙ'] = 'ル', ['ﾚ'] = 'レ', ['ﾛ'] = 'ロ',
    ['ﾜ'] = 'ワ', ['ｦ'] = 'ヲ', ['ﾝ'] = 'ン',
    ['ｧ'] = 'ァ', ['ｨ'] = 'ィ', ['ｩ'] = 'ゥ', ['ｪ'] = 'ェ', ['ｫ'] = 'ォ',
    ['ｯ'] = 'ッ', ['ｬ'] = 'ャ', ['ｭ'] = 'ュ', ['ｮ'] = 'ョ',
    ['ﾞ'] = '゛', ['ﾟ'] = '゜'
}

-- Function to convert half-width katakana to full-width
local function convert_halfwidth_to_fullwidth(text)
    return text:gsub(".", function(c)
        return half_to_full[c] or c
    end)
end

-- Hook into the subtitle text before display
mp.add_hook("on_sub_text", 50, function(subtitle)
    if subtitle and subtitle.text then
        mp.msg.info("Original subtitle text: " .. subtitle.text)
        subtitle.text = convert_halfwidth_to_fullwidth(subtitle.text)
        mp.msg.info("Converted subtitle text: " .. subtitle.text)
    end
    return subtitle
end)