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

ai_bot/keyboards/CreateVideoKeyboard.py

@@ -0,0 +1,14 @@
+from bot.Keyboard import Keyboard
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup
+
+
+class CreateVideoKeyboard(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

ai_bot/keyboards/MainKeyboard.py

@@ -0,0 +1,12 @@
+from bot.Keyboard import Keyboard
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup
+
+
+class MainKeyboard(Keyboard):
+
+    def get_keyboard(self):
+        return [
+            [
+                InlineKeyboardButton("Меню", callback_data='menu'),
+            ],
+        ]

ai_bot/keyboards/MenuKeyboard.py

@@ -0,0 +1,13 @@
+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')
+            ],
+        ]

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')
+            ]
+        ]