Create server.js

server.js

@@ -0,0 +1,54 @@
+const express = require('express');
+const proxy = require('express-http-proxy');
+const app = express();
+const bodyParser = require('body-parser');
+const targetUrl = 'https://api.openai.com';
+const openaiKey = process.env.OPENAI_KEY;
+const proxyKey = process.env.PROXY_KEY; // Your secret proxy key
+const port = 7860;
+const baseUrl = getExternalUrl(process.env.SPACE_ID);
+
+app.use(bodyParser.json({ limit: '50mb' }));
+
+// Middleware to authenticate requests with the proxy key and check the model
+function authenticateProxyKeyAndModel(req, res, next) {
+  const providedKey = req.headers['auro']; // Assuming the key is sent in the 'x-proxy-key' header
+  const requestedModel = req.body.model;
+
+  // List of allowed models
+  const allowedModels = ['gpt-3.5-turbo', 'gpt-4-vision-preview','text-moderation-latest', 'gpt-3.5-turbo-1106'];
+
+  if (providedKey && providedKey === proxyKey && allowedModels.includes(requestedModel)) {
+    // If the provided key matches the expected key and the requested model is allowed, allow the request to proceed
+    next();
+  } else {
+    // If the key is missing or incorrect, or the model is not allowed, reject the request with an error response
+    res.status(401).json({ error: 'Unauthorized or invalid model' });
+  }
+}
+
+
+app.use('/api', authenticateProxyKeyAndModel, proxy(targetUrl, {
+  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
+    // Modify the request headers if necessary
+    proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
+    return proxyReqOpts;
+  },
+}));
+
+app.get("/", (req, res) => {
+  // res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}`);
+});
+
+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}`);
+});