Update server.js

server.js

@@ -2,14 +2,28 @@ const express = require('express');
 const proxy = require('express-http-proxy');
 const app = express();
 const targetUrl = 'https://api.openai.com'; // OpenAI API endpoint
-const openai_key = process.env.OPENAI_KEY
+const openai_key = process.env.OPENAI_KEY;
+
+if (!openai_key) {
+  console.error('Error: OPENAI_KEY is not set.');
+  process.exit(1);
+}
+
 app.use('/', proxy(targetUrl, {
   proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
     // Modify the request headers if necessary
-    proxyReqOpts.headers['Authorization'] = 'Bearer '+openai_key;
+    proxyReqOpts.headers['Authorization'] = 'Bearer ' + openai_key;
     return proxyReqOpts;
   },
+  userResDecorator: (proxyRes, proxyResData, userReq, userRes) => {
+    return proxyResData;
+  },
+  proxyErrorHandler: (err, res, next) => {
+    console.error('Proxy error:', err);
+    res.status(500).send('Proxy error');
+  }
 }));
+
 const port = 7860;
 app.listen(port, () => {
   console.log(`Reverse proxy server listening on port ${port}`);