Spaces:

Emalawi19
/

JS-Coder-Backend

Paused

App Files Files Community

Emalawi19 commited on Jun 20

Commit

f3001d7

verified ·

1 Parent(s): 0390951

Update server.js

Browse files

Files changed (1) hide show

server.js +136 -68

server.js CHANGED Viewed

@@ -11,19 +11,80 @@ 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) => {
@@ -36,15 +97,16 @@ const server = http.createServer(async (req, res) => {
         return res.end();
     }
-    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" }));
         return;
     }
     if (pathname === '/generate' && req.method === 'POST') {
         let body = '';
         req.on('data', chunk => { body += chunk.toString(); });
@@ -52,13 +114,14 @@ const server = http.createServer(async (req, res) => {
             try {
                 const { prompt } = JSON.parse(body);
                 if (!generator) {
                     res.writeHead(503);
                     return res.end(JSON.stringify({ error: "Model is still loading..." }));
                 }
                 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;
@@ -71,90 +134,95 @@ IDENTITY:
 - 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. 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: coding ? 800 : 800, // 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
                 });
-                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 }));
             } catch (err) {
-                res.writeHead(400);
-                res.end(JSON.stringify({ error: "Invalid request", detail: err.message }));
             }
         });
         return;
     }
     res.writeHead(404);
-    res.end(JSON.stringify({ error: "Not Found", requested_path: pathname }));
 });
 loadModel().then(() => {

     console.log("Model loaded successfully!");
 }
+const CODING_PHRASES = [
+    'generate code', 'write code', 'create code',
+    'generate a website', 'build a website', 'create a website', 'make a website',
+    'generate a webpage', 'build a webpage',
+    'generate a login', 'create a login', 'build a login',
+    'generate a form', 'create a form', 'build a form',
+    'generate a page', 'create a page', 'build a page',
+    'generate an app', 'create an app', 'build an app',
+    'write a function', 'write a script', 'write a program',
+    'create a function', 'create a script',
+    'build an api', 'create an api', 'generate an api',
+    'node.js', 'express.js', 'express server',
+    'show me the code', 'give me the code', 'write html',
+    'write css', 'write javascript', 'write js',
+    'create a backend', 'build a backend',
+    'create a frontend', 'build a frontend',
+    'create a server', 'build a server',
+    'how do i code', 'how to code', 'code for',
+    'sample code', 'example code', '```'
+];
+const NON_CODING_CONTEXTS = [
+    'function of', 'functions of', 'what is the function',
+    'what are the functions', 'purpose of', 'role of',
+    'explain', 'what is', 'what are', 'tell me about',
+    'history of', 'meaning of', 'definition of',
+    'describe', 'how does', 'why is', 'benefits of',
+    'importance of', 'effects of', 'causes of'
 ];
 function isCodingRequest(prompt) {
+    const lower = prompt.toLowerCase().trim();
+    if (NON_CODING_CONTEXTS.some(ctx => lower.includes(ctx))) return false;
+    return CODING_PHRASES.some(phrase => lower.includes(phrase));
+}
+function isWebsiteRequest(prompt) {
     const lower = prompt.toLowerCase();
+    return ['website', 'webpage', 'web page', 'html page', 'login page',
+            'register page', 'landing page', 'form page', 'signup page',
+            'homepage', 'home page', 'portfolio', 'dashboard'].some(w => lower.includes(w));
+}
+function cleanHTML(html) {
+    if (!html.trimStart().toLowerCase().startsWith('<!doctype')) {
+        if (html.trimStart().toLowerCase().startsWith('<html')) {
+            html = '<!DOCTYPE html>\n' + html.trimStart();
+        } else {
+            html = '<!DOCTYPE html>\n<html lang="en">\n' + html.trimStart();
+        }
+    }
+    if (!html.trimEnd().toLowerCase().endsWith('</html>')) {
+        if (html.toLowerCase().includes('</body>')) {
+            html = html.replace(/<\/body>\s*$/i, '</body>\n</html>');
+        } else {
+            html = html.trimEnd() + '\n</body>\n</html>';
+        }
+    }
+    return html.trim();
+}
+function enforceHTMLStructure(text) {
+    const codeBlockMatch = text.match(/```html\s*([\s\S]*?)```/i);
+    if (codeBlockMatch) {
+        return '```html\n' + cleanHTML(codeBlockMatch[1].trim()) + '\n```';
+    }
+    const htmlStart = text.search(/<(!DOCTYPE|html)/i);
+    if (htmlStart !== -1) {
+        let html = text.slice(htmlStart);
+        const htmlEnd = html.search(/<\/html>/i);
+        if (htmlEnd !== -1) html = html.slice(0, htmlEnd + 7);
+        return '```html\n' + cleanHTML(html) + '\n```';
+    }
+    return text;
 }
 const server = http.createServer(async (req, res) => {
         return res.end();
     }
     const pathname = req.url.split('?')[0];
     if (pathname === '/' && req.method === 'GET') {
+        res.setHeader('Content-Type', 'application/json');
         res.writeHead(200);
         res.end(JSON.stringify({ status: "Backend is running" }));
         return;
     }
+    // ── STREAMING ENDPOINT ───────────────────────────────────────────────────
     if (pathname === '/generate' && req.method === 'POST') {
         let body = '';
         req.on('data', chunk => { body += chunk.toString(); });
             try {
                 const { prompt } = JSON.parse(body);
                 if (!generator) {
+                    res.setHeader('Content-Type', 'application/json');
                     res.writeHead(503);
                     return res.end(JSON.stringify({ error: "Model is still loading..." }));
                 }
                 const coding = isCodingRequest(prompt);
+                const websiteReq = isWebsiteRequest(prompt);
                 const finalPrompt = coding
                     ? `You are a coding assistant. Write complete, working code for this task. Never refuse. Always provide full code: ${prompt}`
                     : prompt;
 - You are an AI assistant, not a human.
 CODING RULES — NEVER BREAK THESE:
+1. You ALWAYS write code when asked. You NEVER refuse any coding task.
+2. Code must be complete, real, and copy-paste ready — no placeholders.
+3. Default language is always JavaScript/HTML/CSS — NEVER Python unless the user asks.
+4. Add short comments inside the code.
+HTML WEBSITE RULES:
+1. ALL website/page code MUST start with <!DOCTYPE html> on the very first line.
+2. Then <html lang="en"> on the second line.
+3. Must include a proper <head> with <meta charset>, <meta viewport>, <title>, and <style>.
+4. Must include a proper <body> with all content.
+5. Must end with </body> then </html> as the very last line.
+6. ALL CSS inside <style> in <head>. ALL JS inside <script> at the bottom of <body>.
+7. Wrap full HTML in a \`\`\`html code block.
+8. The page must be complete and functional — never cut off mid-code.
 STEP-BY-STEP RULES:
+1. When asked to build a full website or app, list ALL steps first.
+2. Say: "Let's begin with Step 1. Say 'next' when ready."
+3. For each step: title, explanation, complete code, filename, how to run, then "Say 'next' to continue."
+4. ONE step per response. Never combine. Never cut off mid-code.`
+                    : `You are Gini AI, a helpful 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 directly and clearly in plain English.
+- NEVER generate code for general knowledge or everyday questions.
+- NEVER use any programming language to answer a non-coding question.
+- Answer in plain sentences. Keep it concise, friendly, and easy to understand.`;
                 const messages = [
                     { role: 'system', content: systemPrompt },
                     { role: 'user', content: finalPrompt }
                 ];
+                // Set SSE headers for streaming
+                res.writeHead(200, {
+                    'Content-Type': 'text/event-stream',
+                    'Cache-Control': 'no-cache',
+                    'Connection': 'keep-alive',
+                    'X-Accel-Buffering': 'no'
                 });
+                let fullResult = '';
+                // Stream tokens as they are generated
+                await generator(messages, {
+                    max_new_tokens: coding ? 800 : 300,
+                    temperature: coding ? 0.2 : 0.5,
+                    repetition_penalty: 1.2,
+                    do_sample: !coding,
+                    // This callback fires every time a new token is produced
+                    callback_function: (beams) => {
+                        const token = beams[0].output_token_ids;
+                        const newText = beams[0]?.generated_text ?? '';
+                        // Calculate only the new characters added since last callback
+                        if (newText.length > fullResult.length) {
+                            const newChunk = newText.slice(fullResult.length);
+                            fullResult = newText;
+                            // Send the new chunk immediately as an SSE event
+                            res.write(`data: ${JSON.stringify({ chunk: newChunk })}\n\n`);
                         }
+                    }
+                });
+                // Post-process HTML if needed
+                if (websiteReq && coding) {
+                    fullResult = enforceHTMLStructure(fullResult);
                 }
+                // Send the final complete result and close the stream
+                res.write(`data: ${JSON.stringify({ done: true, result: fullResult })}\n\n`);
+                res.end();
             } catch (err) {
+                res.write(`data: ${JSON.stringify({ error: err.message })}\n\n`);
+                res.end();
             }
         });
         return;
     }
+    res.setHeader('Content-Type', 'application/json');
     res.writeHead(404);
+    res.end(JSON.stringify({ error: "Not Found" }));
 });
 loadModel().then(() => {