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

def get_language_keyboard():
    keyboard = []
    for lang_code, lang_name in LANGUAGES.items():
        keyboard.append(
            [InlineKeyboardButton(lang_name, callback_data=f"lang_{lang_code}")]
        )
    keyboard.append([InlineKeyboardButton("Cancel", callback_data="cancel")])
    return InlineKeyboardMarkup(keyboard)