Update index.js

index.js

@@ -7,18 +7,10 @@ const app = express();
 const server = http.createServer(app);
 const wss = new WebSocket.Server({ server });
 
-const ollama = new Ollama({ host: "http://127.0.0.1:11434" });
+const ollama = new Ollama({ host: "[http://127.0.0.1:11434](http://127.0.0.1:11434)" });
 
-const SYSTEM = `
-You are Exocore AI Code Engine.
-
-RULES:
-- Support multiple languages: Node.js, Python, PHP, Java, Go
-- Always detect best language automatically
-- NEVER refuse coding requests
-- If unsure → assume programming
-- Output must be structured project files
-`;
+// Make sure Exocore's persona is consistent
+const EXOCORE_PERSONA = `You are Exocore AI, an advanced, brilliant, and highly precise code engine and assistant. You always provide clear, accurate, and functional answers.`;
 
 function send(ws, type, data) {
   ws.send(JSON.stringify({ type, data }));
@@ -34,22 +26,13 @@ async function detectLanguage(prompt) {
     messages: [
       {
         role: "system",
-        content: `
-Detect best programming language.
-
-Return ONLY one word:
-nodejs | python | php | java | go | unknown
-
-Rules:
-- base on user request
-- if web api → nodejs or python
-- if unsure → nodejs
-        `
+        content: `Detect the programming language needed for the user's request.
+Return ONLY one single word from this list: nodejs, python, php, java, go, typescript.
+If unsure, return nodejs.`
       },
       { role: "user", content: prompt }
     ]
   });
-
   return res.message.content.trim().toLowerCase();
 }
 
@@ -59,29 +42,24 @@ async function planFiles(prompt, lang) {
     messages: [
       {
         role: "system",
-        content: `
-You are a senior ${lang} architect.
-
-Return ONLY JSON:
-{
-  "files": ["file1", "file2"]
-}
-
-Rules:
-- include proper structure for ${lang}
-- include entry file
-- include dependencies file (package.json / requirements.txt etc)
-- include folders if needed
-- no explanation
-        `
+        content: `You are a senior ${lang} architect.
+Return ONLY a valid JSON object containing an array of actual file names needed for this project.
+Example format: {"files": ["src/index.ts", "package.json", "tsconfig.json"]}
+RULES:
+- Do NOT use placeholder names like "file1" or "file2".
+- Name the actual entry files, dependency files, and routes based on the user's prompt.
+- Do NOT include any explanations or markdown formatting outside the JSON.`
       },
       { role: "user", content: prompt }
     ]
   });
 
   try {
-    return JSON.parse(res.message.content);
-  } catch {
+    // Strip markdown if the AI mistakenly wraps the JSON in ```json
+    const cleanJson = res.message.content.replace(/```json/g, "").replace(/```/g, "").trim();
+    return JSON.parse(cleanJson);
+  } catch (err) {
+    console.error("JSON Parse Error on Planning:", err.message);
     return { files: lang === "python" ? ["main.py", "requirements.txt"] : ["index.js", "package.json"] };
   }
 }
@@ -92,56 +70,48 @@ async function generateFile(file, prompt, lang) {
     messages: [
       {
         role: "system",
-        content: `
-You are a ${lang} code generator.
-
-Generate ONLY this file: ${file}
-
-Rules:
-- No explanation
-- ONLY valid code
-- Must match ${lang} project standards
-        `
+        content: `You are a strict ${lang} code generator.
+Write ONLY the complete, functional, and production-ready code for the file named: ${file}.
+RULES:
+- Implement all required functions as requested in the prompt.
+- Use strict typing and correct syntax.
+- Do NOT include explanations, greetings, or comments outside the code.`
       },
       { role: "user", content: prompt }
     ]
   });
 
-  return res.message.content;
+  // Strip markdown code blocks so the raw file is clean
+  let code = res.message.content.trim();
+  code = code.replace(/^```[\w]*\n/, '').replace(/\n```$/, '').trim();
+  
+  return code;
 }
 
+// --- WEBSOCKET FOR FILE GENERATION ---
 wss.on("connection", (ws) => {
   ws.on("message", async (msg) => {
     const prompt = msg.toString();
 
     send(ws, "connected", "Exocore MultiLang WS Active");
 
-    // GREETING MODE
     if (isGreeting(prompt)) {
       send(ws, "greeting", "Hello! Exocore AI is ready for coding.");
       return;
     }
 
-    // STEP 1 - DETECT LANGUAGE
     send(ws, "step", "Detecting language...");
-
     const lang = await detectLanguage(prompt);
-
     send(ws, "language", lang);
 
-    // STEP 2 - PLAN PROJECT
     send(ws, "step", `Planning ${lang} project structure...`);
-
     const plan = await planFiles(prompt, lang);
-
     send(ws, "plan_done", plan);
 
     const files = plan.files || [];
 
-    // STEP 3 - GENERATE FILES
     for (const file of files) {
       send(ws, "step", `Generating ${file} (${lang})...`);
-
       const code = await generateFile(file, prompt, lang);
 
       send(ws, "file_done", {
@@ -151,7 +121,6 @@ wss.on("connection", (ws) => {
       });
     }
 
-    // STEP 4 - DONE
     send(ws, "done", {
       message: "Project complete",
       language: lang
@@ -159,12 +128,42 @@ wss.on("connection", (ws) => {
   });
 });
 
-// ROUTE
+// --- EXPRESS ROUTES ---
+
+// Health Check
 app.get("/", (req, res) => {
   res.send("Exocore MultiLang WS AI Running");
 });
 
-// START
+// New Chat Route
+app.get("/chat", async (req, res) => {
+  const userPrompt = req.query.prompt || req.query.q;
+
+  if (!userPrompt) {
+    return res.status(400).json({ error: "Please provide a prompt. Example: /chat?prompt=hello" });
+  }
+
+  try {
+    const response = await ollama.chat({
+      model: "qwen3:0.6b",
+      messages: [
+        { role: "system", content: EXOCORE_PERSONA },
+        { role: "user", content: userPrompt }
+      ]
+    });
+
+    res.json({
+      status: "success",
+      engine: "Exocore AI",
+      response: response.message.content
+    });
+  } catch (error) {
+    console.error("Chat Error:", error);
+    res.status(500).json({ error: "Exocore Engine failed to respond." });
+  }
+});
+
+// --- START SERVER ---
 server.listen(process.env.PORT || 7860, "0.0.0.0", () => {
-  console.log("Exocore MultiLang WS Live");
+  console.log("Exocore MultiLang WS Live on Port 7860");
 });