Update server.js

server.js

@@ -1,120 +1,50 @@
 const express = require('express');
 const proxy = require('express-http-proxy');
-const FormData = require('form-data');
-const https = require('https');
-const url = require('url');
-const bodyParser = require('body-parser'); // Add bodyParser to parse request body
-var multer  = require('multer')();
 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;
+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' })); // Use bodyParser to parse JSON request body
+app.use(bodyParser.json({ limit: '50mb' }));
 
-function authenticateProxyKey(req, res, next) {
-  const providedKey = req.headers['auro'];
+// 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;
 
-  if (providedKey && providedKey === proxyKey) {
+  if (providedKey && providedKey === proxyKey && requestedModel === 'gpt-3.5-turbo') {
+    // If the provided key matches the expected key and the requested model is 'gpt-3.5', allow the request to proceed
     next();
   } else {
-    res.status(401).json({ error: 'Unauthorized' });
+    // If the key is missing or incorrect, or the model is not 'gpt-3.5', reject the request with an error response
+    res.status(401).json({ error: 'Unauthorized or invalid model' });
   }
 }
 
-app.use('/api', (req, res, next) => {
-  if (req.method === 'POST') {
-    console.log(`Incoming POST Request to ${req.originalUrl}`);
-    console.log('Request Headers:', req.headers);
-    console.log('Request Body:', req.body);
-    console.log('Query Parameters:', req.query);
-  }
-  next();
-});
-
-
-app.post('/api/proxy', multer.single('fileFieldName'), async (req, res) => {
-  const audioBase64 = req.fileFieldName; // Get audio data from the request body
-
-  if (!audioBase64) {
-    res.status(400).json({ error: 'Missing audio data' });
-    return;
-  }
-
-  const audioBuffer = Buffer.from(audioBase64, 'base64');
-
-  const headers = {
-    auro: openaiKey,
-    'Content-Type': 'multipart/form-data',
-  };
-
-  const form = new FormData();
-
-  // Append the audio data as a buffer with the appropriate content type
-  form.append('fileFieldName', audioBuffer, {
-    filename: 'audioFile',
-    contentType: 'audio/m4a',
-  });
-
-  // Append other form fields
-  form.append('model', 'whisper-1');
-  form.append('response_format', 'text');
+app.use('/api', authenticateProxyKeyAndModel, proxy(targetUrl, {
+  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
+    // Modify the request headers if necessary
+    proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
+    return proxyReqOpts;
+  },
+}));
 
-  const requestOptions = {
-    method: 'POST',
-    headers,
-    path: '/audio/transcriptions',
-    host: targetUrl.replace('https://', ''),
-    port: 443,
-  };
-
-  try {
-    const request = https.request(requestOptions, (response) => {
-      let data = '';
-
-      response.on('data', (chunk) => {
-        data += chunk;
-      });
-
-      response.on('end', () => {
-        if (response.statusCode === 200) {
-          res.status(200).json({ transcription: data });
-        } else {
-          res.status(response.statusCode).json({ error: data });
-        }
-      });
-    });
-
-    form.pipe(request);
-
-    request.on('error', (error) => {
-      console.error(error);
-      res.status(500).json({ error: 'Internal Server Error' });
-    });
-
-    request.end();
-  } catch (error) {
-    console.error(error);
-    res.status(500).json({ error: 'Internal Server Error' });
-  }
+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`;
+    const [username, spacename] = spaceId.split("/");
+    return `https://${username}-${spacename.replace(/_/g, "-")}.hf.space/api/v1`;
   } catch (e) {
-    return '';
+    return "";
   }
 }
 
-app.get('/', (req, res) => {
-  res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}`);
-});
-
 app.listen(port, () => {
   console.log(`Reverse proxy server running on ${baseUrl}`);
 });