From ee49ea952007209ffb5e7bfe7b917fa4cc0974df Mon Sep 17 00:00:00 2001 From: phixxy Date: Tue, 30 Jan 2024 02:20:09 -0800 Subject: [PATCH] added premium checker back in for dalle --- extensions/chatgpt.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/extensions/chatgpt.py b/extensions/chatgpt.py index 399272f..30d718e 100644 --- a/extensions/chatgpt.py +++ b/extensions/chatgpt.py @@ -205,6 +205,8 @@ class ChatGPT(commands.Cog): return "Error occurred in dalle" async def download_image(self, url, destination): + if url == "Error occurred in dalle": + return async with self.bot.http_session.get(url) as resp: if resp.status == 200: f = await aiofiles.open(destination, mode='wb') @@ -213,7 +215,7 @@ class ChatGPT(commands.Cog): return destination 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] image_url = await self.dalle_api_call(prompt, model=model, quality=quality, size=size) my_filename = str(time.time_ns()) + ".png"