Update server.js

server.js

@@ -15,20 +15,25 @@ function randomFilename() {
 }
 
 app.post("/v1/start", async (req, res) => {
-  const html = req.body;
-  if (!html) return res.status(400).send("HTML required");
+  try {
+    const html = req.body;
+    if (!html) return res.status(400).json({ error: "HTML required" });
 
-  const browser = await chromium.launch({ args: ["--no-sandbox"] });
-  const page = await browser.newPage();
-  await page.setContent(html);
+    const browser = await chromium.launch({ args: ["--no-sandbox", "--disable-setuid-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 filename = randomFilename();
+    const filepath = path.join(tmpDir, filename);
+    await page.screenshot({ path: filepath });
 
-  await browser.close();
+    await browser.close();
 
-  res.json({ url: `${req.protocol}://${req.get("host")}/t/${filename}` });
+    res.json({ url: `${req.protocol}://${req.get("host")}/t/${filename}` });
+  } catch (error) {
+    console.error("Error:", error);
+    res.status(500).json({ error: "Internal Server Error" });
+  }
 });
 
 app.use("/t", (req, res, next) => {
@@ -48,4 +53,4 @@ app.use("/t", (req, res, next) => {
   next();
 }, express.static(tmpDir));
 
-app.listen(process.env.PORT || 7860);
+app.listen(process.env.PORT || 7860, () => console.log("Server running..."));