|
-
-
- from bot.bot import Bot
- from config import conf
- from common.log import logger
- import openai
-
-
-
- class OpenAIBot(Bot):
- def __init__(self):
- openai.api_key = conf().get('open_ai_api_key')
-
- def reply(self, query, context=None):
- if not context or not context.get('type') or context.get('type') == 'TEXT':
- return self.reply_text(query)
- elif context.get('type', None) == 'IMAGE_CREATE':
- return self.create_img(query)
-
- def reply_text(self, query):
- logger.info("[OPEN_AI] query={}".format(query))
- try:
- response = openai.Completion.create(
- model="text-davinci-003",
- prompt=query,
- temperature=0.9,
- max_tokens=1200,
- top_p=1,
- frequency_penalty=0.0,
- presence_penalty=0.6,
- stop=["#"]
- )
- res_content = response.choices[0]["text"].strip()
- except Exception as e:
- logger.exception(e)
- return None
- logger.info("[OPEN_AI] reply={}".format(res_content))
- return res_content
-
- def create_img(self, query):
- try:
- logger.info("[OPEN_AI] image_query={}".format(query))
- response = openai.Image.create(
- prompt=query,
- n=1,
- size="256x256"
- )
- image_url = response['data'][0]['url']
- logger.info("[OPEN_AI] image_url={}".format(image_url))
- except Exception as e:
- logger.exception(e)
- return None
- return image_url
-
- def edit_img(self, query, src_img):
- try:
- response = openai.Image.create_edit(
- image=open(src_img, 'rb'),
- mask=open('cat-mask.png', 'rb'),
- prompt=query,
- n=1,
- size='512x512'
- )
- image_url = response['data'][0]['url']
- logger.info("[OPEN_AI] image_url={}".format(image_url))
- except Exception as e:
- logger.exception(e)
- return None
- return image_url
-
- def migration_img(self, query, src_img):
-
- try:
- response = openai.Image.create_variation(
- image=open(src_img, 'rb'),
- n=1,
- size="512x512"
- )
- image_url = response['data'][0]['url']
- logger.info("[OPEN_AI] image_url={}".format(image_url))
- except Exception as e:
- logger.exception(e)
- return None
- return image_url
|