Update app.js

app.js

@@ -2148,13 +2148,22 @@ app.post("/playwright", async (req, res) => {
     if (!code) return res.status(400).json({ output: "code is required" });
 
     try {
-        const vm = new NodeVM({
-            require: { external: ["playwright"], builtin: [] },
-        });
+        const logs = [];
+        const originalConsoleLog = console.log;
+        console.log = (...args) => {
+            logs.push(args.map(arg => (typeof arg === "object" ? JSON.stringify(arg, null, 2) : String(arg))).join(" "));
+            originalConsoleLog(...args);
+        };
+
+        const vm = new NodeVM({ require: { external: ["playwright"], builtin: [] } });
+        const wrappedCode = `(async () => { ${code} })();`;
+        const result = await vm.run(wrappedCode, import.meta.url);
+        if (result !== undefined) logs.push(String(result));
 
-        const functionInSandbox = vm.run(`module.exports = async (callback) => { ${code} };`, import.meta.url);
-        functionInSandbox((output) => res.json({ output }));
+        console.log = originalConsoleLog;
+        res.json({ output: logs.length ? logs.join("\n") : "No output captured" });
     } catch (error) {
+        console.log = originalConsoleLog;
         res.status(500).json({ output: error.message });
     }
 });