first

ai_bot/keyboards/flux/SelectTypeRequestKeyboard.py

@@ -0,0 +1,15 @@
+from bot.Keyboard import Keyboard
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup
+
+
+class SelectTypeRequestKeyboard(Keyboard):
+
+    def get_keyboard(self):
+        return [
+            [
+                InlineKeyboardButton("Сгенерировать изображение из текста", callback_data='set_type_request?type=txt2img'),
+            ],
+            [
+                InlineKeyboardButton("Сгенерировать изображение на основе изображения", callback_data='set_type_request?type=img2img')
+            ]
+        ]