Update server.js

server.js

@@ -8,6 +8,7 @@ import bcrypt from 'bcrypt';
 import jwt from 'jsonwebtoken';
 import FormData from 'form-data';
 import pg from 'pg';
+import fetch from 'node-fetch';
 
 dotenv.config();
 
@@ -17,7 +18,7 @@ const app = express();
 const port = process.env.PORT || 3000;
 
 // Middleware
-app.use(cors());
+app.use(cors({ origin: '*', credentials: true }));
 app.use(express.json());
 app.use(express.static('public'));
 
@@ -921,6 +922,44 @@ app.post('/api/admin/process-old', async (req, res) => {
   }
 });
 
+// Proxy route for HuggingFace/Gradio images
+app.get('/proxy-image', async (req, res) => {
+    const { url, proxied } = req.query;
+    if (!url) {
+        return res.status(400).send('Missing url parameter');
+    }
+    if (!url.startsWith('https://yanze-pulid-flux.hf.space/')) {
+        return res.status(403).send('Forbidden: Only allowed to proxy from yanze-pulid-flux.hf.space');
+    }
+    try {
+        const response = await fetch(url, {
+            method: 'GET',
+            headers: {
+                'Accept': 'image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8',
+                'Accept-Language': 'en-GB,en-US;q=0.9,en;q=0.8,hi;q=0.7',
+                'Cache-Control': 'no-cache',
+                'Pragma': 'no-cache',
+                'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36',
+            }
+        });
+        if (!response.ok) {
+            return res.status(response.status).send('Failed to fetch image');
+        }
+        res.set('Access-Control-Allow-Origin', '*');
+        res.set('Access-Control-Allow-Credentials', 'true');
+        res.set('Content-Type', response.headers.get('content-type') || 'image/png');
+        res.set('Content-Disposition', response.headers.get('content-disposition') || 'inline');
+        // If proxied param is set, respond with the proxied URL
+        if (proxied === '1') {
+            const proxiedUrl = `https://shashwatidr-vision.hf.space/proxy-image?url=${encodeURIComponent(url)}`;
+            return res.json({ proxiedUrl });
+        }
+        response.body.pipe(res);
+    } catch (err) {
+        res.status(500).send('Proxy error: ' + err.message);
+    }
+});
+
 app.listen(port, () => {
   console.log(`Server running on port ${port}`);
 });