Image Generation Fix (This prompt has been blocked by Bing)

[LucasDiscordCzech]

Fix for #11 issue

If you're having issues with generating images, eg 'Bing has blocked this prompt.... ".

I have found a fix, aswell I have fixed an issue where it would save images that are not working (corrupted) or just shouldn't be downloaded.

All you need to do is add more cookies from the Bing website, which seems to fix it permanently.

Code (bimg/node_modules/dist/imgen.js):

import axios from "axios";
import fs from "fs";
import path from "path";
import querystring from "querystring";
import { performance } from "perf_hooks";
import Config from "./config.js";
const BING_URL = "https://www.bing.com";
const createSession = (authCookie) => {
    const session = axios.create({
        headers: {
            accept: "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
            "accept-encoding": "gzip, deflate, br",
            "accept-language": "en-US,en;q=0.9,zh-CN;q=0.8,zh-TW;q=0.7,zh;q=0.6",
            "cache-control": "max-age=0",
            "content-type": "application/x-www-form-urlencoded",
            "Referrer-Policy": "origin-when-cross-origin",
            referrer: "https://www.bing.com/images/create/",
            origin: "https://www.bing.com",
            "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36 Edg/111.0.1661.54",
          cookie: `_U=${authCookie}; SRCHHPGUSR=${process.env.SRCHHPGUSR}`, // do this for all the cookies after getting all the cookies from a network request using dev tools, should be about 14 cookies here
            "sec-ch-ua": `"Microsoft Edge";v="111", "Not(A:Brand";v="8", "Chromium";v="111"`,
            "sec-ch-ua-mobile": "?0",
            "sec-fetch-dest": "document",
            "sec-fetch-mode": "navigate",
            "sec-fetch-site": "same-origin",
            "sec-fetch-user": "?1",
            "upgrade-insecure-requests": "1",
        },
    });
    return session;
};
const getImages = async (session, prompt) => {
    console.log("Sending request...");
    const urlEncodedPrompt = querystring.escape(prompt);
    const url = `${BING_URL}/images/create?q=${urlEncodedPrompt}&rt=3&FORM=GENCRE`; // force use rt=3
    console.log(url);
const response = await session.post(url, {
        maxRedirects: 0,
        validateStatus: function (status) {
            return status >= 200 && status < 303;
        },
        timeout: 200000,
    });
    let redirectUrl;
    // toHTMLfile(response, "example.html");
    if (response.status == 200) {
        redirectUrl = response.request.res.responseUrl.replace("&nfy=1", "");
    }
    else if (response.status !== 302) {
        console.error(`BINGERR:  the status is ${response.status} instead of 302 or 200`);
        throw new Error("BINGERR: Redirect failed");
    }
    console.log("Redirected to", redirectUrl);
    const requestId = redirectUrl.split("id=")[1];
    await session.get(redirectUrl);
    const pollingUrl = `${BING_URL}/images/create/async/results/${requestId}?q=${urlEncodedPrompt}`;
    console.log("Waiting for results...");
    const startWait = performance.now();
    let imagesResponse;
    while (true) {
        if (performance.now() - startWait > 300000) {
            throw new Error("BINGERR: Timeout error");
        }
        console.log(".", { end: "", flush: true });
        imagesResponse = await session.get(pollingUrl);
        if (imagesResponse.status !== 200) {
            throw new Error("BINGERR: Could not get results");
        }
        if (imagesResponse.data === "") {
            await new Promise((resolve) => setTimeout(resolve, 1000));
            continue;
        }
        else {
            break;
        }
    }
    if (imagesResponse.data.errorMessage === "Pending") {
        throw new Error("BINGERR: This prompt has been blocked by Bing. Bing's system flagged this prompt because it may conflict with their content policy. More policy violations may lead to automatic suspension of your access.");
    }
    else if (imagesResponse.data.errorMessage) {
        throw new Error("BINGERR: Bing returned an error: " + imagesResponse.data.errorMessage);
    }
    const imageLinks = imagesResponse.data
        .match(/src="([^"]+)"/g)
        .map((src) => src.slice(5, -1));
    const normalImageLinks = Array.from(new Set(imageLinks.map((link) => link.split("?w=")[0])));
    const badImages = [
        "https://r.bing.com/rp/in-2zU3AJUdkgFe7ZKv19yPBHVs.png",
        "https://r.bing.com/rp/TX9QuO3WzcCJz1uaaSwQAz39Kb0.jpg",
    ];
    for (const im of normalImageLinks) {
        if (badImages.includes(im)) {
            throw new Error("BINGERR: Bad images");
        }
    }
    if (normalImageLinks.length === 0) {
        throw new Error("BINGERR: No images");
    }
    return normalImageLinks;
};
const saveImages = async (session, links, outputDir) => {
    console.log("\nDownloading images...");
    try {
        fs.mkdirSync(outputDir, { recursive: true });
    }
    catch (err) {
        if (err.code !== "EEXIST")
            throw err;
    }
          let imageNum = 0;
          for (const link of links) {
              try {
                  const response = await session.get(link, { responseType: "stream" });

                  // Check if the content type is JPEG
                  if (response.headers['content-type'] === 'image/jpeg') {
                      const outputPath = path.join(outputDir, `${imageNum}.jpeg`);
                      const writer = fs.createWriteStream(outputPath);
                      response.data.pipe(writer);
                      await new Promise((resolve, reject) => {
                          writer.on("finish", resolve);
                          writer.on("error", reject);
                      });
                      imageNum += 1;
                  } else {
                      console.log("Skipped a non-JPEG image");
                  }
              }
              catch (err) {
            if (err instanceof axios.AxiosError) {
                throw new Error("BINGERR: Inappropriate contents found in the generated images. Please try again or try another prompt.");
            }
            else {
                throw err.message;
            }
        }
    }
};
export const generateImagesLinks = async (prompt) => {
    const authCookie = Config.bingImageCookie;
    const outputDir = `${Config.tempDir}/${prompt}`;
    if (!authCookie || !prompt) {
        throw new Error("BINGERR: Missing parameters");
    }
    // Create image generator session
    const session = createSession(authCookie);
    const imageLinks = await getImages(session, prompt);
    return imageLinks;
};
export const generateImageFiles = async (prompt) => {
    const authCookie = Config.bingImageCookie;
    const outputDir = `${Config.tempDir}/${prompt}`;
    if (!authCookie || !prompt) {
        throw new Error("BINGERR: Missing parameters");
    }
    // Create image generator session
    const session = createSession(authCookie);
    const imageLinks = await getImages(session, prompt);
    await saveImages(session, imageLinks, outputDir);
    // Read saved images from the output directory
    const imageFiles = fs.readdirSync(outputDir);
    const images = imageFiles.map((filename) => {
        const filePath = path.join(outputDir, filename);
        const fileData = fs.readFileSync(filePath);
        return {
            filename,
            data: fileData.toString("base64"),
        };
    });
    return images;
};

Hope this helps! 😊

[prismspecs]

I appreciate your fix but it still gives me the same error message