Reupload OmniDev clean version

app/api/augment/route.ts

@@ -6,26 +6,26 @@ import { InferenceClient } from "@huggingface/inference";
 import type { AugmentRequest, AugmentResponse } from "@/types";
 
 const SYS = `You are Omni Engine, an autonomous code evolution system.
-You receive:
-- project context (tree + critical file excerpts)
+INPUTS:
+- project context (tree + critical files)
 - a user instruction
 
-You MUST respond ONLY with strict JSON, no comments, no markdown fences.
-JSON schema:
+OUTPUT:
+- STRICT JSON ONLY. No prose. No code fences. No leading or trailing text.
+- Prefer this object shape:
 {
   "ok": true,
   "summary": string,
   "logs": string[],
-  "files": [
-    { "path": string, "action": "add"|"update"|"delete", "content"?: string, "note"?: string }
-  ]
+  "files": [ { "path": string, "action": "add"|"update"|"delete", "content"?: string, "note"?: string } ]
 }
+- If you cannot produce an object, return a top-level JSON array of the same file objects (files array).
 
-Constraints:
-- Ensure paths are POSIX-like and rooted from repo (e.g., /frontend/src/App.tsx)
-- For add/update, provide full file content (ready to run)
-- Keep changes minimal and consistent
-- Produce compilable code for the chosen language/framework
+RULES:
+- Paths must be POSIX-like and rooted from repo (e.g., /frontend/src/App.tsx)
+- For add/update include full file content (runnable)
+- Keep changes consistent and compilable for the chosen language/framework
+- DO NOT include markdown fences or explanation outside JSON
 `;
 
 function cleanText(raw: string): string {
@@ -66,6 +66,46 @@ function tryParseAny(jsonish: string): any {
   }
 }
 
+function findJsonCandidates(text: string, maxCandidates = 8): string[] {
+  const candidates: string[] = [];
+  const cleaned = cleanText(text);
+
+  // 1) Code-fenced JSON blocks (already stripped in cleanText, so skip)
+
+  // 2) Balanced scanning for objects and arrays
+  const openers = ['{', '['] as const;
+  for (let start = 0; start < cleaned.length; start++) {
+    const ch = cleaned[start];
+    if (!openers.includes(ch as any)) continue;
+    const isArray = ch === '[';
+    const closer = isArray ? ']' : '}';
+
+    let depth = 0;
+    let inStr: null | '"' | "'" = null;
+    let prev = '';
+    for (let i = start; i < cleaned.length; i++) {
+      const c = cleaned[i];
+      if (inStr) {
+        if (c === inStr && prev !== '\\') inStr = null as any;
+      } else {
+        if (c === '"' || c === "'") inStr = c as any;
+        else if (c === ch) depth++;
+        else if (c === closer) depth--;
+        if (depth === 0) {
+          const slice = cleaned.slice(start, i + 1).trim();
+          if ((isArray && slice.startsWith('[') && slice.endsWith(']')) || (!isArray && slice.startsWith('{') && slice.endsWith('}'))) {
+            candidates.push(slice);
+            if (candidates.length >= maxCandidates) return candidates;
+          }
+          break;
+        }
+      }
+      prev = c;
+    }
+  }
+  return candidates;
+}
+
 export async function POST(req: NextRequest) {
   try {
     const body = (await req.json()) as AugmentRequest;
@@ -123,10 +163,29 @@ export async function POST(req: NextRequest) {
     }
 
     let json: any;
+    let parseErr: any = null;
     try {
       json = tryParseAny(cleaned);
     } catch (e: any) {
-      return NextResponse.json({ ok: false, message: e?.message || "Invalid JSON from model", raw: text } as any, { status: 500 });
+      parseErr = e;
+      // Try candidates extracted by balanced scanning
+      const cands = findJsonCandidates(cleaned, 12);
+      for (const c of cands) {
+        try { json = tryParseAny(c); parseErr = null; break; } catch {}
+      }
+      // Last resort: attempt to coerce by simple replacements
+      if (parseErr) {
+        try {
+          const coerced = cleaned
+            .replace(/(\w+)\s*:/g, '"$1":') // quote keys
+            .replace(/'([^']*)'/g, '"$1"'); // single to double quotes
+          json = tryParseAny(coerced);
+          parseErr = null;
+        } catch {}
+      }
+      if (parseErr) {
+        return NextResponse.json({ ok: false, message: e?.message || "Invalid JSON from model", raw: text } as any, { status: 500 });
+      }
     }
 
     // Minimal validation
@@ -137,6 +196,23 @@ export async function POST(req: NextRequest) {
     if (Array.isArray(json)) {
       return NextResponse.json({ ok: true, files: json } as AugmentResponse, { status: 200 });
     }
+    // If the object contains files nested under any key, lift it
+    if (json && typeof json === 'object') {
+      const stack: any[] = [json];
+      while (stack.length) {
+        const cur = stack.pop();
+        if (cur && typeof cur === 'object') {
+          if (Array.isArray(cur.files)) {
+            return NextResponse.json({ ok: true, files: cur.files, summary: cur.summary, logs: cur.logs } as AugmentResponse, { status: 200 });
+          }
+          for (const k of Object.keys(cur)) {
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const v: any = (cur as any)[k];
+            if (v && typeof v === 'object') stack.push(v);
+          }
+        }
+      }
+    }
 
     return NextResponse.json({ ok: false, message: "Model returned unexpected shape", raw: json } as any, { status: 500 });
   } catch (e: any) {

components/new/scaffold.tsx

@@ -118,7 +118,7 @@ export default function ScaffoldNew() {
   return (
     <section className="max-w-3xl mx-auto p-6 text-neutral-200">
       <h1 className="text-2xl font-semibold mb-1">Create New Project</h1>
-      <p className="text-sm text-neutral-400 mb-4">Choose your stack and hero style, then OmniDev will scaffold a complete project.</p>
+      <p className="text-sm text-neutral-400 mb-4">Choose your stack, then OmniDev will scaffold a complete project using AI.</p>
       <div className="grid grid-cols-1 md:grid-cols-2 gap-4">
         <div>
           <label className="text-sm block mb-1">Project Title</label>