Create flux.js

plugins/flux.js

@@ -0,0 +1,105 @@
+const axios = require('axios');
+
+async function fluxImage(prompt, width = 1024, height = 1024, server = "NSFW-Core: Uncensored Server 2") {
+  try {
+    const { data: init } = await axios.post(
+      "https://nihalgazi-flux-unlimited.hf.space/gradio_api/call/generate_image",
+      { data: [prompt, width, height, 3, true, server] },
+      {
+        headers: {
+          "Content-Type": "application/json",
+          Origin: "https://chrunos.com",
+          Referer: "https://chrunos.com/",
+        },
+      }
+    );
+
+    const eventId = init.event_id;
+    if (!eventId) throw new Error("Failed to obtain event_id.");
+
+    const streamUrl = `https://nihalgazi-flux-unlimited.hf.space/gradio_api/call/generate_image/${eventId}`;
+    let imageUrl = null;
+
+    for (let i = 0; i < 15; i++) {
+      const { data } = await axios.get(streamUrl, {
+        headers: { Accept: "text/event-stream" },
+      });
+
+      const match = data.match(/"url":\s*"([^"]+)"/);
+      if (match) {
+        imageUrl = match[1];
+        break;
+      }
+
+      await new Promise(r => setTimeout(r, 2000));
+    }
+
+    if (!imageUrl) throw new Error("Failed to retrieve image URL from stream.");
+    return imageUrl;
+  } catch (err) {
+    throw new Error(`fluxImage error: ${err.message}`);
+  }
+}
+
+const handler = async (req, res) => {
+  try {
+    const { prompt, key, width, height, server } = req.query;
+
+    if (!prompt) {
+      return res.status(400).json({
+        author: "Herza",
+        success: false,
+        error: 'Missing required parameter: prompt'
+      });
+    }
+
+    if (!key) {
+      return res.status(400).json({
+        author: "Herza",
+        success: false,
+        error: 'Missing required parameter: key'
+      });
+    }
+
+    const imgWidth = width ? parseInt(width) : 1024;
+    const imgHeight = height ? parseInt(height) : 1024;
+    const selectedServer = server || "NSFW-Core: Uncensored Server 2";
+
+    const result = await fluxImage(prompt, imgWidth, imgHeight, selectedServer);
+    
+    if (!result) {
+      return res.status(500).json({
+        author: "Herza",
+        success: false,
+        error: 'Failed to generate image'
+      });
+    }
+
+    res.json({
+      author: "Herza",
+      success: true,
+      data: {
+        result: result
+      }
+    });
+
+  } catch (error) {
+    res.status(500).json({
+      author: "Herza",
+      success: false,
+      error: error.message
+    });
+  }
+};
+
+module.exports = {
+  name: 'Flux Image Generator',
+  description: 'Generate images using Flux AI model',
+  type: 'GET',
+  routes: ['api/AI/Flux'],
+  tags: ['ai', 'image', 'flux', 'generator'],
+  main: ['AI'],
+  parameters: ['prompt', 'width', 'height', 'server', 'key'],
+  enabled: true,
+  handler
+};