some fixes in nsfw detection

Author: divamgupta

electron_app/src/StableDiffusion.vue

@@ -59,25 +59,29 @@ export default {
 
                 // check for nsfw content
                 img.onload = () => {
-                    if (this.$parent.app_state.app_data.settings.nsfw_filter == false) {
+                    if (this.$parent.app_state.app_data.settings.nsfw_filter == true) {
                         if (img.attached_cbs) {
                             if (img.attached_cbs.on_img)
                                 img.attached_cbs.on_img(impath);
                         }
                     }
-                    nsfwjs.load()
-                        .then(model => model.classify(img))
-                        .then(predictions => {
-                            switch (predictions[0].className) {
-                                case 'Hentai':
-                                case 'Porn':
-                                    impath = "nsfw";
-                            }
-                            if (img.attached_cbs) {
-                                if (img.attached_cbs.on_img)
-                                    img.attached_cbs.on_img(impath);
-                            }
-                        });
+                    else {
+                        
+                        nsfwjs.load()
+                            .then(model => model.classify(img))
+                            .catch(() => {
+                                console.log("cant run nsfw")
+                            })
+                            .then(predictions => {
+                                if (predictions && (predictions[0].className ==  'Hentai' || predictions[0].className ==  'Porn')) {
+                                        impath = "nsfw_" + Math.random();
+                                }
+                                if (img.attached_cbs) {
+                                    if (img.attached_cbs.on_img)
+                                        img.attached_cbs.on_img(impath);
+                                }
+                            });
+                    }
                 }
             }
 

electron_app/src/components/ImageItem.vue

@@ -18,7 +18,7 @@
         </div>
 
 
-        <img v-if="path != 'nsfw'" @click="open_image_popup( path )"  class="gal_img" :src="'file://' + path " style="max-height: 100% ; max-width: 100%;" >
+        <img v-if="!path.startsWith('nsfw_')" @click="open_image_popup( path )"  class="gal_img" :src="'file://' + path " style="max-height: 100% ; max-width: 100%;" >
         <img v-else @click="open_image_popup( path )"  class="gal_img" :src="require('@/assets/nsfw.png')" style="max-height: 100% ; max-width: 100%;" >
         <br>
         <div v-if="!hide_extra_save_button" @click="save_image(path)" class="l_button">Save Image</div>
@@ -51,6 +51,8 @@ export default {
     },
     methods: {
         open_image_popup(img){
+            if(img.startsWith('nsfw_'))
+                return;
             open_popup("file://"+img , undefined);
         },
 

electron_app/src/components/Settings.vue

@@ -8,8 +8,8 @@
             <br>
             <div class="setting_box">
             <div class="settings_left">
-            <h2>Block Adult Content</h2>
-            <p>Prevents image generation considered NSFW</p>
+            <h2>Allow NSFW Content</h2>
+            <p>Allows image generation considered NSFW</p>
             </div>
             <hr>
             <div style="float:right;margin-right: 9px;align-self: center;" >