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

reduce code duplication and reuse image_to_data_uri

Browse Source
This commit is contained in:
F4ria
2024-04-17 14:03:26 +08:00
parent 19bb441533
commit 2abcc90a32
3 changed files with 16 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
handlers/yi.py
View File

@ -1,6 +1,4 @@
import base64
from os import environ
from pathlib import Path
from openai import OpenAI
import requests
@ -88,12 +86,6 @@ def yi_handler(message: Message, bot: TeleBot) -> None:
bot_reply_markdown(reply_id, who, yi_reply_text, bot)
def _image_to_data_uri(file_path):
with open(file_path, "rb") as image_file:
encoded_image = base64.b64encode(image_file.read()).decode("utf-8")
return f"data:image/png;base64,{encoded_image}"
def yi_photo_handler(message: Message, bot: TeleBot) -> None:
s = message.caption
prompt = s.strip()
@ -121,7 +113,7 @@ def yi_photo_handler(message: Message, bot: TeleBot) -> None:
{"type": "text", "text": prompt},
{
"type": "image_url",
"image_url": {"url": _image_to_data_uri("yi_temp.jpg")},
"image_url": {"url": image_to_data_uri("yi_temp.jpg")},
},
],
}