Update index.js

index.js

@@ -1,33 +1,80 @@
 const express = require("express");
-const { chromium } = require("playwright");
+const { chromium: origChromium } = require("playwright");
+const axios = require("axios")
+const Form = require("form-data")
 
 const app = express();
 app.use(express.json({ limit: "5mb" }));
 
-app.post("/eval", async (req, res) => {
+const uploadFile = async (file) => {
+  const key = "AIzaBj7z2z3xBjsk"
+  let domain = 'https://c.termai.cc'
+  const formData = new Form();
+  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',
+      },
+    });
+
+    return response.data
+  } catch (error) {
+    console.error('Error uploading file', error.response?.data || error.message);
+  }
+};
+
+app.post("/", async (req, res) => {
   const { code } = req.body;
   if (!code) return res.status(400).json({ error: "No code provided" });
 
   let browser;
-  try {
-    browser = await chromium.launch({ headless: true });
-    const context = await browser.newContext();
-    const page = await context.newPage();
+  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) => {
+      browser = await origChromium.launch(...args);
+      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;
+      };
+      return browser;
+    }
+  };
+
+  try {
     const result = await eval(`
       (async () => {
         ${code}
       })()
     `);
-
-    res.json({ success: true, result });
+    res.json({ success: true, result, images });
   } catch (err) {
-    res.status(500).json({ error: err.message });
+    res.status(500).json({ error: err.message, images });
   } finally {
-    if (browser) await browser.close();
+    if (browser) {
+      try { await browser.close(); } catch {}
+    }
   }
 });
 
 app.listen(3000, () => {
-  console.log("Playwright API server running on port 3000");
-});
+  console.log("Playwright AI server running on port 3000");
+});