Update plugins/fluxai.js

plugins/fluxai.js

@@ -33,6 +33,66 @@ async function schellwithflux(args) {
     }
 }
 
+/**
+ * @swagger
+ * /api/v1/flux/black-forest-labs/flux-1-schnell:
+ *   get:
+ *     summary: Deepseek R1
+ *     tags: [FLUX]
+ *     parameters:
+ *       - in: query
+ *         name: query
+ *         required: true
+ *         description: User's input query
+ *         schema:
+ *           type: string
+ *       - in: header
+ *         name: x-api-key
+ *         required: true
+ *         description: API key for authentication
+ *         schema:
+ *           type: string
+ *     responses:
+ *       200:
+ *         description: Success
+ */
+/**
+ * @swagger
+ * /api/v1/flux/black-forest-labs/flux-1-schnell:
+ *   get:
+ *     summary: black forest labs
+ *     tags: [FLUX]
+ *     parameters:
+ *       - in: query
+ *         name: query
+ *         required: true
+ *         description: User's input query
+ *         schema:
+ *           type: string
+ *       - in: header
+ *         name: x-api-key
+ *         required: true
+ *         description: API key for authentication
+ *         schema:
+ *           type: string
+ *     responses:
+ *       200:
+ *         description: Success
+ */
+FluxRoutes.get('/api/v1/flux/black-forest-labs/flux-1-schnell', authenticateApiKeyPremium, apiLimiter, async (req, res) => {
+    try {
+        const query = req.query.query;
+        const result = await FluxSchnell(query);
+        const imageBytes = Buffer.from(result, "base64");
+        const processedImage = await sharp(imageBytes).jpeg().toBuffer();
+        res.set("Content-Type", "image/jpeg");
+        Readable.from(processedImage).pipe(res);
+    } catch (error) {
+        res.status(401).json({ error: error.message });
+    }
+});
+
+
 /**
  * @swagger
  * /api/v1/flux/fluxai-ai: