Update server.js

server.js

@@ -1,59 +1,22 @@
 import express from 'express';
 import { render } from '@antv/gpt-vis-ssr';
-import path from 'path';
-import fs from 'fs/promises';
-import crypto from 'crypto';
 
 const app = express();
 app.use(express.json());
 
-// Serve static files in /images (adjust if you use another folder)
-app.use('/images', express.static(path.join(process.cwd(), 'images')));
-
 app.post('/api/chart', async (req, res) => {
   try {
-    const options = req.body;
+    const options = req.body; // chart options (type, data, etc.)
     const vis = await render(options);
     const buffer = vis.toBuffer();
-
-    // Generate a unique filename
-    const filename = crypto.randomBytes(16).toString('hex') + '.png';
-    const filePath = path.join(process.cwd(), 'images', filename);
-
-    // Optional: Check if the images directory exists before writing (non-destructive)
-    // This helps catch issues without trying to create it.
-    const imagesDir = path.join(process.cwd(), 'images');
-    await fs.access(imagesDir); // Throws if directory doesn't exist or is inaccessible
-
-    // If you want to add back a creation check (uncomment if needed):
-    // try {
-    //   await fs.access(imagesDir);
-    // } catch {
-    //   await fs.mkdir(imagesDir, { recursive: true });
-    // }
-
-    // Save image to disk
-    await fs.writeFile(filePath, buffer);
-
-    // Build public URL (HuggingFace Space exposes root as https://<space>.hf.space/)
-    const imageUrl = `/images/${filename}`;
-    const fullUrl = `${req.protocol}://${req.get('host')}${imageUrl}`;
-
-    res.json({
-      success: true,
-      resultObj: fullUrl,
-      errorMessage: ''
-    });
+    res.setHeader('Content-Type', 'image/png');
+    res.send(buffer);
   } catch (err) {
-    res.json({
-      success: false,
-      resultObj: '',
-      errorMessage: err.message
-    });
+    res.status(400).json({ error: err.message });
   }
 });
 
 const port = process.env.PORT || 7860;
 app.listen(port, () => {
   console.log('API running on port ' + port);
-});
+});