Update plugins/flux.js

plugins/flux.js

@@ -1,4 +1,6 @@
 const axios = require('axios');
+const fs = require('fs');
+const path = require('path');
 
 async function fluxImage(prompt, width = 1024, height = 1024, server = "NSFW-Core: Uncensored Server 2") {
   try {
@@ -41,6 +43,31 @@ async function fluxImage(prompt, width = 1024, height = 1024, server = "NSFW-Cor
   }
 }
 
+async function downloadAndSaveImage(imageUrl, fileId) {
+  const tmpDir = path.join(process.cwd(), 'tmp');
+  
+  if (!fs.existsSync(tmpDir)) {
+    fs.mkdirSync(tmpDir, { recursive: true });
+  }
+
+  const filePath = path.join(tmpDir, `flux-${fileId}.webp`);
+
+  const response = await axios.get(imageUrl, {
+    responseType: 'arraybuffer',
+    timeout: 30000
+  });
+
+  fs.writeFileSync(filePath, Buffer.from(response.data));
+
+  setTimeout(() => {
+    if (fs.existsSync(filePath)) {
+      fs.unlinkSync(filePath);
+    }
+  }, 60000);
+
+  return filePath;
+}
+
 const handler = async (req, res) => {
   try {
     const { prompt, key, width, height, server } = req.query;
@@ -65,9 +92,9 @@ const handler = async (req, res) => {
     const imgHeight = height ? parseInt(height) : 1024;
     const selectedServer = server || "NSFW-Core: Uncensored Server 2";
 
-    const result = await fluxImage(prompt, imgWidth, imgHeight, selectedServer);
+    const imageUrl = await fluxImage(prompt, imgWidth, imgHeight, selectedServer);
     
-    if (!result) {
+    if (!imageUrl) {
       return res.status(500).json({
         author: "Herza",
         success: false,
@@ -75,11 +102,17 @@ const handler = async (req, res) => {
       });
     }
 
+    const fileId = Date.now() + '_' + Math.random().toString(36).substring(7);
+    await downloadAndSaveImage(imageUrl, fileId);
+
+    const baseUrl = `${req.protocol}://${req.get('host')}`;
+    const localUrl = `${baseUrl}/tmp/flux-${fileId}.webp`;
+
     res.json({
       author: "Herza",
       success: true,
       data: {
-        result: result
+        result: localUrl
       }
     });
 
@@ -99,7 +132,7 @@ module.exports = {
   routes: ['api/AI/Flux'],
   tags: ['ai', 'image', 'flux', 'generator'],
   main: ['AI'],
-  parameters: ['prompt', 'width', 'height', 'server', 'key'],
+  parameters: ['prompt', 'key', 'width', 'height', 'server'],
   enabled: true,
   handler
 };