Reupload OmniDev clean version

app/api/augment/route.ts

@@ -0,0 +1,104 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import { NextRequest, NextResponse } from "next/server";
+import { GoogleGenAI } from "@google/genai";
+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)
+- a user instruction
+
+You MUST respond ONLY with strict JSON, no comments, no markdown fences.
+JSON schema:
+{
+  "ok": true,
+  "summary": string,
+  "logs": string[],
+  "files": [
+    { "path": string, "action": "add"|"update"|"delete", "content"?: string, "note"?: string }
+  ]
+}
+
+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
+`;
+
+function extractJson(text: string): any {
+  try {
+    return JSON.parse(text);
+  } 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);
+    }
+    throw new Error("Model did not return valid JSON");
+  }
+}
+
+export async function POST(req: NextRequest) {
+  try {
+    const body = (await req.json()) as AugmentRequest;
+    const { context, instruction, language = "javascript", framework = "express-react", response_type = "file_updates", model, provider } = body || {} as AugmentRequest;
+    if (!context || !instruction) {
+      return NextResponse.json({ ok: false, message: "Missing context or instruction" } as AugmentResponse, { status: 400 });
+    }
+
+    const userPrompt = `Project Context:\n${context}\n\nInstruction:\n${instruction}\n\nTarget:\n- language: ${language}\n- framework: ${framework}\n- response_type: ${response_type}`;
+
+    let text = "";
+    if ((provider || "").toLowerCase() === "google" || (model || "").toLowerCase().startsWith("gemini-")) {
+      const apiKey = process.env.GEMINI_API_KEY;
+      if (!apiKey) return NextResponse.json({ ok: false, message: "Missing GEMINI_API_KEY" } as AugmentResponse, { status: 500 });
+      const ai = new GoogleGenAI({ apiKey });
+      const res = await ai.models.generateContent({
+        model: model || "gemini-2.5-flash",
+        contents: [
+          { role: "user", parts: [{ text: SYS }] },
+          { role: "user", parts: [{ text: userPrompt }] },
+        ],
+        config: { maxOutputTokens: 4096 },
+      } as any);
+      text = (res as any)?.candidates?.[0]?.content?.parts?.[0]?.text || "";
+    } else {
+      const token = process.env.HF_TOKEN || process.env.DEFAULT_HF_TOKEN;
+      const client = new InferenceClient(token);
+      const res = await client.chatCompletion({
+        model: model || "deepseek-ai/DeepSeek-V3.1",
+        provider: (provider as any) || undefined,
+        messages: [
+          { role: "system", content: SYS },
+          { role: "user", content: userPrompt },
+        ],
+      });
+      text = res.choices?.[0]?.message?.content || "";
+    }
+
+    if (!text.trim()) {
+      return NextResponse.json({ ok: false, message: "Empty model response" } as AugmentResponse, { status: 500 });
+    }
+
+    let json: any;
+    try {
+      json = extractJson(text);
+    } catch (e: any) {
+      return NextResponse.json({ ok: false, message: e?.message || "Invalid JSON from model", raw: text } as any, { status: 500 });
+    }
+
+    // Minimal validation
+    if (json && json.ok && Array.isArray(json.files)) {
+      return NextResponse.json(json as AugmentResponse, { status: 200 });
+    }
+
+    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 });
+  }
+}
+

app/api/me/projects/[namespace]/[repoId]/apply/route.ts

@@ -0,0 +1,59 @@
+import { NextRequest, NextResponse } from "next/server";
+import { uploadFiles, deleteFiles } from "@huggingface/hub";
+
+import { isAuthenticated } from "@/lib/auth";
+import type { FileUpdate } from "@/types";
+
+export async function POST(
+  req: NextRequest,
+  { params }: { params: Promise<{ namespace: string; repoId: string }> }
+) {
+  const user = await isAuthenticated();
+  if (user instanceof NextResponse || !user) {
+    return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
+  }
+
+  const { files, commitTitle = "OmniDev: Apply changes" } = await req.json();
+  if (!Array.isArray(files) || files.length === 0) {
+    return NextResponse.json(
+      { ok: false, error: "files array is required" },
+      { status: 400 }
+    );
+  }
+
+  const { namespace, repoId } = await params;
+  const repo = { type: "space" as const, name: `${namespace}/${repoId}` };
+
+  try {
+    const toDelete: string[] = [];
+    const toUpload: File[] = [];
+
+    (files as FileUpdate[]).forEach((f) => {
+      const path = (f.path || '').replace(/^\/+/, '');
+      if (f.action === 'delete') {
+        toDelete.push(path);
+      } else if ((f.action === 'add' || f.action === 'update') && typeof f.content === 'string') {
+        const type = path.endsWith('.html') ? 'text/html' : 'text/plain';
+        toUpload.push(new File([f.content], path, { type }));
+      }
+    });
+
+    const results: any = { ok: true, commits: [] };
+
+    if (toDelete.length > 0) {
+      const delRes = await deleteFiles({ repo, accessToken: user.token as string, paths: toDelete });
+      results.commits.push({ type: 'delete', ...delRes.commit });
+    }
+
+    if (toUpload.length > 0) {
+      const upRes = await uploadFiles({ repo, files: toUpload, accessToken: user.token as string, commitTitle });
+      results.commits.push({ type: 'upload', ...upRes.commit, title: commitTitle });
+    }
+
+    return NextResponse.json(results);
+  } catch (error: any) {
+    console.error('Apply changes failed:', error);
+    return NextResponse.json({ ok: false, error: error.message || 'Failed to apply changes' }, { status: 500 });
+  }
+}
+

app/console/page.tsx

@@ -0,0 +1,10 @@
+import OmniConsole from "@/components/omni-console";
+
+export default function ConsolePage() {
+  return (
+    <main className="max-w-5xl mx-auto p-6">
+      <OmniConsole />
+    </main>
+  );
+}
+

app/layout.tsx

@@ -28,14 +28,14 @@ const ptSans = PT_Sans({
 
 export const metadata: Metadata = {
   ...generateSEO({
-    title: "DeepSite | Build with AI ✨",
+    title: "OmniDev | AI Project Builder",
     description:
-      "DeepSite is a web development tool that helps you build websites with AI, no code required. Let's deploy your website with DeepSite and enjoy the magic of AI.",
+      "OmniDev is an AI engineering environment that builds complete web projects (frontend + backend) from natural language.",
     path: "/",
   }),
   appleWebApp: {
     capable: true,
-    title: "DeepSite",
+    title: "OmniDev",
     statusBarStyle: "black-translucent",
   },
   icons: {
@@ -77,16 +77,16 @@ export default async function RootLayout({
 }>) {
   const data = await getMe();
 
-  // Generate structured data
+  // Structured data with OmniDev
   const structuredData = generateStructuredData("WebApplication", {
-    name: "DeepSite",
-    description: "Build websites with AI, no code required",
-    url: "https://deepsite.hf.co",
+    name: "OmniDev",
+    description: "AI-powered full-stack project builder",
+    url: process.env.PUBLIC_BASE_URL || "https://omnidev.hf.co",
   });
 
   const organizationData = generateStructuredData("Organization", {
-    name: "DeepSite",
-    url: "https://deepsite.hf.co",
+    name: "OmniDev",
+    url: process.env.PUBLIC_BASE_URL || "https://omnidev.hf.co",
   });
 
   return (
@@ -105,11 +105,15 @@ export default async function RootLayout({
           }}
         />
       </head>
-      <Script
-        defer
-        data-domain="deepsite.hf.co"
-        src="https://plausible.io/js/script.js"
-      ></Script>
+      {process.env.PUBLIC_BASE_URL && (
+        <Script
+          defer
+          data-domain={(process.env.PUBLIC_BASE_URL as string)
+            .replace("https://", "")
+            .replace("http://", "")}
+          src="https://plausible.io/js/script.js"
+        />
+      )}
       <body
         className={`${inter.variable} ${ptSans.variable} antialiased bg-black dark h-[100dvh] overflow-hidden`}
       >
@@ -128,3 +132,4 @@ export default async function RootLayout({
     </html>
   );
 }
+

components/editor/index.tsx

@@ -49,8 +49,8 @@ export const AppEditor = ({
   const editorRef = useRef<editor.IStandaloneCodeEditor | null>(null);
 
   useMount(() => {
+    // Always start clean for new projects; clear any leftover local storage pages
     if (isNew && pagesStorage) {
-      setPages(pagesStorage);
       removePagesStorage();
     }
   });

components/iframe-detector/modal.tsx

@@ -59,3 +59,4 @@ IframeWarningModalProps) {
     </Dialog>
   );
 }
+

components/omni-console/README.md

@@ -0,0 +1,9 @@
+Omni Console
+
+This console demonstrates the Omni Engine augment flow:
+
+- Use the Project Context to paste a tree and key file excerpts.
+- Enter an Instruction (e.g., Add JWT auth with /auth/login route).
+- Click Run to get an AugmentResponse including file updates.
+- Provide the target Space "namespace/repoId" then click Apply Changes to commit to your Space.
+

components/omni-console/index.tsx

@@ -0,0 +1,76 @@
+"use client";
+import { useState } from "react";
+import { Button } from "@/components/ui/button";
+import { TextareaHTMLAttributes } from "react";
+import { useAi } from "@/hooks/useAi";
+
+export default function OmniConsole() {
+  const [instruction, setInstruction] = useState("");
+  const [context, setContext] = useState("/frontend\n  /src\n  /components\n/backend\n  /routes\n  /controllers\n");
+  const [out, setOut] = useState<any>(null);
+  const { model, provider } = useAi();
+  const [loading, setLoading] = useState(false);
+  const [space, setSpace] = useState("");
+  const [commitTitle, setCommitTitle] = useState("OmniDev: Apply changes");
+
+  const run = async () => {
+    setLoading(true);
+    setOut(null);
+    try {
+      const res = await fetch('/api/augment', {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ context, instruction, language: 'javascript', framework: 'express-react', response_type: 'file_updates', model, provider }),
+      });
+      const data = await res.json();
+      setOut(data);
+    } finally {
+      setLoading(false);
+    }
+  };
+
+  const apply = async () => {
+    if (!out?.files || !Array.isArray(out.files) || out.files.length === 0) return;
+    if (!space.includes('/')) {
+      alert('Specify target space as namespace/repoId');
+      return;
+    }
+    const [namespace, repoId] = space.split('/');
+    setLoading(true);
+    try {
+      const res = await fetch(`/api/me/projects/${namespace}/${repoId}/apply`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ files: out.files, commitTitle }),
+      });
+      const data = await res.json();
+      setOut((prev: any) => ({ ...prev, apply: data }));
+    } finally {
+      setLoading(false);
+    }
+  }
+
+  return (
+    <div className="p-4 border border-neutral-800 bg-neutral-900 rounded-xl text-neutral-200">
+      <h2 className="text-xl font-semibold mb-2">Omni Console</h2>
+      <p className="text-sm text-neutral-400 mb-4">Type engineering instructions; the engine returns structured file updates.</p>
+      <div className="grid gap-2">
+        <label className="text-sm">Project Context</label>
+        <textarea className="min-h-24 bg-neutral-800 rounded p-2 text-sm" value={context} onChange={(e) => setContext(e.target.value)} />
+        <label className="text-sm">Instruction</label>
+        <textarea className="min-h-24 bg-neutral-800 rounded p-2 text-sm" placeholder="e.g., Add JWT auth with /auth/login route" value={instruction} onChange={(e) => setInstruction(e.target.value)} />
+        <label className="text-sm">Target Space (namespace/repoId)</label>
+        <input className="bg-neutral-800 rounded p-2 text-sm" placeholder="yourname/your-space" value={space} onChange={(e) => setSpace(e.target.value)} />
+        <label className="text-sm">Commit Title</label>
+        <input className="bg-neutral-800 rounded p-2 text-sm" value={commitTitle} onChange={(e) => setCommitTitle(e.target.value)} />
+      </div>
+      <div className="flex items-center gap-2 mt-3">
+        <Button size="sm" onClick={run} disabled={loading || !instruction.trim()}>Run</Button>
+        <Button size="sm" variant="outline" onClick={apply} disabled={loading || !out?.files?.length}>Apply Changes</Button>
+      </div>
+      {out && (
+        <pre className="mt-4 text-xs bg-neutral-950 p-3 rounded overflow-x-auto max-h-80">{JSON.stringify(out, null, 2)}</pre>
+      )}
+    </div>
+  );
+}

lib/seo.ts

@@ -10,14 +10,14 @@ interface SEOParams {
 }
 
 export function generateSEO({
-  title = "DeepSite | Build with AI ✨",
-  description = "DeepSite is a web development tool that helps you build websites with AI, no code required. Let's deploy your website with DeepSite and enjoy the magic of AI.",
+  title = "OmniDev | AI Project Builder",
+  description = "OmniDev is an AI engineering environment that builds complete web projects (frontend + backend) from natural language.",
   path = "",
   image = "/banner.png",
   noIndex = false,
   canonical,
 }: SEOParams = {}): Metadata {
-  const baseUrl = process.env.PUBLIC_BASE_URL || "https://deepsite.hf.co";
+  const baseUrl = process.env.PUBLIC_BASE_URL || "https://omnidev.hf.co";
   const fullUrl = `${baseUrl}${path}`;
   const canonicalUrl = canonical || fullUrl;
   
@@ -54,13 +54,13 @@ export function generateSEO({
       title,
       description,
       url: canonicalUrl,
-      siteName: "DeepSite",
+      siteName: "OmniDev",
       images: [
         {
           url: `${baseUrl}${image}`,
           width: 1200,
           height: 630,
-          alt: `${title} - DeepSite`,
+          alt: `${title} - OmniDev`,
         },
       ],
       locale: "en_US",
@@ -71,14 +71,11 @@ export function generateSEO({
       title,
       description,
       images: [`${baseUrl}${image}`],
-      creator: "@deepsite",
+      creator: "@omnidev",
     },
     other: {
-      // Prevent iframe embedding from unauthorized domains
       'X-Frame-Options': 'SAMEORIGIN',
-      // Control how the page appears when shared
       'og:image:secure_url': `${baseUrl}${image}`,
-      // Help search engines understand the primary URL
       'rel': 'canonical',
     },
   };
@@ -94,9 +91,9 @@ export function generateStructuredData(type: 'WebApplication' | 'Organization' |
     case 'WebApplication':
       return {
         ...baseStructuredData,
-        name: 'DeepSite',
-        description: 'Build websites with AI, no code required',
-        url: process.env.PUBLIC_BASE_URL || 'https://deepsite.hf.co',
+        name: 'OmniDev',
+        description: 'AI-powered full-stack project builder',
+        url: process.env.PUBLIC_BASE_URL || 'https://omnidev.hf.co',
         applicationCategory: 'DeveloperApplication',
         operatingSystem: 'Web',
         offers: {
@@ -106,8 +103,8 @@ export function generateStructuredData(type: 'WebApplication' | 'Organization' |
         },
         creator: {
           '@type': 'Organization',
-          name: 'DeepSite',
-          url: process.env.PUBLIC_BASE_URL || 'https://deepsite.hf.co',
+          name: 'OmniDev',
+          url: process.env.PUBLIC_BASE_URL || 'https://omnidev.hf.co',
         },
         ...data,
       };
@@ -115,12 +112,11 @@ export function generateStructuredData(type: 'WebApplication' | 'Organization' |
     case 'Organization':
       return {
         ...baseStructuredData,
-        name: 'DeepSite',
-        url: process.env.PUBLIC_BASE_URL || 'https://deepsite.hf.co',
-        logo: `${process.env.PUBLIC_BASE_URL || 'https://deepsite.hf.co'}/logo.svg`,
+        name: 'OmniDev',
+        url: process.env.PUBLIC_BASE_URL || 'https://omnidev.hf.co',
+        logo: `${process.env.PUBLIC_BASE_URL || 'https://omnidev.hf.co'}/logo.svg`,
         description: 'AI-powered web development platform',
         sameAs: [
-          // Add social media links here if available
         ],
         ...data,
       };
@@ -129,3 +125,4 @@ export function generateStructuredData(type: 'WebApplication' | 'Organization' |
       return { ...baseStructuredData, ...data };
   }
 }
+

types/index.ts

@@ -55,3 +55,31 @@ export interface EnhancedSettings {
 }
 
 export type Theme = "light" | "dark" | undefined;
+
+// OmniDev Augmentation types
+export type FileAction = "add" | "update" | "delete";
+
+export interface FileUpdate {
+  path: string; // e.g., /frontend/src/App.tsx or /backend/routes/auth.ts
+  action: FileAction;
+  content?: string; // required for add/update; omit for delete
+  note?: string; // optional human-readable note
+}
+
+export interface AugmentRequest {
+  context: string; // project structure + important file snippets
+  instruction: string; // user instruction (e.g., add JWT auth)
+  language?: string; // javascript, typescript, python
+  framework?: string; // express-react, nextjs, nestjs, etc.
+  response_type?: "file_updates" | "explanation";
+  model?: string;
+  provider?: string; // 'google' to force Gemini
+}
+
+export interface AugmentResponse {
+  ok: boolean;
+  files?: FileUpdate[];
+  logs?: string[];
+  summary?: string;
+  message?: string;
+}