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

added indicator that dalle is generating image

Browse source
This commit is contained in:
phixxy 2024-01-30 02:35:45 -08:00
parent fcb02562f8
commit 167b1b8ea0
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
extensions/chatgpt.py
View file

@ -220,6 +220,7 @@ class ChatGPT(commands.Cog):
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 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]
await ctx.send(f"Please be patient this may take some time! Generating: {prompt}.")
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"
image_filepath = f"{self.data_dir}dalle/{my_filename}" image_filepath = f"{self.data_dir}dalle/{my_filename}"