Update index.js

index.js

@@ -14,18 +14,30 @@ const uploadFile = async (file) => {
   formData.append('file', file, { filename: 'file.png'});
 
   try {
-    const response = await axios.post(`${domain}/api/upload?uploader=drive&key=`+key, formData, {
-      headers: {
-        'Content-Type': 'multipart/form-data',
-      },
+    const response = await axios.post(`${domain}/api/upload?uploader=drive&key=` + key, formData, {
+      headers: formData.getHeaders(), 
+      maxBodyLength: Infinity,
     });
 
-    return response.data
+    return response.data || {};
   } catch (error) {
     console.error('Error uploading file', error.response?.data || error.message);
+    return {};
   }
 };
 
+const patchScreenshot = (page) => {
+  const orig = page.screenshot;
+  page.screenshot = async function (options = {}) {
+    const buffer = await orig.apply(this, [options]);
+    const url = await uploadFile(buffer);
+    if (url && url.path) { 
+      images.push(url.path);
+    }
+    return buffer;
+  };
+};
+
 app.all("/", async (req, res) => {
   const code = req.body?.code || req.query?.code;
   if (!code) return res.status(400).json({ error: "No code provided" });
@@ -33,17 +45,6 @@ app.all("/", async (req, res) => {
   let browser;
   const images = [];
 
-  const patchScreenshot = (target) => {
-    const orig = target.screenshot;
-    target.screenshot = async function (...args) {
-      const buffer = await orig.apply(this, args);
-      const url = await uploadFile(buffer); 
-      delete url.status
-      images.push(url);
-      return buffer; 
-    };
-  };
-
   const chromium = {
     ...origChromium,
     launch: async (...args) => {
@@ -51,8 +52,6 @@ app.all("/", async (req, res) => {
       const origNewPage = browser.newPage;
       browser.newPage = async (...a) => {
         const page = await origNewPage.apply(browser, a);
-        patchScreenshot(page); 
-        const origElemScreenshot = page.locator("").screenshot;
         patchScreenshot(page);
         return page;
       };