cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-04-29 00:27:09 +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
No known key found for this signature in database
GPG Key ID: 5BFA9CB4DDA943BF
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
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: def gemini_photo_handler(message: Message, bot: TeleBot) -> None:
s = message.caption s = message.caption
bot.reply_to( reply_message = bot.reply_to(
message, message,
"Generating google gemini vision answer please wait,", "Generating google gemini vision answer please wait,",
) )
@ -93,9 +93,11 @@ def gemini_photo_handler(message: Message, bot: TeleBot) -> None:
contents = { contents = {
"parts": [{"mime_type": "image/jpeg", "data": image_data}, {"text": prompt}] "parts": [{"mime_type": "image/jpeg", "data": image_data}, {"text": prompt}]
} }
response = model.generate_content(contents=contents) try:
print(response.text) response = model.generate_content(contents=contents)
bot.reply_to(message, "Gemini vision answer:\n" + 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: def register(bot: TeleBot) -> None:
@ -104,6 +106,6 @@ def register(bot: TeleBot) -> None:
bot.register_message_handler( bot.register_message_handler(
gemini_photo_handler, gemini_photo_handler,
content_types=["photo"], 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, pass_bot=True,
) )