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

should upload all dalle images to website now

Browse source
This commit is contained in:
phixxy 2024-01-27 21:01:24 -08:00
parent 254fbdd8ec
commit e2e42e8139
2 changed files with 34 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

4
extensions/chatgpt.py
View file

@ -32,6 +32,8 @@ class ChatGPT(commands.Cog):
os.mkdir(self.data_dir + "logs")
if not os.path.exists(self.data_dir + "dalle"):
os.mkdir(self.data_dir + "dalle")
if not os.path.exists(self.data_dir + "dalle2"):
os.mkdir(self.data_dir + "dalle2")
except:
self.bot.logger.exception("ChatGPT failed to make directories")
@ -219,7 +221,7 @@ class ChatGPT(commands.Cog):
prompt = ctx.message.content.split(" ", maxsplit=1)[1]
img_url = await self.get_dalle(prompt)
my_filename = str(time.time_ns()) + ".png"
filepath = f"{self.data_dir}dalle/{my_filename}"
filepath = f"{self.data_dir}dalle2/{my_filename}"
await self.download_image(img_url, filepath)
with open(filepath, "rb") as fh:
f = discord.File(fh, filename=filepath)