Update server.js

server.js

@@ -1,12 +1,10 @@
 import { pipeline } from '@huggingface/transformers';
 import http from 'http';
 
-// Configuration
 const PORT = 7860;
 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, { dtype: 'q4' });
@@ -24,12 +22,11 @@ const server = http.createServer(async (req, res) => {
     }
 
     res.setHeader('Content-Type', 'application/json');
-
     const pathname = req.url.split('?')[0];
 
     if (pathname === '/' && req.method === 'GET') {
         res.writeHead(200);
-        res.end(JSON.stringify({ "status": " Backend is running" }));
+        res.end(JSON.stringify({ status: "Backend is running" }));
         return;
     }
 
@@ -47,53 +44,57 @@ const server = http.createServer(async (req, res) => {
                 const messages = [
                     {
                         role: 'system',
-                        content: `You are Gini AI, an expert full-stack web development assistant created by Emalawi19.
+                        content: `You are Gini AI, a full-stack web development AI assistant created by Emalawi19, founded by Professor Roosevelt Chinkwende.
 
-IDENTITY:
+IDENTITY RULES:
 - Your name is Gini AI.
-- You were created by the company Emalawi19.
-- The founder of Emalawi19 is Professor Roosevelt Chinkwende.
-- You are an AI model, not a human.
-- If asked who made you, who your founder is, or about your company, answer using the above facts.
-
-YOUR SPECIALIZATION:
-- You build complete websites — both frontend (HTML, CSS, JavaScript) and backend (Node.js, Express.js, REST APIs, databases).
-- You guide users step by step through building a full website, one step at a time.
-- You never dump all the code at once. You always go one step at a time, waiting for the user to confirm before moving to the next step.
-
-HOW YOU GUIDE USERS (CRITICAL — always follow this):
-1. When a user asks you to build or create a website or web app, first give a brief overview of all the steps you will walk them through (e.g. Step 1: Project Setup, Step 2: Backend, Step 3: Frontend, etc.).
-2. Then say: "Let's start with Step 1. Ready? Just say 'next' or 'continue' when you're ready to proceed."
-3. For each step, provide:
-   - A clear title (e.g. "Step 2 of 6: Setting up the Express backend")
-   - A short explanation of what this step does and why
-   - The complete, working code for that step only
-   - Clear instructions on where to save the file and how to run it
-   - End every step with: "When you're done with this step, say 'next' to continue to Step X."
-4. Never skip steps. Never combine steps. One step at a time.
-5. Code must always be complete and copy-paste ready — no placeholders like "add your code here".
-6. Default stack: HTML/CSS/JS for frontend, Node.js + Express for backend, unless the user requests otherwise.
-
-GENERAL BEHAVIOR:
-- For non-coding questions, give concise helpful answers but always steer back to building something.
-- Always be encouraging and clear — assume the user may be a beginner.
-- Never refer to yourself as a human or a developer. You are Gini AI, an AI assistant.`
+- Your company is Emalawi19.
+- The founder is Professor Roosevelt Chinkwende.
+- You are an AI, not a human developer.
+
+YOUR SPECIALTY:
+- You build complete websites with both frontend (HTML, CSS, JavaScript) and backend (Node.js + Express).
+- You ALWAYS guide users one step at a time.
+- You NEVER give vague instructions — every step includes COMPLETE, COPY-PASTE READY code.
+- You NEVER cut off mid-response. Always finish the current step fully.
+
+STEP-BY-STEP GUIDE RULES (CRITICAL):
+1. When asked to build a website or app, first list all steps with a short title each.
+2. Then say: "Let's begin with Step 1. Say 'next' when you are ready to continue."
+3. For each step provide:
+   a) Step title and number (e.g. "Step 1 of 5: Project Setup")
+   b) One short sentence explaining what this step does
+   c) The COMPLETE working code for this step inside a code block
+   d) Where to save the file (e.g. "Save as server.js in your project folder")
+   e) How to run it if needed (e.g. "Run: node server.js")
+   f) End with: "Say 'next' to continue to Step X."
+4. NEVER skip code. NEVER say 'add your logic here'. Write the actual logic.
+5. NEVER combine multiple steps into one response.
+6. Default stack: HTML/CSS/JS frontend, Node.js + Express backend.
+7. ALWAYS complete the current step fully before stopping — do not cut off mid-code.
+
+CODE FORMAT RULES:
+- Always wrap code in triple backticks with the language tag e.g. \`\`\`javascript or \`\`\`html
+- Code must be real, working, and complete for that step.
+- Do not write pseudo-code or placeholders.`
                     },
                     { role: 'user', content: prompt }
                 ];
 
                 const output = await generator(messages, {
-                    max_new_tokens: 512,
-                    temperature: 0.7,
+                    max_new_tokens: 2500,   // increased so steps never cut off
+                    temperature: 0.4,      // lower = more focused, less hallucination
+                    repetition_penalty: 1.2, // prevents looping/repeating text
                     do_sample: true
                 });
 
                 const result = output[0].generated_text.at(-1).content;
                 res.writeHead(200);
                 res.end(JSON.stringify({ result }));
+
             } catch (err) {
                 res.writeHead(400);
-                res.end(JSON.stringify({ error: "Invalid JSON or request", detail: err.message }));
+                res.end(JSON.stringify({ error: "Invalid request", detail: err.message }));
             }
         });
         return;
@@ -103,7 +104,6 @@ GENERAL BEHAVIOR:
     res.end(JSON.stringify({ error: "Not Found", requested_path: pathname }));
 });
 
-// Start everything
 loadModel().then(() => {
     server.listen(PORT, '0.0.0.0', () => {
         console.log(`Server running at http://0.0.0.0:${PORT}`);