Update server.js

server.js

@@ -11,44 +11,44 @@ let requestDetails = []; // Store details of each request
 let ipRequestCounts = {}; // Dictionary to keep track of requests per IP
 
 app.use('/api', (req, res, next) => {
-  // Check if the request is specifically for POST /v1/chat/completions
-  if (req.method === 'POST' && req.url === '/v1/chat/completions') {
+  // Continue only if it's a POST to /v1/chat/completions
+  if (req.method === 'POST' && req.url.startsWith('/v1/chat/completions')) {
     const ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
-    const currentCount = ipRequestCounts[ip] || 0;
+    ipRequestCounts[ip] = (ipRequestCounts[ip] || 0) + 1;
 
-    if (currentCount >= 10) {
-      // When the limit is reached, return a 429 status
-      return res.status(429).send('You have reached the 10 message limit. Contact me on reddit for more.');
+    if (ipRequestCounts[ip] > 10) {
+      console.log(`IP ${ip} hit rate limit with ${ipRequestCounts[ip]} requests.`);
+      return res.status(429).send('You have reached the 10 message limit. Please contact support for more information.');
     }
 
-    // Increment the count only if the condition is true
-    ipRequestCounts[ip] = currentCount + 1;
+    console.log(`IP ${ip} has made ${ipRequestCounts[ip]} requests so far.`);
   }
 
-  // Proceed with logging request details regardless of the endpoint
+  // Log request details
   const timestamp = new Date().toISOString();
   const requestData = {
-    requestNumber: ipRequestCounts[req.connection.remoteAddress] || 0, // Update to show request count
+    requestNumber: ipRequestCounts[req.connection.remoteAddress] || 0,
     ip: req.connection.remoteAddress,
     timestamp: timestamp,
     text: req.method + ' ' + req.url
   };
-
   requestDetails.push(requestData);
   console.log(`Request ${requestData.requestNumber} on ${requestData.timestamp} from IP ${requestData.ip} with method "${req.method}" and text "${requestData.text}"`);
 
+  // Process the request as usual
   next();
 }, proxy(targetUrl, {
-  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
+  proxyReqOptDecorator: function(proxyReqOpts, srcReq) {
+    // Set authorization headers for OpenAI API
     proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
     return proxyReqOpts;
-  },
+  }
 }));
 
 app.get("/", (req, res) => {
   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}`);
+  res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}.<br><br>Requests Overview:<br>${requestsInfo}`);
 });
 
 function getExternalUrl(spaceId) {
@@ -56,10 +56,10 @@ function getExternalUrl(spaceId) {
     const [username, spacename] = spaceId.split("/");
     return `https://${username}-${spacename.replace(/_/g, "-")}.hf.space/api/v1`;
   } catch (e) {
-    return "";
+    return "Error generating external URL";
   }
 }
 
 app.listen(port, () => {
-  console.log(`Reverse proxy server running on ${baseUrl}.`);
+  console.log(`Reverse proxy server running on ${baseUrl}`);
 });