from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
from chatbot.plugins.languages import LANGUAGES

def get_language_keyboard():
    keyboard = []

    # Row 1: English, Indonesian
    keyboard.append([
        InlineKeyboardButton("English", callback_data="lang_en"),
        InlineKeyboardButton("Indonesian", callback_data="lang_id")
    ])

    # Row 2: Spanish
    keyboard.append([
        InlineKeyboardButton("Spanish", callback_data="lang_es")
    ])

    # Row 3: French, German, Japanese
    keyboard.append([
        InlineKeyboardButton("French", callback_data="lang_fr"),
        InlineKeyboardButton("German", callback_data="lang_de"),
        InlineKeyboardButton("Japanese", callback_data="lang_jp")
    ])

    keyboard.append([
        InlineKeyboardButton("Malagasy", callback_data="lang_mg"),
        InlineKeyboardButton("Russian", callback_data="lang_ru"),
        InlineKeyboardButton("Hindi", callback_data="lang_hi"),
    ])
    keyboard.append([
        InlineKeyboardButton("Cancel", callback_data="cancel")
    ])

    return InlineKeyboardMarkup(keyboard)