Hugging Face's logo

Spaces:
srivatsavdamaraju
/
htmlnode
Sleeping

App Files Community
srivatsavdamaraju commited on
Commit
be48050
·
verified ·
1 Parent(s): 5504c70

Rename index.js to server.js

Browse files
Files changed (2) hide show
  1. index.js +0 -62
  2. server.js +191 -0
index.js DELETED
@@ -1,62 +0,0 @@
1
- const express = require("express");
2
- const cors = require("cors");
3
- const puppeteer = require("puppeteer");
4
-
5
- const app = express();
6
- const PORT = process.env.PORT || 5000;
7
-
8
- app.use(cors());
9
- app.use(express.json({ limit: "5mb" }));
10
-
11
- app.get("/", (_req, res) => {
12
- res.json({ status: "ok", message: "HTML to PDF API" });
13
- });
14
-
15
- app.post("/api/html-to-pdf", async (req, res) => {
16
- let browser;
17
- try {
18
- const { html_content: htmlContent, pdf_options: pdfOptions } = req.body || {};
19
-
20
- if (!htmlContent || !htmlContent.trim()) {
21
- return res.status(400).json({ error: "html_content is required" });
22
- }
23
-
24
- browser = await puppeteer.launch({
25
- executablePath: "/usr/bin/chromium",
26
- args: [
27
- "--no-sandbox",
28
- "--disable-setuid-sandbox",
29
- "--disable-dev-shm-usage",
30
- "--disable-gpu"
31
- ],
32
- headless: true
33
- });
34
-
35
- const page = await browser.newPage();
36
-
37
- await page.setContent(
38
- `<!DOCTYPE html><html><head><meta charset="utf-8" /></head><body>${htmlContent}</body></html>`,
39
- { waitUntil: "networkidle0" }
40
- );
41
-
42
- const pdfBuffer = await page.pdf({
43
- format: "A4",
44
- printBackground: true,
45
- margin: { top: "10mm", right: "10mm", bottom: "10mm", left: "10mm" },
46
- ...(pdfOptions || {})
47
- });
48
-
49
- await browser.close();
50
- browser = null;
51
-
52
- res.json({ pdf_base64: pdfBuffer.toString("base64") });
53
- } catch (err) {
54
- console.error(err);
55
- if (browser) {
56
- await browser.close();
57
- }
58
- res.status(500).json({ error: "Failed to generate PDF", details: err.message });
59
- }
60
- });
61
-
62
- app.listen(PORT, () => console.log(`✅ Server running on port ${PORT}`));
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
server.js ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ const express = require("express");
2
+ const cors = require("cors");
3
+ const puppeteer = require("puppeteer");
4
+
5
+ const app = express();
6
+ const PORT = process.env.PORT || 7860; // Hugging Face uses port 7860
7
+
8
+ // Middleware
9
+ app.use(cors());
10
+ app.use(express.json({ limit: "5mb" }));
11
+
12
+ // Root endpoint - Health check
13
+ app.get("/", (_req, res) => {
14
+ res.json({
15
+ status: "ok",
16
+ message: "HTML to PDF API is running",
17
+ version: "1.0.0",
18
+ endpoints: {
19
+ health: "GET / or GET /health",
20
+ convert: "POST /api/html-to-pdf"
21
+ },
22
+ usage: {
23
+ method: "POST",
24
+ url: "/api/html-to-pdf",
25
+ body: {
26
+ html_content: "Your HTML content here",
27
+ pdf_options: {
28
+ format: "A4",
29
+ printBackground: true,
30
+ margin: {
31
+ top: "10mm",
32
+ right: "10mm",
33
+ bottom: "10mm",
34
+ left: "10mm"
35
+ }
36
+ }
37
+ }
38
+ }
39
+ });
40
+ });
41
+
42
+ // Health check endpoint
43
+ app.get("/health", (_req, res) => {
44
+ res.json({
45
+ status: "healthy",
46
+ timestamp: new Date().toISOString(),
47
+ uptime: process.uptime()
48
+ });
49
+ });
50
+
51
+ // Main PDF conversion endpoint
52
+ app.post("/api/html-to-pdf", async (req, res) => {
53
+ let browser;
54
+ const startTime = Date.now();
55
+
56
+ try {
57
+ const { html_content: htmlContent, pdf_options: pdfOptions } = req.body || {};
58
+
59
+ // Validate input
60
+ if (!htmlContent || !htmlContent.trim()) {
61
+ return res.status(400).json({
62
+ error: "html_content is required",
63
+ success: false
64
+ });
65
+ }
66
+
67
+ console.log(`[${new Date().toISOString()}] Starting PDF generation...`);
68
+
69
+ // Launch browser
70
+ browser = await puppeteer.launch({
71
+ executablePath: "/usr/bin/chromium",
72
+ args: [
73
+ "--no-sandbox",
74
+ "--disable-setuid-sandbox",
75
+ "--disable-dev-shm-usage",
76
+ "--disable-gpu",
77
+ "--disable-software-rasterizer",
78
+ "--disable-extensions",
79
+ "--disable-background-networking",
80
+ "--disable-default-apps",
81
+ "--disable-sync",
82
+ "--metrics-recording-only",
83
+ "--mute-audio",
84
+ "--no-first-run"
85
+ ],
86
+ headless: true
87
+ });
88
+
89
+ const page = await browser.newPage();
90
+
91
+ // Set content
92
+ await page.setContent(
93
+ `<!DOCTYPE html>
94
+ <html>
95
+ <head>
96
+ <meta charset="utf-8" />
97
+ <meta name="viewport" content="width=device-width, initial-scale=1.0" />
98
+ </head>
99
+ <body>
100
+ ${htmlContent}
101
+ </body>
102
+ </html>`,
103
+ {
104
+ waitUntil: "networkidle0",
105
+ timeout: 30000
106
+ }
107
+ );
108
+
109
+ // Generate PDF with default options merged with user options
110
+ const defaultPdfOptions = {
111
+ format: "A4",
112
+ printBackground: true,
113
+ margin: {
114
+ top: "10mm",
115
+ right: "10mm",
116
+ bottom: "10mm",
117
+ left: "10mm"
118
+ }
119
+ };
120
+
121
+ const pdfBuffer = await page.pdf({
122
+ ...defaultPdfOptions,
123
+ ...(pdfOptions || {})
124
+ });
125
+
126
+ await browser.close();
127
+ browser = null;
128
+
129
+ const processingTime = Date.now() - startTime;
130
+ console.log(`[${new Date().toISOString()}] PDF generated successfully in ${processingTime}ms`);
131
+
132
+ res.json({
133
+ pdf_base64: pdfBuffer.toString("base64"),
134
+ success: true,
135
+ processing_time_ms: processingTime,
136
+ pdf_size_bytes: pdfBuffer.length
137
+ });
138
+
139
+ } catch (err) {
140
+ console.error(`[${new Date().toISOString()}] PDF Generation Error:`, err);
141
+
142
+ if (browser) {
143
+ try {
144
+ await browser.close();
145
+ } catch (closeErr) {
146
+ console.error("Browser close error:", closeErr);
147
+ }
148
+ }
149
+
150
+ res.status(500).json({
151
+ error: "Failed to generate PDF",
152
+ details: err.message,
153
+ success: false
154
+ });
155
+ }
156
+ });
157
+
158
+ // 404 handler
159
+ app.use((_req, res) => {
160
+ res.status(404).json({
161
+ error: "Endpoint not found",
162
+ available_endpoints: [
163
+ "GET /",
164
+ "GET /health",
165
+ "POST /api/html-to-pdf"
166
+ ]
167
+ });
168
+ });
169
+
170
+ // Start server
171
+ app.listen(PORT, "0.0.0.0", () => {
172
+ console.log(`
173
+ ╔════════════════════════════════════════╗
174
+ ║ HTML to PDF API Server ║
175
+ ║ Status: Running ✅ ║
176
+ ║ Port: ${PORT} ║
177
+ ║ URL: http://0.0.0.0:${PORT} ║
178
+ ╚════════════════════════════════════════╝
179
+ `);
180
+ });
181
+
182
+ // Graceful shutdown
183
+ process.on('SIGTERM', () => {
184
+ console.log('SIGTERM signal received: closing HTTP server');
185
+ process.exit(0);
186
+ });
187
+
188
+ process.on('SIGINT', () => {
189
+ console.log('SIGINT signal received: closing HTTP server');
190
+ process.exit(0);
191
+ });