ai-char/ai-char-bot
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

+ assets

Browse Source
This commit is contained in:
xds
2026-02-03 14:20:22 +03:00
parent 739f027742
commit 7050999ed8
6 changed files with 109 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
routers/assets_router.py Normal file
View File

@@ -0,0 +1,26 @@
from aiogram import Router
from aiogram.filters import Command
from aiogram.types import Message, InputMediaPhoto, InputMedia, BufferedInputFile
from repos.dao import DAO
router = Router()
@router.message(Command("assets"))
async def assets_command(msg: Message, dao: DAO):
assets = await dao.assets.get_assets(limit=10, offset=0)
media_group = []
for asset in assets:
if asset.tg_photo_file_id:
media_group.append(InputMediaPhoto(media=asset.tg_photo_file_id))
elif asset.tg_doc_file_id:
asset_full_info = await dao.assets.get_asset(asset.id)
media_group.append(InputMediaPhoto(media=BufferedInputFile(asset_full_info.data, asset_full_info.name)))
else:
continue
mg = await msg.answer_media_group(media_group)
for media_index, media in enumerate(mg):
if assets[media_index].tg_photo_file_id is None:
assets[media_index].tg_photo_file_id = media.photo[-1].file_id
await dao.assets.update_asset(assets[media_index].id, assets[media_index])

30
routers/gen_router.py
View File

@@ -14,6 +14,7 @@ from aiogram.types import *
import keyboards
from adapters.google_adapter import GoogleAdapter
from models.Asset import Asset, AssetType
from models.Character import Character
from models.enums import AspectRatios, Quality, GenType
from repos.dao import DAO
@@ -37,16 +38,20 @@ async def init_gen_mode(state: FSMContext, dao: DAO):
@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,
res = await generate_image(prompt=message.caption, 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 💫")
await wait_msg.delete()
doc = await message.answer_document(res[0], caption="Generated result 💫")
await dao.assets.save_asset(Asset(id=None, name=res[0].filename, type=AssetType.IMAGE, data=res[0].data,
tg_doc_file_id=doc.document.file_id, tg_photo_file_id=None, linked_char_id=None))
@router.message(Command("gen_mode"))
@@ -250,14 +255,17 @@ async def handle_album(
)
await wait_msg.delete()
data = await state.get_data()
# 4. Отправляем результат
if generated_files:
for file in generated_files:
await message.answer_document(file, caption="✨ Generated result")
doc = await message.answer_document(file, caption="✨ Generated result")
await dao.assets.save_asset(Asset(id=None, name=file.filename, type=AssetType.IMAGE, data=file.data,
tg_doc_file_id = doc.document.file_id, tg_photo_file_id = None,
linked_char_id = data["char_id"]))
else:
await message.answer("❌ Генерация не вернула изображений.")
await gen_mode_base_msg(message=message, state=state, dao=dao, call_type="start")
await gen_mode_base_msg(message=message, state=state, dao=dao, call_type="start")
except Exception as e:
await wait_msg.edit_text(f"❌ Ошибка: {e}")
@@ -290,7 +298,6 @@ async def gen_mode_start(
media_ids.append(message.reply_to_message.photo[-1].file_id)
wait_msg = await message.answer("🎨 Генерирую...")
data = await state.get_data()
try:
if GenType[data['type']] is GenType.IMAGE:
@@ -307,7 +314,10 @@ async def gen_mode_start(
if generated_files:
for file in generated_files:
await message.answer_document(file, caption="✨ Generated result")
doc = await message.answer_document(file, caption="✨ Generated result")
await dao.assets.save_asset(Asset(id=None, name=file.filename, type=AssetType.IMAGE, data=file.data,
tg_doc_file_id=doc.document.file_id, tg_photo_file_id=None,
linked_char_id=data["char_id"]))
else:
await message.answer("❌ Ничего не сгенерировалось.")