Update server.js

server.js

@@ -1,6 +1,11 @@
 const express = require('express');
+const http = require('http');
 const proxy = require('express-http-proxy');
+const socketIo = require('socket.io');
+
 const app = express();
+const server = http.createServer(app);
+const io = socketIo(server);
 
 const targetUrl = 'https://api.openai.com';
 const openaiKey = process.env.OPENAI_KEY;
@@ -10,36 +15,45 @@ const baseUrl = getExternalUrl(process.env.SPACE_ID);
 let requestDetails = []; // Store details of each request
 let ipRequestCounts = {}; // Dictionary to keep track of requests per IP
 
+io.on('connection', (socket) => {
+    console.log('A user connected to the websocket for live logs.');
+});
+
+function logAndEmit(message) {
+    console.log(message);  // Continue to log to the console
+    io.emit('log', message);  // Emit this log to the frontend via WebSocket
+}
+
 app.use('/api', (req, res, next) => {
-  // 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;
     ipRequestCounts[ip] = (ipRequestCounts[ip] || 0) + 1;
 
     if (ipRequestCounts[ip] > 10) {
-      console.log(`IP ${ip} hit rate limit with ${ipRequestCounts[ip]} requests.`);
+      const rateLimitMessage = `IP ${ip} hit rate limit with ${ipRequestCounts[ip]} requests.`;
+      logAndEmit(rateLimitMessage);
       return res.status(429).send('You have reached the 10 message limit. Please contact support for more information.');
     }
-
-    console.log(`IP ${ip} has made ${ipRequestCounts[ip]} requests so far.`);
+    
+    const logMessage = `IP ${ip} has made ${ipRequestCounts[ip]} requests so far.`;
+    logAndEmit(logMessage);
   }
 
-  // Log request details
   const timestamp = new Date().toISOString();
   const requestData = {
-    requestNumber: ipRequestCounts[req.connection.remoteAddress] || 0,
-    ip: req.connection.remoteAddress,
+    requestNumber: ipRequestCounts[req.headers['x-forwarded-for'] || req.connection.remoteAddress],
+    ip: req.headers['x-forwarded-for'] || 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}"`);
+  
+  const detailedLog = `Request ${requestData.requestNumber} on ${requestData.timestamp} from IP ${requestData.ip} with method "${req.method}" and text "${requestData.text}"`;
+  logAndEmit(detailedLog);
 
-  // Process the request as usual
   next();
 }, proxy(targetUrl, {
   proxyReqOptDecorator: function(proxyReqOpts, srcReq) {
-    // Set authorization headers for OpenAI API
     proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
     return proxyReqOpts;
   }
@@ -51,6 +65,10 @@ app.get("/", (req, res) => {
   res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}.<br><br>Requests Overview:<br>${requestsInfo}`);
 });
 
+app.get('/logs', (req, res) => {
+    res.sendFile(__dirname + '/index.html');
+});
+
 function getExternalUrl(spaceId) {
   try {
     const [username, spacename] = spaceId.split("/");
@@ -60,6 +78,6 @@ function getExternalUrl(spaceId) {
   }
 }
 
-app.listen(port, () => {
-  console.log(`Reverse proxy server running on ${baseUrl}`);
+server.listen(port, () => {
+   logAndEmit(`Reverse proxy server and WebSocket running on port ${port}`);
 });