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

added premium checker back in for dalle

Browse source
This commit is contained in:
phixxy 2024-01-30 02:20:09 -08:00
parent 15adde4b23
commit ee49ea9520
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
extensions/chatgpt.py
View file

@ -205,6 +205,8 @@ class ChatGPT(commands.Cog):
return "Error occurred in dalle" return "Error occurred in dalle"
async def download_image(self, url, destination): async def download_image(self, url, destination):
if url == "Error occurred in dalle":
return
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:
f = await aiofiles.open(destination, mode='wb') f = await aiofiles.open(destination, mode='wb')
@ -213,7 +215,7 @@ class ChatGPT(commands.Cog):
return destination return destination
async def generate_dalle_image(self, ctx, model, quality="standard", size="1024x1024"): async def generate_dalle_image(self, ctx, model, quality="standard", size="1024x1024"):
if True:#ctx.author.get_role(self.premium_role): if 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.dalle_api_call(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"