Update server.js

server.js

@@ -6,45 +6,45 @@ const cors = require('cors');
 const app = express();
 const PORT = process.env.PORT || 7860;
 
-// Enable CORS for all routes (useful if frontend is served separately)
 app.use(cors());
-
-// Increase payload limit for base64 image data (up to 10 MB)
 app.use(express.json({ limit: '10mb' }));
 
-// Define the snapshots directory (relative to this file)
 const SNAPSHOT_DIR = path.join(__dirname, 'snapshots');
 
-// Create snapshots folder if it doesn't exist
+// Ensure directory exists with proper permissions
 if (!fs.existsSync(SNAPSHOT_DIR)) {
   fs.mkdirSync(SNAPSHOT_DIR, { recursive: true });
   console.log(`[SERVER] Created snapshots directory at ${SNAPSHOT_DIR}`);
 }
 
-// Serve static files from the "public" folder at the root URL
+// Serve static files from public
 app.use(express.static(path.join(__dirname, 'public')));
 
-// Serve the snapshots folder as a static endpoint (so users can view saved images)
-app.use('/snapshots', express.static(SNAPSHOT_DIR));
+// Explicit route for snapshots with logging
+app.get('/snapshots/:filename', (req, res) => {
+  const filename = req.params.filename;
+  if (filename.includes('..') || filename.includes('/')) {
+    return res.status(400).send('Invalid filename');
+  }
+  const filePath = path.join(SNAPSHOT_DIR, filename);
+  if (fs.existsSync(filePath)) {
+    console.log(`[SERVER] Serving snapshot: ${filename}`);
+    res.sendFile(filePath);
+  } else {
+    console.log(`[SERVER] Snapshot not found: ${filename}`);
+    res.status(404).send('Snapshot not found');
+  }
+});
 
-// POST endpoint to save a snapshot
+// POST endpoint with enhanced logging
 app.post('/api/snapshot', (req, res) => {
   try {
-    const { image } = req.body; // expects base64 string (without the data:image/png;base64, prefix)
-    
+    const { image } = req.body;
     if (!image) {
-      return res.status(400).json({ success: false, error: 'No image data provided' });
+      return res.status(400).json({ success: false, error: 'No image data' });
     }
 
-    // Validate base64 format (basic check)
-    if (typeof image !== 'string' || image.length < 100) {
-      return res.status(400).json({ success: false, error: 'Invalid image data format' });
-    }
-
-    // Convert base64 to binary buffer
     const buffer = Buffer.from(image, 'base64');
-
-    // Generate timestamp-based filename
     const now = new Date();
     const dateStr = now.getFullYear() + '-' +
                     String(now.getMonth() + 1).padStart(2, '0') + '-' +
@@ -56,42 +56,27 @@ app.post('/api/snapshot', (req, res) => {
     const filename = `snapshot_${dateStr}.png`;
     const filePath = path.join(SNAPSHOT_DIR, filename);
 
-    // Write the file synchronously (can also use async with callback)
     fs.writeFileSync(filePath, buffer);
+    console.log(`[SERVER] Snapshot saved: ${filename} (${buffer.length} bytes) at ${filePath}`);
 
-    console.log(`[SERVER] Saved snapshot: ${filename} (${buffer.length} bytes)`);
-
-    // Return success response with the file path (relative URL)
-    res.json({
-      success: true,
-      path: `/snapshots/${filename}`,
-      filename: filename
-    });
-
-  } catch (err) {
-    console.error('[SERVER] Error saving snapshot:', err.message);
-    res.status(500).json({ success: false, error: 'Internal server error: ' + err.message });
-  }
-});
+    // Verify write
+    if (fs.existsSync(filePath)) {
+      console.log(`[SERVER] Verification: file exists, size ${fs.statSync(filePath).size}`);
+    } else {
+      console.error(`[SERVER] Verification FAILED: file not found after write`);
+    }
 
-// Optional: endpoint to list all saved snapshots (for debugging)
-app.get('/api/snapshots/list', (req, res) => {
-  try {
-    const files = fs.readdirSync(SNAPSHOT_DIR).filter(f => f.endsWith('.png'));
-    res.json({ success: true, files });
+    res.json({ success: true, path: `/snapshots/${filename}`, filename });
   } catch (err) {
+    console.error('[SERVER] Save error:', err);
     res.status(500).json({ success: false, error: err.message });
   }
 });
 
-// Health check endpoint (useful for Hugging Face)
-app.get('/health', (req, res) => {
-  res.status(200).send('OK');
-});
+// Health check
+app.get('/health', (req, res) => res.send('OK'));
 
-// Start the server
 app.listen(PORT, () => {
-  console.log(`[SERVER] Gesture Drawing Space running on port ${PORT}`);
-  console.log(`[SERVER] Serving static files from ${path.join(__dirname, 'public')}`);
-  console.log(`[SERVER] Snapshots will be saved to ${SNAPSHOT_DIR}`);
+  console.log(`[SERVER] Running on port ${PORT}`);
+  console.log(`[SERVER] Snapshots directory: ${SNAPSHOT_DIR}`);
 });