Update index.js

index.js

@@ -95,7 +95,9 @@ app.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" });
         }