first

ai_bot/keyboards/CreateImageKeyboard.py

@@ -0,0 +1,14 @@
+from bot.Keyboard import Keyboard
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup
+
+
+class CreateImageKeyboard(Keyboard):
+
+    def get_keyboard(self):
+        keyboard = []
+        models = self.get_option("models_arr")
+
+        for model in models:
+            keyboard.append([InlineKeyboardButton(model, callback_data='start_model_scenario?model={model}'.format(model=model))])
+
+        return keyboard