Update server.js

server.js

@@ -10,12 +10,29 @@ const baseUrl = getExternalUrl(process.env.SPACE_ID);
 let requestDetails = [];
 let ipRequestCounts = {};
 
+// Declare the proxy as middleware which can be reused
+const openAiProxy = proxy(targetUrl, {
+    proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
+        proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
+        return proxyReqOpts;
+    },
+    userResDecorator: (proxyRes, proxyResData, userReq, userRes) => {
+        const data = proxyResData.toString('utf8');
+        if (proxyRes.statusCode === 429) {
+            // Handling at userResDecorator might be skipped based on your situation, thus handling it later
+            return data;
+        }
+        return data;  // Pass through all other non-429 responses
+    }
+});
+
+// Apply rate limiting and API proxying
 app.use('/api', (req, res, next) => {
     const ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
     const currentCount = ipRequestCounts[ip] || 0;
 
     if (currentCount >= 1) {
-        res.status(429).send('You have reached your request limit of 10 messages. Please contact support for more details.');
+        res.status(429).send('You have reached your request limit of 1 message. Please contact support for more details.');
         return;
     }
 
@@ -31,20 +48,19 @@ app.use('/api', (req, res, next) => {
     console.log(`Request ${requestData.requestNumber} from IP ${requestData.ip} on ${requestData.timestamp}`);
 
     next();
-}, proxy(targetUrl, {
-    proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
-        proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
-        return proxyReqOpts;
-    },
-    userResDecorator: (proxyRes, proxyResData, userReq, userRes) => {
-        if (proxyRes.statusCode === 429) {
-            return JSON.stringify({
-                message: 'Request limit reached on the proxied service. Please try again later.'
+}, openAiProxy);
+
+// Modify proxy responses after the proxy has finished
+app.use('/api', function(req, res, next) {
+    res.on('finish', function() {
+        if (res.statusCode === 429 && !res.headersSent) {
+            res.status(429).json({
+                message: 'Custom message for 429 status.'
             });
         }
-        return proxyResData;  // Return the original response body if not a 429 error
-    }
-}));
+    });
+    next();
+});
 
 app.get("/", (req, res) => {
     let requestsInfo = requestDetails.map(detail =>