+ fixes

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()