Deep_Solo

Running

App Files Files Community

kalhdrawi commited on Oct 20

Commit

2271467

1 Parent(s): ef7bca1

Reupload OmniDev clean version

Browse files

Files changed (1) hide show

app/api/augment/route.ts +67 -3

app/api/augment/route.ts CHANGED Viewed

@@ -3,7 +3,7 @@ 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";
 const SYS = `You are Omni Engine, an autonomous code evolution system.
 INPUTS:
@@ -183,8 +183,36 @@ export async function POST(req: NextRequest) {
           parseErr = null;
         } catch {}
       }
       if (parseErr) {
-        return NextResponse.json({ ok: false, message: e?.message || "Invalid JSON from model", raw: text } as any, { status: 500 });
       }
     }
@@ -214,8 +242,44 @@ export async function POST(req: NextRequest) {
       }
     }
-    return NextResponse.json({ ok: false, message: "Model returned unexpected shape", raw: json } as any, { status: 500 });
   } catch (e: any) {
     return NextResponse.json({ ok: false, message: e?.message || "Internal error" } as AugmentResponse, { status: 500 });
   }
 }

 import { GoogleGenAI } from "@google/genai";
 import JSON5 from "json5";
 import { InferenceClient } from "@huggingface/inference";
+import type { AugmentRequest, AugmentResponse, FileUpdate } from "@/types";
 const SYS = `You are Omni Engine, an autonomous code evolution system.
 INPUTS:
           parseErr = null;
         } catch {}
       }
+      // REPAIR PASS via Gemini: convert arbitrary text to strict JSON per schema
       if (parseErr) {
+        try {
+          const apiKey = process.env.GEMINI_API_KEY;
+          if (apiKey) {
+            const ai = new GoogleGenAI({ apiKey });
+            const repairSys = `You are a JSON repair tool. Convert the user's content into STRICT JSON matching either: { ok, files[] } or just files[]. files[] is an array of { path, action, content?, note? }. No prose, no code fences.`;
+            const resR = await ai.models.generateContent({
+              model: "gemini-2.5-flash",
+              contents: [
+                { role: 'user', parts: [{ text: repairSys }] },
+                { role: 'user', parts: [{ text: cleaned }] },
+              ],
+              config: { maxOutputTokens: 2048 },
+            } as any);
+            const repaired = ((resR as any)?.response?.text && (resR as any).response.text())
+              || (resR as any)?.text
+              || ((resR as any)?.candidates?.[0]?.content?.parts?.map((p: any) => p?.text || "").join("") || "");
+            const repairedClean = cleanText(repaired || "");
+            if (repairedClean) {
+              json = tryParseAny(repairedClean);
+              parseErr = null;
+            }
+          }
+        } catch {}
+      }
+      if (parseErr) {
+        // As a last-resort, generate a minimal runnable scaffold so the flow never breaks
+        const files = generateFallbackFiles(framework, language, (instruction.match(/Title:\s*(.*)/)?.[1] || "OmniDev App"));
+        return NextResponse.json({ ok: true, files } as AugmentResponse, { status: 200 });
       }
     }
       }
     }
+    // Shape still unexpected: provide runnable fallback
+    const files = generateFallbackFiles(framework, language, (instruction.match(/Title:\s*(.*)/)?.[1] || "OmniDev App"));
+    return NextResponse.json({ ok: true, files } as AugmentResponse, { status: 200 });
   } catch (e: any) {
     return NextResponse.json({ ok: false, message: e?.message || "Internal error" } as AugmentResponse, { status: 500 });
   }
 }
+function generateFallbackFiles(framework: string, language: string, title: string): FileUpdate[] {
+  const isTs = language.toLowerCase().includes('ts');
+  if (framework?.startsWith('express')) {
+    const server = `import express from 'express';\nimport cors from 'cors';\nconst app = express();\napp.use(cors());\napp.get('/', (_, res) => res.send('OK'));\napp.get('/health', (_, res) => res.json({ ok: true }));\nconst PORT = process.env.PORT || 3000;\napp.listen(PORT, () => console.log('Server listening on', PORT));\n`;
+    const indexHtml = `<!DOCTYPE html>\n<html lang=\"en\">\n<head>\n<meta charset=\"UTF-8\" />\n<meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\" />\n<title>${title}</title>\n<script src=\"https://cdn.tailwindcss.com\"></script>\n</head>\n<body class=\"bg-neutral-950 text-white\">\n<div id=\"root\"></div>\n<script type=\"module\" src=\"/src/main.${isTs ? 'tsx' : 'jsx'}\"></script>\n</body>\n</html>`;
+    const main = `import React from 'react';\nimport { createRoot } from 'react-dom/client';\nimport App from './App';\ncreateRoot(document.getElementById('root')).render(<App />);\n`;
+    const app = `export default function App(){\n  return (<main className=\"min-h-screen grid place-content-center\">\n    <h1 className=\"text-3xl font-bold\">${title}</h1>\n    <p className=\"text-neutral-400\">Frontend + Backend scaffolded by OmniDev</p>\n  </main>);\n}\n`;
+    const fePkg = { name: "frontend", private: true, scripts: { dev: "vite", build: "vite build", preview: "vite preview" }, dependencies: { react: "^18.3.1", "react-dom": "^18.3.1" }, devDependencies: { vite: "^5.4.0" } };
+    const bePkg = { name: "backend", private: true, type: "module", scripts: { start: "node server.js" }, dependencies: { express: "^4.19.0", cors: "^2.8.5" } };
+    const files: FileUpdate[] = [
+      { path: "/backend/server.js", action: "add", content: server },
+      { path: "/backend/package.json", action: "add", content: JSON.stringify(bePkg, null, 2) },
+      { path: "/frontend/index.html", action: "add", content: indexHtml },
+      { path: `/frontend/src/main.${isTs ? 'tsx' : 'jsx'}`, action: "add", content: main },
+      { path: `/frontend/src/App.${isTs ? 'tsx' : 'jsx'}`, action: "add", content: app },
+      { path: "/frontend/package.json", action: "add", content: JSON.stringify(fePkg, null, 2) },
+      { path: "/README.md", action: "add", content: `# ${title}\n\nGenerated by OmniDev scaffold.` },
+    ];
+    return files;
+  }
+  if (framework?.startsWith('next')) {
+    const page = `export default function Page(){ return (<main style={{minHeight:'100vh',display:'grid',placeContent:'center'}}><h1>${title}</h1><p>Next.js scaffold by OmniDev</p></main>); }`;
+    const api = `export async function GET(){ return Response.json({ ok: true }); }`;
+    const pkg = { name: "next-app", private: true, scripts: { dev: "next dev", build: "next build", start: "next start" }, dependencies: { next: "^15.0.0", react: "^18.3.1", "react-dom": "^18.3.1" } };
+    return [
+      { path: "/app/page.tsx", action: "add", content: page },
+      { path: "/app/api/health/route.ts", action: "add", content: api },
+      { path: "/package.json", action: "add", content: JSON.stringify(pkg, null, 2) },
+      { path: "/README.md", action: "add", content: `# ${title}\n\nGenerated by OmniDev scaffold.` },
+    ];
+  }
+  return [ { path: "/README.md", action: "add", content: `# ${title}\n\nOmniDev scaffold.` } ];
+}