From befd1a66f7e7d1beacf61dc6fc01813bb3ec1fda Mon Sep 17 00:00:00 2001 From: xds Date: Tue, 3 Feb 2026 10:18:27 +0300 Subject: [PATCH] + fixes --- routers/gen_router.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/routers/gen_router.py b/routers/gen_router.py index dc709ea..36c2140 100644 --- a/routers/gen_router.py +++ b/routers/gen_router.py @@ -34,6 +34,21 @@ async def init_gen_mode(state: FSMContext, dao: DAO): await state.update_data(data) +@router.message(Command("image")) +async def generate_image_cmd(message: Message, state: FSMContext, dao: DAO, gemini: GoogleAdapter, bot: Bot): + wait_msg = await message.answer("Генерирую...") + if message.photo: + res = await generate_image(prompt=message.text, media=[message.photo[0].file_id], state=state, dao=dao, bot=bot, + gemini=gemini) + await wait_msg.delete() + await message.answer_document(res[0], caption="Generated result 💫") + else: + res = await generate_image(prompt=message.text, media=None, state=state, dao=dao, bot=bot, + gemini=gemini) + await wait_msg.delete() + await message.answer_document(res[0], caption="Generated result 💫") + + @router.message(Command("gen_mode")) async def gen_mode(message: Message, state: FSMContext, dao: DAO): state_on = await state.get_state()