Update index.js

index.js

@@ -26,7 +26,7 @@ const uploadFile = async (file) => {
   }
 };
 
-const patchScreenshot = (page) => {
+const patchScreenshot = (page, images) => {
   const orig = page.screenshot;
   page.screenshot = async function (options = {}) {
     const buffer = await orig.apply(this, [options]);
@@ -38,12 +38,13 @@ const patchScreenshot = (page) => {
   };
 };
 
+
 app.all("/", async (req, res) => {
   const code = req.body?.code || req.query?.code;
   if (!code) return res.status(400).json({ error: "No code provided" });
 
   let browser;
-  const images = [];
+  let images = [];
 
   const chromium = {
     ...origChromium,
@@ -52,7 +53,7 @@ app.all("/", async (req, res) => {
       const origNewPage = browser.newPage;
       browser.newPage = async (...a) => {
         const page = await origNewPage.apply(browser, a);
-        patchScreenshot(page);
+        patchScreenshot(page, images);
         return page;
       };
       return browser;