Update server.js

server.js

@@ -7,24 +7,32 @@ const openaiKey = process.env.OPENAI_KEY;
 const port = 7860;
 const baseUrl = getExternalUrl(process.env.SPACE_ID);
 
-let requestCount = 0; // Counter to keep track of the number of proxied requests
-let requestDetails = []; // Array to store details of each request
+let requestDetails = []; // Store details of each request
+let ipRequestCounts = {}; // Dictionary to keep track of requests per IP
 
 app.use('/api', (req, res, next) => {
-  requestCount++; // Increment request count for every request passed to /api
-  
-  // Capture the request details
   const ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
+  const currentCount = ipRequestCounts[ip] || 0;
+
+  if (currentCount >= 20) {
+    // Send a more generic message without mentioning IP-based limitation
+    return res.status(429).send('Too many requests. Please try again later.');
+  }
+
+  // Increment request counter for the IP
+  ipRequestCounts[ip] = currentCount + 1;
+
+  // Log request with timestamp
+  const timestamp = new Date().toISOString();
   const requestData = {
-      requestNumber: requestCount,
-      ip: ip,
-      text: req.method + ' ' + req.url // Example: captures request method and path
+    requestNumber: ipRequestCounts[ip],
+    ip: ip,
+    timestamp: timestamp,
+    text: req.method + ' ' + req.url
   };
-  
-  // Push details to the array
   requestDetails.push(requestData);
-  
-  console.log(`Request ${requestData.requestNumber} from IP ${requestData.ip} with text "${requestData.text}"`);
+  console.log(`Request ${requestData.requestNumber} on ${requestData.timestamp} from IP ${requestData.ip} with text "${requestData.text}"`);
+
   next();
 }, proxy(targetUrl, {
   proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
@@ -34,7 +42,8 @@ app.use('/api', (req, res, next) => {
 }));
 
 app.get("/", (req, res) => {
-  let requestsInfo = requestDetails.map(detail => `Request ${detail.requestNumber} from IP ${detail.ip} with text "${detail.text}"`).join('<br>');
+  let requestsInfo = requestDetails.map(detail =>
+    `Request ${detail.requestNumber} on ${detail.timestamp} from IP ${detail.ip} with text "${detail.text}"`).join('<br>');
   res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}.<br><br>${requestsInfo}`);
 });