From 44d656c15de4be552e98f75853e8966703b1552e Mon Sep 17 00:00:00 2001
From: phixxy <bottlecap@protonmail.com>
Date: Fri, 19 Jan 2024 18:49:07 -0800
Subject: [PATCH] made nsfw and sfw folders on plugin startup

---
 plugins/stable_diffusion.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/plugins/stable_diffusion.py b/plugins/stable_diffusion.py
index 28c8c8c..2b1a4c2 100644
--- a/plugins/stable_diffusion.py
+++ b/plugins/stable_diffusion.py
@@ -24,6 +24,8 @@ class StableDiffusion(commands.Cog):
         try:
             if not os.path.exists(self.working_dir):
                 os.mkdir(self.working_dir)
+                os.mkdir(f"{self.working_dir}sfw")
+                os.mkdir(f"{self.working_dir}nsfw")
             if not os.path.exists(self.db_dir):
                 os.mkdir(self.db_dir)
         except:
@@ -297,7 +299,7 @@ class StableDiffusion(commands.Cog):
             pnginfo.add_text("parameters", response2.get("info"))
             try:
                 if ctx.channel.is_nsfw():
-                    folder = self.working_dir
+                    folder = self.working_dir + "nsfw/"
                 else:
                     folder = self.working_dir + "sfw/"
             except: