+ fixes

routers/gen_router.py

@@ -142,12 +142,13 @@ async def change_quality(call: CallbackQuery, state: FSMContext, dao: DAO):
 @router.callback_query(States.gen_mode, F.data == 'gen_mode_change_type')
 async def gen_mode_change_type(call: CallbackQuery, state: FSMContext, dao: DAO):
     await call.answer()
-    keyboards = []
-    for gen_type in GenType:
-        keyboards.append(InlineKeyboardButton(text=gen_type.value, callback_data=f'select_type_{gen_type.name}'))
-    await call.message.edit_caption(caption="Выбери тип генерации", reply_markup=InlineKeyboardMarkup(
-        inline_keyboard=[keyboards,
-                         [InlineKeyboardButton(text="⬅️ Назад", callback_data="gen_mode_cancel_type_change")]]))
+    data = await state.get_data()
+    if GenType(data['type']) is GenType.IMAGE:
+        await state.update_data({"type": GenType.TEXT.name})
+    else:
+        await state.update_data({"type": GenType.IMAGE.name})
+
+    await gen_mode_base_msg(call.message, state=state, dao=dao)
 
 
 @router.callback_query(States.gen_mode, F.data.startswith('select_type_'))
@@ -346,7 +347,6 @@ async def generate_image(
         quality=Quality[data['quality']],
     )
 
-
     images = []
     if generated_images_io:
         for img_io in generated_images_io: