This commit is contained in:
xds
2026-02-03 09:45:27 +03:00
parent 2693675e85
commit 21f86afa38

View File

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