server.js

const express = require('express');
const proxy = require('express-http-proxy');
const app = express();

const targetUrl = 'https://api.openai.com';
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

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 requestData = {
      requestNumber: requestCount,
      ip: ip,
      text: req.method + ' ' + req.url // Example: captures request method and path
  };
  
  // Push details to the array
  requestDetails.push(requestData);
  
  console.log(`Request ${requestData.requestNumber} from IP ${requestData.ip} with text "${requestData.text}"`);
  next();
}, proxy(targetUrl, {
  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
    proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
    return proxyReqOpts;
  },
}));

app.get("/", (req, res) => {
  let requestsInfo = requestDetails.map(detail => `Request ${detail.requestNumber} from IP ${detail.ip} with text "${detail.text}"`).join('<br>');
  res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}.<br><br>${requestsInfo}`);
});

function getExternalUrl(spaceId) {
  try {
    const [username, spacename] = spaceId.split("/");
    return `https://${username}-${spacename.replace(/_/g, "-")}.hf.space/api/v1`;
  } catch (e) {
    return "";
  }
}

app.listen(port, () => {
  console.log(`Reverse proxy server running on ${baseUrl}.`);
});