phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

dalle functions entirely cleaned up and working

Browse source
This commit is contained in:
phixxy 2024-01-30 02:16:28 -08:00
parent 2b47c9b10d
commit 15adde4b23
1 changed files with 27 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

53
extensions/chatgpt.py
View file

@ -103,29 +103,6 @@ class ChatGPT(commands.Cog):
self.bot.logger.exception("Error occurred in answer_question") self.bot.logger.exception("Error occurred in answer_question")
return "Error occurred in answer_question" return "Error occurred in answer_question"
async def get_dalle(self, prompt, model="dall-e-2", quality="standard", size="1024x1024"):
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {os.getenv("openai.api_key")}',
}
data = {
"model": model,
"prompt": prompt,
"size": size,
"quality": quality,
"n":1,
}
url = "https://api.openai.com/v1/images/generations"
try:
async with self.bot.http_session.post(url, headers=headers, json=data) as resp:
response_data = await resp.json()
response = response_data['data'][0]['url']
return response
except Exception as error:
self.bot.logger.exception("Error occurred in dalle")
return "Error occurred in dalle"
@commands.command( @commands.command(
description="Personality", description="Personality",
@ -203,6 +180,30 @@ class ChatGPT(commands.Cog):
else: else:
await ctx.send("Sorry you must be a premium member to use this command. (!donate)") await ctx.send("Sorry you must be a premium member to use this command. (!donate)")
async def dalle_api_call(self, prompt, model="dall-e-2", quality="standard", size="1024x1024"):
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {os.getenv("openai.api_key")}',
}
data = {
"model": model,
"prompt": prompt,
"size": size,
"quality": quality,
"n":1,
}
url = "https://api.openai.com/v1/images/generations"
try:
async with self.bot.http_session.post(url, headers=headers, json=data) as resp:
response_data = await resp.json()
response = response_data['data'][0]['url']
return response
except Exception as error:
self.bot.logger.exception("Error occurred in dalle")
return "Error occurred in dalle"
async def download_image(self, url, destination): async def download_image(self, url, destination):
async with self.bot.http_session.get(url) as resp: async with self.bot.http_session.get(url) as resp:
if resp.status == 200: if resp.status == 200:
@ -211,10 +212,10 @@ class ChatGPT(commands.Cog):
await f.close() await f.close()
return destination return destination
async def generate_dalle_image(self, ctx, model, quality=None, size=None): async def generate_dalle_image(self, ctx, model, quality="standard", size="1024x1024"):
if ctx.author.get_role(self.premium_role): if True:#ctx.author.get_role(self.premium_role):
prompt = ctx.message.content.split(" ", maxsplit=1)[1] prompt = ctx.message.content.split(" ", maxsplit=1)[1]
image_url = await self.get_dalle(prompt, model=model, quality=quality, size=size) image_url = await self.dalle_api_call(prompt, model=model, quality=quality, size=size)
my_filename = str(time.time_ns()) + ".png" my_filename = str(time.time_ns()) + ".png"
image_filepath = f"{self.data_dir}dalle/{my_filename}" image_filepath = f"{self.data_dir}dalle/{my_filename}"
await self.download_image(image_url, image_filepath) await self.download_image(image_url, image_filepath)