Spaces:

Emalawi19
/

JS-Coder-Backend

Paused

App Files Files Community

Emalawi19 commited on Jun 21

Commit

7c2521a

verified ·

1 Parent(s): 91128a1

Update server.js

Browse files

Files changed (1) hide show

server.js +135 -88

server.js CHANGED Viewed

@@ -11,6 +11,47 @@ async function loadModel() {
     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',
@@ -47,7 +88,6 @@ function isCodingRequest(prompt) {
     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',
@@ -55,34 +95,7 @@ function isWebsiteRequest(prompt) {
             '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();
@@ -90,20 +103,32 @@ function cleanHTML(html) {
             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) => {
     res.setHeader('Access-Control-Allow-Origin', '*');
     res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
@@ -114,12 +139,12 @@ 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;
     }
@@ -129,13 +154,25 @@ const server = http.createServer(async (req, res) => {
         req.on('end', async () => {
             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}`
@@ -154,100 +191,110 @@ 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 — 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.
 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 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 : 800,
                     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(() => {

     console.log("Model loaded successfully!");
 }
+// ── WIKIPEDIA SEARCH ─────────────────────────────────────────────────────────
+// Step 1: Search Wikipedia for the best matching article title
+async function searchWikipedia(query) {
+    try {
+        const url = `https://en.wikipedia.org/w/api.php?action=search&list=search&srsearch=${encodeURIComponent(query)}&srlimit=1&format=json&origin=*`;
+        const res  = await fetch(url);
+        const data = await res.json();
+        const results = data?.query?.search;
+        if (!results || results.length === 0) return null;
+        return results[0].title; // Return best matching article title
+    } catch {
+        return null;
+    }
+}
+// Step 2: Fetch the article summary (first 600 chars) using the title
+async function fetchWikipediaSummary(title) {
+    try {
+        const url = `https://en.wikipedia.org/api/rest_v1/page/summary/${encodeURIComponent(title)}`;
+        const res  = await fetch(url);
+        const data = await res.json();
+        if (data.extract) {
+            // Trim to 600 chars so it fits the model's context without overloading it
+            return data.extract.slice(0, 600);
+        }
+        return null;
+    } catch {
+        return null;
+    }
+}
+// Step 3: Combine search + fetch into one call
+async function getWikipediaContext(prompt) {
+    const title   = await searchWikipedia(prompt);
+    if (!title) return null;
+    const summary = await fetchWikipediaSummary(title);
+    if (!summary) return null;
+    return { title, summary };
+}
+// ── CODING KEYWORD DETECTION ─────────────────────────────────────────────────
 const CODING_PHRASES = [
     'generate code', 'write code', 'create code',
     'generate a website', 'build a website', 'create a website', 'make a website',
     return CODING_PHRASES.some(phrase => lower.includes(phrase));
 }
 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));
 }
 function cleanHTML(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();
         }
     }
     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;
+}
+// ── HTTP SERVER ───────────────────────────────────────────────────────────────
 const server = http.createServer(async (req, res) => {
     res.setHeader('Access-Control-Allow-Origin', '*');
     res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
         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", model: MODEL_NAME }));
         return;
     }
         req.on('end', async () => {
             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);
+                // ── Fetch Wikipedia context for non-coding questions ──────────
+                let wikipediaContext = '';
+                if (!coding) {
+                    const wiki = await getWikipediaContext(prompt);
+                    if (wiki) {
+                        wikipediaContext = `\n\nWIKIPEDIA REFERENCE (from article: "${wiki.title}"):\n${wiki.summary}\n\nUse the above Wikipedia information to improve your answer. Rewrite it in your own words — do not copy it directly. Add your own explanation on top of it.`;
+                        console.log(`Wikipedia context fetched: "${wiki.title}"`);
+                    }
+                }
                 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:
 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.
+${wikipediaContext}
 GENERAL BEHAVIOR:
 - Answer the question directly and clearly in plain English.
+- If Wikipedia reference data is provided above, use it to make your answer accurate and informative.
+- Rewrite the Wikipedia information in your own words — never copy it word for word.
+- Add your own helpful explanation beyond just what Wikipedia says.
 - NEVER generate code for general knowledge or everyday questions.
+- Keep answers concise, friendly, and easy to understand.`;
                 const messages = [
                     { role: 'system', content: systemPrompt },
+                    { role: 'user',   content: finalPrompt  }
                 ];
+                // ── SSE streaming response ────────────────────────────────────
+                res.writeHead(200, {
+                    'Content-Type': 'text/event-stream',
+                    'Cache-Control': 'no-cache',
+                    'Connection': 'keep-alive',
+                    'X-Accel-Buffering': 'no'
+                });
+                let fullResult = '';
+                await generator(messages, {
+                    max_new_tokens: coding ? 800 : 300,
                     temperature: coding ? 0.2 : 0.5,
                     repetition_penalty: 1.2,
+                    do_sample: !coding,
+                    callback_function: (beams) => {
+                        const newText = beams[0]?.generated_text ?? '';
+                        if (newText.length > fullResult.length) {
+                            const chunk = newText.slice(fullResult.length);
+                            fullResult  = newText;
+                            res.write(`data: ${JSON.stringify({ chunk })}\n\n`);
+                        }
+                    }
                 });
                 // 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 => fullResult.toLowerCase().startsWith(r));
                 if (refused && coding) {
+                    fullResult = '';
+                    await generator([
+                        { role: 'system', content: 'You are a code generator. Output only complete working code. Never refuse.' },
+                        { role: 'user',   content: `Write complete working HTML code starting with <!DOCTYPE html> and ending with </html> for: ${prompt}` }
+                    ], {
                         max_new_tokens: 800,
                         temperature: 0.1,
+                        do_sample: false,
+                        callback_function: (beams) => {
+                            const newText = beams[0]?.generated_text ?? '';
+                            if (newText.length > fullResult.length) {
+                                const chunk = newText.slice(fullResult.length);
+                                fullResult  = newText;
+                                res.write(`data: ${JSON.stringify({ chunk })}\n\n`);
+                            }
+                        }
                     });
                 }
+                if (websiteReq && coding) {
+                    fullResult = enforceHTMLStructure(fullResult);
                 }
+                // Send final done event with source info
+                const wikiSource = (!coding && wikipediaContext) ? '📖 Wikipedia' : null;
+                res.write(`data: ${JSON.stringify({ done: true, result: fullResult, source: wikiSource })}\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(() => {