cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-11-04 16:56:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: delete reply after done

Browse Source 
Signed-off-by: Frost Ming <me@frostming.com>
This commit is contained in:
Frost Ming
2023-12-15 15:54:28 +08:00
parent 828530cc10
commit f37dd9371c
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
handlers/gemini.py
View File

@ -76,7 +76,7 @@ def gemini_handler(message: Message, bot: TeleBot) -> None:
def gemini_photo_handler(message: Message, bot: TeleBot) -> None:
s = message.caption
bot.reply_to(
reply_message = bot.reply_to(
message,
"Generating google gemini vision answer please wait,",
)
@ -93,9 +93,11 @@ def gemini_photo_handler(message: Message, bot: TeleBot) -> None:
contents = {
"parts": [{"mime_type": "image/jpeg", "data": image_data}, {"text": prompt}]
}
try:
response = model.generate_content(contents=contents)
print(response.text)
bot.reply_to(message, "Gemini vision answer:\n" + response.text)
finally:
bot.delete_message(reply_message.chat.id, reply_message.message_id)
def register(bot: TeleBot) -> None:
@ -104,6 +106,6 @@ def register(bot: TeleBot) -> None:
bot.register_message_handler(
gemini_photo_handler,
content_types=["photo"],
func=lambda m: m.caption and m.caption.startswith("gemini:"),
func=lambda m: m.caption and m.caption.startswith(("gemini:", "/gemini")),
pass_bot=True,
)