Update plugins/fluxai.js

plugins/fluxai.js

@@ -1,5 +1,6 @@
 import fetch from "node-fetch";
 import * as config from './config.js';
+const FluxRoutes = express.Router();
 
 /**
  * @param {string} args - The input string
@@ -27,4 +28,60 @@ async function schellwithflux(args) {
     }
 }
 
-export { schellwithflux };
+/**
+ * @swagger
+ * tags:
+ *   name: FLUX
+ *   description: etc
+ */
+
+/**
+ * @swagger
+ * /api/v1/flux-ai:
+ *   get:
+ *     summary: Get a generator image from Flux AI image
+ *     tags: [FLUX]
+ *     description: This endpoint interacts with the Flux AI
+ *     parameters:
+ *       - in: query
+ *         name: query
+ *         required: true
+ *         description: The query to be processed by the Flux AI.
+ *         schema:
+ *           type: string
+ *       500:
+ *         description: Internal server error.
+ *         content:
+ *           application/json:
+ *             schema:
+ *               type: object
+ *               properties:
+ *                 error:
+ *                   type: string
+ *                   description: Error message.
+ */
+FluxRoutes.post("/api/v1/fluxai-ai", async (req, res) => {
+    try {
+        const query = req.body.query;
+        const imageBytes = await schellwithflux(query);
+        if (!query) {
+          return res.status(400).send('Query parameter is missing');
+        }
+        if (!imageBytes) {
+            return res.status(500).json({ error: "Failed to fetch image bytes" });
+        }
+        const buffer = Buffer.isBuffer(imageBytes) ? imageBytes : Buffer.from(imageBytes);
+        
+        const processedImage = await sharp(buffer)
+            .jpeg()
+            .toBuffer();
+        res.set("Content-Type", "image/jpeg");
+        const stream = Readable.from(processedImage);
+        stream.pipe(res);
+    } catch (error) {
+        console.error("Error processing image:", error.message);
+        res.status(500).json({ error: "Internal server error" });
+    }
+});
+
+export { FluxRoutes };