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

Merge pull request #7 from frostming/fix/delete

Browse Source 
fix: delete reply after done
This commit is contained in:
yihong 2023-12-15 16:30:54 +08:00 committed by GitHub
commit 384cc4656f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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:
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}]
}
response = model.generate_content(contents=contents)
print(response.text)
bot.reply_to(message, "Gemini vision answer:\n" + response.text)
try:
response = model.generate_content(contents=contents)
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,
)