Update app.js

app.js

@@ -1,5 +1,5 @@
 const express = require('express');
-const { chromium } = require('playwright');
+const { chromium, devices } = require('playwright');
 const cors = require('cors');
 const dotenv = require('dotenv');
 const os = require('os');
@@ -53,6 +53,46 @@ const utils = {
     }
 };
 
+const utils2 = {
+    async initialize() {
+        if (!browser) {
+            browser = await chromium.launch({ headless: true });
+            const context = await browser.newContext({
+                viewport: config.viewport,
+                userAgent: config.userAgent,
+              ...devices['iPhone 13'],
+            });
+
+            await context.route('**/*', (route) => {
+                const url = route.request().url();
+                if (url.endsWith('.png') || url.endsWith('.jpg') || url.includes('google-analytics')) {
+                    return route.abort();
+                }
+                route.continue();
+            });
+
+            page = await context.newPage();
+            await page.goto('https://www.bratgenerator.com/', { waitUntil: 'domcontentloaded', timeout: 10000 });
+
+            try {
+                await page.click('#onetrust-accept-btn-handler', { timeout: 2000 });
+            } catch { }
+
+            await page.evaluate(() => setupTheme('white'));
+        }
+    },
+
+    async generateBrat(text) {
+        await page.fill('#textInput', text);
+        const overlay = page.locator('#textOverlay');
+        return overlay.screenshot({ timeout: 3000 });
+    },
+
+    async close() {
+        if (browser) await browser.close();
+    }
+};
+
 const app = express();
 app.use(express.json());
 app.use(cors());
@@ -60,7 +100,7 @@ app.set('json spaces', 3);
 
 app.get('*', async (req, res) => {
     try {
-        const { q } = req.query;
+        const { q, type } = req.query;
         if (!q) {
             return res.status(200).json({
                 name: 'HD Bart Generator API',
@@ -76,9 +116,15 @@ app.get('*', async (req, res) => {
                 }
             });
         }
-        const imageBuffer = await utils.generateBrat(q);
-        res.set('Content-Type', 'image/png');
-        res.send(imageBuffer);
+        if (type === "iphone") {
+            const imageBuffer = await utils2.generateBrat(q);
+            res.set('Content-Type', 'image/png');
+            res.send(imageBuffer);
+        } else {
+            const imageBuffer = await utils.generateBrat(q);
+            res.set('Content-Type', 'image/png');
+            res.send(imageBuffer);
+        }
     } catch (error) {
         console.error(error);
         res.status(500).json({
@@ -94,9 +140,11 @@ const PORT = process.env.PORT || 7860;
 app.listen(PORT, async () => {
     console.log(`Server running on port ${PORT}`);
     await utils.initialize();
+    await utils2.initialize();
 });
 
 process.on('SIGINT', async () => {
     await utils.close();
+    await utils2.close();
     process.exit(0);
 });