Reupload OmniDev clean version

app/api/augment/route.ts

@@ -1,6 +1,7 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import { NextRequest, NextResponse } from "next/server";
 import { GoogleGenAI } from "@google/genai";
+import JSON5 from "json5";
 import { InferenceClient } from "@huggingface/inference";
 import type { AugmentRequest, AugmentResponse } from "@/types";
 
@@ -27,18 +28,41 @@ Constraints:
 - Produce compilable code for the chosen language/framework
 `;
 
-function extractJson(text: string): any {
+function cleanText(raw: string): string {
+  let t = raw || "";
+  t = t.replace(/```[a-zA-Z]*\n?/g, "");
+  t = t.replace(/```/g, "");
+  return t.trim();
+}
+
+function tryParseAny(jsonish: string): any {
+  // Try strict JSON first
   try {
-    return JSON.parse(text);
+    return JSON.parse(jsonish);
   } catch {
-    // Try to strip code fences or extraneous text
-    const start = text.indexOf("{");
-    const end = text.lastIndexOf("}");
-    if (start >= 0 && end > start) {
-      const candidate = text.slice(start, end + 1);
-      return JSON.parse(candidate);
+    // Relaxed JSON5 parse (handles trailing commas, etc.)
+    try {
+      // Prefer full string
+      return JSON5.parse(jsonish);
+    } catch {
+      // Try to extract object
+      const sObj = jsonish.indexOf("{");
+      const eObj = jsonish.lastIndexOf("}");
+      if (sObj >= 0 && eObj > sObj) {
+        const candidate = jsonish.slice(sObj, eObj + 1);
+        try { return JSON.parse(candidate); } catch {}
+        try { return JSON5.parse(candidate); } catch {}
+      }
+      // Try to extract array
+      const sArr = jsonish.indexOf("[");
+      const eArr = jsonish.lastIndexOf("]");
+      if (sArr >= 0 && eArr > sArr) {
+        const candidate = jsonish.slice(sArr, eArr + 1);
+        try { return JSON.parse(candidate); } catch {}
+        try { return JSON5.parse(candidate); } catch {}
+      }
+      throw new Error("Model did not return valid JSON");
     }
-    throw new Error("Model did not return valid JSON");
   }
 }
 
@@ -93,13 +117,14 @@ export async function POST(req: NextRequest) {
       text = res.choices?.[0]?.message?.content || "";
     }
 
-    if (!text.trim()) {
+    const cleaned = cleanText(text);
+    if (!cleaned) {
       return NextResponse.json({ ok: false, message: "Empty model response" } as AugmentResponse, { status: 500 });
     }
 
     let json: any;
     try {
-      json = extractJson(text);
+      json = tryParseAny(cleaned);
     } catch (e: any) {
       return NextResponse.json({ ok: false, message: e?.message || "Invalid JSON from model", raw: text } as any, { status: 500 });
     }
@@ -108,6 +133,10 @@ export async function POST(req: NextRequest) {
     if (json && json.ok && Array.isArray(json.files)) {
       return NextResponse.json(json as AugmentResponse, { status: 200 });
     }
+    // If model returned array at root, assume it's files
+    if (Array.isArray(json)) {
+      return NextResponse.json({ ok: true, files: json } as AugmentResponse, { status: 200 });
+    }
 
     return NextResponse.json({ ok: false, message: "Model returned unexpected shape", raw: json } as any, { status: 500 });
   } catch (e: any) {

components/new/scaffold.tsx

@@ -6,9 +6,8 @@ import { useRouter } from "next/navigation";
 import { Button } from "@/components/ui/button";
 import { toast } from "sonner";
 import { NEW_STACKS, NewStackId } from "@/lib/new-stacks";
-import { HERO_STYLES } from "@/lib/hero-presets";
 
-function buildInstruction(stack: NewStackId, lang: "js" | "ts", hero: string, title?: string) {
+function buildInstruction(stack: NewStackId, lang: "js" | "ts", title?: string) {
   const jsOrTs = lang === 'ts' ? 'TypeScript' : 'JavaScript';
   const reactEntry = lang === 'ts' ? "/frontend/src/main.tsx and /frontend/src/App.tsx" : "/frontend/src/main.jsx and /frontend/src/App.jsx";
 
@@ -21,7 +20,7 @@ Title: ${title || 'OmniDev Full-Stack App'}
 Requirements:
 - Frontend under /frontend using React + Vite (${jsOrTs}), TailwindCSS preconfigured.
 - Backend under /backend using Express (${jsOrTs}, ESM), with basic routes (GET /, GET /health) and CORS enabled.
-- Use a Hero section with style: ${hero}. Keep it performant and accessible.
+- Create a visually striking Hero section (animated) appropriate to the app, optimized and accessible.
 - Add a minimal README.md at root with start instructions.
 - Provide package.json in both /frontend and /backend with scripts to start dev/prod.
 - Provide /frontend/index.html and ${reactEntry}.
@@ -36,7 +35,7 @@ Title: ${title || 'OmniDev Next App'}
 
 Requirements:
 - Next.js (${jsOrTs}), App Router, TailwindCSS.
-- Implement a landing page with a Hero section style: ${hero}.
+- Implement a landing page with an animated Hero section suitable for the domain.
 - Add /api/health route that returns { ok: true }.
 - Provide package.json with dev/build scripts.
 - Keep it simple and runnable with \'next dev\'.
@@ -48,7 +47,7 @@ Title: ${title || 'OmniDev Nest + React'}
 
 Requirements:
 - Backend under /backend using NestJS (${jsOrTs}). Generate AppModule, AppController with GET / and GET /health. Enable CORS.
-- Frontend under /frontend using React + Vite (${jsOrTs}), TailwindCSS with a modern Hero section style: ${hero}.
+- Frontend under /frontend using React + Vite (${jsOrTs}), TailwindCSS with a modern animated Hero section appropriate to the app.
 - Provide package.json in both apps with start/build scripts.
 - Return STRICT JSON ONLY as file updates (no markdown), paths rooted from repo.`;
   }
@@ -63,7 +62,6 @@ export default function ScaffoldNew() {
   const [logs, setLogs] = useState<string[]>([]);
   const [stack, setStack] = useState<NewStackId>("express-react");
   const [lang, setLang] = useState<"js" | "ts">("js");
-  const [hero, setHero] = useState<string>(HERO_STYLES[0].id);
   const [title, setTitle] = useState<string>("");
 
   async function runScaffold() {
@@ -76,7 +74,7 @@ export default function ScaffoldNew() {
     setLogs(["Starting scaffold via Augment..."]);
     try {
       // 1) Call augment to generate full-stack files
-      const instruction = buildInstruction(stack, lang, hero, title);
+      const instruction = buildInstruction(stack, lang, title);
       const aug = await fetch('/api/augment', {
         method: 'POST',
         headers: { 'Content-Type': 'application/json' },
@@ -86,8 +84,9 @@ export default function ScaffoldNew() {
           language: lang === 'ts' ? 'typescript' : 'javascript',
           framework: stack,
           response_type: 'file_updates',
-          model,
-          provider,
+          // Force Gemini for scaffold stage
+          model: 'gemini-2.5-flash',
+          provider: 'google',
         }),
       }).then(r => r.json());
       if (!aug?.ok || !Array.isArray(aug.files) || aug.files.length === 0) {
@@ -141,14 +140,7 @@ export default function ScaffoldNew() {
           </select>
           <p className="text-xs text-neutral-500 mt-1">{NEW_STACKS.find(s => s.id === stack)?.description}</p>
         </div>
-        <div>
-          <label className="text-sm block mb-1">Hero Style</label>
-          <select className="w-full bg-neutral-800 rounded p-2 text-sm" value={hero} onChange={(e) => setHero(e.target.value)}>
-            {HERO_STYLES.map(h => (
-              <option key={h.id} value={h.id}>{h.label}</option>
-            ))}
-          </select>
-        </div>
+        {/* Hero selection removed: AI decides the best hero automatically */}
       </div>
       {error && <p className="text-red-400 text-sm mt-3">{error}</p>}
       <ul className="text-sm text-neutral-400 space-y-1 mt-3">

package-lock.json

@@ -34,6 +34,7 @@
         "clsx": "^2.1.1",
         "date-fns": "^4.1.0",
         "framer-motion": "^12.23.22",
+        "json5": "^2.2.3",
         "log4js": "^6.9.1",
         "log4js-json-layout": "^2.2.3",
         "lucide-react": "^0.542.0",

package.json

@@ -8,6 +8,7 @@
     "start": "next start"
   },
   "dependencies": {
+    "json5": "^2.2.3",
     "@google/genai": "^0.11.0",
     "mime": "^4.0.4",
     "@huggingface/hub": "^2.6.2",