Update server.js

server.js

@@ -1,16 +1,54 @@
 const express = require('express');
-const proxy = require('express-http-proxy');
+const rateLimit = require('express-rate-limit');
+const axios = require('axios');
+require('dotenv').config();
+
 const app = express();
-const targetUrl = 'https://api.openai.com'; // OpenAI API endpoint
-const openai_key = process.env.OPENAI_KEY
-app.use('/', proxy(targetUrl, {
-  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
-    // Modify the request headers if necessary
-   // proxyReqOpts.headers['Authorization'] = 'Bearer '+openai_key;
-    return proxyReqOpts;
-  },
-}));
+const openai_key = process.env.OPENAI_KEY;
+
+// Rate limiter configuration
+const limiter = rateLimit({
+  windowMs: 30 * 1000, // 30 seconds
+  max: 1, // limit each IP to 1 request per windowMs
+  onLimitReached: (req, res) => {
+    console.error(`Rate limit exceeded for ${req.ip}`);
+  }
+});
+
+// Apply rate limiting to all requests
+app.use(limiter);
+
+// Enable JSON body parsing
+app.use(express.json());
+
+// DALL-E 3 image generation endpoint
+app.post('/', async (req, res) => {
+  try {
+    const { prompt } = req.body;
+    // Check if the prompt is provided
+    if (!prompt) {
+      return res.status(400).send({ message: 'No prompt provided' });
+    }
+
+    // Send the request to OpenAI's DALL-E 3 API
+    const response = await axios.post('https://api.openai.com/v1/images/generations', {
+      prompt: prompt
+    }, {
+      headers: {
+        'Authorization': `Bearer ${openai_key}`,
+        'Content-Type': 'application/json'
+      }
+    });
+
+    // Send back the generated image
+    res.send(response.data);
+  } catch (error) {
+    console.error('Error generating image:', error);
+    res.status(500).send({ message: 'Error generating image' });
+  }
+});
+
 const port = 7860;
 app.listen(port, () => {
-  console.log(`Reverse proxy server listening on port ${port}`);
+  console.log(`API server listening on port ${port}`);
 });