Create nsfw-text-to-image.js

nsfw-text-to-image.js

@@ -0,0 +1,132 @@
+/*
+
+credits rynn buat kode nya
+
+*/
+
+
+const axios = require('axios');
+
+const ProxyAgent = require('@rynn-k/proxy-agent');
+
+const proxy = new ProxyAgent('./proxies.txt', { random: true }); // Adjust proxy path
+
+
+
+async function nsfwimage(prompt, options = {}) {
+
+    try {
+
+        const {
+
+            negative_prompt = 'lowres, bad anatomy, bad hands, text, error, missing finger, extra digits, fewer digits, cropped, worst quality, low quality, low score, bad score, average score, signature, watermark, username, blurry',
+
+            style = 'anime',
+
+            width = 1024,
+
+            height = 1024,
+
+            guidance_scale = 7,
+
+            inference_steps = 28
+
+        } = options;
+
+        
+
+        const _style = ['anime', 'real', 'photo'];
+
+        
+
+        if (!prompt) throw new Error('Prompt is required');
+
+        if (!_style.includes(style)) throw new Error(`Available styles: ${_style.join(', ')}`);
+
+        if (width < 256 || width > 1216) throw new Error('Min width: 256, Max width: 1216');
+
+        if (height < 256 || height > 1216) throw new Error('Min height: 256, Max height: 1216');
+
+        if (guidance_scale < 0 || guidance_scale > 20) throw new Error('Min guidance scale: 0, Max guidance scale: 20');
+
+        if (inference_steps < 1 || inference_steps > 28) throw new Error('Max inference steps: 20');
+
+        
+
+        const agent = proxy.config();
+
+        const session_hash = Math.random().toString(36).substring(2);
+
+        const d = await axios.post(`https://heartsync-nsfw-uncensored${style !== 'anime' ? `-${style}` : ''}.hf.space/gradio_api/queue/join?`, {
+
+            data: [
+
+                prompt,
+
+                negative_prompt,
+
+                0,
+
+                true,
+
+                width,
+
+                height,
+
+                guidance_scale,
+
+                inference_steps
+
+            ],
+
+            event_data: null,
+
+            fn_index: 2,
+
+            trigger_id: 16,
+
+            session_hash: session_hash
+
+        }, agent);
+
+        
+
+        const { data } = await axios.get(`https://heartsync-nsfw-uncensored${style !== 'anime' ? `-${style}` : ''}.hf.space/gradio_api/queue/data?session_hash=${session_hash}`, agent);
+
+        
+
+        let result;
+
+        const lines = data.split('\n\n');
+
+        for (const line of lines) {
+
+            if (line.startsWith('data:')) {
+
+                const d = JSON.parse(line.substring(6));
+
+                if (d.msg === 'process_completed') result = d.output.data[0].url;
+
+            }
+
+        }
+
+        
+
+        return result;
+
+    } catch (error) {
+
+        throw new Error(error.message);
+
+    }
+
+}
+
+
+
+// Usage:
+
+const resp = await nsfwimage('1girl, Shiroko (Blue Archive), take a bath', { style: 'anime', width: 1216, height: 720 });
+
+console.log(resp);