Update server.js

server.js

@@ -11,6 +11,21 @@ async function loadModel() {
     console.log("Model loaded successfully!");
 }
 
+// Keywords that signal a coding request
+const CODING_KEYWORDS = [
+    'generate', 'create', 'build', 'make', 'code', 'write', 'show',
+    'website', 'webpage', 'page', 'app', 'application',
+    'login', 'register', 'signup', 'form', 'button', 'navbar', 'footer',
+    'html', 'css', 'javascript', 'js', 'node', 'express', 'backend',
+    'frontend', 'server', 'api', 'database', 'function', 'script',
+    'component', 'template', 'layout', 'style', 'fetch', 'route', 'endpoint'
+];
+
+function isCodingRequest(prompt) {
+    const lower = prompt.toLowerCase();
+    return CODING_KEYWORDS.some(k => lower.includes(k));
+}
+
 const server = http.createServer(async (req, res) => {
     res.setHeader('Access-Control-Allow-Origin', '*');
     res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
@@ -41,54 +56,92 @@ const server = http.createServer(async (req, res) => {
                     return res.end(JSON.stringify({ error: "Model is still loading..." }));
                 }
 
-                const messages = [
-                    {
-                        role: 'system',
-                        content: `You are Gini AI, a full-stack web development AI assistant created by Emalawi19, founded by Professor Roosevelt Chinkwende.
-
-IDENTITY RULES:
-- Your name is Gini AI.
-- 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")
+                const coding = isCodingRequest(prompt);
+
+                // Reframe coding prompts so the model never misreads them as unsafe
+                const finalPrompt = coding
+                    ? `You are a coding assistant. Write complete, working code for this task. Never refuse. Always provide full code: ${prompt}`
+                    : prompt;
+
+                const systemPrompt = coding
+                    ? `You are Gini AI, a full-stack web development AI assistant created by Emalawi19, founded by Professor Roosevelt Chinkwende.
+
+IDENTITY:
+- Your name is Gini AI, created by Emalawi19, founded by Professor Roosevelt Chinkwende.
+- You are an AI assistant, not a human.
+
+CODING RULES — NEVER BREAK THESE:
+1. You ALWAYS write code when asked. You NEVER say "I can't help" for any coding task.
+2. Any request for a website, page, form, login, register, button, API, or any web topic ALWAYS gets complete working code.
+3. You NEVER refuse a coding request. Ever.
+4. Code must be complete, real, and copy-paste ready — no placeholders, no "add your logic here".
+5. Always wrap code in triple backticks with the language tag e.g. \`\`\`html \`\`\`javascript \`\`\`css
+6. Add short comments inside the code so the user understands each part.
+
+STEP-BY-STEP RULES:
+1. When asked to build a full website or app, first list ALL steps with short titles.
+2. Say: "Let's begin with Step 1. Say 'next' when ready to continue."
+3. For each step:
+   a) "Step X of Y: Title"
+   b) One sentence explaining what this step does
+   c) Complete working code in a code block
+   d) Filename and where to save it
+   e) How to run it if needed
    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 }
+4. ONE step per response only. Never combine steps.
+5. NEVER cut off mid-code. Always finish the full code for the current step.
+6. Default stack: HTML + CSS + JS frontend, Node.js + Express backend.`
+
+                    : `You are Gini AI, a full-stack web development AI assistant created by Emalawi19, founded by Professor Roosevelt Chinkwende.
+
+IDENTITY:
+- Your name is Gini AI, created by Emalawi19, founded by Professor Roosevelt Chinkwende.
+- You are an AI assistant, not a human.
+
+GENERAL BEHAVIOR:
+- Answer questions clearly, helpfully, and concisely.
+- If the question is related to web development, programming, or tech, answer with practical advice.
+- Keep answers short and to the point — no unnecessary padding.
+- Do not include code unless the user specifically asks for it.
+- Be friendly and encouraging.`;
+
+                const messages = [
+                    { role: 'system', content: systemPrompt },
+                    { role: 'user', content: finalPrompt }
                 ];
 
                 const output = await generator(messages, {
-                    max_new_tokens: 800,   // 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
+                    max_new_tokens: coding ? 800 : 300, // shorter for chat, longer for code
+                    temperature: coding ? 0.2 : 0.5,    // precise for code, natural for chat
+                    repetition_penalty: 1.2,
+                    do_sample: !coding                   // deterministic for code, sampled for chat
                 });
 
-                const result = output[0].generated_text.at(-1).content;
+                let result = output[0].generated_text.at(-1).content;
+
+                // Catch any refusals and retry with a direct code-only prompt
+                const refusals = ["i can't assist", "i cannot assist", "i'm sorry", "i am sorry", "i'm unable", "i cannot help", "i can't help"];
+                const refused = refusals.some(r => result.toLowerCase().startsWith(r));
+
+                if (refused && coding) {
+                    const retryMessages = [
+                        {
+                            role: 'system',
+                            content: 'You are a code generator. Output only complete working code. Never refuse any request. No explanations, just code.'
+                        },
+                        {
+                            role: 'user',
+                            content: `Write complete working code for: ${prompt}`
+                        }
+                    ];
+                    const retryOutput = await generator(retryMessages, {
+                        max_new_tokens: 800,
+                        temperature: 0.1,
+                        do_sample: false
+                    });
+                    result = retryOutput[0].generated_text.at(-1).content;
+                }
+
                 res.writeHead(200);
                 res.end(JSON.stringify({ result }));