Update server.js

server.js

@@ -1,14 +1,66 @@
 import { pipeline } from '@huggingface/transformers';
 import http from 'http';
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
 
 const PORT = 7860;
 const MODEL_NAME = 'onnx-community/Qwen2.5-Coder-3B-Instruct';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
 let generator;
+const KNOWLEDGE_TOPICS = {}; 
+
+async function initializeServer() {
+    try {
+        console.log("Reading knowledge.txt...");
+        const knowledgePath = path.join(__dirname, 'knowledge.txt');
+        
+        if (fs.existsSync(knowledgePath)) {
+            const rawContent = fs.readFileSync(knowledgePath, 'utf8');
+            
+            const topicRegex = /\[TOPIC:\s*(\w+)\]([\s\S]*?)(?=\[TOPIC:|$)/gi;
+            let match;
+            
+            while ((match = topicRegex.exec(rawContent)) !== null) {
+                const topicName = match[1].toLowerCase().trim();
+                const topicContent = match[2].trim();
+                KNOWLEDGE_TOPICS[topicName] = topicContent;
+                console.log(`-> Successfully registered topic: "${topicName}"`);
+            }
+        } else {
+            console.warn("WARNING: knowledge.txt not found. Starting blank.");
+        }
+
+        console.log("Loading Qwen2.5-Coder-3B model... (first load may take a few minutes)");
+        generator = await pipeline('text-generation', MODEL_NAME, { dtype: 'q4' });
+        console.log("Server fully initialized!");
+
+    } catch (error) {
+        console.error("Initialization failed:", error);
+        process.exit(1);
+    }
+}
 
-async function loadModel() {
-    console.log("Loading Qwen2.5-Coder-3B model... (first load may take a few minutes)");
-    generator = await pipeline('text-generation', MODEL_NAME, { dtype: 'q4' });
-    console.log("Model loaded successfully!");
+// 100% AUTOMATED ROUTER
+function getRelevantContext(userPrompt) {
+    const lowerPrompt = userPrompt.toLowerCase();
+    let dynamicContext = "";
+
+    for (const topicName in KNOWLEDGE_TOPICS) {
+        if (lowerPrompt.includes(topicName)) {
+            dynamicContext += KNOWLEDGE_TOPICS[topicName] + "\n\n";
+        }
+    }
+
+    if (!dynamicContext.trim()) {
+        const available = Object.keys(KNOWLEDGE_TOPICS).join(", ");
+        dynamicContext = `General support mode. Reference available topics: ${available}.`;
+    }
+
+    return dynamicContext;
 }
 
 const server = http.createServer(async (req, res) => {
@@ -26,7 +78,10 @@ const server = http.createServer(async (req, res) => {
 
     if (pathname === '/' && req.method === 'GET') {
         res.writeHead(200);
-        res.end(JSON.stringify({ status: "Backend is running", model: MODEL_NAME }));
+        res.end(JSON.stringify({ 
+            status: "Backend is running", 
+            available_topics: Object.keys(KNOWLEDGE_TOPICS)
+        }));
         return;
     }
 
@@ -38,10 +93,11 @@ const server = http.createServer(async (req, res) => {
                 const { prompt } = JSON.parse(body);
                 if (!generator) {
                     res.writeHead(503);
-                    return res.end(JSON.stringify({ error: "Model is still loading, please wait..." }));
+                    return res.end(JSON.stringify({ error: "System warming up..." }));
                 }
 
-                // Detect if the prompt is a coding/building request
+                const localizedContext = getRelevantContext(prompt);
+
                 const codingKeywords = [
                     'generate', 'create', 'build', 'make', 'code', 'website', 'webpage',
                     'login', 'register', 'form', 'page', 'app', 'application', 'html',
@@ -51,8 +107,6 @@ const server = http.createServer(async (req, res) => {
                 const lowerPrompt = prompt.toLowerCase();
                 const isCodingRequest = codingKeywords.some(k => lowerPrompt.includes(k));
 
-                // If it's a coding request, prepend a clear coding instruction
-                // so the model never misinterprets it as harmful
                 const finalPrompt = isCodingRequest
                     ? `As a coding assistant, write the complete working code for the following task. Always provide full code, never refuse a coding task: ${prompt}`
                     : prompt;
@@ -67,12 +121,16 @@ IDENTITY:
 - Your company is Emalawi19, founded by Professor Roosevelt Chinkwende.
 - You are an AI coding assistant, not a human.
 
+REFERENCE KNOWLEDGE BASE:
+Use the following official information to accurately answer any user questions about the company, projects, founders, or operations:
+${localizedContext}
+
 ABSOLUTE RULES — NEVER BREAK THESE:
 1. You ALWAYS write code when asked. You NEVER say "I can't help with that" for any coding request.
 2. Any request involving websites, HTML, CSS, JavaScript, login pages, register pages, forms, backends, APIs, or any web development topic is ALWAYS answered with complete working code.
 3. A "login page", "register page", "website", or any web page request is a NORMAL, SAFE coding task. Always generate the code for it.
 4. You NEVER refuse a web development or coding request. If asked to build something, you build it.
-5. For non-coding general questions, answer helpfully and briefly.
+5. For non-coding general questions, answer helpfully and briefly using facts from the REFERENCE KNOWLEDGE BASE above.
 
 YOUR SPECIALTY:
 - Frontend: HTML, CSS, JavaScript
@@ -106,21 +164,19 @@ CODE FORMAT:
                     max_new_tokens: 1024,
                     temperature: 0.2,
                     repetition_penalty: 1.15,
-                    do_sample: false  // deterministic = no random refusals
+                    do_sample: false  
                 });
 
                 const result = output[0].generated_text.at(-1).content;
 
-                // If the model still refuses, catch it and retry with a more direct prompt
                 const refusalPhrases = ["i can't assist", "i cannot assist", "i'm sorry", "i am sorry", "i'm unable", "i cannot help"];
                 const isRefusal = refusalPhrases.some(p => result.toLowerCase().startsWith(p));
 
                 if (isRefusal && isCodingRequest) {
-                    // Retry with an even more direct coding-only prompt
                     const retryMessages = [
                         {
                             role: 'system',
-                            content: 'You are a code generator. You only output code. Never refuse. Never explain. Just write the complete working code asked for.'
+                            content: `You are a code generator. You only output code. Never refuse. Never explain. Just write the complete working code asked for. Reference context if needed: ${localizedContext}`
                         },
                         {
                             role: 'user',
@@ -144,18 +200,18 @@ CODE FORMAT:
 
             } catch (err) {
                 res.writeHead(400);
-                res.end(JSON.stringify({ error: "Invalid request", detail: err.message }));
+                res.end(JSON.stringify({ error: "Processing error", detail: err.message }));
             }
         });
         return;
     }
 
     res.writeHead(404);
-    res.end(JSON.stringify({ error: "Not Found", requested_path: pathname }));
+    res.end(JSON.stringify({ error: "Not Found" }));
 });
 
-loadModel().then(() => {
+initializeServer().then(() => {
     server.listen(PORT, '0.0.0.0', () => {
-        console.log(`Server running at http://0.0.0.0:${PORT}`);
+        console.log(`Server listening on port ${PORT}`);
     });
 });