Update app.js

app.js

@@ -1,13 +1,33 @@
 const express = require('express');
 const puppeteer = require('puppeteer-extra');
 const StealthPlugin = require('puppeteer-extra-plugin-stealth');
+const multer = require('multer');
+const path = require('path');
+const fs = require('fs');
 
 puppeteer.use(StealthPlugin());
 
 const app = express();
 const PORT = process.env.PORT || 7860;
 
+const storage = multer.diskStorage({
+    destination: (req, file, cb) => {
+        const uploadsDir = './uploads';
+        if (!fs.existsSync(uploadsDir)) {
+            fs.mkdirSync(uploadsDir);
+        }
+        cb(null, uploadsDir);
+    },
+    filename: (req, file, cb) => {
+        const timestamp = Date.now();
+        cb(null, `screenshot-${timestamp}.png`);
+    }
+});
+
+const upload = multer({ storage });
+
 app.use(express.json());
+app.use('/uploads', express.static('uploads'));
 
 const getRandomUserAgent = () => {
     const userAgents = [
@@ -200,6 +220,22 @@ app.get('/extract-token', async (req, res) => {
             }
         }
 
+        const timestamp = Date.now();
+        const filename = `screenshot-${timestamp}.png`;
+        const filepath = path.join('./uploads', filename);
+        
+        if (!fs.existsSync('./uploads')) {
+            fs.mkdirSync('./uploads');
+        }
+
+        const screenshot = await page.screenshot({ 
+            path: filepath,
+            fullPage: false,
+            type: 'png'
+        });
+
+        const screenshotUrl = `${req.protocol}://${req.get('host')}/uploads/${filename}`;
+
         if (capturedToken) {
             console.log('Token extracted successfully');
             res.json({ 
@@ -207,6 +243,8 @@ app.get('/extract-token', async (req, res) => {
                 accessToken: capturedToken,
                 apiData: capturedData,
                 source: capturedData ? 'network_request' : 'storage',
+                screenshotUrl: screenshotUrl,
+                screenshotPath: filepath,
                 url: url,
                 timestamp: new Date().toISOString()
             });
@@ -215,6 +253,8 @@ app.get('/extract-token', async (req, res) => {
             res.json({ 
                 success: false,
                 message: 'No accessToken found in network requests or storage',
+                screenshotUrl: screenshotUrl,
+                screenshotPath: filepath,
                 url: url,
                 timestamp: new Date().toISOString()
             });