Update server.js

server.js

@@ -3,15 +3,14 @@ import http from 'http';
 
 // Configuration
 const PORT = 7860;
-const MODEL_NAME = 'Xenova/codegen-350M-multi'; // ← Multi-language: JS, Python, C++, etc.
+// Instruction-following model: understands "generate a login page" and responds correctly
+const MODEL_NAME = 'onnx-community/Qwen2.5-0.5B-Instruct';
 let generator;
 
 // Initialize the model on startup
 async function loadModel() {
     console.log("Loading coding model...");
-    generator = await pipeline('text-generation', MODEL_NAME, {
-        model_file_name: 'decoder_model_merged' // ← Required for codegen models in Transformers.js v3
-    });
+    generator = await pipeline('text-generation', MODEL_NAME, { dtype: 'q4' });
     console.log("Model loaded successfully!");
 }
 
@@ -50,16 +49,26 @@ const server = http.createServer(async (req, res) => {
                     res.writeHead(503);
                     return res.end(JSON.stringify({ error: "Model is still loading..." }));
                 }
-                // Generate code based on the prompt
-                const output = await generator(prompt, {
-                    max_new_tokens: 100,
-                    temperature: 0.7
+
+                // Use chat format so the model understands instructions properly
+                const messages = [
+                    { role: 'system', content: 'You are an expert JavaScript and web developer. When asked to generate code, respond with clean, complete, working code only. No explanations unless asked.' },
+                    { role: 'user', content: prompt }
+                ];
+
+                const output = await generator(messages, {
+                    max_new_tokens: 512,  // More tokens for complete code output
+                    temperature: 0.7,
+                    do_sample: true
                 });
+
+                // Extract only the assistant's reply
+                const result = output[0].generated_text.at(-1).content;
                 res.writeHead(200);
-                res.end(JSON.stringify({ result: output[0].generated_text }));
+                res.end(JSON.stringify({ result }));
             } catch (err) {
                 res.writeHead(400);
-                res.end(JSON.stringify({ error: "Invalid JSON or request" }));
+                res.end(JSON.stringify({ error: "Invalid JSON or request", detail: err.message }));
             }
         });
         return;