From 64fc777a2be8ee456c9119d34e1eae75b3bbb175 Mon Sep 17 00:00:00 2001
From: ManInDark <61268856+ManInDark@users.noreply.github.com>
Date: Wed, 9 Aug 2023 22:21:36 +0200
Subject: [PATCH] Added nsfw flag, that way there won't be issues later on if
 it was to be implemented

---
 ui/easydiffusion/easydb/mappings.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ui/easydiffusion/easydb/mappings.py b/ui/easydiffusion/easydb/mappings.py
index 31549709..035f2ee8 100644
--- a/ui/easydiffusion/easydb/mappings.py
+++ b/ui/easydiffusion/easydb/mappings.py
@@ -25,6 +25,7 @@ class GalleryImage(Base):
     prompt = Column(String)
     negative_prompt = Column(String)
     time_created = Column(DateTime(timezone=True), server_default=func.now())
+    nsfw = Column(String, server_default='unknown')
 
     def __repr__(self):
         return "<GalleryImage(path='%s', seed='%s', use_stable_diffusion_model='%s', clip_skip='%s', use_vae_model='%s', sampler_name='%s', width='%s', height='%s', num_inference_steps='%s', guidance_scale='%s', lora='%s', use_hypernetwork_model='%s', tiling='%s', use_face_correction='%s', use_upscale='%s', prompt='%s', negative_prompt='%s')>" % (