Update plugins/fluxai.js

plugins/fluxai.js

@@ -32,35 +32,57 @@ async function schellwithflux(args) {
  * @swagger
  * tags:
  *   name: FLUX
- *   description: etc
+ *   description: Flux AI image generation endpoints.
  */
 
 /**
  * @swagger
- * /api/v1/flux-ai:
- *   get:
- *     summary: Get a generator image from Flux AI image
+ * /api/v1/fluxai-ai:
+ *   post:
+ *     summary: Generate an image using Flux AI.
  *     tags: [FLUX]
- *     description: This endpoint interacts with the Flux AI
+ *     description: This endpoint interacts with Flux AI to generate an image based on the query.
+ *     requestBody:
+ *       required: true
+ *       content:
+ *         application/json:
+ *           schema:
+ *             type: object
+ *             properties:
+ *               query:
+ *                 type: string
+ *                 description: The input query for image generation.
+ *     responses:
+ *       200:
+ *         description: A successfully processed image.
+ *         content:
+ *           image/jpeg:
+ *             schema:
+ *               type: string
+ *               format: binary
+ *       400:
+ *         description: Missing query parameter.
+ *       500:
+ *         description: Internal server error.
  */
 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');
+            return res.status(400).send("Query parameter is missing");
         }
+
+        const imageBytes = await schellwithflux(query);
         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();
+
+        const processedImage = await sharp(buffer).jpeg().toBuffer();
+
         res.set("Content-Type", "image/jpeg");
-        const stream = Readable.from(processedImage);
-        stream.pipe(res);
+        Readable.from(processedImage).pipe(res);
     } catch (error) {
         console.error("Error processing image:", error.message);
         res.status(500).json({ error: "Internal server error" });