Update server.js

server.js

@@ -15,18 +15,31 @@ async function loadModel() {
 }
 
 const server = http.createServer(async (req, res) => {
-    // Set headers for JSON response
+    // 1. Add CORS headers so external websites can talk to this API
+    res.setHeader('Access-Control-Allow-Origin', '*');
+    res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
+    res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
+    
+    // Handle preflight requests for the API
+    if (req.method === 'OPTIONS') {
+        res.writeHead(200);
+        return res.end();
+    }
+
     res.setHeader('Content-Type', 'application/json');
 
-    // 1. Your requested Status Check
-    if (req.url === '/' && req.method === 'GET') {
+    // Clean the URL (Removes Hugging Face's ?__theme=light additions)
+    const pathname = req.url.split('?')[0];
+
+    // 2. Your requested Status Check
+    if (pathname === '/' && req.method === 'GET') {
         res.writeHead(200);
         res.end(JSON.stringify({ "status": " Backend is running" }));
         return;
     }
 
-    // 2. Code Generation Endpoint
-    if (req.url === '/generate' && req.method === 'POST') {
+    // 3. Code Generation Endpoint
+    if (pathname === '/generate' && req.method === 'POST') {
         let body = '';
         req.on('data', chunk => { body += chunk.toString(); });
         req.on('end', async () => {
@@ -55,7 +68,7 @@ const server = http.createServer(async (req, res) => {
 
     // Default 404
     res.writeHead(404);
-    res.end(JSON.stringify({ error: "Not Found" }));
+    res.end(JSON.stringify({ error: "Not Found", requested_path: pathname }));
 });
 
 // Start everything