Update server.js

server.js

@@ -1,56 +1,28 @@
 const express = require("express");
 const { chromium } = require("playwright");
 const bodyParser = require("body-parser");
-const fs = require("fs");
-const path = require("path");
-const os = require("os");
 
 const app = express();
 app.use(bodyParser.text({ type: "text/html" }));
 
-const tmpDir = os.tmpdir();
-
-function randomFilename() {
-  return Math.random().toString(36).slice(2, 6) + ".png";
-}
-
 app.post("/v1/start", async (req, res) => {
   try {
     const html = req.body;
-    if (!html) return res.status(400).json({ error: "HTML required" });
+    if (!html) return res.sendStatus(400);
 
-    const browser = await chromium.launch({ args: ["--no-sandbox", "--disable-setuid-sandbox"] });
+    const browser = await chromium.launch({ args: ["--no-sandbox"] });
     const page = await browser.newPage();
     await page.setContent(html, { waitUntil: "load" });
 
-    const filename = randomFilename();
-    const filepath = path.join(tmpDir, filename);
-    await page.screenshot({ path: filepath });
+    const buffer = await page.screenshot({ type: "png" });
 
     await browser.close();
 
-    res.json({ url: `${req.protocol}://${req.get("host")}/t/${filename}` });
-  } catch (error) {
-    console.error("Error:", error);
-    res.status(500).json({ error: "Internal Server Error" });
+    res.setHeader("Content-Type", "image/png");
+    res.send(buffer);
+  } catch {
+    res.sendStatus(500);
   }
 });
 
-app.use("/t", (req, res, next) => {
-  fs.readdir(tmpDir, (err, files) => {
-    if (err) return next();
-    const now = Date.now();
-    files.forEach((file) => {
-      const filePath = path.join(tmpDir, file);
-      fs.stat(filePath, (err, stats) => {
-        if (err) return;
-        if (now - stats.ctimeMs > 5 * 60 * 1000) {
-          fs.unlink(filePath, () => {});
-        }
-      });
-    });
-  });
-  next();
-}, express.static(tmpDir));
-
-app.listen(process.env.PORT || 7860, () => console.log("Server running..."));
+app.listen(7860);