from bot.Keyboard import Keyboard
from telegram import InlineKeyboardButton, InlineKeyboardMarkup


class MenuKeyboard(Keyboard):

    def get_keyboard(self):
        return [
            [
                InlineKeyboardButton("Сгенерировать видео", callback_data='create_video'),
                InlineKeyboardButton("Сгенерировать изображение", callback_data='create_img')
            ],
        ]