Update server.js

server.js

@@ -1,62 +1,60 @@
 const express = require('express');
 const proxy = require('express-http-proxy');
-const bodyParser = require('body-parser');
 const app = express();
 
 const targetUrl = 'https://api.openai.com';
 const openaiKey = process.env.OPENAI_KEY;
-const adminPassword = 'securepassword'; // Never use hard-coded sensitive passwords in production
 const port = 7860;
+const baseUrl = getExternalUrl(process.env.SPACE_ID);
 
-app.use(bodyParser.json());
+let requestDetails = [];
+let ipRequestCounts = {};
 
-// In-memory configuration "database"
-let ipConfigurations = {};
-
-// Rate limit middleware
 app.use('/api', (req, res, next) => {
   const ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
-  const config = ipConfigurations[ip] || { limit: 2, requestCount: 0, nextAvailableTime: Date.now() };
-
-  if (new Date() < new Date(config.nextAvailableTime) || config.requestCount >= config.limit) {
-    return res.status(429).json({
-      error: true,
-      message: 'You have reached your maximum request limit.',
-      nextAvailableRequestTime: getNextAvailableRequestTime(config)
-    });
+  const currentCount = ipRequestCounts[ip] || 0;
+
+  if (currentCount >= 2) {
+    return res.status(429).send('You have reached your request limit of 10 messages. Please contact support for more details.');
   }
 
-  // Increment the request count
-  config.requestCount++;
-  ipConfigurations[ip] = config; // Update config for the IP
-  console.log(`Allowed request for ${ip}: ${config.requestCount} of ${config.limit}`);
+  ipRequestCounts[ip] = currentCount + 1;
+  const timestamp = new Date().toISOString();
+  const requestData = {
+    requestNumber: ipRequestCounts[ip],
+    ip: ip,
+    timestamp: timestamp,
+    text: req.method + ' ' + req.url
+  };
+  requestDetails.push(requestData);
+  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;
   },
-}));
-
-// Admin route to update IP configurations
-app.post('/admin/config', (req, res) => {
-  const { password, ip, limit } = req.body;
-
-  if (password !== adminPassword) {
-    return res.status(403).send("Unauthorized");
+  userResDecorator: (proxyRes, proxyResData, userReq, userRes) => {
+    if (proxyRes.statusCode === 429) {
+      // Return a custom message for the 429 error from the proxy attempt
+      return Buffer.from('Request limit reached on the proxied service. Please try again later.');
+    }
+    return proxyResData;  // Return original data if response is not 429
   }
+}));
 
-  ipConfigurations[ip] = { limit: limit, requestCount: 0, nextAvailableTime: Date.now() };
-  res.send(`Configuration set for IP ${ip} with limit ${limit}`);
+app.get("/", (req, res) => {
+  let requestsInfo = requestDetails.map(detail =>
+    `Request ${detail.requestNumber} from IP ${detail.ip} on ${detail.timestamp}`).join('<br>');
+  res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}.<br><br>${requestsInfo}`);
 });
 
-// Helper function to get next available request time
-function getNextAvailableRequestTime(config) {
-  const resetTime = new Date(config.nextAvailableTime);
-  return resetTime.toISOString();
+function getExternalUrl(spaceId) {
+  const [username, spacename] = spaceId.split("/");
+  return `https://${username}-${spacename.replace(/_/g, "-")}.hf.space/api/v1`;
 }
 
-// Start the express server
 app.listen(port, () => {
-  console.log(`Reverse proxy server running on port ${port}.`);
+  console.log(`Reverse proxy server running on ${baseUrl}.`);
 });