Update app/server.js

app/server.js

@@ -43,6 +43,8 @@
 // });
 
 // module.exports = router;
+
+
 const express = require('express');
 const multer = require('multer');
 const {
@@ -56,30 +58,25 @@ const app = express();
 const router = express.Router();
 const upload = multer();
 
-// Debug: Server initialization
 console.log('🔹 Initializing Shadow Backend...');
 
 // ---------------- CORS Middleware ----------------
 app.use((req, res, next) => {
-  res.setHeader('Access-Control-Allow-Origin', '*'); // Allow all origins
+  res.setHeader('Access-Control-Allow-Origin', '*');
   res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
   res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization');
 
-  // Handle preflight requests
-  if (req.method === 'OPTIONS') {
-    return res.sendStatus(200);
-  }
-
+  if (req.method === 'OPTIONS') return res.sendStatus(200);
   next();
 });
 
+// ---------------- Root route ----------------
+app.get('/', (req, res) => res.send('Shadow Backend is running ✅'));
+
 // ---------------- Submit new shadow task ----------------
 router.post('/add-shadow', upload.single('file'), async (req, res) => {
   console.log('🔹 /add-shadow called');
-  if (!req.file) {
-    console.warn('⚠️ No file uploaded');
-    return res.status(400).json({ error: 'No image uploaded' });
-  }
+  if (!req.file) return res.status(400).json({ error: 'No image uploaded' });
 
   const options = {
     type: req.body.type || 'cast',
@@ -90,46 +87,31 @@ 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);
+    console.error(err);
     res.status(500).json({ error: 'Server busy, try again' });
   }
 });
 
-// ---------------- Server load / status ----------------
+// ---------------- Server status ----------------
 router.get('/status', (req, res) => {
-  const queueLength = getQueueLength();
-  console.log('🔹 /status called, queue_length:', queueLength);
-  res.json({
-    status: 'ok',
-    queue_length: queueLength,
-  });
+  res.json({ status: 'ok', queue_length: getQueueLength() });
 });
 
 // ---------------- Poll for result ----------------
 router.get('/result/:id', (req, res) => {
   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(taskId);
-  if (!result) {
-    console.warn('⚠️ Result not found for task:', taskId);
-    return res.status(404).json({ error: 'Result not found' });
-  }
+  if (!result) 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);
 });
@@ -137,10 +119,6 @@ router.get('/result/:id', (req, res) => {
 app.use('/api/shadow', router);
 
 // ---------------- Start server ----------------
-const PORT = process.env.PORT; // HF Spaces dynamic port
+const PORT = process.env.PORT || 7860;
 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;