diff options
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 43 |
1 files changed, 39 insertions, 4 deletions
@@ -137,7 +137,22 @@ async def guest_middleware(handler, event, data): replied_text = None prompt = "" - + + if replied and replied.location: + loc = replied.location + prompt += ( + f"<ответ на локацию>\n" + f"{loc.latitude}, {loc.longitude}\n" + f"</ответ на локацию>\n" + ) + + if message.location: + loc = message.location + prompt += ( + f"<локация>\n" + f"{loc.latitude}, {loc.longitude}\n" + f"</локация>\n" + ) if replied and replied.poll: poll = replied.poll if poll.allows_multiple_answers: @@ -222,7 +237,7 @@ async def guest_middleware(handler, event, data): return await handler(event, data) -@router.message(F.content_type.in_({"text", "photo", "document", "poll"})) +@router.message(F.content_type.in_({"text", "photo", "document", "poll", "location"})) async def text_handler(message: Message): chat_type = message.sender_chat.type if message.sender_chat else "" user_id = message.sender_chat.id if message.sender_chat else message.from_user.id @@ -327,9 +342,28 @@ async def text_handler(message: Message): print(f"{RED} -- {e}{RESET}") return + if replied and replied.location: + loc = replied.location + + prompt += ( + f"<ответ на локацию>\n" + f"{loc.latitude}, {loc.longitude}\n" + f"</ответ на локацию>\n" + ) + + if message.location: + loc = message.location + + prompt += ( + f"<локация>\n" + f"{loc.latitude}, {loc.longitude}\n" + f"</локация>\n" + ) + elif replied and replied.photo and IMAGE_MODEL != "": try: - msg = await message.reply("👀") + if not msg: + msg = await message.reply("👀") except Exception: return try: @@ -345,7 +379,8 @@ async def text_handler(message: Message): prompt = f"<ответ на фото>{image_text}</ответ на фото>\n<подпись в ответе>{replied.caption}</подпись в ответе>\n" else: prompt = f"<ответ на фото>{image_text}</ответ на фото>\n<подпись в ответе>{replied.caption}\n" - await msg.edit_text("💬") + if not msg: + await msg.edit_text("💬") elif replied and replied.text: replied_text = message.reply_to_message.text |