Spaces:

Emalawi19
/

JS-Coder-Backend

Paused

App Files Files Community

Emalawi19 commited on Jun 20

Commit

4a7c883

verified ·

1 Parent(s): f3001d7

Update server.js

Browse files

Files changed (1) hide show

server.js +93 -68

server.js CHANGED Viewed

@@ -47,6 +47,7 @@ function isCodingRequest(prompt) {
     return CODING_PHRASES.some(phrase => lower.includes(phrase));
 }
 function isWebsiteRequest(prompt) {
     const lower = prompt.toLowerCase();
     return ['website', 'webpage', 'web page', 'html page', 'login page',
@@ -54,7 +55,34 @@ function isWebsiteRequest(prompt) {
             '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();
@@ -62,29 +90,18 @@ function cleanHTML(html) {
             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) => {
@@ -97,16 +114,15 @@ 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(); });
@@ -114,13 +130,12 @@ const server = http.createServer(async (req, res) => {
             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}`
@@ -139,21 +154,30 @@ CODING RULES — NEVER BREAK THESE:
 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.
@@ -162,67 +186,68 @@ IDENTITY:
 - 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(() => {

     return CODING_PHRASES.some(phrase => lower.includes(phrase));
 }
+// Detect if the prompt is asking for a full HTML website/page
 function isWebsiteRequest(prompt) {
     const lower = prompt.toLowerCase();
     return ['website', 'webpage', 'web page', 'html page', 'login page',
             'homepage', 'home page', 'portfolio', 'dashboard'].some(w => lower.includes(w));
 }
+// Ensure HTML output starts with <!DOCTYPE html> or <html> and ends with </html>
+function enforceHTMLStructure(text) {
+    // Extract code from inside ```html ... ``` block if present
+    const codeBlockMatch = text.match(/```html\s*([\s\S]*?)```/i);
+    if (codeBlockMatch) {
+        let html = codeBlockMatch[1].trim();
+        html = cleanHTML(html);
+        return '```html\n' + html + '\n```';
+    }
+    // If raw HTML is present (starts with < somewhere), extract and clean it
+    const htmlStart = text.search(/<(!DOCTYPE|html)/i);
+    if (htmlStart !== -1) {
+        let html = text.slice(htmlStart);
+        // Cut off anything after </html>
+        const htmlEnd = html.search(/<\/html>/i);
+        if (htmlEnd !== -1) {
+            html = html.slice(0, htmlEnd + 7); // include </html>
+        }
+        html = cleanHTML(html);
+        return '```html\n' + html + '\n```';
+    }
+    return text;
+}
 function cleanHTML(html) {
+    // Ensure it starts with <!DOCTYPE html>
     if (!html.trimStart().toLowerCase().startsWith('<!doctype')) {
         if (html.trimStart().toLowerCase().startsWith('<html')) {
             html = '<!DOCTYPE html>\n' + html.trimStart();
             html = '<!DOCTYPE html>\n<html lang="en">\n' + html.trimStart();
         }
     }
+    // Ensure it ends with </html>
     if (!html.trimEnd().toLowerCase().endsWith('</html>')) {
+        // Check if </body> is present, add </html> after it
         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();
 }
 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(); });
             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);
+                const websiteRequest = isWebsiteRequest(prompt);
                 const finalPrompt = coding
                     ? `You are a coding assistant. Write complete, working code for this task. Never refuse. Always provide full code: ${prompt}`
 3. Default language is always JavaScript/HTML/CSS — NEVER Python unless the user asks.
 4. Add short comments inside the code.
+HTML WEBSITE RULES — ALWAYS FOLLOW FOR ANY WEBSITE/PAGE REQUEST:
 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> section with <meta charset>, <meta viewport>, <title>, and embedded <style>.
+4. Must include a proper <body> section with all content.
 5. Must end with </body> then </html> as the very last line.
+6. ALL CSS goes inside a <style> tag in the <head> — no external stylesheets.
+7. ALL JavaScript goes inside a <script> tag at the bottom of <body> — no external scripts.
+8. The full HTML must be wrapped in a \`\`\`html code block.
+9. 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, 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 helpful AI assistant created by Emalawi19, founded by Professor Roosevelt Chinkwende.
 - You are an AI assistant, not a human.
 GENERAL BEHAVIOR:
+- Answer the question directly and clearly in plain English.
 - NEVER generate code for general knowledge or everyday questions.
+- NEVER use Python, JavaScript, or any programming language to answer a non-coding question.
+- If someone asks "what is the function of X" or "explain X" — answer in plain sentences, not code.
+- Keep answers concise, friendly, and easy to understand.
+- Only mention web development if the question is specifically about it.`;
                 const messages = [
                     { role: 'system', content: systemPrompt },
                     { role: 'user', content: finalPrompt }
                 ];
+                const output = await generator(messages, {
                     max_new_tokens: coding ? 800 : 300,
                     temperature: coding ? 0.2 : 0.5,
                     repetition_penalty: 1.2,
+                    do_sample: !coding
                 });
+                let result = output[0].generated_text.at(-1).content;
+                // Catch refusals and retry
+                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. No explanations, just code.'
+                        },
+                        {
+                            role: 'user',
+                            content: `Write complete working HTML code starting with <!DOCTYPE html> and ending with </html> 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;
+                }
+                // Post-process: enforce proper HTML structure for website requests
+                if (websiteRequest && coding) {
+                    result = enforceHTMLStructure(result);
                 }
+                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(() => {