File size: 2,043 Bytes
2889205 7f84064 2889205 7f84064 2889205 cb34952 7f84064 c619948 cb34952 a627ace cb34952 a627ace cb34952 c619948 cb34952 c619948 cb34952 7f84064 2889205 7f84064 2889205 cb34952 c619948 2889205 c619948 7f84064 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
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 requestDetails = []; // Store details of each request
let ipRequestCounts = {}; // Dictionary to keep track of requests per IP
app.use('/api', (req, res, next) => {
const ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
const currentCount = ipRequestCounts[ip] || 0;
if (currentCount >= 10) {
// Send a more generic message without mentioning IP-based limitation
return res.status(429).send('You reached 10 message limit. Contact me on reddit for more.');
}
// Increment request counter for the IP
ipRequestCounts[ip] = currentCount + 1;
// Log request with timestamp
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} on ${requestData.timestamp} 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} 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}`);
});
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}.`);
});
|