Update app/server.js

app/server.js

@@ -43,7 +43,6 @@
 // });
 
 // module.exports = router;
-
 const express = require('express');
 const multer = require('multer');
 const cors = require('cors');
@@ -58,11 +57,18 @@ const app = express();
 const router = express.Router();
 const upload = multer();
 
+// Debug: Server initialization
+console.log('🔹 Initializing Shadow Backend...');
+
 app.use(cors());
 
 // ---------------- Submit new shadow task ----------------
 router.post('/add-shadow', upload.single('file'), async (req, res) => {
-  if (!req.file) return res.status(400).json({ error: 'No image uploaded' });
+  console.log('🔹 /add-shadow called');
+  if (!req.file) {
+    console.warn('⚠️ No file uploaded');
+    return res.status(400).json({ error: 'No image uploaded' });
+  }
 
   const options = {
     type: req.body.type || 'cast',
@@ -73,8 +79,11 @@ router.post('/add-shadow', upload.single('file'), async (req, res) => {
     lightY: req.body.lightY ? parseFloat(req.body.lightY) : undefined,
   };
 
+  console.log('🔹 Shadow options:', options);
+
   try {
     const taskId = await addTask(req.file.buffer, options);
+    console.log(`✅ Task added with ID: ${taskId}`);
     res.json({ task_id: taskId });
   } catch (err) {
     console.error('❌ Error adding shadow task:', err);
@@ -84,22 +93,32 @@ router.post('/add-shadow', upload.single('file'), async (req, res) => {
 
 // ---------------- Server load / status ----------------
 router.get('/status', (req, res) => {
+  const queueLength = getQueueLength();
+  console.log('🔹 /status called, queue_length:', queueLength);
   res.json({
     status: 'ok',
-    queue_length: getQueueLength(),
+    queue_length: queueLength,
   });
 });
 
 // ---------------- Poll for result ----------------
 router.get('/result/:id', (req, res) => {
-  const status = getTaskStatus(req.params.id);
+  const taskId = req.params.id;
+  console.log('🔹 /result called for task:', taskId);
+
+  const status = getTaskStatus(taskId);
+  console.log('🔹 Task status:', status);
 
   if (status === 'pending') return res.sendStatus(202);
   if (status === 'failed') return res.status(500).json({ error: 'Shadow generation failed' });
 
-  const result = getTaskResult(req.params.id);
-  if (!result) return res.status(404).json({ error: 'Result not found' });
+  const result = getTaskResult(taskId);
+  if (!result) {
+    console.warn('⚠️ Result not found for task:', taskId);
+    return res.status(404).json({ error: 'Result not found' });
+  }
 
+  console.log('🔹 Returning result for task:', taskId);
   res.set('Content-Type', 'image/png');
   res.send(result);
 });
@@ -108,7 +127,9 @@ app.use('/api/shadow', router);
 
 // ---------------- Start server ----------------
 const PORT = process.env.PORT || 5000;
-const HOST = '0.0.0.0';  // Bind to all network interfaces for HF Spaces
-app.listen(PORT, HOST, () => console.log(`🚀 Shadow backend running on port ${PORT}`));
+const HOST = '0.0.0.0';
+
+console.log('🔹 Starting server...');
+app.listen(PORT, HOST, () => console.log(`🚀 Shadow backend running on ${HOST}:${PORT}`));
 
 module.exports = app;