ai-codegen-cli-vrk 1.0.1 → 2.0.1

package/bin/index.js

@@ -1,2 +1,2 @@
 #!/usr/bin/env node
-import "../src/runner.js";
+import "../src/runner.js";

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "1.0.1",
-  "description": "Terminal-based AI code generator",
+  "version": "2.0.1",
+  "description": "Minimalist Terminal-based AI code generator",
   "type": "module",
   "bin": {
     "ai-codegen": "bin/index.js"
@@ -11,8 +11,11 @@
     "src"
   ],
   "dependencies": {
+    "@google/generative-ai": "^0.21.0",
     "fs-extra": "^11.2.0",
-    "groq-sdk": "^0.7.0",
     "readline-sync": "^1.4.10"
+  },
+  "engines": {
+    "node": ">=18.0.0"
   }
-}
+}

package/src/aiClient.js

@@ -1,110 +1,56 @@
-import Groq from "groq-sdk";
+import { GoogleGenerativeAI } from "@google/generative-ai";
 
-let groqClient = null;
+let genAI = null;
+let model = null;
 
 export function setApiKey(apiKey) {
-  groqClient = new Groq({ apiKey });
+  genAI = new GoogleGenerativeAI(apiKey);
+  // Using 1.5-flash for speed and large context window
+  model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
 }
 
-async function call(messages) {
-  if (!groqClient) {
-    throw new Error("API key not initialized");
-  }
-
-  return groqClient.chat.completions.create({
-    model: "llama-3.1-8b-instant",
-    temperature: 0,
-    messages
-  });
-}
-
-/* ========= PHASE 1: FILE PLAN ========= */
-export async function generateFilePlan(task, tests) {
-  const res = await call([
-    {
-      role: "system",
-      content: `
-Return ONLY JSON.
-
-Generate ONLY file paths required to solve the task.
-
-FORMAT:
-{
-  "files": [
-    { "path": "file.ext" }
-  ]
-}
-`
-    },
-    {
-      role: "user",
-      content: `TASK:\n${task}\n\nTESTS:\n${tests}`
-    }
-  ]);
-
-  const parsed = JSON.parse(res.choices[0].message.content);
-
-  if (!Array.isArray(parsed.files)) {
-    throw new Error("Invalid file plan from AI");
-  }
-
-  return parsed.files.map(f => f.path);
-}
-
-/* ========= PHASE 2: FILE CONTENT ========= */
-export async function generateFileContent(filePath, task, tests) {
-  try {
-    const res = await call([
-      {
-        role: "system",
-        content: `
-Generate ONLY the source code for this file:
-
-${filePath}
-
-No explanations.
-No markdown.
-Return RAW CODE ONLY.
-`
-      },
-      {
-        role: "user",
-        content: `TASK:\n${task}\n\nTESTS:\n${tests}`
-      }
-    ]);
-
-    const content = res.choices[0].message.content?.trim();
-
-    if (content && content.length > 0) {
-      return { path: filePath, content };
-    }
-  } catch {
-    /* fall through */
-  }
-
-  /* ========= GENERIC FALLBACK ========= */
-
-  if (filePath.endsWith(".py")) {
-    return {
-      path: filePath,
-      content: `def main():
-    pass
-
-if __name__ == "__main__":
-    main()
-`
-    };
-  }
-
-  if (filePath.endsWith(".js")) {
-    return {
-      path: filePath,
-      content: `module.exports = {};`
-    };
-  }
-
-  return {
-    path: filePath,
-    content: ""
-  };
-}
+export async function generateFullProject(task, tests) {
+  if (!genAI) throw new Error("API Key not set");
+
+  const prompt = `
+You are an expert automated coding exam solver. 
+Generate the ENTIRE project in a SINGLE continuous text response.
+
+### RULES:
+1. ABSOLUTE PRIORITY: Test cases are the ONLY specification.
+2. MINIMALIST: Generate the bare minimum code to pass. No extra logic.
+3. FLAT STRUCTURE: Use simple, standard patterns.
+4. DEPENDENCIES: Only use 'express', 'mongoose', 'jsonwebtoken', and 'cookie-parser' if needed.
+
+### FORMATTING (STRICT):
+Every file MUST start with this EXACT header style:
+
+/**
+ * ==========================================
+ * FILE: package.json (Reference)
+ * ==========================================
+ * [List Dependencies here]
+ */
+
+// ==========================================
+// FILE: path/to/filename.js
+// ==========================================
+[CODE HERE]
+
+### TASK:
+${task}
+
+### TESTS:
+${tests}
+
+### OUTPUT:
+Return ONLY the code sections. No talk. No markdown backticks.
+`;
+
+  const result = await model.generateContent(prompt);
+  const response = await result.response;
+  const text = response.text();
+  
+  // Strip markdown backticks in case AI adds them
+  return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
+}

package/src/fileWriter.js

@@ -1,10 +1,7 @@
 import fs from "fs-extra";
 import path from "path";
 
-export async function writeFiles(baseDir, files) {
-  for (const file of files) {
-    const fullPath = path.join(baseDir, file.path);
-    await fs.ensureDir(path.dirname(fullPath));
-    await fs.writeFile(fullPath, file.content, "utf8");
-  }
-}
+export async function writeSingleFile(baseDir, content) {
+  const outputPath = path.join(baseDir, "generated_project.txt");
+  await fs.writeFile(outputPath, content, "utf8");
+}

package/src/runner.js

@@ -1,63 +1,44 @@
 import readlineSync from "readline-sync";
 import fs from "fs";
+import { setApiKey, generateFullProject } from "./aiClient.js";
+import { writeSingleFile } from "./fileWriter.js";
 
-import { setApiKey, generateFilePlan, generateFileContent } from "./aiClient.js";
-import { writeFiles } from "./fileWriter.js";
+async function main() {
+  // Visible prompt for API key
+  const apiKey = readlineSync.question("--- ");
+  if (!apiKey || apiKey.trim().length === 0) process.exit(1);
+  setApiKey(apiKey.trim());
 
-/* -------- API KEY (---) -------- */
-const apiKey = readlineSync.question("--- ", {
-  hideEchoBack: true
-});
+  const task = readlineSync.question("- ");
+  if (!task || task.trim().length === 0) process.exit(1);
 
-if (!apiKey || apiKey.trim().length === 0) {
-  process.exit(1);
-}
-
-setApiKey(apiKey);
-
-/* -------- TASK (-) -------- */
-const task = readlineSync.question("- ");
-
-if (!task || task.trim().length === 0) {
-  process.exit(1);
-}
-
-/* -------- TESTS (--) OPTIONAL -------- */
-const testsInput = readlineSync.question("-- ");
-let tests = "";
-
-try {
-  if (
-    testsInput &&
-    (testsInput.endsWith(".js") || testsInput.endsWith(".test.js")) &&
-    fs.existsSync(testsInput)
-  ) {
-    tests = fs.readFileSync(testsInput, "utf8");
-  } else {
+  const testsInput = readlineSync.question("-- ");
+  let tests = "";
+  try {
+    if (testsInput && fs.existsSync(testsInput)) {
+      tests = fs.readFileSync(testsInput, "utf8");
+    } else {
+      tests = testsInput || "Implement correctly.";
+    }
+  } catch {
     tests = testsInput;
   }
-} catch {
-  tests = testsInput;
-}
-
-/* ---- SILENT FALLBACK ---- */
-if (!tests || tests.trim().length === 0) {
-  tests = "No tests provided. Generate a simple correct implementation.";
-}
 
-(async () => {
   try {
-    const paths = await generateFilePlan(task, tests);
-
-    const files = [];
-    for (const p of paths) {
-      const file = await generateFileContent(p, task, tests);
-      files.push(file);
+    console.log(".....");
+    const projectContent = await generateFullProject(task, tests);
+    
+    if (!projectContent || projectContent.length < 20) {
+      throw new Error("AI returned no project content. Check your prompt.");
     }
 
-    await writeFiles(process.cwd(), files);
+    await writeSingleFile(process.cwd(), projectContent);
     process.exit(0);
-  } catch {
+  } catch (error) {
+    // Report errors explicitly instead of exiting silently
+    console.error("\n❌ Error:", error.message);
     process.exit(1);
   }
-})();
+}
+
+main();

package/src/promptTemplate.js

@@ -1,34 +0,0 @@
-export const systemPrompt = `
-You are solving an automated coding exam.
-
-ABSOLUTE PRIORITY:
-The given test cases are the ONLY specification.
-
-RULES:
-- Generate the MINIMUM code needed
-- No extra logic
-- No comments
-- No console logs
-- No refactoring
-- No optimization
-
-ARCHITECTURE:
-- Create ONLY required files
-- Names must match tests EXACTLY
-- Error messages must match EXACTLY
-
-OUTPUT:
-- ONLY valid JSON
-- No markdown
-- No explanations
-
-FORMAT:
-{
-  "files": [
-    {
-      "path": "relative/path/file.js",
-      "content": "exact code"
-    }
-  ]
-}
-`;