add-qwen3-coder

MCP-SERVER.md

@@ -1,428 +0,0 @@
-# DeepSite MCP Server
-
-DeepSite is now available as an MCP (Model Context Protocol) server, enabling AI assistants like Claude to create websites directly using natural language.
-
-## Two Ways to Use DeepSite MCP
-
-**Quick Comparison:**
-
-| Feature | Option 1: HTTP Server | Option 2: Local Server |
-|---------|----------------------|------------------------|
-| **Setup Difficulty** | ✅ Easy (just config) | ⚠️ Requires installation |
-| **Authentication** | HF Token in config header | HF Token or session cookie in env |
-| **Best For** | Most users | Developers, custom modifications |
-| **Maintenance** | ✅ Always up-to-date | Need to rebuild for updates |
-
-**Recommendation:** Use Option 1 (HTTP Server) unless you need to modify the MCP server code.
-
----
-
-### 🌐 Option 1: HTTP Server (Recommended)
-
-**No installation required!** Use DeepSite's hosted MCP server.
-
-#### Setup for Claude Desktop
-
-Add to your Claude Desktop configuration file:
-
-**MacOS**: `~/Library/Application Support/Claude/claude_desktop_config.json`  
-**Windows**: `%APPDATA%\Claude\claude_desktop_config.json`
-
-```json
-{
-  "mcpServers": {
-    "deepsite": {
-      "url": "https://huggingface.co/deepsite/api/mcp",
-      "transport": {
-        "type": "sse"
-      },
-      "headers": {
-        "Authorization": "Bearer hf_your_token_here"
-      }
-    }
-  }
-}
-```
-
-**Getting Your Hugging Face Token:**
-
-1. Go to https://huggingface.co/settings/tokens
-2. Create a new token with `write` access
-3. Copy the token
-4. Add it to the `Authorization` header in your config (recommended for security)
-5. Alternatively, you can pass it as the `hf_token` parameter when using the tool
-
-**⚠️ Security Recommendation:** Use the `Authorization` header in your config instead of passing the token in chat. This keeps your token secure and out of conversation history.
-
-#### Example Usage with Claude
-
-> "Create a portfolio website using DeepSite. Include a hero section, about section, and contact form."
-
-Claude will automatically:
-1. Use the `create_project` tool
-2. Authenticate using the token from your config
-3. Create the website on Hugging Face Spaces
-4. Return the URLs to access your new site
-
----
-
-### 💻 Option 2: Local Server
-
-Run the MCP server locally for more control or offline use.
-
-> **Note:** Most users should use Option 1 (HTTP Server) instead. Option 2 is only needed if you want to run the MCP server locally or modify its behavior.
-
-#### Installation
-
-```bash
-cd mcp-server
-npm install
-npm run build
-```
-
-#### Setup for Claude Desktop
-
-**Method A: Using HF Token (Recommended)**
-
-```json
-{
-  "mcpServers": {
-    "deepsite-local": {
-      "command": "node",
-      "args": ["/absolute/path/to/deepsite-v3/mcp-server/dist/index.js"],
-      "env": {
-        "HF_TOKEN": "hf_your_token_here",
-        "DEEPSITE_API_URL": "https://huggingface.co/deepsite"
-      }
-    }
-  }
-}
-```
-
-**Method B: Using Session Cookie (Alternative)**
-
-```json
-{
-  "mcpServers": {
-    "deepsite-local": {
-      "command": "node",
-      "args": ["/absolute/path/to/deepsite-v3/mcp-server/dist/index.js"],
-      "env": {
-        "DEEPSITE_AUTH_COOKIE": "your-session-cookie",
-        "DEEPSITE_API_URL": "https://huggingface.co/deepsite"
-      }
-    }
-  }
-}
-```
-
-**Getting Your Session Cookie (Method B only):**
-
-1. Log in to https://huggingface.co/deepsite
-2. Open Developer Tools (F12)
-3. Go to Application → Cookies
-4. Copy the session cookie value
-5. Set as `DEEPSITE_AUTH_COOKIE` in the config
-
----
-
-## Available Tools
-
-### `create_project`
-
-Creates a new DeepSite project with HTML/CSS/JS files.
-
-**Parameters:**
-
-| Parameter | Type | Required | Description |
-|-----------|------|----------|-------------|
-| `title` | string | No | Project title (defaults to "DeepSite Project") |
-| `pages` | array | Yes | Array of file objects with `path` and `html` |
-| `prompt` | string | No | Commit message/description |
-| `hf_token` | string | No* | Hugging Face API token (*optional if provided via Authorization header in config) |
-
-**Page Object:**
-```typescript
-{
-  path: string;  // e.g., "index.html", "styles.css", "script.js"
-  html: string;  // File content
-}
-```
-
-**Returns:**
-```json
-{
-  "success": true,
-  "message": "Project created successfully!",
-  "projectUrl": "https://huggingface.co/deepsite/username/project-name",
-  "spaceUrl": "https://huggingface.co/spaces/username/project-name",
-  "liveUrl": "https://username-project-name.hf.space",
-  "spaceId": "username/project-name",
-  "projectId": "space-id",
-  "files": ["index.html", "styles.css"]
-}
-```
-
----
-
-## Example Prompts for Claude
-
-### Simple Landing Page
-> "Create a modern landing page for my SaaS product using DeepSite. Include a hero section with CTA, features grid, and footer. Use gradient background."
-
-### Portfolio Website
-> "Build a portfolio website with DeepSite. I need:
-> - Hero section with my name and photo
-> - Projects gallery with 3 sample projects
-> - Skills section with tech stack
-> - Contact form
-> Use dark mode with accent colors."
-
-### Blog Homepage
-> "Create a blog homepage using DeepSite. Include:
-> - Header with navigation
-> - Featured post section
-> - Grid of recent posts (3 cards)
-> - Sidebar with categories
-> - Footer with social links
-> Clean, minimal design."
-
-### Interactive Dashboard
-> "Make an analytics dashboard with DeepSite:
-> - Sidebar navigation
-> - 4 metric cards at top
-> - 2 chart placeholders
-> - Data table
-> - Modern, professional UI with charts.css"
-
----
-
-## Direct API Usage
-
-You can also call the HTTP endpoint directly:
-
-### Using Authorization Header (Recommended)
-
-```bash
-curl -X POST https://huggingface.co/deepsite/api/mcp \
-  -H "Content-Type: application/json" \
-  -H "Authorization: Bearer hf_your_token_here" \
-  -d '{
-    "jsonrpc": "2.0",
-    "id": 1,
-    "method": "tools/call",
-    "params": {
-      "name": "create_project",
-      "arguments": {
-        "title": "My Website",
-        "pages": [
-          {
-            "path": "index.html",
-            "html": "<!DOCTYPE html><html><head><title>Hello</title></head><body><h1>Hello World!</h1></body></html>"
-          }
-        ]
-      }
-    }
-  }'
-```
-
-### Using Token Parameter (Fallback)
-
-```bash
-curl -X POST https://huggingface.co/deepsite/api/mcp \
-  -H "Content-Type: application/json" \
-  -d '{
-    "jsonrpc": "2.0",
-    "id": 1,
-    "method": "tools/call",
-    "params": {
-      "name": "create_project",
-      "arguments": {
-        "title": "My Website",
-        "pages": [
-          {
-            "path": "index.html",
-            "html": "<!DOCTYPE html><html><head><title>Hello</title></head><body><h1>Hello World!</h1></body></html>"
-          }
-        ],
-        "hf_token": "hf_xxxxx"
-      }
-    }
-  }'
-```
-
-### List Available Tools
-
-```bash
-curl -X POST https://huggingface.co/deepsite/api/mcp \
-  -H "Content-Type: application/json" \
-  -d '{
-    "jsonrpc": "2.0",
-    "id": 1,
-    "method": "tools/list",
-    "params": {}
-  }'
-```
-
----
-
-## Testing
-
-### Test Local Server
-
-```bash
-cd mcp-server
-./test.sh
-```
-
-### Test HTTP Server
-
-```bash
-curl -X POST https://huggingface.co/deepsite/api/mcp \
-  -H "Content-Type: application/json" \
-  -d '{"jsonrpc":"2.0","id":1,"method":"tools/list","params":{}}'
-```
-
----
-
-## Migration Guide: From Parameter to Header Auth
-
-If you're currently passing the token as a parameter in your prompts, here's how to migrate to the more secure header-based authentication:
-
-### Step 1: Update Your Config
-
-Edit your Claude Desktop config file and add the `headers` section:
-
-```json
-{
-  "mcpServers": {
-    "deepsite": {
-      "url": "https://huggingface.co/deepsite/api/mcp",
-      "transport": {
-        "type": "sse"
-      },
-      "headers": {
-        "Authorization": "Bearer hf_your_actual_token_here"
-      }
-    }
-  }
-}
-```
-
-### Step 2: Restart Claude Desktop
-
-Completely quit and restart Claude Desktop for the changes to take effect.
-
-### Step 3: Use Simpler Prompts
-
-Now you can simply say:
-> "Create a portfolio website with DeepSite"
-
-Instead of:
-> "Create a portfolio website with DeepSite using token `hf_xxxxx`"
-
-Your token is automatically included in all requests via the header!
-
----
-
-## Security Notes
-
-### HTTP Server (Option 1)
-- **✅ Recommended:** Store your HF token in the `Authorization` header in your Claude Desktop config
-- The token is stored locally on your machine and never exposed in chat
-- The token is sent with each request but only used to authenticate with Hugging Face API
-- DeepSite does not store your token
-- Use tokens with minimal required permissions (write access to spaces)
-- You can revoke tokens anytime at https://huggingface.co/settings/tokens
-- **⚠️ Fallback:** You can still pass the token as a parameter, but this is less secure as it appears in conversation history
-
-### Local Server (Option 2)
-- Use `HF_TOKEN` environment variable (same security as Option 1)
-- Or use `DEEPSITE_AUTH_COOKIE` if you prefer session-based auth
-- All authentication data stays on your local machine
-- Better for development and testing
-- No need for both HTTP Server and Local Server - choose one!
-
----
-
-## Troubleshooting
-
-### "Invalid Hugging Face token"
-- Verify your token at https://huggingface.co/settings/tokens
-- Ensure the token has write permissions
-- Check that you copied the full token (starts with `hf_`)
-
-### "At least one page is required"
-- Make sure you're providing the `pages` array
-- Each page must have both `path` and `html` properties
-
-### "Failed to create project"
-- Check your token permissions
-- Ensure the project title doesn't conflict with existing spaces
-- Verify your Hugging Face account is in good standing
-
-### Claude doesn't see the tool
-- Restart Claude Desktop after modifying the config
-- Check that the JSON config is valid (no trailing commas)
-- For HTTP: verify the URL is correct
-- For local: check the absolute path to index.js
-
----
-
-## Architecture
-
-### HTTP Server Flow
-```
-Claude Desktop
-      ↓
- (HTTP Request)
-      ↓
-huggingface.co/deepsite/api/mcp
-      ↓
-Hugging Face API (with user's token)
-      ↓
-New Space Created
-      ↓
-URLs returned to Claude
-```
-
-### Local Server Flow
-```
-Claude Desktop
-      ↓
- (stdio transport)
-      ↓
-Local MCP Server
-      ↓
- (HTTP to DeepSite API)
-      ↓
-huggingface.co/deepsite/api/me/projects
-      ↓
-New Space Created
-```
-
----
-
-## Contributing
-
-The MCP server implementation lives in:
-- HTTP Server: `/app/api/mcp/route.ts`
-- Local Server: `/mcp-server/index.ts`
-
-Both use the same core DeepSite logic for creating projects - no duplication!
-
----
-
-## License
-
-MIT
-
----
-
-## Resources
-
-- [Model Context Protocol Spec](https://modelcontextprotocol.io/)
-- [DeepSite Documentation](https://huggingface.co/deepsite)
-- [Hugging Face Spaces](https://huggingface.co/docs/hub/spaces)
-- [Claude Desktop](https://claude.ai/desktop)
-

README.md

@@ -1,5 +1,5 @@
 ---
-title: DeepSite v3
+title: DeepSite v2
 emoji: 🐳
 colorFrom: blue
 colorTo: blue
@@ -7,22 +7,16 @@ sdk: docker
 pinned: true
 app_port: 3000
 license: mit
-failure_strategy: rollback
-short_description: Generate any application by Vibe Coding
+short_description: Generate any application with DeepSeek
 models:
   - deepseek-ai/DeepSeek-V3-0324
   - deepseek-ai/DeepSeek-R1-0528
-  - deepseek-ai/DeepSeek-V3.1
-  - deepseek-ai/DeepSeek-V3.1-Terminus
-  - deepseek-ai/DeepSeek-V3.2-Exp
-  - Qwen/Qwen3-Coder-480B-A35B-Instruct
-  - moonshotai/Kimi-K2-Instruct
-  - moonshotai/Kimi-K2-Instruct-0905
-  - zai-org/GLM-4.6
-  - MiniMaxAI/MiniMax-M2
-  - moonshotai/Kimi-K2-Thinking
 ---
 
 # DeepSite 🐳
 
-DeepSite is a Vibe Coding Platform designed to make coding smarter and more efficient. Tailored for developers, data scientists, and AI engineers, it integrates generative AI into your coding projects to enhance creativity and productivity.
+DeepSite is a coding platform powered by DeepSeek AI, designed to make coding smarter and more efficient. Tailored for developers, data scientists, and AI engineers, it integrates generative AI into your coding projects to enhance creativity and productivity.
+
+## How to use it locally
+
+Follow [this discussion](https://huggingface.co/spaces/enzostvs/deepsite/discussions/74)

app/(public)/layout.tsx

@@ -6,7 +6,7 @@ export default async function PublicLayout({
   children: React.ReactNode;
 }>) {
   return (
-    <div className="h-screen bg-neutral-950 z-1 relative overflow-auto scroll-smooth">
+    <div className="min-h-screen bg-black z-1 relative">
       <div className="background__noisy" />
       <Navigation />
       {children}

app/(public)/page.tsx

@@ -1,5 +1,44 @@
-import { MyProjects } from "@/components/my-projects";
-
-export default async function HomePage() {
-  return <MyProjects />;
+import { AskAi } from "@/components/space/ask-ai";
+import { redirect } from "next/navigation";
+export default function Home() {
+  redirect("/projects/new");
+  return (
+    <>
+      <header className="container mx-auto pt-20 px-6 relative flex flex-col items-center justify-center text-center">
+        <div className="rounded-full border border-neutral-100/10 bg-neutral-100/5 text-xs text-neutral-300 px-3 py-1 max-w-max mx-auto mb-2">
+          ✨ DeepSite Public Beta
+        </div>
+        <h1 className="text-8xl font-semibold text-white font-mono max-w-4xl">
+          Code your website with AI in seconds
+        </h1>
+        <p className="text-2xl text-neutral-300/80 mt-4 text-center max-w-2xl">
+          Vibe Coding has never been so easy.
+        </p>
+        <div className="mt-14 max-w-2xl w-full mx-auto">
+          <AskAi />
+        </div>
+        <div className="absolute inset-0 pointer-events-none -z-[1]">
+          <div className="w-full h-full bg-gradient-to-r from-purple-500 to-pink-500 opacity-10 blur-3xl rounded-full" />
+          <div className="w-2/3 h-3/4 bg-gradient-to-r from-blue-500 to-teal-500 opacity-24 blur-3xl absolute -top-20 right-10 transform rotate-12" />
+          <div className="w-1/2 h-1/2 bg-gradient-to-r from-amber-500 to-rose-500 opacity-20 blur-3xl absolute bottom-0 left-10 rounded-3xl" />
+          <div className="w-48 h-48 bg-gradient-to-r from-cyan-500 to-indigo-500 opacity-20 blur-3xl absolute top-1/3 right-1/3 rounded-lg transform -rotate-15" />
+        </div>
+      </header>
+      <div id="community" className="h-screen flex items-center justify-center">
+        <h1 className="text-7xl font-extrabold text-white font-mono">
+          Community Driven
+        </h1>
+      </div>
+      <div id="deploy" className="h-screen flex items-center justify-center">
+        <h1 className="text-7xl font-extrabold text-white font-mono">
+          Deploy your website in seconds
+        </h1>
+      </div>
+      <div id="features" className="h-screen flex items-center justify-center">
+        <h1 className="text-7xl font-extrabold text-white font-mono">
+          Features that make you smile
+        </h1>
+      </div>
+    </>
+  );
 }

app/(public)/projects/page.tsx

@@ -0,0 +1,13 @@
+import { redirect } from "next/navigation";
+
+import { MyProjects } from "@/components/my-projects";
+import { getProjects } from "@/app/actions/projects";
+
+export default async function ProjectsPage() {
+  const { ok, projects } = await getProjects();
+  if (!ok) {
+    redirect("/");
+  }
+
+  return <MyProjects projects={projects} />;
+}

app/[namespace]/[repoId]/page.tsx

@@ -1,28 +0,0 @@
-import { AppEditor } from "@/components/editor";
-import { generateSEO } from "@/lib/seo";
-import { Metadata } from "next";
-
-export async function generateMetadata({
-  params,
-}: {
-  params: Promise<{ namespace: string; repoId: string }>;
-}): Promise<Metadata> {
-  const { namespace, repoId } = await params;
-
-  return generateSEO({
-    title: `${namespace}/${repoId} - DeepSite Editor`,
-    description: `Edit and build ${namespace}/${repoId} with AI-powered tools on DeepSite. Create stunning websites with no code required.`,
-    path: `/${namespace}/${repoId}`,
-    // Prevent indexing of individual project editor pages if they contain sensitive content
-    noIndex: false, // Set to true if you want to keep project pages private
-  });
-}
-
-export default async function ProjectNamespacePage({
-  params,
-}: {
-  params: Promise<{ namespace: string; repoId: string }>;
-}) {
-  const { namespace, repoId } = await params;
-  return <AppEditor namespace={namespace} repoId={repoId} />;
-}

app/actions/auth.ts

@@ -11,7 +11,7 @@ export async function getAuth() {
   const redirect_uri =
     `${host.includes("localhost") ? "http://" : "https://"}` +
     url +
-    "/deepsite/auth/callback";
+    "/auth/callback";
 
   const loginRedirectUrl = `https://huggingface.co/oauth/authorize?client_id=${process.env.OAUTH_CLIENT_ID}&redirect_uri=${redirect_uri}&response_type=code&scope=openid%20profile%20write-repos%20manage-repos%20inference-api&prompt=consent&state=1234567890`;
   return loginRedirectUrl;

app/actions/projects.ts

@@ -2,13 +2,13 @@
 
 import { isAuthenticated } from "@/lib/auth";
 import { NextResponse } from "next/server";
-import { listSpaces } from "@huggingface/hub";
-import { ProjectType } from "@/types";
+import dbConnect from "@/lib/mongodb";
+import Project from "@/models/Project";
+import { Project as ProjectType } from "@/types";
 
 export async function getProjects(): Promise<{
   ok: boolean;
   projects: ProjectType[];
-  isEmpty?: boolean;
 }> {
   const user = await isAuthenticated();
 
@@ -19,29 +19,45 @@ export async function getProjects(): Promise<{
     };
   }
 
-  const projects = [];
-  for await (const space of listSpaces({
-    accessToken: user.token as string,
-    additionalFields: ["author", "cardData"],
-    search: {
-      owner: user.name,
-    }
-  })) {
-    if (
-      !space.private &&
-      space.sdk === "static" &&
-      Array.isArray((space.cardData as { tags?: string[] })?.tags) &&
-      (
-        ((space.cardData as { tags?: string[] })?.tags?.includes("deepsite-v3")) ||
-        ((space.cardData as { tags?: string[] })?.tags?.includes("deepsite"))
-      )
-    ) {
-      projects.push(space);
-    }
+  await dbConnect();
+  const projects = await Project.find({
+    user_id: user?.id,
+  })
+    .sort({ _createdAt: -1 })
+    .limit(100)
+    .lean();
+  if (!projects) {
+    return {
+      ok: false,
+      projects: [],
+    };
   }
-
   return {
     ok: true,
-    projects,
+    projects: JSON.parse(JSON.stringify(projects)) as ProjectType[],
   };
 }
+
+export async function getProject(
+  namespace: string,
+  repoId: string
+): Promise<ProjectType | null> {
+  const user = await isAuthenticated();
+
+  if (user instanceof NextResponse || !user) {
+    return null;
+  }
+
+  await dbConnect();
+  const project = await Project.findOne({
+    user_id: user.id,
+    namespace,
+    repoId,
+  }).lean();
+
+  if (!project) {
+    return null;
+  }
+
+  return JSON.parse(JSON.stringify(project)) as ProjectType;
+}

app/api/{ask → ask-ai}/route.ts

@@ -4,29 +4,25 @@ import { NextResponse } from "next/server";
 import { headers } from "next/headers";
 import { InferenceClient } from "@huggingface/inference";
 
-import { MODELS } from "@/lib/providers";
+import { MODELS, PROVIDERS } from "@/lib/providers";
 import {
+  DIVIDER,
   FOLLOW_UP_SYSTEM_PROMPT,
-  FOLLOW_UP_SYSTEM_PROMPT_LIGHT,
   INITIAL_SYSTEM_PROMPT,
-  INITIAL_SYSTEM_PROMPT_LIGHT,
   MAX_REQUESTS_PER_IP,
-  PROMPT_FOR_PROJECT_NAME,
+  REPLACE_END,
+  SEARCH_START,
 } from "@/lib/prompts";
 import MY_TOKEN_KEY from "@/lib/get-cookie-name";
-import { Page } from "@/types";
-import { isAuthenticated } from "@/lib/auth";
-import { getBestProvider } from "@/lib/best-provider";
 
 const ipAddresses = new Map();
 
 export async function POST(request: NextRequest) {
   const authHeaders = await headers();
-  const tokenInHeaders = authHeaders.get("Authorization");
-  const userToken = tokenInHeaders ? tokenInHeaders.replace("Bearer ", "") : request.cookies.get(MY_TOKEN_KEY())?.value;
+  const userToken = request.cookies.get(MY_TOKEN_KEY())?.value;
 
   const body = await request.json();
-  const { prompt, provider, model, redesignMarkdown, enhancedSettings, pages } = body;
+  const { prompt, provider, model, redesignMarkdown, html } = body;
 
   if (!model || (!prompt && !redesignMarkdown)) {
     return NextResponse.json(
@@ -38,7 +34,6 @@ export async function POST(request: NextRequest) {
   const selectedModel = MODELS.find(
     (m) => m.value === model || m.label === model
   );
-
   if (!selectedModel) {
     return NextResponse.json(
       { ok: false, error: "Invalid model selected" },
@@ -46,8 +41,18 @@ export async function POST(request: NextRequest) {
     );
   }
 
-  let token: string | null = null;
-  if (userToken) token = userToken;
+  if (!selectedModel.providers.includes(provider) && provider !== "auto") {
+    return NextResponse.json(
+      {
+        ok: false,
+        error: `The selected model does not support the ${provider} provider.`,
+        openSelectProvider: true,
+      },
+      { status: 400 }
+    );
+  }
+
+  let token = userToken;
   let billTo: string | null = null;
 
   /**
@@ -63,7 +68,7 @@ export async function POST(request: NextRequest) {
     ? authHeaders.get("x-forwarded-for")?.split(",")[1].trim()
     : authHeaders.get("x-forwarded-for");
 
-  if (!token || token === "null" || token === "") {
+  if (!token) {
     ipAddresses.set(ip, (ipAddresses.get(ip) || 0) + 1);
     if (ipAddresses.get(ip) > MAX_REQUESTS_PER_IP) {
       return NextResponse.json(
@@ -80,11 +85,19 @@ export async function POST(request: NextRequest) {
     billTo = "huggingface";
   }
 
+  const DEFAULT_PROVIDER = PROVIDERS.novita;
+  const selectedProvider =
+    provider === "auto"
+      ? PROVIDERS[selectedModel.autoProvider as keyof typeof PROVIDERS]
+      : PROVIDERS[provider as keyof typeof PROVIDERS] ?? DEFAULT_PROVIDER;
+
   try {
+    // Create a stream response
     const encoder = new TextEncoder();
     const stream = new TransformStream();
     const writer = stream.writable.getWriter();
 
+    // Start the response
     const response = new NextResponse(stream.readable, {
       headers: {
         "Content-Type": "text/plain; charset=utf-8",
@@ -94,54 +107,75 @@ export async function POST(request: NextRequest) {
     });
 
     (async () => {
+      let completeResponse = "";
       try {
         const client = new InferenceClient(token);
-        
-        const systemPrompt = selectedModel.value.includes('MiniMax') 
-          ? INITIAL_SYSTEM_PROMPT_LIGHT 
-          : INITIAL_SYSTEM_PROMPT;
-        
-        const userPrompt = prompt;
-                
         const chatCompletion = client.chatCompletionStream(
           {
-            model: selectedModel.value + (provider !== "auto" ? `:${provider}` : ""),
+            model: selectedModel.value,
+            provider: selectedProvider.id as any,
             messages: [
               {
                 role: "system",
-                content: systemPrompt,
+                content: INITIAL_SYSTEM_PROMPT,
               },
-              ...(redesignMarkdown ? [{
-                role: "assistant",
-                content: `User will ask you to redesign the site based on this markdown. Use the same images as the site, but you can improve the content and the design. Here is the markdown: ${redesignMarkdown}`
-              }] : []),
               {
                 role: "user",
-                content: userPrompt + (enhancedSettings.isActive ? `1. I want to use the following primary color: ${enhancedSettings.primaryColor} (eg: bg-${enhancedSettings.primaryColor}-500).
-2. I want to use the following secondary color: ${enhancedSettings.secondaryColor} (eg: bg-${enhancedSettings.secondaryColor}-500).
-3. I want to use the following theme: ${enhancedSettings.theme} mode.` : "")
+                content: redesignMarkdown
+                  ? `Here is my current design as a markdown:\n\n${redesignMarkdown}\n\nNow, please create a new design based on this markdown.`
+                  : html
+                  ? `Here is my current HTML code:\n\n\`\`\`html\n${html}\n\`\`\`\n\nNow, please create a new design based on this HTML.`
+                  : prompt,
               },
             ],
-            ...(selectedModel.top_k ? { top_k: selectedModel.top_k } : {}),
-            ...(selectedModel.temperature ? { temperature: selectedModel.temperature } : {}),
-            ...(selectedModel.top_p ? { top_p: selectedModel.top_p } : {}),
+            max_tokens: selectedProvider.max_tokens,
           },
           billTo ? { billTo } : {}
         );
 
         while (true) {
-          const { done, value } = await chatCompletion.next()
+          const { done, value } = await chatCompletion.next();
           if (done) {
             break;
           }
 
           const chunk = value.choices[0]?.delta?.content;
           if (chunk) {
-            await writer.write(encoder.encode(chunk));
+            let newChunk = chunk;
+            if (!selectedModel?.isThinker) {
+              if (provider !== "sambanova") {
+                await writer.write(encoder.encode(chunk));
+                completeResponse += chunk;
+
+                if (completeResponse.includes("</html>")) {
+                  break;
+                }
+              } else {
+                if (chunk.includes("</html>")) {
+                  newChunk = newChunk.replace(/<\/html>[\s\S]*/, "</html>");
+                }
+                completeResponse += newChunk;
+                await writer.write(encoder.encode(newChunk));
+                if (newChunk.includes("</html>")) {
+                  break;
+                }
+              }
+            } else {
+              const lastThinkTagIndex =
+                completeResponse.lastIndexOf("</think>");
+              completeResponse += newChunk;
+              await writer.write(encoder.encode(newChunk));
+              if (lastThinkTagIndex !== -1) {
+                const afterLastThinkTag = completeResponse.slice(
+                  lastThinkTagIndex + "</think>".length
+                );
+                if (afterLastThinkTag.includes("</html>")) {
+                  break;
+                }
+              }
+            }
           }
         }
-        
-        await writer.close();
       } catch (error: any) {
         if (error.message?.includes("exceeded your monthly included credits")) {
           await writer.write(
@@ -153,18 +187,7 @@ export async function POST(request: NextRequest) {
               })
             )
           );
-        } else if (error?.message?.includes("inference provider information")) {
-          await writer.write(
-            encoder.encode(
-              JSON.stringify({
-                ok: false,
-                openSelectProvider: true,
-                message: error.message,
-              })
-            )
-          );
-        }
-        else {
+        } else {
           await writer.write(
             encoder.encode(
               JSON.stringify({
@@ -177,10 +200,7 @@ export async function POST(request: NextRequest) {
           );
         }
       } finally {
-        try {
-          await writer?.close();
-        } catch {
-        }
+        await writer?.close();
       }
     })();
 
@@ -199,18 +219,14 @@ export async function POST(request: NextRequest) {
 }
 
 export async function PUT(request: NextRequest) {
-  const user = await isAuthenticated();
-  if (user instanceof NextResponse || !user) {
-    return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
-  }
-
   const authHeaders = await headers();
+  const userToken = request.cookies.get(MY_TOKEN_KEY())?.value;
 
   const body = await request.json();
-  const { prompt, provider, selectedElementHtml, model, pages, files, repoId, isNew } =
+  const { prompt, html, previousPrompt, provider, selectedElementHtml, model } =
     body;
 
-  if (!prompt || pages.length === 0) {
+  if (!prompt || !html) {
     return NextResponse.json(
       { ok: false, error: "Missing required fields" },
       { status: 400 }
@@ -227,7 +243,7 @@ export async function PUT(request: NextRequest) {
     );
   }
 
-  let token = user.token as string;
+  let token = userToken;
   let billTo: string | null = null;
 
   /**
@@ -243,7 +259,7 @@ export async function PUT(request: NextRequest) {
     ? authHeaders.get("x-forwarded-for")?.split(",")[1].trim()
     : authHeaders.get("x-forwarded-for");
 
-  if (!token || token === "null" || token === "") {
+  if (!token) {
     ipAddresses.set(ip, (ipAddresses.get(ip) || 0) + 1);
     if (ipAddresses.get(ip) > MAX_REQUESTS_PER_IP) {
       return NextResponse.json(
@@ -260,131 +276,136 @@ export async function PUT(request: NextRequest) {
     billTo = "huggingface";
   }
 
+  const client = new InferenceClient(token);
+
+  const DEFAULT_PROVIDER = PROVIDERS.novita;
+  const selectedProvider =
+    provider === "auto"
+      ? PROVIDERS[selectedModel.autoProvider as keyof typeof PROVIDERS]
+      : PROVIDERS[provider as keyof typeof PROVIDERS] ?? DEFAULT_PROVIDER;
+
   try {
-    const encoder = new TextEncoder();
-    const stream = new TransformStream();
-    const writer = stream.writable.getWriter();
+    const response = await client.chatCompletion(
+      {
+        model: selectedModel.value,
+        provider: selectedProvider.id as any,
+        messages: [
+          {
+            role: "system",
+            content: FOLLOW_UP_SYSTEM_PROMPT,
+          },
+          {
+            role: "user",
+            content: previousPrompt
+              ? previousPrompt
+              : "You are modifying the HTML file based on the user's request.",
+          },
+          {
+            role: "assistant",
 
-    const response = new NextResponse(stream.readable, {
-      headers: {
-        "Content-Type": "text/plain; charset=utf-8",
-        "Cache-Control": "no-cache",
-        Connection: "keep-alive",
+            content: `The current code is: \n\`\`\`html\n${html}\n\`\`\` ${
+              selectedElementHtml
+                ? `\n\nYou have to update ONLY the following element, NOTHING ELSE: \n\n\`\`\`html\n${selectedElementHtml}\n\`\`\``
+                : ""
+            }`,
+          },
+          {
+            role: "user",
+            content: prompt,
+          },
+        ],
+        ...(selectedProvider.id !== "sambanova"
+          ? {
+              max_tokens: selectedProvider.max_tokens,
+            }
+          : {}),
       },
-    });
+      billTo ? { billTo } : {}
+    );
 
-    (async () => {
-      try {
-        const client = new InferenceClient(token);
+    const chunk = response.choices[0]?.message?.content;
+    if (!chunk) {
+      return NextResponse.json(
+        { ok: false, message: "No content returned from the model" },
+        { status: 400 }
+      );
+    }
 
-        const basePrompt = selectedModel.value.includes('MiniMax') 
-          ? FOLLOW_UP_SYSTEM_PROMPT_LIGHT 
-          : FOLLOW_UP_SYSTEM_PROMPT;
-        const systemPrompt = basePrompt + (isNew ? PROMPT_FOR_PROJECT_NAME : "");
-        const userContext = "You are modifying the HTML file based on the user's request.";
+    if (chunk) {
+      const updatedLines: number[][] = [];
+      let newHtml = html;
+      let position = 0;
+      let moreBlocks = true;
+
+      while (moreBlocks) {
+        const searchStartIndex = chunk.indexOf(SEARCH_START, position);
+        if (searchStartIndex === -1) {
+          moreBlocks = false;
+          continue;
+        }
 
-        const allPages = pages || [];
-        const pagesContext = allPages
-          .map((p: Page) => `- ${p.path}\n${p.html}`)
-          .join("\n\n");
+        const dividerIndex = chunk.indexOf(DIVIDER, searchStartIndex);
+        if (dividerIndex === -1) {
+          moreBlocks = false;
+          continue;
+        }
 
-        const assistantContext = `${selectedElementHtml
-            ? `\n\nYou have to update ONLY the following element, NOTHING ELSE: \n\n\`\`\`html\n${selectedElementHtml}\n\`\`\` Could be in multiple pages, if so, update all the pages.`
-            : ""
-          }. Current pages (${allPages.length} total): ${pagesContext}. ${files?.length > 0 ? `Available images: ${files?.map((f: string) => f).join(', ')}.` : ""}`;
+        const replaceEndIndex = chunk.indexOf(REPLACE_END, dividerIndex);
+        if (replaceEndIndex === -1) {
+          moreBlocks = false;
+          continue;
+        }
 
-        const chatCompletion = client.chatCompletionStream(
-          {
-            model: selectedModel.value + (provider !== "auto" ? `:${provider}` : ""),
-            messages: [
-              {
-                role: "system",
-                content: systemPrompt,
-              },
-              {
-                role: "user",
-                content: userContext,
-              },
-              {
-                role: "assistant",
-                content: assistantContext,
-              },
-              {
-                role: "user",
-                content: prompt,
-              },
-            ],
-            ...(selectedModel.top_k ? { top_k: selectedModel.top_k } : {}),
-            ...(selectedModel.temperature ? { temperature: selectedModel.temperature } : {}),
-            ...(selectedModel.top_p ? { top_p: selectedModel.top_p } : {}),
-          },
-          billTo ? { billTo } : {}
+        const searchBlock = chunk.substring(
+          searchStartIndex + SEARCH_START.length,
+          dividerIndex
+        );
+        const replaceBlock = chunk.substring(
+          dividerIndex + DIVIDER.length,
+          replaceEndIndex
         );
 
-        // Stream the response chunks to the client
-        while (true) {
-          const { done, value } = await chatCompletion.next();
-          if (done) {
-            break;
-          }
-
-          const chunk = value.choices[0]?.delta?.content;
-          if (chunk) {
-            await writer.write(encoder.encode(chunk));
+        if (searchBlock.trim() === "") {
+          newHtml = `${replaceBlock}\n${newHtml}`;
+          updatedLines.push([1, replaceBlock.split("\n").length]);
+        } else {
+          const blockPosition = newHtml.indexOf(searchBlock);
+          if (blockPosition !== -1) {
+            const beforeText = newHtml.substring(0, blockPosition);
+            const startLineNumber = beforeText.split("\n").length;
+            const replaceLines = replaceBlock.split("\n").length;
+            const endLineNumber = startLineNumber + replaceLines - 1;
+
+            updatedLines.push([startLineNumber, endLineNumber]);
+            newHtml = newHtml.replace(searchBlock, replaceBlock);
           }
         }
 
-        await writer.write(encoder.encode(`\n___METADATA_START___\n${JSON.stringify({
-          repoId,
-          isNew,
-          userName: user.name,
-        })}\n___METADATA_END___\n`));
-
-        await writer.close();
-      } catch (error: any) {
-        if (error.message?.includes("exceeded your monthly included credits")) {
-          await writer.write(
-            encoder.encode(
-              JSON.stringify({
-                ok: false,
-                openProModal: true,
-                message: error.message,
-              })
-            )
-          );
-        } else if (error?.message?.includes("inference provider information")) {
-          await writer.write(
-            encoder.encode(
-              JSON.stringify({
-                ok: false,
-                openSelectProvider: true,
-                message: error.message,
-              })
-            )
-          );
-        } else {
-          await writer.write(
-            encoder.encode(
-              JSON.stringify({
-                ok: false,
-                message:
-                  error.message ||
-                  "An error occurred while processing your request.",
-              })
-            )
-          );
-        }
-      } finally {
-        try {
-          await writer?.close();
-        } catch {
-          // ignore
-        }
+        position = replaceEndIndex + REPLACE_END.length;
       }
-    })();
 
-    return response;
+      return NextResponse.json({
+        ok: true,
+        html: newHtml,
+        updatedLines,
+      });
+    } else {
+      return NextResponse.json(
+        { ok: false, message: "No content returned from the model" },
+        { status: 400 }
+      );
+    }
   } catch (error: any) {
+    if (error.message?.includes("exceeded your monthly included credits")) {
+      return NextResponse.json(
+        {
+          ok: false,
+          openProModal: true,
+          message: error.message,
+        },
+        { status: 402 }
+      );
+    }
     return NextResponse.json(
       {
         ok: false,
@@ -396,4 +417,3 @@ export async function PUT(request: NextRequest) {
     );
   }
 }
-

app/api/auth/login-url/route.ts

@@ -1,21 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-
-export async function GET(req: NextRequest) {
-  const host = req.headers.get("host") ?? "localhost:3000";
-  
-  let url: string;
-  if (host.includes("localhost")) {
-    url = host;
-  } else {
-    url = "huggingface.co";
-  }
-  
-  const redirect_uri =
-    `${host.includes("localhost") ? "http://" : "https://"}` +
-    url +
-    "/deepsite/auth/callback";
-
-  const loginRedirectUrl = `https://huggingface.co/oauth/authorize?client_id=${process.env.OAUTH_CLIENT_ID}&redirect_uri=${redirect_uri}&response_type=code&scope=openid%20profile%20write-repos%20manage-repos%20inference-api&prompt=consent&state=1234567890`;
-  
-  return NextResponse.json({ loginUrl: loginRedirectUrl });
-}

app/api/auth/logout/route.ts

@@ -1,25 +0,0 @@
-import { NextResponse } from "next/server";
-import MY_TOKEN_KEY from "@/lib/get-cookie-name";
-
-export async function POST() {
-  const cookieName = MY_TOKEN_KEY();
-  const isProduction = process.env.NODE_ENV === "production";
-  
-  const response = NextResponse.json(
-    { message: "Logged out successfully" },
-    { status: 200 }
-  );
-  
-  // Clear the HTTP-only cookie
-  const cookieOptions = [
-    `${cookieName}=`,
-    "Max-Age=0",
-    "Path=/",
-    "HttpOnly",
-    ...(isProduction ? ["Secure", "SameSite=None"] : ["SameSite=Lax"])
-  ].join("; ");
-  
-  response.headers.set("Set-Cookie", cookieOptions);
-  
-  return response;
-}

app/api/auth/route.ts

@@ -24,12 +24,12 @@ export async function POST(req: NextRequest) {
     req.headers.get("host") ?? req.headers.get("origin") ?? "localhost:3000";
 
   const url = host.includes("/spaces/enzostvs")
-    ? "huggingface.co/deepsite"
+    ? "enzostvs-deepsite.hf.space"
     : host;
   const redirect_uri =
     `${host.includes("localhost") ? "http://" : "https://"}` +
     url +
-    "/deepsite/auth/callback";
+    "/auth/callback";
   const request_auth = await fetch("https://huggingface.co/oauth/token", {
     method: "POST",
     headers: {

app/api/mcp/route.ts

@@ -1,435 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-import { RepoDesignation, createRepo, uploadFiles, spaceInfo, listCommits } from "@huggingface/hub";
-import { COLORS } from "@/lib/utils";
-import { injectDeepSiteBadge, isIndexPage } from "@/lib/inject-badge";
-import { Commit, Page } from "@/types";
-
-// Timeout configuration (in milliseconds)
-const OPERATION_TIMEOUT = 120000; // 2 minutes for HF operations
-
-// Extend the maximum execution time for this route
-export const maxDuration = 180; // 3 minutes
-
-// Utility function to wrap promises with timeout
-async function withTimeout<T>(
-  promise: Promise<T>,
-  timeoutMs: number,
-  errorMessage: string = "Operation timed out"
-): Promise<T> {
-  let timeoutId: NodeJS.Timeout;
-  
-  const timeoutPromise = new Promise<never>((_, reject) => {
-    timeoutId = setTimeout(() => {
-      reject(new Error(errorMessage));
-    }, timeoutMs);
-  });
-
-  try {
-    const result = await Promise.race([promise, timeoutPromise]);
-    clearTimeout(timeoutId!);
-    return result;
-  } catch (error) {
-    clearTimeout(timeoutId!);
-    throw error;
-  }
-}
-
-interface MCPRequest {
-  jsonrpc: "2.0";
-  id: number | string;
-  method: string;
-  params?: any;
-}
-
-interface MCPResponse {
-  jsonrpc: "2.0";
-  id: number | string;
-  result?: any;
-  error?: {
-    code: number;
-    message: string;
-    data?: any;
-  };
-}
-
-interface CreateProjectParams {
-  title?: string;
-  pages: Page[];
-  prompt?: string;
-  hf_token?: string; // Optional - can come from header instead
-}
-
-// MCP Server over HTTP
-export async function POST(req: NextRequest) {
-  try {
-    const body: MCPRequest = await req.json();
-    const { jsonrpc, id, method, params } = body;
-
-    // Validate JSON-RPC 2.0 format
-    if (jsonrpc !== "2.0") {
-      return NextResponse.json({
-        jsonrpc: "2.0",
-        id: id || null,
-        error: {
-          code: -32600,
-          message: "Invalid Request: jsonrpc must be '2.0'",
-        },
-      });
-    }
-
-    let response: MCPResponse;
-
-    switch (method) {
-      case "initialize":
-        response = {
-          jsonrpc: "2.0",
-          id,
-          result: {
-            protocolVersion: "2024-11-05",
-            capabilities: {
-              tools: {},
-            },
-            serverInfo: {
-              name: "deepsite-mcp-server",
-              version: "1.0.0",
-            },
-          },
-        };
-        break;
-
-      case "tools/list":
-        response = {
-          jsonrpc: "2.0",
-          id,
-          result: {
-            tools: [
-              {
-                name: "create_project",
-                description: `Create a new DeepSite project. This will create a new Hugging Face Space with your HTML/CSS/JS files.
-
-Example usage:
-- Create a simple website with HTML, CSS, and JavaScript files
-- Each page needs a 'path' (filename like "index.html", "styles.css", "script.js") and 'html' (the actual content)
-- The title will be formatted to a valid repository name
-- Returns the project URL and metadata`,
-                inputSchema: {
-                  type: "object",
-                  properties: {
-                    title: {
-                      type: "string",
-                      description: "Project title (optional, defaults to 'DeepSite Project'). Will be formatted to a valid repo name.",
-                    },
-                    pages: {
-                      type: "array",
-                      description: "Array of files to include in the project",
-                      items: {
-                        type: "object",
-                        properties: {
-                          path: {
-                            type: "string",
-                            description: "File path (e.g., 'index.html', 'styles.css', 'script.js')",
-                          },
-                          html: {
-                            type: "string",
-                            description: "File content",
-                          },
-                        },
-                        required: ["path", "html"],
-                      },
-                    },
-                    prompt: {
-                      type: "string",
-                      description: "Optional prompt/description for the commit message",
-                    },
-                    hf_token: {
-                      type: "string",
-                      description: "Hugging Face API token (optional if provided via Authorization header)",
-                    },
-                  },
-                  required: ["pages"],
-                },
-              },
-            ],
-          },
-        };
-        break;
-
-      case "tools/call":
-        const { name, arguments: toolArgs } = params;
-
-        if (name === "create_project") {
-          try {
-            // Extract token from Authorization header if present
-            const authHeader = req.headers.get("authorization");
-            let hf_token = toolArgs.hf_token;
-            
-            if (authHeader && authHeader.startsWith("Bearer ")) {
-              hf_token = authHeader.substring(7); // Remove "Bearer " prefix
-            }
-            
-            const result = await handleCreateProject({
-              ...toolArgs,
-              hf_token,
-            } as CreateProjectParams);
-            response = {
-              jsonrpc: "2.0",
-              id,
-              result,
-            };
-          } catch (error: any) {
-            response = {
-              jsonrpc: "2.0",
-              id,
-              error: {
-                code: -32000,
-                message: error.message || "Failed to create project",
-                data: error.data,
-              },
-            };
-          }
-        } else {
-          response = {
-            jsonrpc: "2.0",
-            id,
-            error: {
-              code: -32601,
-              message: `Unknown tool: ${name}`,
-            },
-          };
-        }
-        break;
-
-      default:
-        response = {
-          jsonrpc: "2.0",
-          id,
-          error: {
-            code: -32601,
-            message: `Method not found: ${method}`,
-          },
-        };
-    }
-
-    return NextResponse.json(response);
-  } catch (error: any) {
-    return NextResponse.json({
-      jsonrpc: "2.0",
-      id: null,
-      error: {
-        code: -32700,
-        message: "Parse error",
-        data: error.message,
-      },
-    });
-  }
-}
-
-// Handle OPTIONS for CORS
-export async function OPTIONS() {
-  return new NextResponse(null, {
-    status: 200,
-    headers: {
-      "Access-Control-Allow-Origin": "*",
-      "Access-Control-Allow-Methods": "POST, OPTIONS",
-      "Access-Control-Allow-Headers": "Content-Type",
-    },
-  });
-}
-
-async function handleCreateProject(params: CreateProjectParams) {
-  const { title: titleFromRequest, pages, prompt, hf_token } = params;
-
-  // Validate required parameters
-  if (!hf_token || typeof hf_token !== "string") {
-    throw new Error("hf_token is required and must be a string");
-  }
-
-  if (!pages || !Array.isArray(pages) || pages.length === 0) {
-    throw new Error("At least one page is required");
-  }
-
-  // Validate that each page has required fields
-  for (const page of pages) {
-    if (!page.path || !page.html) {
-      throw new Error("Each page must have 'path' and 'html' properties");
-    }
-  }
-
-  // Get user info from HF token
-  let username: string;
-  try {
-    const userResponse = await withTimeout(
-      fetch("https://huggingface.co/api/whoami-v2", {
-        headers: {
-          Authorization: `Bearer ${hf_token}`,
-        },
-      }),
-      30000, // 30 seconds for authentication
-      "Authentication timeout: Unable to verify Hugging Face token"
-    );
-
-    if (!userResponse.ok) {
-      throw new Error("Invalid Hugging Face token");
-    }
-
-    const userData = await userResponse.json();
-    username = userData.name;
-  } catch (error: any) {
-    if (error.message?.includes('timeout')) {
-      throw new Error(`Authentication timeout: ${error.message}`);
-    }
-    throw new Error(`Authentication failed: ${error.message}`);
-  }
-
-  const title = titleFromRequest ?? "DeepSite Project";
-
-  const formattedTitle = title
-    .toLowerCase()
-    .replace(/[^a-z0-9]+/g, "-")
-    .split("-")
-    .filter(Boolean)
-    .join("-")
-    .slice(0, 96);
-
-  const repo: RepoDesignation = {
-    type: "space",
-    name: `${username}/${formattedTitle}`,
-  };
-
-  const colorFrom = COLORS[Math.floor(Math.random() * COLORS.length)];
-  const colorTo = COLORS[Math.floor(Math.random() * COLORS.length)];
-  const README = `---
-title: ${title}
-colorFrom: ${colorFrom}
-colorTo: ${colorTo}
-emoji: 🐳
-sdk: static
-pinned: false
-tags:
-  - deepsite-v3
----
-
-# Welcome to your new DeepSite project!
-This project was created with [DeepSite](https://huggingface.co/deepsite).
-`;
-
-  const files: File[] = [];
-  const readmeFile = new File([README], "README.md", { type: "text/markdown" });
-  files.push(readmeFile);
-
-  pages.forEach((page: Page) => {
-    // Determine MIME type based on file extension
-    let mimeType = "text/html";
-    if (page.path.endsWith(".css")) {
-      mimeType = "text/css";
-    } else if (page.path.endsWith(".js")) {
-      mimeType = "text/javascript";
-    } else if (page.path.endsWith(".json")) {
-      mimeType = "application/json";
-    }
-
-    // Inject the DeepSite badge script into index pages only
-    const content = mimeType === "text/html" && isIndexPage(page.path) 
-      ? injectDeepSiteBadge(page.html) 
-      : page.html;
-    const file = new File([content], page.path, { type: mimeType });
-    files.push(file);
-  });
-
-  try {
-    const { repoUrl } = await withTimeout(
-      createRepo({
-        repo,
-        accessToken: hf_token,
-      }),
-      60000, // 1 minute for repo creation
-      "Timeout creating repository. Please try again."
-    );
-
-    const commitTitle = !prompt || prompt.trim() === "" ? "Initial project creation via MCP" : prompt;
-    
-    await withTimeout(
-      uploadFiles({
-        repo,
-        files,
-        accessToken: hf_token,
-        commitTitle,
-      }),
-      OPERATION_TIMEOUT,
-      "Timeout uploading files. The repository was created but files may not have been uploaded."
-    );
-
-    const path = repoUrl.split("/").slice(-2).join("/");
-
-    const commits: Commit[] = [];
-    const commitIterator = listCommits({ repo, accessToken: hf_token });
-    
-    // Wrap the commit listing with a timeout
-    const commitTimeout = new Promise<void>((_, reject) => {
-      setTimeout(() => reject(new Error("Timeout listing commits")), 30000);
-    });
-    
-    try {
-      await Promise.race([
-        (async () => {
-          for await (const commit of commitIterator) {
-            if (commit.title.includes("initial commit") || commit.title.includes("image(s)") || commit.title.includes("Promote version")) {
-              continue;
-            }
-            commits.push({
-              title: commit.title,
-              oid: commit.oid,
-              date: commit.date,
-            });
-          }
-        })(),
-        commitTimeout
-      ]);
-    } catch (error: any) {
-      // If listing commits times out, continue with empty commits array
-      console.error("Failed to list commits:", error.message);
-    }
-
-    const space = await withTimeout(
-      spaceInfo({
-        name: repo.name,
-        accessToken: hf_token,
-      }),
-      30000, // 30 seconds for space info
-      "Timeout fetching space information"
-    );
-
-    const projectUrl = `https://huggingface.co/deepsite/${path}`;
-    const spaceUrl = `https://huggingface.co/spaces/${path}`;
-    const liveUrl = `https://${username}-${formattedTitle}.hf.space`;
-
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(
-            {
-              success: true,
-              message: "Project created successfully!",
-              projectUrl,
-              spaceUrl,
-              liveUrl,
-              spaceId: space.name,
-              projectId: space.id,
-              files: pages.map((p) => p.path),
-              updatedAt: space.updatedAt,
-            },
-            null,
-            2
-          ),
-        },
-      ],
-    };
-  } catch (err: any) {
-    if (err.message?.includes('timeout') || err.message?.includes('Timeout')) {
-      throw new Error(err.message || "Operation timed out. Please try again.");
-    }
-    throw new Error(err.message || "Failed to create project");
-  }
-}
-

app/api/me/projects/[namespace]/[repoId]/commits/[commitId]/promote/route.ts

@@ -1,230 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-import { RepoDesignation, listFiles, spaceInfo, uploadFiles, deleteFiles, downloadFile } from "@huggingface/hub";
-
-import { isAuthenticated } from "@/lib/auth";
-import { Page } from "@/types";
-
-export async function POST(
-  req: NextRequest,
-  { params }: { 
-    params: Promise<{ 
-      namespace: string; 
-      repoId: string; 
-      commitId: string; 
-    }> 
-  }
-) {
-  const user = await isAuthenticated();
-
-  if (user instanceof NextResponse || !user) {
-    return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
-  }
-
-  const param = await params;
-  const { namespace, repoId, commitId } = param;
-
-  try {
-    const repo: RepoDesignation = {
-      type: "space",
-      name: `${namespace}/${repoId}`,
-    };
-
-    const space = await spaceInfo({
-      name: `${namespace}/${repoId}`,
-      accessToken: user.token as string,
-      additionalFields: ["author"],
-    });
-
-    if (!space || space.sdk !== "static") {
-      return NextResponse.json(
-        { ok: false, error: "Space is not a static space." },
-        { status: 404 }
-      );
-    }
-    
-    if (space.author !== user.name) {
-      return NextResponse.json(
-        { ok: false, error: "Space does not belong to the authenticated user." },
-        { status: 403 }
-      );
-    }
-
-    const files: File[] = [];
-    const pages: Page[] = [];
-    const mediaFiles: string[] = [];
-    const allowedExtensions = ["html", "md", "css", "js", "json", "txt"];
-    const allowedFilesExtensions = ["jpg", "jpeg", "png", "gif", "svg", "webp", "avif", "heic", "heif", "ico", "bmp", "tiff", "tif", "mp4", "webm", "ogg", "avi", "mov", "mp3", "wav", "ogg", "aac", "m4a"];
-    const commitFilePaths: Set<string> = new Set();
-    
-    for await (const fileInfo of listFiles({
-      repo,
-      accessToken: user.token as string,
-      revision: commitId,
-    })) {
-      const fileExtension = fileInfo.path.split('.').pop()?.toLowerCase();
-      
-      if (fileInfo.path.endsWith(".html") || fileInfo.path.endsWith(".css") || fileInfo.path.endsWith(".js") || fileInfo.path.endsWith(".json")) {
-        commitFilePaths.add(fileInfo.path);
-        
-        const blob = await downloadFile({ 
-          repo, 
-          accessToken: user.token as string, 
-          path: fileInfo.path, 
-          revision: commitId,
-          raw: true 
-        });
-        const content = await blob?.text();
-        
-        if (content) {
-          let mimeType = "text/plain";
-          
-          switch (fileExtension) {
-            case "html":
-              mimeType = "text/html";
-              break;
-            case "css":
-              mimeType = "text/css";
-              break;
-            case "js":
-              mimeType = "application/javascript";
-              break;
-            case "json":
-              mimeType = "application/json";
-              break;
-          }
-          
-          if (fileInfo.path === "index.html") {
-            pages.unshift({
-              path: fileInfo.path,
-              html: content,
-            });
-          } else {
-            pages.push({
-              path: fileInfo.path,
-              html: content,
-            });
-          }
-          
-          const file = new File([content], fileInfo.path, { type: mimeType });
-          files.push(file);
-        }
-      }
-      else if (fileInfo.type === "directory" && (["videos", "images", "audio"].includes(fileInfo.path) || fileInfo.path === "components")) {
-        for await (const subFileInfo of listFiles({
-          repo,
-          accessToken: user.token as string,
-          revision: commitId,
-          path: fileInfo.path,
-        })) {
-          if (subFileInfo.path.includes("components")) {
-            commitFilePaths.add(subFileInfo.path);
-            const blob = await downloadFile({ 
-              repo, 
-              accessToken: user.token as string, 
-              path: subFileInfo.path, 
-              revision: commitId,
-              raw: true 
-            });
-            const content = await blob?.text();
-            
-            if (content) {
-              pages.push({
-                path: subFileInfo.path,
-                html: content,
-              });
-              
-              const file = new File([content], subFileInfo.path, { type: "text/html" });
-              files.push(file);
-            }
-          } else if (allowedFilesExtensions.includes(subFileInfo.path.split(".").pop() || "")) {
-            commitFilePaths.add(subFileInfo.path);
-            mediaFiles.push(`https://huggingface.co/spaces/${namespace}/${repoId}/resolve/main/${subFileInfo.path}`);
-          }
-        }
-      }
-      else if (allowedExtensions.includes(fileExtension || "")) {
-        commitFilePaths.add(fileInfo.path);
-      }
-    }
-
-    const mainBranchFilePaths: Set<string> = new Set();
-    for await (const fileInfo of listFiles({
-      repo,
-      accessToken: user.token as string,
-      revision: "main",
-    })) {
-      const fileExtension = fileInfo.path.split('.').pop()?.toLowerCase();
-      
-      if (allowedExtensions.includes(fileExtension || "")) {
-        mainBranchFilePaths.add(fileInfo.path);
-      }
-    }
-
-    const filesToDelete: string[] = [];
-    for (const mainFilePath of mainBranchFilePaths) {
-      if (!commitFilePaths.has(mainFilePath)) {
-        filesToDelete.push(mainFilePath);
-      }
-    }
-
-    if (files.length === 0 && filesToDelete.length === 0) {
-      return NextResponse.json(
-        { ok: false, error: "No files found in the specified commit and no files to delete" },
-        { status: 404 }
-      );
-    }
-
-    if (filesToDelete.length > 0) {
-      await deleteFiles({
-        repo,
-        paths: filesToDelete,
-        accessToken: user.token as string,
-        commitTitle: `Removed files from promoting ${commitId.slice(0, 7)}`,
-        commitDescription: `Removed files that don't exist in commit ${commitId}:\n${filesToDelete.map(path => `- ${path}`).join('\n')}`,
-      });
-    }
-
-    if (files.length > 0) {
-      await uploadFiles({
-        repo,
-        files,
-        accessToken: user.token as string,
-        commitTitle: `Promote version ${commitId.slice(0, 7)} to main`,
-        commitDescription: `Promoted commit ${commitId} to main branch`,
-      });
-    }
-
-    return NextResponse.json(
-      { 
-        ok: true, 
-        message: "Version promoted successfully",
-        promotedCommit: commitId,
-        pages: pages,
-        files: mediaFiles,
-      },
-      { status: 200 }
-    );
-
-  } catch (error: any) {
-    
-    // Handle specific HuggingFace API errors
-    if (error.statusCode === 404) {
-      return NextResponse.json(
-        { ok: false, error: "Commit not found" },
-        { status: 404 }
-      );
-    }
-    
-    if (error.statusCode === 403) {
-      return NextResponse.json(
-        { ok: false, error: "Access denied to repository" },
-        { status: 403 }
-      );
-    }
-
-    return NextResponse.json(
-      { ok: false, error: error.message || "Failed to promote version" },
-      { status: 500 }
-    );
-  }
-}

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

@@ -1,102 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-import { RepoDesignation, listFiles, spaceInfo, downloadFile } from "@huggingface/hub";
-
-import { isAuthenticated } from "@/lib/auth";
-import { Page } from "@/types";
-
-export async function GET(
-  req: NextRequest,
-  { params }: { 
-    params: Promise<{ 
-      namespace: string; 
-      repoId: string; 
-      commitId: string; 
-    }> 
-  }
-) {
-  const user = await isAuthenticated();
-
-  if (user instanceof NextResponse || !user) {
-    return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
-  }
-
-  const param = await params;
-  const { namespace, repoId, commitId } = param;
-
-  try {
-    const repo: RepoDesignation = {
-      type: "space",
-      name: `${namespace}/${repoId}`,
-    };
-
-    const space = await spaceInfo({
-      name: `${namespace}/${repoId}`,
-      accessToken: user.token as string,
-      additionalFields: ["author"],
-    });
-
-    if (!space || space.sdk !== "static") {
-      return NextResponse.json(
-        { ok: false, error: "Space is not a static space." },
-        { status: 404 }
-      );
-    }
-    
-    if (space.author !== user.name) {
-      return NextResponse.json(
-        { ok: false, error: "Space does not belong to the authenticated user." },
-        { status: 403 }
-      );
-    }
-
-    const pages: Page[] = [];
-    
-    for await (const fileInfo of listFiles({
-      repo,
-      accessToken: user.token as string,
-      revision: commitId,
-    })) {
-      const fileExtension = fileInfo.path.split('.').pop()?.toLowerCase();
-      
-      if (fileInfo.path.endsWith(".html") || fileInfo.path.endsWith(".css") || fileInfo.path.endsWith(".js") || fileInfo.path.endsWith(".json")) {
-        const blob = await downloadFile({ 
-          repo, 
-          accessToken: user.token as string, 
-          path: fileInfo.path, 
-          revision: commitId,
-          raw: true 
-        });
-        const content = await blob?.text();
-        
-        if (content) {
-          if (fileInfo.path === "index.html") {
-            pages.unshift({
-              path: fileInfo.path,
-              html: content,
-            });
-          } else {
-            pages.push({
-              path: fileInfo.path,
-              html: content,
-            });
-          }
-        }
-      }
-    }
-    
-    return NextResponse.json({
-      ok: true,
-      pages,
-    });
-  } catch (error: any) {
-    console.error("Error fetching commit pages:", error);
-    return NextResponse.json(
-      {
-        ok: false,
-        error: error.message || "Failed to fetch commit pages",
-      },
-      { status: 500 }
-    );
-  }
-}
-

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

@@ -1,105 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-import { RepoDesignation, listFiles, spaceInfo, downloadFile } from "@huggingface/hub";
-import JSZip from "jszip";
-
-import { isAuthenticated } from "@/lib/auth";
-
-export async function GET(
-  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 param = await params;
-  const { namespace, repoId } = param;
-
-  try {
-    const space = await spaceInfo({
-      name: `${namespace}/${repoId}`,
-      accessToken: user.token as string,
-      additionalFields: ["author"],
-    });
-
-    if (!space || space.sdk !== "static") {
-      return NextResponse.json(
-        {
-          ok: false,
-          error: "Space is not a static space",
-        },
-        { status: 404 }
-      );
-    }
-
-    if (space.author !== user.name) {
-      return NextResponse.json(
-        {
-          ok: false,
-          error: "Space does not belong to the authenticated user",
-        },
-        { status: 403 }
-      );
-    }
-
-    const repo: RepoDesignation = {
-      type: "space",
-      name: `${namespace}/${repoId}`,
-    };
-
-    const zip = new JSZip();
-
-    for await (const fileInfo of listFiles({ 
-      repo, 
-      accessToken: user.token as string, 
-      recursive: true,
-    })) {
-      if (fileInfo.type === "directory" || fileInfo.path.startsWith(".")) {
-        continue;
-      }
-
-      try {
-        const blob = await downloadFile({ 
-          repo, 
-          accessToken: user.token as string, 
-          path: fileInfo.path, 
-          raw: true 
-        });
-
-        if (blob) {
-          const arrayBuffer = await blob.arrayBuffer();
-          zip.file(fileInfo.path, arrayBuffer);
-        }
-      } catch (error) {
-        console.error(`Error downloading file ${fileInfo.path}:`, error);
-      }
-    }
-
-    const zipBlob = await zip.generateAsync({ 
-      type: "blob",
-      compression: "DEFLATE",
-      compressionOptions: {
-        level: 6
-      }
-    });
-
-    const projectName = `${namespace}-${repoId}`.replace(/[^a-zA-Z0-9-_]/g, '_');
-    const filename = `${projectName}.zip`;
-
-    return new NextResponse(zipBlob, {
-      headers: {
-        "Content-Type": "application/zip",
-        "Content-Disposition": `attachment; filename="${filename}"`,
-        "Content-Length": zipBlob.size.toString(),
-      },
-    });
-  } catch (error: any) {
-    return NextResponse.json(
-      { ok: false, error: error.message || "Failed to create ZIP file" },
-      { status: 500 }
-    );
-  }
-}
-

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

@@ -1,125 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-import { RepoDesignation, spaceInfo, uploadFiles } from "@huggingface/hub";
-
-import { isAuthenticated } from "@/lib/auth";
-import Project from "@/models/Project";
-import dbConnect from "@/lib/mongodb";
-
-export async function POST(
-  req: NextRequest,
-  { params }: { params: Promise<{ namespace: string; repoId: string }> }
-) {
-  try {
-    const user = await isAuthenticated();
-
-    if (user instanceof NextResponse || !user) {
-      return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
-    }
-
-    const param = await params;
-    const { namespace, repoId } = param;
-
-    const space = await spaceInfo({
-      name: `${namespace}/${repoId}`,
-      accessToken: user.token as string,
-      additionalFields: ["author"],
-    });
-
-    if (!space || space.sdk !== "static") {
-      return NextResponse.json(
-        { ok: false, error: "Space is not a static space." },
-        { status: 404 }
-      );
-    }
-    
-    if (space.author !== user.name) {
-      return NextResponse.json(
-        { ok: false, error: "Space does not belong to the authenticated user." },
-        { status: 403 }
-      );
-    }
-
-    // Parse the FormData to get the media files
-    const formData = await req.formData();
-    const mediaFiles = formData.getAll("images") as File[];
-
-    if (!mediaFiles || mediaFiles.length === 0) {
-      return NextResponse.json(
-        {
-          ok: false,
-          error: "At least one media file is required under the 'images' key",
-        },
-        { status: 400 }
-      );
-    }
-
-    const files: File[] = [];
-    for (const file of mediaFiles) {
-      if (!(file instanceof File)) {
-        return NextResponse.json(
-          {
-            ok: false,
-            error: "Invalid file format - all items under 'images' key must be files",
-          },
-          { status: 400 }
-        );
-      }
-
-      // Check if file is a supported media type
-      const isImage = file.type.startsWith('image/');
-      const isVideo = file.type.startsWith('video/');
-      const isAudio = file.type.startsWith('audio/');
-      
-      if (!isImage && !isVideo && !isAudio) {
-        return NextResponse.json(
-          {
-            ok: false,
-            error: `File ${file.name} is not a supported media type (image, video, or audio)`,
-          },
-          { status: 400 }
-        );
-      }
-
-      // Create File object with appropriate folder prefix
-      let folderPrefix = 'images/';
-      if (isVideo) {
-        folderPrefix = 'videos/';
-      } else if (isAudio) {
-        folderPrefix = 'audio/';
-      }
-      
-      const fileName = `${folderPrefix}${file.name}`;
-      const processedFile = new File([file], fileName, { type: file.type });
-      files.push(processedFile);
-    }
-
-    // Upload files to HuggingFace space
-    const repo: RepoDesignation = {
-      type: "space",
-      name: `${namespace}/${repoId}`,
-    };
-
-    await uploadFiles({
-      repo,
-      files,
-      accessToken: user.token as string,
-      commitTitle: `Upload ${files.length} media file(s)`,
-    });
-
-    return NextResponse.json({ 
-      ok: true, 
-      message: `Successfully uploaded ${files.length} media file(s) to ${namespace}/${repoId}/`,
-      uploadedFiles: files.map((file) => `https://huggingface.co/spaces/${namespace}/${repoId}/resolve/main/${file.name}`),
-    }, { status: 200 });
-
-  } catch (error) {
-    console.error('Error uploading media files:', error);
-    return NextResponse.json(
-      {
-        ok: false,
-        error: "Failed to upload media files",
-      },
-      { status: 500 }
-    );
-  }
-}

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

@@ -1,10 +1,12 @@
 import { NextRequest, NextResponse } from "next/server";
-import { RepoDesignation, spaceInfo, listFiles, deleteRepo, listCommits, downloadFile } from "@huggingface/hub";
+import { RepoDesignation, spaceInfo, uploadFile } from "@huggingface/hub";
 
 import { isAuthenticated } from "@/lib/auth";
-import { Commit, Page } from "@/types";
+import Project from "@/models/Project";
+import dbConnect from "@/lib/mongodb";
+import { getPTag } from "@/lib/utils";
 
-export async function DELETE(
+export async function GET(
   req: NextRequest,
   { params }: { params: Promise<{ namespace: string; repoId: string }> }
 ) {
@@ -14,63 +16,24 @@ export async function DELETE(
     return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
   }
 
+  await dbConnect();
   const param = await params;
   const { namespace, repoId } = param;
 
-  try {
-    const space = await spaceInfo({
-      name: `${namespace}/${repoId}`,
-      accessToken: user.token as string,
-      additionalFields: ["author"],
-    });
-    
-    if (!space || space.sdk !== "static") {
-      return NextResponse.json(
-        { ok: false, error: "Space is not a static space." },
-        { status: 404 }
-      );
-    }
-    
-    if (space.author !== user.name) {
-      return NextResponse.json(
-        { ok: false, error: "Space does not belong to the authenticated user." },
-        { status: 403 }
-      );
-    }
-    
-    const repo: RepoDesignation = {
-      type: "space",
-      name: `${namespace}/${repoId}`,
-    };
-    
-    await deleteRepo({
-      repo,
-      accessToken: user.token as string,
-    });
-
-    
-    return NextResponse.json({ ok: true }, { status: 200 });
-  } catch (error: any) {
+  const project = await Project.findOne({
+    user_id: user.id,
+    space_id: `${namespace}/${repoId}`,
+  }).lean();
+  if (!project) {
     return NextResponse.json(
-      { ok: false, error: error.message },
-      { status: 500 }
+      {
+        ok: false,
+        error: "Project not found",
+      },
+      { status: 404 }
     );
   }
-}
-
-export async function GET(
-  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 param = await params;
-  const { namespace, repoId } = param;
-
+  const space_url = `https://huggingface.co/spaces/${namespace}/${repoId}/raw/main/index.html`;
   try {
     const space = await spaceInfo({
       name: namespace + "/" + repoId,
@@ -97,85 +60,26 @@ export async function GET(
       );
     }
 
-    const repo: RepoDesignation = {
-      type: "space",
-      name: `${namespace}/${repoId}`,
-    };
-
-    const htmlFiles: Page[] = [];
-    const files: string[] = [];
-
-    const allowedFilesExtensions = ["jpg", "jpeg", "png", "gif", "svg", "webp", "avif", "heic", "heif", "ico", "bmp", "tiff", "tif", "mp4", "webm", "ogg", "avi", "mov", "mp3", "wav", "ogg", "aac", "m4a"];
-    
-    for await (const fileInfo of listFiles({repo, accessToken: user.token as string})) {
-      if (fileInfo.path.endsWith(".html") || fileInfo.path.endsWith(".css") || fileInfo.path.endsWith(".js") || fileInfo.path.endsWith(".json")) {
-        const blob = await downloadFile({ repo, accessToken: user.token as string, path: fileInfo.path, raw: true });
-        const html = await blob?.text();
-        if (!html) {
-          continue;
-        }
-        if (fileInfo.path === "index.html") {
-          htmlFiles.unshift({
-            path: fileInfo.path,
-            html,
-          });
-        } else {
-          htmlFiles.push({
-            path: fileInfo.path,
-            html,
-          });
-        }
-      }
-      if (fileInfo.type === "directory" && (["videos", "images", "audio"].includes(fileInfo.path) || fileInfo.path === "components")) {
-        for await (const subFileInfo of listFiles({repo, accessToken: user.token as string, path: fileInfo.path})) {
-          if (subFileInfo.path.includes("components")) {
-            const blob = await downloadFile({ repo, accessToken: user.token as string, path: subFileInfo.path, raw: true });
-            const html = await blob?.text();
-            if (!html) {
-              continue;
-            }
-            htmlFiles.push({
-              path: subFileInfo.path,
-              html,
-            });
-          } else if (allowedFilesExtensions.includes(subFileInfo.path.split(".").pop() || "")) {
-            files.push(`https://huggingface.co/spaces/${namespace}/${repoId}/resolve/main/${subFileInfo.path}`);
-          }
-        }
-      }
-    }
-    const commits: Commit[] = [];
-    for await (const commit of listCommits({ repo, accessToken: user.token as string })) {
-      if (commit.title.includes("initial commit") || commit.title.includes("image(s)") || commit.title.includes("Removed files from promoting")) {
-        continue;
-      }
-      commits.push({
-        title: commit.title,
-        oid: commit.oid,
-        date: commit.date,
-      });
-    }
-    
-    if (htmlFiles.length === 0) {
+    const response = await fetch(space_url);
+    if (!response.ok) {
       return NextResponse.json(
         {
           ok: false,
-          error: "No HTML files found",
+          error: "Failed to fetch space HTML",
         },
         { status: 404 }
       );
     }
+    let html = await response.text();
+    // remove the last p tag including this url https://enzostvs-deepsite.hf.space
+    html = html.replace(getPTag(namespace + "/" + repoId), "");
+
     return NextResponse.json(
       {
         project: {
-          id: space.id,
-          space_id: space.name,
-          private: space.private,
-          _updatedAt: space.updatedAt,
+          ...project,
+          html,
         },
-        pages: htmlFiles,
-        files,
-        commits,
         ok: true,
       },
       { status: 200 }
@@ -184,6 +88,10 @@ export async function GET(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
   } catch (error: any) {
     if (error.statusCode === 404) {
+      await Project.deleteOne({
+        user_id: user.id,
+        space_id: `${namespace}/${repoId}`,
+      });
       return NextResponse.json(
         { error: "Space not found", ok: false },
         { status: 404 }
@@ -195,3 +103,135 @@ export async function GET(
     );
   }
 }
+
+export async function PUT(
+  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 });
+  }
+
+  await dbConnect();
+  const param = await params;
+  const { namespace, repoId } = param;
+  const { html, prompts } = await req.json();
+
+  const project = await Project.findOne({
+    user_id: user.id,
+    space_id: `${namespace}/${repoId}`,
+  }).lean();
+  if (!project) {
+    return NextResponse.json(
+      {
+        ok: false,
+        error: "Project not found",
+      },
+      { status: 404 }
+    );
+  }
+
+  const repo: RepoDesignation = {
+    type: "space",
+    name: `${namespace}/${repoId}`,
+  };
+
+  const newHtml = html.replace(/<\/body>/, `${getPTag(repo.name)}</body>`);
+  const file = new File([newHtml], "index.html", { type: "text/html" });
+  await uploadFile({
+    repo,
+    file,
+    accessToken: user.token as string,
+    commitTitle: `${prompts[prompts.length - 1]} - Follow Up Deployment`,
+  });
+
+  await Project.updateOne(
+    { user_id: user.id, space_id: `${namespace}/${repoId}` },
+    {
+      $set: {
+        prompts: [
+          ...(project && "prompts" in project ? project.prompts : []),
+          ...prompts,
+        ],
+      },
+    }
+  );
+  return NextResponse.json({ ok: true }, { status: 200 });
+}
+
+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 });
+  }
+
+  await dbConnect();
+  const param = await params;
+  const { namespace, repoId } = param;
+
+  const space = await spaceInfo({
+    name: namespace + "/" + repoId,
+    accessToken: user.token as string,
+    additionalFields: ["author"],
+  });
+
+  if (!space || space.sdk !== "static") {
+    return NextResponse.json(
+      {
+        ok: false,
+        error: "Space is not a static space",
+      },
+      { status: 404 }
+    );
+  }
+  if (space.author !== user.name) {
+    return NextResponse.json(
+      {
+        ok: false,
+        error: "Space does not belong to the authenticated user",
+      },
+      { status: 403 }
+    );
+  }
+
+  const project = await Project.findOne({
+    user_id: user.id,
+    space_id: `${namespace}/${repoId}`,
+  }).lean();
+  if (project) {
+    // redirect to the project page if it already exists
+    return NextResponse.json(
+      {
+        ok: false,
+        error: "Project already exists",
+        redirect: `/projects/${namespace}/${repoId}`,
+      },
+      { status: 400 }
+    );
+  }
+
+  const newProject = new Project({
+    user_id: user.id,
+    space_id: `${namespace}/${repoId}`,
+    prompts: [],
+  });
+
+  await newProject.save();
+  return NextResponse.json(
+    {
+      ok: true,
+      project: {
+        id: newProject._id,
+        space_id: newProject.space_id,
+        prompts: newProject.prompts,
+      },
+    },
+    { status: 201 }
+  );
+}

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

@@ -1,72 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-import { uploadFiles } from "@huggingface/hub";
-
-import { isAuthenticated } from "@/lib/auth";
-import { Page } from "@/types";
-
-export async function PUT(
-  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 param = await params;
-  const { namespace, repoId } = param;
-  const { pages, commitTitle = "Manual changes saved" } = await req.json();
-
-  if (!pages || !Array.isArray(pages) || pages.length === 0) {
-    return NextResponse.json(
-      { ok: false, error: "Pages are required" },
-      { status: 400 }
-    );
-  }
-
-  try {
-    // Prepare files for upload
-    const files: File[] = [];
-    pages.forEach((page: Page) => {
-      // Determine MIME type based on file extension
-      let mimeType = "text/html";
-      if (page.path.endsWith(".css")) {
-        mimeType = "text/css";
-      } else if (page.path.endsWith(".js")) {
-        mimeType = "text/javascript";
-      } else if (page.path.endsWith(".json")) {
-        mimeType = "application/json";
-      }
-      const file = new File([page.html], page.path, { type: mimeType });
-      files.push(file);
-    });
-
-    const response = await uploadFiles({
-      repo: {
-        type: "space",
-        name: `${namespace}/${repoId}`,
-      },
-      files,
-      commitTitle,
-      accessToken: user.token as string,
-    });
-
-    return NextResponse.json({
-      ok: true,
-      pages,
-      commit: {
-        ...response.commit,
-        title: commitTitle,
-      }
-    });
-  } catch (error: any) {
-    console.error("Error saving manual changes:", error);
-    return NextResponse.json(
-      {
-        ok: false,
-        error: error.message || "Failed to save changes",
-      },
-      { status: 500 }
-    );
-  }
-}

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

@@ -1,141 +0,0 @@
-import { NextRequest, NextResponse } from "next/server";
-import { createRepo, RepoDesignation, uploadFiles } from "@huggingface/hub";
-
-import { isAuthenticated } from "@/lib/auth";
-import { Page } from "@/types";
-import { COLORS } from "@/lib/utils";
-import { injectDeepSiteBadge, isIndexPage } from "@/lib/inject-badge";
-import { pagesToFiles } from "@/lib/format-ai-response";
-
-/**
- * UPDATE route - for updating existing projects or creating new ones after AI streaming
- * This route handles the HuggingFace upload after client-side AI response processing
- */
-export async function PUT(
-  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 param = await params;
-  let { namespace, repoId } = param;
-  const { pages, commitTitle = "AI-generated changes", isNew, projectName } = await req.json();
-
-  if (!pages || !Array.isArray(pages) || pages.length === 0) {
-    return NextResponse.json(
-      { ok: false, error: "Pages are required" },
-      { status: 400 }
-    );
-  }
-
-  try {
-    let files: File[];
-    
-    if (isNew) {
-      // Creating a new project
-      const title = projectName || "DeepSite Project";
-      const formattedTitle = title
-        .toLowerCase()
-        .replace(/[^a-z0-9]+/g, "-")
-        .split("-")
-        .filter(Boolean)
-        .join("-")
-        .slice(0, 96);
-
-      const repo: RepoDesignation = {
-        type: "space",
-        name: `${user.name}/${formattedTitle}`,
-      };
-
-      try {
-        const { repoUrl } = await createRepo({
-          repo,
-          accessToken: user.token as string,
-        });
-        namespace = user.name;
-        repoId = repoUrl.split("/").slice(-2).join("/").split("/")[1];
-      } catch (createRepoError: any) {
-        return NextResponse.json(
-          {
-            ok: false,
-            error: `Failed to create repository: ${createRepoError.message || 'Unknown error'}`,
-          },
-          { status: 500 }
-        );
-      }
-
-      // Prepare files with badge injection for new projects
-      files = [];
-      pages.forEach((page: Page) => {
-        let mimeType = "text/html";
-        if (page.path.endsWith(".css")) {
-          mimeType = "text/css";
-        } else if (page.path.endsWith(".js")) {
-          mimeType = "text/javascript";
-        } else if (page.path.endsWith(".json")) {
-          mimeType = "application/json";
-        }
-        const content = (mimeType === "text/html" && isIndexPage(page.path))
-          ? injectDeepSiteBadge(page.html)
-          : page.html;
-        const file = new File([content], page.path, { type: mimeType });
-        files.push(file);
-      });
-
-      // Add README.md for new projects
-      const colorFrom = COLORS[Math.floor(Math.random() * COLORS.length)];
-      const colorTo = COLORS[Math.floor(Math.random() * COLORS.length)];
-      const README = `---
-title: ${title}
-colorFrom: ${colorFrom}
-colorTo: ${colorTo}
-emoji: 🐳
-sdk: static
-pinned: false
-tags:
-  - deepsite-v3
----
-
-# Welcome to your new DeepSite project!
-This project was created with [DeepSite](https://huggingface.co/deepsite).
-`;
-      files.push(new File([README], "README.md", { type: "text/markdown" }));
-    } else {
-      // Updating existing project - no badge injection
-      files = pagesToFiles(pages);
-    }
-
-    const response = await uploadFiles({
-      repo: {
-        type: "space",
-        name: `${namespace}/${repoId}`,
-      },
-      files,
-      commitTitle,
-      accessToken: user.token as string,
-    });
-
-    return NextResponse.json({
-      ok: true,
-      pages,
-      repoId: `${namespace}/${repoId}`,
-      commit: {
-        ...response.commit,
-        title: commitTitle,
-      }
-    });
-  } catch (error: any) {
-    console.error("Error updating project:", error);
-    return NextResponse.json(
-      {
-        ok: false,
-        error: error.message || "Failed to update project",
-      },
-      { status: 500 }
-    );
-  }
-}
-

app/api/me/projects/route.ts

@@ -1,121 +1,126 @@
 import { NextRequest, NextResponse } from "next/server";
-import { RepoDesignation, createRepo, listCommits, spaceInfo, uploadFiles } from "@huggingface/hub";
+import { createRepo, RepoDesignation, uploadFiles } from "@huggingface/hub";
 
 import { isAuthenticated } from "@/lib/auth";
-import { Commit, Page } from "@/types";
-import { COLORS } from "@/lib/utils";
-import { injectDeepSiteBadge, isIndexPage } from "@/lib/inject-badge";
+import Project from "@/models/Project";
+import dbConnect from "@/lib/mongodb";
+import { COLORS, getPTag } from "@/lib/utils";
+// import type user
+export async function GET() {
+  const user = await isAuthenticated();
+
+  if (user instanceof NextResponse || !user) {
+    return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
+  }
 
-export async function POST(
-  req: NextRequest,
-) {
+  await dbConnect();
+
+  const projects = await Project.find({
+    user_id: user?.id,
+  })
+    .sort({ _createdAt: -1 })
+    .limit(100)
+    .lean();
+  if (!projects) {
+    return NextResponse.json(
+      {
+        ok: false,
+        projects: [],
+      },
+      { status: 404 }
+    );
+  }
+  return NextResponse.json(
+    {
+      ok: true,
+      projects,
+    },
+    { status: 200 }
+  );
+}
+
+/**
+ * This API route creates a new project in Hugging Face Spaces.
+ * It requires an Authorization header with a valid token and a JSON body with the project details.
+ */
+export async function POST(request: NextRequest) {
   const user = await isAuthenticated();
+
   if (user instanceof NextResponse || !user) {
     return NextResponse.json({ message: "Unauthorized" }, { status: 401 });
   }
 
-  const { title: titleFromRequest, pages, prompt } = await req.json();
+  const { title, html, prompts } = await request.json();
+
+  if (!title || !html) {
+    return NextResponse.json(
+      { message: "Title and HTML content are required.", ok: false },
+      { status: 400 }
+    );
+  }
+
+  await dbConnect();
 
-  const title = titleFromRequest ?? "DeepSite Project";
+  try {
+    let readme = "";
+    let newHtml = html;
 
-  const formattedTitle = title
-  .toLowerCase()
-  .replace(/[^a-z0-9]+/g, "-")
-  .split("-")
-  .filter(Boolean)
-  .join("-")
-  .slice(0, 96);
+    const newTitle = title
+      .toLowerCase()
+      .replace(/[^a-z0-9]+/g, "-")
+      .split("-")
+      .filter(Boolean)
+      .join("-")
+      .slice(0, 96);
 
-  const repo: RepoDesignation = {
-    type: "space",
-    name: `${user.name}/${formattedTitle}`,
-  };
-  const colorFrom = COLORS[Math.floor(Math.random() * COLORS.length)];
-  const colorTo = COLORS[Math.floor(Math.random() * COLORS.length)];
-  const README = `---
-title: ${title}
+    const repo: RepoDesignation = {
+      type: "space",
+      name: `${user.name}/${newTitle}`,
+    };
+
+    const { repoUrl } = await createRepo({
+      repo,
+      accessToken: user.token as string,
+    });
+    const colorFrom = COLORS[Math.floor(Math.random() * COLORS.length)];
+    const colorTo = COLORS[Math.floor(Math.random() * COLORS.length)];
+    readme = `---
+title: ${newTitle}
+emoji: 🐳
 colorFrom: ${colorFrom}
 colorTo: ${colorTo}
-emoji: 🐳
 sdk: static
 pinned: false
 tags:
-  - deepsite-v3
+  - deepsite
 ---
 
-# Welcome to your new DeepSite project!
-This project was created with [DeepSite](https://huggingface.co/deepsite).
-`;
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference`;
 
-  const files: File[] = [];
-  const readmeFile = new File([README], "README.md", { type: "text/markdown" });
-  files.push(readmeFile);
-  pages.forEach((page: Page) => {
-    // Determine MIME type based on file extension
-    let mimeType = "text/html";
-    if (page.path.endsWith(".css")) {
-      mimeType = "text/css";
-    } else if (page.path.endsWith(".js")) {
-      mimeType = "text/javascript";
-    } else if (page.path.endsWith(".json")) {
-      mimeType = "application/json";
-    }
-    // Inject the DeepSite badge script into index pages only (not components or other HTML files)
-    const content = (mimeType === "text/html" && isIndexPage(page.path)) 
-      ? injectDeepSiteBadge(page.html) 
-      : page.html;
-    const file = new File([content], page.path, { type: mimeType });
-    files.push(file);
-  });
-
-  try {
-    const { repoUrl} = await createRepo({
-      repo,
-      accessToken: user.token as string,
+    newHtml = html.replace(/<\/body>/, `${getPTag(repo.name)}</body>`);
+    const file = new File([newHtml], "index.html", { type: "text/html" });
+    const readmeFile = new File([readme], "README.md", {
+      type: "text/markdown",
     });
-    const commitTitle = !prompt || prompt.trim() === "" ? "Redesign my website" : prompt;
+    const files = [file, readmeFile];
     await uploadFiles({
       repo,
       files,
       accessToken: user.token as string,
-      commitTitle
+      commitTitle: `${prompts[prompts.length - 1]} - Initial Deployment`,
     });
-
     const path = repoUrl.split("/").slice(-2).join("/");
-
-    const commits: Commit[] = [];
-    for await (const commit of listCommits({ repo, accessToken: user.token as string })) {
-      if (commit.title.includes("initial commit") || commit.title.includes("image(s)") || commit.title.includes("Promote version")) {
-        continue;
-      }
-      commits.push({
-        title: commit.title,
-        oid: commit.oid,
-        date: commit.date,
-      });
-    }
-
-    const space = await spaceInfo({
-      name: repo.name,
-      accessToken: user.token as string,
+    const project = await Project.create({
+      user_id: user.id,
+      space_id: path,
+      prompts,
     });
-
-    let newProject = {
-      files,
-      pages,
-      commits,
-      project: {
-        id: space.id,
-        space_id: space.name,
-        _updatedAt: space.updatedAt,
-      }
-    }
-    
-    return NextResponse.json({ space: newProject, path, ok: true }, { status: 201 });
+    return NextResponse.json({ project, path, ok: true }, { status: 201 });
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
   } catch (err: any) {
     return NextResponse.json(
       { error: err.message, ok: false },
       { status: 500 }
     );
   }
-}
+}

app/api/me/route.ts

@@ -1,4 +1,3 @@
-import { listSpaces } from "@huggingface/hub";
 import { headers } from "next/headers";
 import { NextResponse } from "next/server";
 
@@ -22,25 +21,5 @@ export async function GET() {
     );
   }
   const user = await userResponse.json();
-  const projects = [];
-  for await (const space of listSpaces({
-    accessToken: token.replace("Bearer ", "") as string,
-    additionalFields: ["author", "cardData"],
-    search: {
-      owner: user.name,
-    }
-  })) {
-    if (
-      space.sdk === "static" &&
-      Array.isArray((space.cardData as { tags?: string[] })?.tags) &&
-      (
-        ((space.cardData as { tags?: string[] })?.tags?.includes("deepsite-v3")) ||
-        ((space.cardData as { tags?: string[] })?.tags?.includes("deepsite"))
-      )
-    ) {
-      projects.push(space);
-    }
-  }
-
-  return NextResponse.json({ user, projects, errCode: null }, { status: 200 });
+  return NextResponse.json({ user, errCode: null }, { status: 200 });
 }

app/api/re-design/route.ts

@@ -1,11 +1,5 @@
 import { NextRequest, NextResponse } from "next/server";
 
-// Timeout configuration (in milliseconds)
-const FETCH_TIMEOUT = 30000; // 30 seconds for external fetch
-
-// Extend the maximum execution time for this route
-export const maxDuration = 60; // 1 minute
-
 export async function PUT(request: NextRequest) {
   const body = await request.json();
   const { url } = body;
@@ -15,54 +9,28 @@ export async function PUT(request: NextRequest) {
   }
 
   try {
-    // Create an AbortController for timeout
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), FETCH_TIMEOUT);
-
-    try {
-      const response = await fetch(
-        `https://r.jina.ai/${encodeURIComponent(url)}`,
-        {
-          method: "POST",
-          signal: controller.signal,
-        }
-      );
-      
-      clearTimeout(timeoutId);
-      
-      if (!response.ok) {
-        return NextResponse.json(
-          { error: "Failed to fetch redesign" },
-          { status: 500 }
-        );
+    const response = await fetch(
+      `https://r.jina.ai/${encodeURIComponent(url)}`,
+      {
+        method: "POST",
       }
-      const markdown = await response.text();
+    );
+    if (!response.ok) {
       return NextResponse.json(
-        {
-          ok: true,
-          markdown,
-        },
-        { status: 200 }
+        { error: "Failed to fetch redesign" },
+        { status: 500 }
       );
-    } catch (fetchError: any) {
-      clearTimeout(timeoutId);
-      
-      if (fetchError.name === 'AbortError') {
-        return NextResponse.json(
-          { error: "Request timeout: The external service took too long to respond. Please try again." },
-          { status: 504 }
-        );
-      }
-      throw fetchError;
     }
+    const markdown = await response.text();
+    return NextResponse.json(
+      {
+        ok: true,
+        markdown,
+      },
+      { status: 200 }
+    );
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
   } catch (error: any) {
-    if (error.name === 'AbortError' || error.message?.includes('timeout')) {
-      return NextResponse.json(
-        { error: "Request timeout: The external service took too long to respond. Please try again." },
-        { status: 504 }
-      );
-    }
     return NextResponse.json(
       { error: error.message || "An error occurred" },
       { status: 500 }

app/auth/callback/page.tsx

@@ -5,92 +5,67 @@ import { use, useState } from "react";
 import { useMount, useTimeoutFn } from "react-use";
 
 import { Button } from "@/components/ui/button";
-import { AnimatedBlobs } from "@/components/animated-blobs";
-import { useBroadcastChannel } from "@/lib/useBroadcastChannel";
 export default function AuthCallback({
   searchParams,
 }: {
   searchParams: Promise<{ code: string }>;
 }) {
   const [showButton, setShowButton] = useState(false);
-  const [isPopupAuth, setIsPopupAuth] = useState(false);
   const { code } = use(searchParams);
   const { loginFromCode } = useUser();
-  const { postMessage } = useBroadcastChannel("auth", () => {});
 
   useMount(async () => {
     if (code) {
-      const isPopup = window.opener || window.parent !== window;
-      setIsPopupAuth(isPopup);
-
-      if (isPopup) {
-        postMessage({
-          type: "user-oauth",
-          code: code,
-        });
-
-        setTimeout(() => {
-          if (window.opener) {
-            window.close();
-          }
-        }, 1000);
-      } else {
-        await loginFromCode(code);
-      }
+      await loginFromCode(code);
     }
   });
 
-  useTimeoutFn(() => setShowButton(true), 7000);
+  useTimeoutFn(
+    () => setShowButton(true),
+    7000 // Show button after 5 seconds
+  );
 
   return (
-    <div className="h-screen flex flex-col justify-center items-center bg-neutral-950 z-1 relative">
-      <div className="background__noisy" />
-      <div className="relative max-w-4xl py-10 flex items-center justify-center w-full">
-        <div className="max-w-lg mx-auto !rounded-2xl !p-0 !bg-white !border-neutral-100 min-w-xs text-center overflow-hidden ring-[8px] ring-white/20">
-          <header className="bg-neutral-50 p-6 border-b border-neutral-200/60">
-            <div className="flex items-center justify-center -space-x-4 mb-3">
-              <div className="size-9 rounded-full bg-pink-200 shadow-2xs flex items-center justify-center text-xl opacity-50">
-                🚀
-              </div>
-              <div className="size-11 rounded-full bg-amber-200 shadow-2xl flex items-center justify-center text-2xl z-2">
-                👋
-              </div>
-              <div className="size-9 rounded-full bg-sky-200 shadow-2xs flex items-center justify-center text-xl opacity-50">
-                🙌
-              </div>
+    <div className="h-screen flex flex-col justify-center items-center">
+      <div className="!rounded-2xl !p-0 !bg-white !border-neutral-100 min-w-xs text-center overflow-hidden ring-[8px] ring-white/20">
+        <header className="bg-neutral-50 p-6 border-b border-neutral-200/60">
+          <div className="flex items-center justify-center -space-x-4 mb-3">
+            <div className="size-9 rounded-full bg-pink-200 shadow-2xs flex items-center justify-center text-xl opacity-50">
+              🚀
             </div>
-            <p className="text-xl font-semibold text-neutral-950">
-              {isPopupAuth
-                ? "Authentication Complete!"
-                : "Login In Progress..."}
-            </p>
-            <p className="text-sm text-neutral-500 mt-1.5">
-              {isPopupAuth
-                ? "You can now close this tab and return to the previous page."
-                : "Wait a moment while we log you in with your code."}
+            <div className="size-11 rounded-full bg-amber-200 shadow-2xl flex items-center justify-center text-2xl z-2">
+              👋
+            </div>
+            <div className="size-9 rounded-full bg-sky-200 shadow-2xs flex items-center justify-center text-xl opacity-50">
+              🙌
+            </div>
+          </div>
+          <p className="text-xl font-semibold text-neutral-950">
+            Login In Progress...
+          </p>
+          <p className="text-sm text-neutral-500 mt-1.5">
+            Wait a moment while we log you in with your code.
+          </p>
+        </header>
+        <main className="space-y-4 p-6">
+          <div>
+            <p className="text-sm text-neutral-700 mb-4 max-w-xs">
+              If you are not redirected automatically in the next 5 seconds,
+              please click the button below
             </p>
-          </header>
-          <main className="space-y-4 p-6">
-            <div>
-              <p className="text-sm text-neutral-700 mb-4 max-w-xs">
-                If you are not redirected automatically in the next 5 seconds,
-                please click the button below
+            {showButton ? (
+              <Link href="/">
+                <Button variant="black" className="relative">
+                  Go to Home
+                </Button>
+              </Link>
+            ) : (
+              <p className="text-xs text-neutral-500">
+                Please wait, we are logging you in...
               </p>
-              {showButton ? (
-                <Link href="/">
-                  <Button variant="black" className="relative">
-                    Go to Home
-                  </Button>
-                </Link>
-              ) : (
-                <p className="text-xs text-neutral-500">
-                  Please wait, we are logging you in...
-                </p>
-              )}
-            </div>
-          </main>
-        </div>
-        <AnimatedBlobs />
+            )}
+          </div>
+        </main>
       </div>
     </div>
   );

app/layout.tsx

@@ -1,18 +1,15 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import type { Metadata, Viewport } from "next";
 import { Inter, PT_Sans } from "next/font/google";
-import Script from "next/script";
-import { headers } from "next/headers";
-import { redirect } from "next/navigation";
+import { cookies } from "next/headers";
 
+import TanstackProvider from "@/components/providers/tanstack-query-provider";
 import "@/assets/globals.css";
 import { Toaster } from "@/components/ui/sonner";
-import IframeDetector from "@/components/iframe-detector";
+import MY_TOKEN_KEY from "@/lib/get-cookie-name";
+import { apiServer } from "@/lib/api";
 import AppContext from "@/components/contexts/app-context";
-import TanstackContext from "@/components/contexts/tanstack-query-context";
-import { LoginProvider } from "@/components/contexts/login-context";
-import { ProProvider } from "@/components/contexts/pro-context";
-import { generateSEO, generateStructuredData } from "@/lib/seo";
+import Script from "next/script";
 
 const inter = Inter({
   variable: "--font-inter-sans",
@@ -26,12 +23,31 @@ const ptSans = PT_Sans({
 });
 
 export const metadata: Metadata = {
-  ...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.",
+  openGraph: {
     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.",
-    path: "/",
-  }),
+    url: "https://deepsite.hf.co",
+    siteName: "DeepSite",
+    images: [
+      {
+        url: "https://deepsite.hf.co/banner.png",
+        width: 1200,
+        height: 630,
+        alt: "DeepSite Open Graph Image",
+      },
+    ],
+  },
+  twitter: {
+    card: "summary_large_image",
+    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.",
+    images: ["https://deepsite.hf.co/banner.png"],
+  },
   appleWebApp: {
     capable: true,
     title: "DeepSite",
@@ -42,9 +58,6 @@ export const metadata: Metadata = {
     shortcut: "/logo.svg",
     apple: "/logo.svg",
   },
-  verification: {
-    google: process.env.GOOGLE_SITE_VERIFICATION,
-  },
 };
 
 export const viewport: Viewport = {
@@ -53,92 +66,42 @@ export const viewport: Viewport = {
   themeColor: "#000000",
 };
 
-// async function getMe() {
-//   const cookieStore = await cookies();
-//   const cookieName = MY_TOKEN_KEY();
-//   const token = cookieStore.get(cookieName)?.value;
-
-//   if (!token) return { user: null, projects: [], errCode: null };
-//   try {
-//     const res = await apiServer.get("/me", {
-//       headers: {
-//         Authorization: `Bearer ${token}`,
-//       },
-//     });
-//     return { user: res.data.user, projects: res.data.projects, errCode: null };
-//   } catch (err: any) {
-//     return { user: null, projects: [], errCode: err.status };
-//   }
-// }
+async function getMe() {
+  const cookieStore = await cookies();
+  const token = cookieStore.get(MY_TOKEN_KEY())?.value;
+  if (!token) return { user: null, errCode: null };
+  try {
+    const res = await apiServer.get("/me", {
+      headers: {
+        Authorization: `Bearer ${token}`,
+      },
+    });
+    return { user: res.data.user, errCode: null };
+  } catch (err: any) {
+    return { user: null, errCode: err.status };
+  }
+}
 
 export default async function RootLayout({
   children,
 }: Readonly<{
   children: React.ReactNode;
 }>) {
-  // Domain redirect check
-  const headersList = await headers();
-  const forwardedHost = headersList.get("x-forwarded-host");
-  const host = headersList.get("host");
-  const hostname = (forwardedHost || host || "").split(":")[0];
-
-  const isLocalDev =
-    hostname === "localhost" ||
-    hostname === "127.0.0.1" ||
-    hostname.startsWith("192.168.");
-  const isHuggingFace =
-    hostname === "huggingface.co" || hostname.endsWith(".huggingface.co");
-
-  if (!isHuggingFace && !isLocalDev) {
-    const pathname = headersList.get("x-invoke-path") || "/deepsite";
-    redirect(`https://huggingface.co${pathname}`);
-  }
-
-  // const data = await getMe();
-
-  // Generate structured data
-  const structuredData = generateStructuredData("WebApplication", {
-    name: "DeepSite",
-    description: "Build websites with AI, no code required",
-    url: "https://huggingface.co/deepsite",
-  });
-
-  const organizationData = generateStructuredData("Organization", {
-    name: "DeepSite",
-    url: "https://huggingface.co/deepsite",
-  });
-
+  const data = await getMe();
   return (
     <html lang="en">
+      <Script
+        defer
+        data-domain="deepsite.hf.co"
+        src="https://plausible.io/js/script.js"
+      ></Script>
       <body
         className={`${inter.variable} ${ptSans.variable} antialiased bg-black dark h-[100dvh] overflow-hidden`}
       >
-        <script
-          type="application/ld+json"
-          dangerouslySetInnerHTML={{
-            __html: JSON.stringify(structuredData),
-          }}
-        />
-        <script
-          type="application/ld+json"
-          dangerouslySetInnerHTML={{
-            __html: JSON.stringify(organizationData),
-          }}
-        />
-        <Script
-          defer
-          data-domain="deepsite.hf.co"
-          src="https://plausible.io/js/script.js"
-        />
-        <IframeDetector />
         <Toaster richColors position="bottom-center" />
-        <TanstackContext>
-          <AppContext>
-            <LoginProvider>
-              <ProProvider>{children}</ProProvider>
-            </LoginProvider>
-          </AppContext>
-        </TanstackContext>
+        <TanstackProvider>
+          <AppContext me={data}>{children}</AppContext>
+        </TanstackProvider>
       </body>
     </html>
   );

app/new/page.tsx

@@ -1,14 +0,0 @@
-import { AppEditor } from "@/components/editor";
-import { Metadata } from "next";
-import { generateSEO } from "@/lib/seo";
-
-export const metadata: Metadata = generateSEO({
-  title: "Create New Project - DeepSite",
-  description:
-    "Start building your next website with AI. Create a new project on DeepSite and experience the power of AI-driven web development.",
-  path: "/new",
-});
-
-export default function NewProjectPage() {
-  return <AppEditor isNew />;
-}

app/projects/[namespace]/[repoId]/page.tsx

@@ -0,0 +1,40 @@
+import { cookies } from "next/headers";
+import { redirect } from "next/navigation";
+
+import { apiServer } from "@/lib/api";
+import MY_TOKEN_KEY from "@/lib/get-cookie-name";
+import { AppEditor } from "@/components/editor";
+
+async function getProject(namespace: string, repoId: string) {
+  // TODO replace with a server action
+  const cookieStore = await cookies();
+  const token = cookieStore.get(MY_TOKEN_KEY())?.value;
+  if (!token) return {};
+  try {
+    const { data } = await apiServer.get(
+      `/me/projects/${namespace}/${repoId}`,
+      {
+        headers: {
+          Authorization: `Bearer ${token}`,
+        },
+      }
+    );
+
+    return data.project;
+  } catch {
+    return {};
+  }
+}
+
+export default async function ProjectNamespacePage({
+  params,
+}: {
+  params: Promise<{ namespace: string; repoId: string }>;
+}) {
+  const { namespace, repoId } = await params;
+  const project = await getProject(namespace, repoId);
+  if (!project?.html) {
+    redirect("/projects");
+  }
+  return <AppEditor project={project} />;
+}

app/projects/new/page.tsx

@@ -0,0 +1,5 @@
+import { AppEditor } from "@/components/editor";
+
+export default function ProjectsNewPage() {
+  return <AppEditor />;
+}

app/sitemap.ts

@@ -1,28 +0,0 @@
-import { MetadataRoute } from 'next';
-
-export default function sitemap(): MetadataRoute.Sitemap {
-  const baseUrl = 'https://huggingface.co/deepsite';
-  
-  return [
-    {
-      url: baseUrl,
-      lastModified: new Date(),
-      changeFrequency: 'daily',
-      priority: 1,
-    },
-    {
-      url: `${baseUrl}/new`,
-      lastModified: new Date(),
-      changeFrequency: 'weekly',
-      priority: 0.8,
-    },
-    {
-      url: `${baseUrl}/auth`,
-      lastModified: new Date(),
-      changeFrequency: 'monthly',
-      priority: 0.5,
-    },
-    // Note: Dynamic project routes will be handled by Next.js automatically
-    // but you can add specific high-priority project pages here if needed
-  ];
-}

assets/globals.css

@@ -112,10 +112,6 @@
   --sidebar-ring: oklch(0.556 0 0);
 }
 
-body {
-  @apply scroll-smooth
-}
-
 @layer base {
   * {
     @apply border-border outline-ring/50;
@@ -131,7 +127,7 @@ body {
 .background__noisy {
   @apply bg-blend-normal pointer-events-none opacity-90;
   background-size: 25ww auto;
-  background-image: url("/deepsite/background_noisy.webp");
+  background-image: url("/background_noisy.webp");
   @apply fixed w-screen h-screen -z-1 top-0 left-0;
 }
 
@@ -148,233 +144,3 @@ body {
 .matched-line {
   @apply bg-sky-500/30;
 }
-
-/* Fast liquid deformation animations */
-@keyframes liquidBlob1 {
-  0%, 100% {
-    border-radius: 40% 60% 50% 50%;
-    transform: scaleX(1) scaleY(1) rotate(0deg);
-  }
-  12.5% {
-    border-radius: 20% 80% 70% 30%;
-    transform: scaleX(1.6) scaleY(0.4) rotate(25deg);
-  }
-  25% {
-    border-radius: 80% 20% 30% 70%;
-    transform: scaleX(0.5) scaleY(2.1) rotate(-15deg);
-  }
-  37.5% {
-    border-radius: 30% 70% 80% 20%;
-    transform: scaleX(1.8) scaleY(0.6) rotate(40deg);
-  }
-  50% {
-    border-radius: 70% 30% 20% 80%;
-    transform: scaleX(0.4) scaleY(1.9) rotate(-30deg);
-  }
-  62.5% {
-    border-radius: 25% 75% 60% 40%;
-    transform: scaleX(1.5) scaleY(0.7) rotate(55deg);
-  }
-  75% {
-    border-radius: 75% 25% 40% 60%;
-    transform: scaleX(0.6) scaleY(1.7) rotate(-10deg);
-  }
-  87.5% {
-    border-radius: 50% 50% 75% 25%;
-    transform: scaleX(1.3) scaleY(0.8) rotate(35deg);
-  }
-}
-
-@keyframes liquidBlob2 {
-  0%, 100% {
-    border-radius: 60% 40% 50% 50%;
-    transform: scaleX(1) scaleY(1) rotate(12deg);
-  }
-  16% {
-    border-radius: 15% 85% 60% 40%;
-    transform: scaleX(0.3) scaleY(2.3) rotate(50deg);
-  }
-  32% {
-    border-radius: 85% 15% 25% 75%;
-    transform: scaleX(2.0) scaleY(0.5) rotate(-20deg);
-  }
-  48% {
-    border-radius: 30% 70% 85% 15%;
-    transform: scaleX(0.4) scaleY(1.8) rotate(70deg);
-  }
-  64% {
-    border-radius: 70% 30% 15% 85%;
-    transform: scaleX(1.9) scaleY(0.6) rotate(-35deg);
-  }
-  80% {
-    border-radius: 40% 60% 70% 30%;
-    transform: scaleX(0.7) scaleY(1.6) rotate(45deg);
-  }
-}
-
-@keyframes liquidBlob3 {
-  0%, 100% {
-    border-radius: 50% 50% 40% 60%;
-    transform: scaleX(1) scaleY(1) rotate(0deg);
-  }
-  20% {
-    border-radius: 10% 90% 75% 25%;
-    transform: scaleX(2.2) scaleY(0.3) rotate(-45deg);
-  }
-  40% {
-    border-radius: 90% 10% 20% 80%;
-    transform: scaleX(0.4) scaleY(2.5) rotate(60deg);
-  }
-  60% {
-    border-radius: 25% 75% 90% 10%;
-    transform: scaleX(1.7) scaleY(0.5) rotate(-25deg);
-  }
-  80% {
-    border-radius: 75% 25% 10% 90%;
-    transform: scaleX(0.6) scaleY(2.0) rotate(80deg);
-  }
-}
-
-@keyframes liquidBlob4 {
-  0%, 100% {
-    border-radius: 45% 55% 50% 50%;
-    transform: scaleX(1) scaleY(1) rotate(-15deg);
-  }
-  14% {
-    border-radius: 90% 10% 65% 35%;
-    transform: scaleX(0.2) scaleY(2.8) rotate(35deg);
-  }
-  28% {
-    border-radius: 10% 90% 20% 80%;
-    transform: scaleX(2.4) scaleY(0.4) rotate(-50deg);
-  }
-  42% {
-    border-radius: 35% 65% 90% 10%;
-    transform: scaleX(0.3) scaleY(2.1) rotate(70deg);
-  }
-  56% {
-    border-radius: 80% 20% 10% 90%;
-    transform: scaleX(2.0) scaleY(0.5) rotate(-40deg);
-  }
-  70% {
-    border-radius: 20% 80% 55% 45%;
-    transform: scaleX(0.5) scaleY(1.9) rotate(55deg);
-  }
-  84% {
-    border-radius: 65% 35% 80% 20%;
-    transform: scaleX(1.6) scaleY(0.6) rotate(-25deg);
-  }
-}
-
-/* Fast flowing movement animations */
-@keyframes liquidFlow1 {
-  0%, 100% { transform: translate(0, 0); }
-  16% { transform: translate(60px, -40px); }
-  32% { transform: translate(-45px, -70px); }
-  48% { transform: translate(80px, 25px); }
-  64% { transform: translate(-30px, 60px); }
-  80% { transform: translate(50px, -20px); }
-}
-
-@keyframes liquidFlow2 {
-  0%, 100% { transform: translate(0, 0); }
-  20% { transform: translate(-70px, 50px); }
-  40% { transform: translate(90px, -30px); }
-  60% { transform: translate(-40px, -55px); }
-  80% { transform: translate(65px, 35px); }
-}
-
-@keyframes liquidFlow3 {
-  0%, 100% { transform: translate(0, 0); }
-  12% { transform: translate(-50px, -60px); }
-  24% { transform: translate(40px, -20px); }
-  36% { transform: translate(-30px, 70px); }
-  48% { transform: translate(70px, 20px); }
-  60% { transform: translate(-60px, -35px); }
-  72% { transform: translate(35px, 55px); }
-  84% { transform: translate(-25px, -45px); }
-}
-
-@keyframes liquidFlow4 {
-  0%, 100% { transform: translate(0, 0); }
-  14% { transform: translate(50px, 60px); }
-  28% { transform: translate(-80px, -40px); }
-  42% { transform: translate(30px, -90px); }
-  56% { transform: translate(-55px, 45px); }
-  70% { transform: translate(75px, -25px); }
-  84% { transform: translate(-35px, 65px); }
-}
-
-/* Light sweep animation for buttons */
-@keyframes lightSweep {
-  0% {
-    transform: translateX(-150%);
-    opacity: 0;
-  }
-  8% {
-    opacity: 0.3;
-  }
-  25% {
-    opacity: 0.8;
-  }
-  42% {
-    opacity: 0.3;
-  }
-  50% {
-    transform: translateX(150%);
-    opacity: 0;
-  }
-  58% {
-    opacity: 0.3;
-  }
-  75% {
-    opacity: 0.8;
-  }
-  92% {
-    opacity: 0.3;
-  }
-  100% {
-    transform: translateX(-150%);
-    opacity: 0;
-  }
-}
-
-.light-sweep {
-  position: relative;
-  overflow: hidden;
-}
-
-.light-sweep::before {
-  content: '';
-  position: absolute;
-  top: 0;
-  left: 0;
-  right: 0;
-  bottom: 0;
-  width: 300%;
-  background: linear-gradient(
-    90deg,
-    transparent 0%,
-    transparent 20%,
-    rgba(56, 189, 248, 0.1) 35%,
-    rgba(56, 189, 248, 0.2) 45%,
-    rgba(255, 255, 255, 0.2) 50%,
-    rgba(168, 85, 247, 0.2) 55%,
-    rgba(168, 85, 247, 0.1) 65%,
-    transparent 80%,
-    transparent 100%
-  );
-  animation: lightSweep 7s cubic-bezier(0.4, 0, 0.2, 1) infinite;
-  pointer-events: none;
-  z-index: 1;
-  filter: blur(1px);
-}
-
-.transparent-scroll {
-  scrollbar-width: none; /* Firefox */
-  -ms-overflow-style: none; /* IE and Edge */
-}
-
-.transparent-scroll::-webkit-scrollbar {
-  display: none; /* Chrome, Safari, Opera */
-}

components.json

@@ -5,7 +5,7 @@
   "tsx": true,
   "tailwind": {
     "config": "",
-    "css": "assets/globals.css",
+    "css": "app/globals.css",
     "baseColor": "neutral",
     "cssVariables": true,
     "prefix": ""

components/animated-blobs/index.tsx

@@ -1,34 +0,0 @@
-export function AnimatedBlobs() {
-  return (
-    <div className="absolute inset-0 pointer-events-none -z-[1]">
-      <div
-        className="w-full h-full bg-gradient-to-r from-purple-500 to-pink-500 opacity-10 blur-3xl"
-        style={{
-          animation:
-            "liquidBlob1 4s ease-in-out infinite, liquidFlow1 6s ease-in-out infinite",
-        }}
-      />
-      <div
-        className="w-2/3 h-3/4 bg-gradient-to-r from-blue-500 to-teal-500 opacity-24 blur-3xl absolute -top-20 right-10"
-        style={{
-          animation:
-            "liquidBlob2 5s ease-in-out infinite, liquidFlow2 7s ease-in-out infinite",
-        }}
-      />
-      <div
-        className="w-1/2 h-1/2 bg-gradient-to-r from-amber-500 to-rose-500 opacity-20 blur-3xl absolute bottom-0 left-10"
-        style={{
-          animation:
-            "liquidBlob3 3.5s ease-in-out infinite, liquidFlow3 8s ease-in-out infinite",
-        }}
-      />
-      <div
-        className="w-48 h-48 bg-gradient-to-r from-cyan-500 to-indigo-500 opacity-20 blur-3xl absolute top-1/3 right-1/3"
-        style={{
-          animation:
-            "liquidBlob4 4.5s ease-in-out infinite, liquidFlow4 6.5s ease-in-out infinite",
-        }}
-      />
-    </div>
-  );
-}

components/animated-text/index.tsx

@@ -1,123 +0,0 @@
-"use client";
-
-import { useState, useEffect } from "react";
-
-interface AnimatedTextProps {
-  className?: string;
-}
-
-export function AnimatedText({ className = "" }: AnimatedTextProps) {
-  const [displayText, setDisplayText] = useState("");
-  const [currentSuggestionIndex, setCurrentSuggestionIndex] = useState(0);
-  const [isTyping, setIsTyping] = useState(true);
-  const [showCursor, setShowCursor] = useState(true);
-  const [lastTypedIndex, setLastTypedIndex] = useState(-1);
-  const [animationComplete, setAnimationComplete] = useState(false);
-
-  // Randomize suggestions on each component mount
-  const [suggestions] = useState(() => {
-    const baseSuggestions = [
-      "create a stunning portfolio!",
-      "build a tic tac toe game!",
-      "design a website for my restaurant!",
-      "make a sleek landing page!",
-      "build an e-commerce store!",
-      "create a personal blog!",
-      "develop a modern dashboard!",
-      "design a company website!",
-      "build a todo app!",
-      "create an online gallery!",
-      "make a contact form!",
-      "build a weather app!",
-    ];
-
-    // Fisher-Yates shuffle algorithm
-    const shuffled = [...baseSuggestions];
-    for (let i = shuffled.length - 1; i > 0; i--) {
-      const j = Math.floor(Math.random() * (i + 1));
-      [shuffled[i], shuffled[j]] = [shuffled[j], shuffled[i]];
-    }
-
-    return shuffled;
-  });
-
-  useEffect(() => {
-    if (animationComplete) return;
-
-    let timeout: NodeJS.Timeout;
-
-    const typeText = () => {
-      const currentSuggestion = suggestions[currentSuggestionIndex];
-
-      if (isTyping) {
-        if (displayText.length < currentSuggestion.length) {
-          setDisplayText(currentSuggestion.slice(0, displayText.length + 1));
-          setLastTypedIndex(displayText.length);
-          timeout = setTimeout(typeText, 80);
-        } else {
-          // Finished typing, wait then start erasing
-          setLastTypedIndex(-1);
-          timeout = setTimeout(() => {
-            setIsTyping(false);
-          }, 2000);
-        }
-      }
-    };
-
-    timeout = setTimeout(typeText, 100);
-    return () => clearTimeout(timeout);
-  }, [
-    displayText,
-    currentSuggestionIndex,
-    isTyping,
-    suggestions,
-    animationComplete,
-  ]);
-
-  // Cursor blinking effect
-  useEffect(() => {
-    if (animationComplete) {
-      setShowCursor(false);
-      return;
-    }
-
-    const cursorInterval = setInterval(() => {
-      setShowCursor((prev) => !prev);
-    }, 600);
-
-    return () => clearInterval(cursorInterval);
-  }, [animationComplete]);
-
-  useEffect(() => {
-    if (lastTypedIndex >= 0) {
-      const timeout = setTimeout(() => {
-        setLastTypedIndex(-1);
-      }, 400);
-
-      return () => clearTimeout(timeout);
-    }
-  }, [lastTypedIndex]);
-
-  return (
-    <p className={`font-mono ${className}`}>
-      Hey DeepSite,&nbsp;
-      {displayText.split("").map((char, index) => (
-        <span
-          key={`${currentSuggestionIndex}-${index}`}
-          className={`transition-colors duration-300 ${
-            index === lastTypedIndex ? "text-neutral-100" : ""
-          }`}
-        >
-          {char}
-        </span>
-      ))}
-      <span
-        className={`${
-          showCursor ? "opacity-100" : "opacity-0"
-        } transition-opacity`}
-      >
-        |
-      </span>
-    </p>
-  );
-}

components/contexts/app-context.tsx

@@ -1,41 +1,42 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 "use client";
-import { useMount } from "react-use";
-import { toast } from "sonner";
-import { usePathname, useRouter } from "next/navigation";
 
 import { useUser } from "@/hooks/useUser";
-import { ProjectType, User } from "@/types";
+import { usePathname, useRouter } from "next/navigation";
+import { useMount } from "react-use";
+import { UserContext } from "@/components/contexts/user-context";
+import { User } from "@/types";
+import { toast } from "sonner";
 import { useBroadcastChannel } from "@/lib/useBroadcastChannel";
 
 export default function AppContext({
   children,
-}: // me: initialData,
-{
+  me: initialData,
+}: {
   children: React.ReactNode;
-  // me?: {
-  //   user: User | null;
-  //   projects: ProjectType[];
-  //   errCode: number | null;
-  // };
+  me?: {
+    user: User | null;
+    errCode: number | null;
+  };
 }) {
-  const { loginFromCode, user, logout, loading, errCode } = useUser();
+  const { loginFromCode, user, logout, loading, errCode } =
+    useUser(initialData);
   const pathname = usePathname();
   const router = useRouter();
 
-  // useMount(() => {
-  //   if (!initialData?.user && !user) {
-  //     if ([401, 403].includes(errCode as number)) {
-  //       logout();
-  //     } else if (pathname.includes("/spaces")) {
-  //       if (errCode) {
-  //         toast.error("An error occured while trying to log in");
-  //       }
-  //       // If we did not manage to log in (probs because api is down), we simply redirect to the home page
-  //       router.push("/");
-  //     }
-  //   }
-  // });
+  useMount(() => {
+    if (!initialData?.user && !user) {
+      if ([401, 403].includes(errCode as number)) {
+        logout();
+      } else if (pathname.includes("/spaces")) {
+        if (errCode) {
+          toast.error("An error occured while trying to log in");
+        }
+        // If we did not manage to log in (probs because api is down), we simply redirect to the home page
+        router.push("/");
+      }
+    }
+  });
 
   const events: any = {};
 
@@ -48,5 +49,9 @@ export default function AppContext({
     }
   });
 
-  return children;
+  return (
+    <UserContext value={{ user, loading, logout } as any}>
+      {children}
+    </UserContext>
+  );
 }

components/contexts/login-context.tsx

@@ -1,62 +0,0 @@
-"use client";
-
-import React, { createContext, useContext, useState, ReactNode } from "react";
-import { LoginModal } from "@/components/login-modal";
-import { Page } from "@/types";
-
-interface LoginContextType {
-  isOpen: boolean;
-  openLoginModal: (options?: LoginModalOptions) => void;
-  closeLoginModal: () => void;
-}
-
-interface LoginModalOptions {
-  pages?: Page[];
-  title?: string;
-  prompt?: string;
-  description?: string;
-}
-
-const LoginContext = createContext<LoginContextType | undefined>(undefined);
-
-export function LoginProvider({ children }: { children: ReactNode }) {
-  const [isOpen, setIsOpen] = useState(false);
-  const [modalOptions, setModalOptions] = useState<LoginModalOptions>({});
-
-  const openLoginModal = (options: LoginModalOptions = {}) => {
-    setModalOptions(options);
-    setIsOpen(true);
-  };
-
-  const closeLoginModal = () => {
-    setIsOpen(false);
-    setModalOptions({});
-  };
-
-  const value = {
-    isOpen,
-    openLoginModal,
-    closeLoginModal,
-  };
-
-  return (
-    <LoginContext.Provider value={value}>
-      {children}
-      <LoginModal
-        open={isOpen}
-        onClose={setIsOpen}
-        title={modalOptions.title}
-        prompt={modalOptions.prompt}
-        description={modalOptions.description}
-      />
-    </LoginContext.Provider>
-  );
-}
-
-export function useLoginModal() {
-  const context = useContext(LoginContext);
-  if (context === undefined) {
-    throw new Error("useLoginModal must be used within a LoginProvider");
-  }
-  return context;
-}

components/contexts/pro-context.tsx

@@ -1,48 +0,0 @@
-"use client";
-
-import React, { createContext, useContext, useState, ReactNode } from "react";
-import { ProModal } from "@/components/pro-modal";
-import { Page } from "@/types";
-import { useEditor } from "@/hooks/useEditor";
-
-interface ProContextType {
-  isOpen: boolean;
-  openProModal: (pages: Page[]) => void;
-  closeProModal: () => void;
-}
-
-const ProContext = createContext<ProContextType | undefined>(undefined);
-
-export function ProProvider({ children }: { children: ReactNode }) {
-  const [isOpen, setIsOpen] = useState(false);
-  const { pages } = useEditor();
-
-  const openProModal = () => {
-    setIsOpen(true);
-  };
-
-  const closeProModal = () => {
-    setIsOpen(false);
-  };
-
-  const value = {
-    isOpen,
-    openProModal,
-    closeProModal,
-  };
-
-  return (
-    <ProContext.Provider value={value}>
-      {children}
-      <ProModal open={isOpen} onClose={setIsOpen} pages={pages} />
-    </ProContext.Provider>
-  );
-}
-
-export function useProModal() {
-  const context = useContext(ProContext);
-  if (context === undefined) {
-    throw new Error("useProModal must be used within a ProProvider");
-  }
-  return context;
-}

components/discord-promo-modal/index.tsx

@@ -1,225 +0,0 @@
-"use client";
-
-import { useEffect, useState } from "react";
-import { useLocalStorage } from "react-use";
-import Image from "next/image";
-import { Button } from "@/components/ui/button";
-import { Dialog, DialogContent, DialogTitle } from "@/components/ui/dialog";
-import { DiscordIcon } from "@/components/icons/discord";
-import Logo from "@/assets/logo.svg";
-
-const DISCORD_PROMO_KEY = "discord-promo-dismissed";
-const DISCORD_URL = "https://discord.gg/KpanwM3vXa";
-
-const Sparkle = ({
-  size = "w-3 h-3",
-  delay = "0s",
-  top = "20%",
-  left = "20%",
-}: {
-  size?: string;
-  delay?: string;
-  top?: string;
-  left?: string;
-}) => (
-  <div
-    className={`absolute ${size}`}
-    style={{ top, left, animationDelay: delay }}
-  >
-    <svg
-      viewBox="0 0 24 24"
-      fill="none"
-      xmlns="http://www.w3.org/2000/svg"
-      className="w-full h-full animate-sparkle"
-    >
-      <path
-        d="M12 0L13.5 8.5L22 10L13.5 11.5L12 20L10.5 11.5L2 10L10.5 8.5L12 0Z"
-        fill="url(#sparkle-gradient)"
-      />
-      <defs>
-        <linearGradient id="sparkle-gradient" x1="2" y1="10" x2="22" y2="10">
-          <stop offset="0%" stopColor="#818cf8" />
-          <stop offset="100%" stopColor="#a5b4fc" />
-        </linearGradient>
-      </defs>
-    </svg>
-  </div>
-);
-
-export const DiscordPromoModal = () => {
-  const [open, setOpen] = useState(false);
-  const [dismissed, setDismissed] = useLocalStorage<boolean>(
-    DISCORD_PROMO_KEY,
-    false
-  );
-
-  useEffect(() => {
-    const cookieDismissed = document.cookie
-      .split("; ")
-      .find((row) => row.startsWith(`${DISCORD_PROMO_KEY}=`))
-      ?.split("=")[1];
-
-    if (dismissed || cookieDismissed === "true") {
-      return;
-    }
-
-    const timer = setTimeout(() => {
-      setOpen(true);
-    }, 60000);
-
-    return () => clearTimeout(timer);
-  }, [dismissed]);
-
-  const handleClose = () => {
-    setOpen(false);
-    setDismissed(true);
-
-    const expiryDate = new Date();
-    expiryDate.setDate(expiryDate.getDate() + 5);
-    document.cookie = `${DISCORD_PROMO_KEY}=true; expires=${expiryDate.toUTCString()}; path=/; SameSite=Lax`;
-  };
-
-  const handleJoinDiscord = () => {
-    window.open(DISCORD_URL, "_blank");
-    handleClose();
-  };
-
-  return (
-    <Dialog open={open} onOpenChange={handleClose}>
-      <DialogContent
-        className="sm:max-w-[480px] lg:!p-0 !rounded-3xl !bg-gradient-to-b !from-indigo-950/40 !via-neutral-900 !to-neutral-900 !border !border-neutral-800 overflow-hidden"
-        showCloseButton={true}
-      >
-        <DialogTitle className="hidden" />
-
-        <div className="relative">
-          <div className="absolute inset-0 bg-gradient-to-br from-indigo-500/10 via-indigo-500/5 to-transparent pointer-events-none" />
-
-          <div className="absolute inset-x-0 top-0 h-48 overflow-hidden pointer-events-none">
-            <Sparkle size="w-2 h-2" delay="0s" top="15%" left="15%" />
-            <Sparkle size="w-3 h-3" delay="0.5s" top="25%" left="75%" />
-            <Sparkle size="w-2 h-2" delay="1s" top="35%" left="20%" />
-            <Sparkle size="w-4 h-4" delay="1.5s" top="10%" left="80%" />
-            <Sparkle size="w-2 h-2" delay="2s" top="30%" left="85%" />
-          </div>
-
-          <div className="relative pt-12 pb-8">
-            <div className="relative z-10 flex justify-center">
-              <div className="relative">
-                <div className="absolute inset-0 bg-gradient-to-br from-indigo-400 via-indigo-500 to-indigo-600 rounded-full blur-md opacity-50" />
-                <div className="relative w-32 h-32 rounded-full bg-gradient-to-br from-neutral-900 via-neutral-800 to-neutral-900 p-1 shadow-2xl">
-                  <div className="w-full h-full rounded-full bg-neutral-900 flex items-center justify-center overflow-hidden">
-                    <div className="relative w-20 h-20 bg-gradient-to-br from-indigo-500 to-indigo-600 rounded-full flex items-center justify-center">
-                      <DiscordIcon className="w-12 h-12 text-white" />
-                    </div>
-                  </div>
-                </div>
-
-                <div className="absolute -bottom-2 -right-2 bg-gradient-to-br from-indigo-500 to-indigo-600 rounded-full p-0.5 shadow-xl border-2 border-neutral-900">
-                  <div className="w-10 h-10 bg-neutral-900 rounded-full flex items-center justify-center">
-                    <Image
-                      src={Logo}
-                      alt="DeepSite"
-                      width={20}
-                      height={20}
-                      className="w-5 h-5"
-                    />
-                  </div>
-                </div>
-              </div>
-            </div>
-          </div>
-
-          <main className="px-8 pb-8 pt-4">
-            <div className="text-center mb-6">
-              <h2 className="text-2xl font-bold text-white mb-2">
-                Ready to level up your DeepSite experience?
-              </h2>
-              <p className="text-neutral-400 text-sm">
-                Get help, share your projects and ask for suggestions!
-              </p>
-            </div>
-
-            <div className="flex flex-col gap-3 mb-6">
-              {[
-                "Get exclusive preview to new features",
-                "Share your projects and get feedback",
-                "Priority support from the team",
-                "Enjoy real-time updates",
-              ].map((benefit, index) => (
-                <div
-                  key={index}
-                  className="flex items-start gap-3 text-neutral-200"
-                  style={{
-                    animation: `fadeIn 0.4s ease-out ${index * 0.1}s both`,
-                  }}
-                >
-                  <div className="flex-shrink-0 w-5 h-5 rounded-full bg-gradient-to-br from-indigo-500 to-indigo-600 flex items-center justify-center mt-0.5">
-                    <svg
-                      className="w-3 h-3 text-white"
-                      fill="none"
-                      viewBox="0 0 24 24"
-                      stroke="currentColor"
-                    >
-                      <path
-                        strokeLinecap="round"
-                        strokeLinejoin="round"
-                        strokeWidth={3}
-                        d="M5 13l4 4L19 7"
-                      />
-                    </svg>
-                  </div>
-                  <span className="text-sm leading-6">{benefit}</span>
-                </div>
-              ))}
-            </div>
-
-            {/* CTA Button */}
-            <div className="flex flex-col gap-3 w-full">
-              <Button
-                onClick={handleJoinDiscord}
-                className="w-full !h-12 !text-base font-semibold !bg-gradient-to-r !from-indigo-500 !to-indigo-600 hover:!from-indigo-600 hover:!to-indigo-700 !text-white !border-0 transform hover:scale-[1.02] transition-all duration-200 shadow-lg shadow-indigo-500/25"
-              >
-                <DiscordIcon className="w-5 h-5 mr-2" />
-                Join Discord Community
-              </Button>
-
-              <p className="text-center text-xs text-neutral-500">
-                Free to join. Connect instantly.
-              </p>
-            </div>
-          </main>
-        </div>
-
-        <style jsx>{`
-          @keyframes fadeIn {
-            from {
-              opacity: 0;
-              transform: translateY(5px);
-            }
-            to {
-              opacity: 1;
-              transform: translateY(0);
-            }
-          }
-
-          @keyframes sparkle {
-            0%,
-            100% {
-              opacity: 0;
-              transform: scale(0) rotate(0deg);
-            }
-            50% {
-              opacity: 1;
-              transform: scale(1) rotate(180deg);
-            }
-          }
-
-          :global(.animate-sparkle) {
-            animation: sparkle 2s ease-in-out infinite;
-          }
-        `}</style>
-      </DialogContent>
-    </Dialog>
-  );
-};

components/editor/ask-ai/context.tsx

@@ -1,128 +0,0 @@
-import { useState, useMemo } from "react";
-import { FileCode, FileText, Braces, AtSign } from "lucide-react";
-
-import { Button } from "@/components/ui/button";
-import { useEditor } from "@/hooks/useEditor";
-import { useAi } from "@/hooks/useAi";
-import {
-  Popover,
-  PopoverContent,
-  PopoverTrigger,
-} from "@/components/ui/popover";
-import classNames from "classnames";
-
-export const Context = () => {
-  const { pages, currentPage, globalEditorLoading } = useEditor();
-  const { contextFile, setContextFile, globalAiLoading } = useAi();
-  const [open, setOpen] = useState(false);
-
-  const selectedFile = contextFile || null;
-
-  const getFileIcon = (filePath: string, size = "size-3.5") => {
-    if (filePath.endsWith(".css")) {
-      return <Braces className={size} />;
-    } else if (filePath.endsWith(".js")) {
-      return <FileCode className={size} />;
-    } else if (filePath.endsWith(".json")) {
-      return <Braces className={size} />;
-    } else {
-      return <FileText className={size} />;
-    }
-  };
-
-  const buttonContent = useMemo(() => {
-    if (selectedFile) {
-      return (
-        <>
-          <span className="truncate max-w-[120px]">{selectedFile}</span>
-        </>
-      );
-    }
-    return <>Add Context</>;
-  }, [selectedFile]);
-
-  return (
-    <Popover open={open} onOpenChange={setOpen}>
-      <PopoverTrigger asChild>
-        <Button
-          size="xs"
-          variant={open ? "default" : "outline"}
-          className={classNames("!rounded-md", {
-            "!bg-blue-500/10 !border-blue-500/30 !text-blue-400":
-              selectedFile && selectedFile.endsWith(".css"),
-            "!bg-orange-500/10 !border-orange-500/30 !text-orange-400":
-              selectedFile && selectedFile.endsWith(".html"),
-            "!bg-amber-500/10 !border-amber-500/30 !text-amber-400":
-              selectedFile && selectedFile.endsWith(".js"),
-            "!bg-yellow-500/10 !border-yellow-500/30 !text-yellow-400":
-              selectedFile && selectedFile.endsWith(".json"),
-          })}
-          disabled={
-            globalAiLoading || globalEditorLoading || pages.length === 0
-          }
-        >
-          <AtSign className="size-3.5" />
-
-          {buttonContent}
-        </Button>
-      </PopoverTrigger>
-      <PopoverContent
-        align="start"
-        className="w-64 !bg-neutral-900 !border-neutral-800 !p-0 !rounded-2xl overflow-hidden"
-      >
-        <header className="flex items-center justify-center text-xs px-2 py-2.5 border-b gap-2 bg-neutral-950 border-neutral-800 font-semibold text-neutral-200">
-          Select a file to send as context
-        </header>
-        <main className="space-y-1 p-2">
-          <div className="max-h-[200px] overflow-y-auto space-y-0.5">
-            {pages.length === 0 ? (
-              <div className="px-2 py-2 text-xs text-neutral-500">
-                No files available
-              </div>
-            ) : (
-              <>
-                <button
-                  onClick={() => {
-                    setContextFile(null);
-                    setOpen(false);
-                  }}
-                  className={`cursor-pointer w-full px-2 py-1.5 text-xs text-left rounded-md hover:bg-neutral-800 transition-colors ${
-                    !selectedFile
-                      ? "bg-neutral-800 text-neutral-200 font-medium"
-                      : "text-neutral-400 hover:text-neutral-200"
-                  }`}
-                >
-                  All files (default)
-                </button>
-                {pages.map((page) => (
-                  <button
-                    key={page.path}
-                    onClick={() => {
-                      setContextFile(page.path);
-                      setOpen(false);
-                    }}
-                    className={`cursor-pointer w-full px-2 py-1.5 text-xs text-left rounded-md hover:bg-neutral-800 transition-colors flex items-center gap-1.5 ${
-                      selectedFile === page.path
-                        ? "bg-neutral-800 text-neutral-200 font-medium"
-                        : "text-neutral-400 hover:text-neutral-200"
-                    }`}
-                  >
-                    <span className="shrink-0">
-                      {getFileIcon(page.path, "size-3")}
-                    </span>
-                    <span className="truncate flex-1">{page.path}</span>
-                    {page.path === currentPage && (
-                      <span className="text-[10px] text-neutral-500 shrink-0">
-                        (current)
-                      </span>
-                    )}
-                  </button>
-                ))}
-              </>
-            )}
-          </div>
-        </main>
-      </PopoverContent>
-    </Popover>
-  );
-};

components/editor/ask-ai/fake-ask.tsx

@@ -1,97 +0,0 @@
-import { useState } from "react";
-import { useLocalStorage } from "react-use";
-import { ArrowUp, Dice6 } from "lucide-react";
-import { useRouter } from "next/navigation";
-
-import { Button } from "@/components/ui/button";
-import { PromptBuilder } from "./prompt-builder";
-import { EnhancedSettings } from "@/types";
-import { Settings } from "./settings";
-import classNames from "classnames";
-import { PROMPTS_FOR_AI } from "@/lib/prompts";
-
-export const FakeAskAi = () => {
-  const router = useRouter();
-  const [prompt, setPrompt] = useState("");
-  const [openProvider, setOpenProvider] = useState(false);
-  const [enhancedSettings, setEnhancedSettings, removeEnhancedSettings] =
-    useLocalStorage<EnhancedSettings>("deepsite-enhancedSettings", {
-      isActive: true,
-      primaryColor: undefined,
-      secondaryColor: undefined,
-      theme: undefined,
-    });
-  const [, setPromptStorage] = useLocalStorage("prompt", "");
-  const [randomPromptLoading, setRandomPromptLoading] = useState(false);
-
-  const callAi = async () => {
-    setPromptStorage(prompt);
-    router.push("/new");
-  };
-
-  const randomPrompt = () => {
-    setRandomPromptLoading(true);
-    setTimeout(() => {
-      setPrompt(
-        PROMPTS_FOR_AI[Math.floor(Math.random() * PROMPTS_FOR_AI.length)]
-      );
-      setRandomPromptLoading(false);
-    }, 400);
-  };
-
-  return (
-    <div className="p-3 w-full max-w-xl mx-auto">
-      <div className="relative bg-neutral-800 border border-neutral-700 rounded-2xl ring-[4px] focus-within:ring-neutral-500/30 focus-within:border-neutral-600 ring-transparent z-20 w-full group">
-        <div className="w-full relative flex items-start justify-between pr-4 pt-4">
-          <textarea
-            className="w-full bg-transparent text-sm outline-none text-white placeholder:text-neutral-400 px-4 pb-4 resize-none"
-            placeholder="Ask DeepSite anything..."
-            value={prompt}
-            onChange={(e) => setPrompt(e.target.value)}
-            onKeyDown={(e) => {
-              if (e.key === "Enter" && !e.shiftKey) {
-                callAi();
-              }
-            }}
-          />
-          <Button
-            size="iconXs"
-            variant="outline"
-            className="!rounded-md"
-            onClick={() => randomPrompt()}
-          >
-            <Dice6
-              className={classNames("size-4", {
-                "animate-spin animation-duration-500": randomPromptLoading,
-              })}
-            />
-          </Button>
-        </div>
-        <div className="flex items-center justify-between gap-2 px-4 pb-3 mt-2">
-          <div className="flex-1 flex items-center justify-start gap-1.5 flex-wrap">
-            <PromptBuilder
-              enhancedSettings={enhancedSettings!}
-              setEnhancedSettings={setEnhancedSettings}
-            />
-            <Settings
-              open={openProvider}
-              isFollowUp={false}
-              error=""
-              onClose={setOpenProvider}
-            />
-          </div>
-          <div className="flex items-center justify-end gap-2">
-            <Button
-              size="iconXs"
-              variant="outline"
-              className="!rounded-md"
-              onClick={() => callAi()}
-            >
-              <ArrowUp className="size-4" />
-            </Button>
-          </div>
-        </div>
-      </div>
-    </div>
-  );
-};

components/editor/ask-ai/follow-up-tooltip.tsx

@@ -0,0 +1,36 @@
+import {
+  Popover,
+  PopoverContent,
+  PopoverTrigger,
+} from "@/components/ui/popover";
+import { Info } from "lucide-react";
+
+export const FollowUpTooltip = () => {
+  return (
+    <Popover>
+      <PopoverTrigger asChild>
+        <Info className="size-3 text-neutral-300 cursor-pointer" />
+      </PopoverTrigger>
+      <PopoverContent
+        align="start"
+        className="!rounded-2xl !p-0 min-w-xs text-center overflow-hidden"
+      >
+        <header className="bg-neutral-950 px-4 py-3 border-b border-neutral-700/70">
+          <p className="text-base text-neutral-200 font-semibold">
+            ⚡ Faster, Smarter Updates
+          </p>
+        </header>
+        <main className="p-4">
+          <p className="text-neutral-300 text-sm">
+            Using the Diff-Patch system, allow DeepSite to intelligently update
+            your project without rewritting the entire codebase.
+          </p>
+          <p className="text-neutral-500 text-sm mt-2">
+            This means faster updates, less data usage, and a more efficient
+            development process.
+          </p>
+        </main>
+      </PopoverContent>
+    </Popover>
+  );
+};

components/editor/ask-ai/index.tsx

@@ -1,144 +1,276 @@
-import { useRef, useState } from "react";
+"use client";
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import { useState, useRef, useMemo } from "react";
 import classNames from "classnames";
-import { ArrowUp, ChevronDown, CircleStop, Dice6 } from "lucide-react";
-import { useLocalStorage, useUpdateEffect, useMount } from "react-use";
 import { toast } from "sonner";
+import { useLocalStorage, useUpdateEffect } from "react-use";
+import { ArrowUp, ChevronDown, Crosshair } from "lucide-react";
+import { FaStopCircle } from "react-icons/fa";
 
-import { useAi } from "@/hooks/useAi";
-import { useEditor } from "@/hooks/useEditor";
-import { EnhancedSettings, Project } from "@/types";
-import { SelectedFiles } from "@/components/editor/ask-ai/selected-files";
-import { SelectedHtmlElement } from "@/components/editor/ask-ai/selected-html-element";
-import { AiLoading } from "@/components/editor/ask-ai/loading";
+import ProModal from "@/components/pro-modal";
 import { Button } from "@/components/ui/button";
-import { Uploader } from "@/components/editor/ask-ai/uploader";
+import { MODELS } from "@/lib/providers";
+import { HtmlHistory } from "@/types";
+import { InviteFriends } from "@/components/invite-friends";
+import { Settings } from "@/components/editor/ask-ai/settings";
+import { LoginModal } from "@/components/login-modal";
 import { ReImagine } from "@/components/editor/ask-ai/re-imagine";
-import { Selector } from "@/components/editor/ask-ai/selector";
-import { PromptBuilder } from "@/components/editor/ask-ai/prompt-builder";
-import { Context } from "@/components/editor/ask-ai/context";
-import { useUser } from "@/hooks/useUser";
-import { useLoginModal } from "@/components/contexts/login-context";
-import { Settings } from "./settings";
-import { useProModal } from "@/components/contexts/pro-context";
-import { MAX_FREE_PROJECTS } from "@/lib/utils";
-import { PROMPTS_FOR_AI } from "@/lib/prompts";
-import { SelectedRedesignUrl } from "./selected-redesign-url";
+import Loading from "@/components/loading";
+import { Checkbox } from "@/components/ui/checkbox";
+import { Tooltip, TooltipTrigger } from "@/components/ui/tooltip";
+import { TooltipContent } from "@radix-ui/react-tooltip";
+import { SelectedHtmlElement } from "./selected-html-element";
+import { FollowUpTooltip } from "./follow-up-tooltip";
+import { isTheSameHtml } from "@/lib/compare-html-diff";
 
-export const AskAi = ({
-  project,
-  isNew,
+export function AskAI({
+  html,
+  setHtml,
   onScrollToBottom,
+  isAiWorking,
+  setisAiWorking,
+  isEditableModeEnabled = false,
+  selectedElement,
+  setSelectedElement,
+  setIsEditableModeEnabled,
+  onNewPrompt,
+  onSuccess,
 }: {
-  project?: Project;
-  files?: string[];
-  isNew?: boolean;
-  onScrollToBottom?: () => void;
-}) => {
-  const { user, projects } = useUser();
-  const { isSameHtml, isUploading, pages, isLoadingProject } = useEditor();
-  const {
-    isAiWorking,
-    isThinking,
-    thinkingContent,
-    selectedFiles,
-    setSelectedFiles,
-    selectedElement,
-    setSelectedElement,
-    callAiNewProject,
-    callAiFollowUp,
-    audio: hookAudio,
-    cancelRequest,
-  } = useAi(onScrollToBottom);
-  const { openLoginModal } = useLoginModal();
-  const { openProModal } = useProModal();
+  html: string;
+  setHtml: (html: string) => void;
+  onScrollToBottom: () => void;
+  isAiWorking: boolean;
+  onNewPrompt: (prompt: string) => void;
+  htmlHistory?: HtmlHistory[];
+  setisAiWorking: React.Dispatch<React.SetStateAction<boolean>>;
+  onSuccess: (h: string, p: string, n?: number[][]) => void;
+  isEditableModeEnabled: boolean;
+  setIsEditableModeEnabled: React.Dispatch<React.SetStateAction<boolean>>;
+  selectedElement?: HTMLElement | null;
+  setSelectedElement: React.Dispatch<React.SetStateAction<HTMLElement | null>>;
+}) {
+  const refThink = useRef<HTMLDivElement | null>(null);
+  const audio = useRef<HTMLAudioElement | null>(null);
+
+  const [open, setOpen] = useState(false);
+  const [prompt, setPrompt] = useState("");
+  const [hasAsked, setHasAsked] = useState(false);
+  const [previousPrompt, setPreviousPrompt] = useState("");
+  const [provider, setProvider] = useLocalStorage("provider", "auto");
+  const [model, setModel] = useLocalStorage("model", MODELS[0].value);
   const [openProvider, setOpenProvider] = useState(false);
   const [providerError, setProviderError] = useState("");
-  const [redesignData, setRedesignData] = useState<
-    undefined | { markdown: string; url: string }
-  >(undefined);
-  const refThink = useRef<HTMLDivElement>(null);
-
-  const [enhancedSettings, setEnhancedSettings, removeEnhancedSettings] =
-    useLocalStorage<EnhancedSettings>("deepsite-enhancedSettings", {
-      isActive: false,
-      primaryColor: undefined,
-      secondaryColor: undefined,
-      theme: undefined,
-    });
-  const [promptStorage, , removePromptStorage] = useLocalStorage("prompt", "");
-
-  const [isFollowUp, setIsFollowUp] = useState(true);
-  const [prompt, setPrompt] = useState(
-    promptStorage && promptStorage.trim() !== "" ? promptStorage : ""
-  );
+  const [openProModal, setOpenProModal] = useState(false);
+  const [think, setThink] = useState<string | undefined>(undefined);
   const [openThink, setOpenThink] = useState(false);
-  const [randomPromptLoading, setRandomPromptLoading] = useState(false);
+  const [isThinking, setIsThinking] = useState(true);
+  const [controller, setController] = useState<AbortController | null>(null);
+  const [isFollowUp, setIsFollowUp] = useState(true);
 
-  useMount(() => {
-    if (promptStorage && promptStorage.trim() !== "") {
-      callAi();
-    }
-  });
+  const selectedModel = useMemo(() => {
+    return MODELS.find((m: { value: string }) => m.value === model);
+  }, [model]);
 
-  const callAi = async (redesignMarkdown?: string | undefined) => {
-    removePromptStorage();
-    if (user && !user.isPro && projects.length >= MAX_FREE_PROJECTS)
-      return openProModal([]);
+  const callAi = async (redesignMarkdown?: string) => {
     if (isAiWorking) return;
     if (!redesignMarkdown && !prompt.trim()) return;
+    setisAiWorking(true);
+    setProviderError("");
+    setThink("");
+    setOpenThink(false);
+    setIsThinking(true);
 
-    if (isFollowUp && !redesignMarkdown && !isSameHtml) {
-      if (!user) return openLoginModal({ prompt });
-      const result = await callAiFollowUp(prompt, enhancedSettings, isNew);
+    let contentResponse = "";
+    let thinkResponse = "";
+    let lastRenderTime = 0;
 
-      if (result?.error) {
-        handleError(result.error, result.message);
-        return;
-      }
+    const abortController = new AbortController();
+    setController(abortController);
+    try {
+      onNewPrompt(prompt);
+      if (isFollowUp && !redesignMarkdown && !isSameHtml) {
+        const selectedElementHtml = selectedElement
+          ? selectedElement.outerHTML
+          : "";
+        const request = await fetch("/api/ask-ai", {
+          method: "PUT",
+          body: JSON.stringify({
+            prompt,
+            provider,
+            previousPrompt,
+            model,
+            html,
+            selectedElementHtml,
+          }),
+          headers: {
+            "Content-Type": "application/json",
+            "x-forwarded-for": window.location.hostname,
+          },
+          signal: abortController.signal,
+        });
+        if (request && request.body) {
+          const res = await request.json();
+          if (!request.ok) {
+            if (res.openLogin) {
+              setOpen(true);
+            } else if (res.openSelectProvider) {
+              setOpenProvider(true);
+              setProviderError(res.message);
+            } else if (res.openProModal) {
+              setOpenProModal(true);
+            } else {
+              toast.error(res.message);
+            }
+            setisAiWorking(false);
+            return;
+          }
+          setHtml(res.html);
+          toast.success("AI responded successfully");
+          setPreviousPrompt(prompt);
+          setPrompt("");
+          setisAiWorking(false);
+          onSuccess(res.html, prompt, res.updatedLines);
+          if (audio.current) audio.current.play();
+        }
+      } else {
+        const request = await fetch("/api/ask-ai", {
+          method: "POST",
+          body: JSON.stringify({
+            prompt,
+            provider,
+            model,
+            html: isSameHtml ? "" : html,
+            redesignMarkdown,
+          }),
+          headers: {
+            "Content-Type": "application/json",
+            "x-forwarded-for": window.location.hostname,
+          },
+          signal: abortController.signal,
+        });
+        if (request && request.body) {
+          const reader = request.body.getReader();
+          const decoder = new TextDecoder("utf-8");
+          const selectedModel = MODELS.find(
+            (m: { value: string }) => m.value === model
+          );
+          let contentThink: string | undefined = undefined;
+          const read = async () => {
+            const { done, value } = await reader.read();
+            if (done) {
+              const isJson =
+                contentResponse.trim().startsWith("{") &&
+                contentResponse.trim().endsWith("}");
+              const jsonResponse = isJson ? JSON.parse(contentResponse) : null;
+              if (jsonResponse && !jsonResponse.ok) {
+                if (jsonResponse.openLogin) {
+                  setOpen(true);
+                } else if (jsonResponse.openSelectProvider) {
+                  setOpenProvider(true);
+                  setProviderError(jsonResponse.message);
+                } else if (jsonResponse.openProModal) {
+                  setOpenProModal(true);
+                } else {
+                  toast.error(jsonResponse.message);
+                }
+                setisAiWorking(false);
+                return;
+              }
 
-      if (result?.success) {
-        setPrompt("");
-      }
-    } else {
-      const result = await callAiNewProject(
-        prompt,
-        enhancedSettings,
-        redesignMarkdown,
-        !!user,
-        user?.name
-      );
+              toast.success("AI responded successfully");
+              setPreviousPrompt(prompt);
+              setPrompt("");
+              setisAiWorking(false);
+              setHasAsked(true);
+              if (selectedModel?.isThinker) {
+                setModel(MODELS[0].value);
+              }
+              if (audio.current) audio.current.play();
 
-      if (result?.error) {
-        handleError(result.error, result.message);
-        return;
-      }
+              // Now we have the complete HTML including </html>, so set it to be sure
+              const finalDoc = contentResponse.match(
+                /<!DOCTYPE html>[\s\S]*<\/html>/
+              )?.[0];
+              if (finalDoc) {
+                setHtml(finalDoc);
+              }
+              onSuccess(finalDoc ?? contentResponse, prompt);
+
+              return;
+            }
 
-      if (result?.success) {
-        setPrompt("");
+            const chunk = decoder.decode(value, { stream: true });
+            thinkResponse += chunk;
+            if (selectedModel?.isThinker) {
+              const thinkMatch = thinkResponse.match(/<think>[\s\S]*/)?.[0];
+              if (thinkMatch && !thinkResponse?.includes("</think>")) {
+                if ((contentThink?.length ?? 0) < 3) {
+                  setOpenThink(true);
+                }
+                setThink(thinkMatch.replace("<think>", "").trim());
+                contentThink += chunk;
+                return read();
+              }
+            }
+
+            contentResponse += chunk;
+
+            const newHtml = contentResponse.match(
+              /<!DOCTYPE html>[\s\S]*/
+            )?.[0];
+            if (newHtml) {
+              setIsThinking(false);
+              let partialDoc = newHtml;
+              if (
+                partialDoc.includes("<head>") &&
+                !partialDoc.includes("</head>")
+              ) {
+                partialDoc += "\n</head>";
+              }
+              if (
+                partialDoc.includes("<body") &&
+                !partialDoc.includes("</body>")
+              ) {
+                partialDoc += "\n</body>";
+              }
+              if (!partialDoc.includes("</html>")) {
+                partialDoc += "\n</html>";
+              }
+
+              // Throttle the re-renders to avoid flashing/flicker
+              const now = Date.now();
+              if (now - lastRenderTime > 300) {
+                setHtml(partialDoc);
+                lastRenderTime = now;
+              }
+
+              if (partialDoc.length > 200) {
+                onScrollToBottom();
+              }
+            }
+            read();
+          };
+
+          read();
+        }
+      }
+    } catch (error: any) {
+      setisAiWorking(false);
+      toast.error(error.message);
+      if (error.openLogin) {
+        setOpen(true);
       }
     }
   };
 
-  const handleError = (error: string, message?: string) => {
-    switch (error) {
-      case "login_required":
-        openLoginModal();
-        break;
-      case "provider_required":
-        setOpenProvider(true);
-        setProviderError(message || "");
-        break;
-      case "pro_required":
-        openProModal([]);
-        break;
-      case "api_error":
-        toast.error(message || "An error occurred");
-        break;
-      case "network_error":
-        toast.error(message || "Network error occurred");
-        break;
-      default:
-        toast.error("An unexpected error occurred");
+  const stopController = () => {
+    if (controller) {
+      controller.abort();
+      setController(null);
+      setisAiWorking(false);
+      setThink("");
+      setOpenThink(false);
+      setIsThinking(false);
     }
   };
 
@@ -146,30 +278,22 @@ export const AskAi = ({
     if (refThink.current) {
       refThink.current.scrollTop = refThink.current.scrollHeight;
     }
-    // Auto-open dropdown when thinking content appears
-    if (thinkingContent && isThinking && !openThink) {
-      setOpenThink(true);
-    }
-    // Auto-collapse when thinking is complete
-    if (thinkingContent && !isThinking && openThink) {
+  }, [think]);
+
+  useUpdateEffect(() => {
+    if (!isThinking) {
       setOpenThink(false);
     }
-  }, [thinkingContent, isThinking]);
+  }, [isThinking]);
 
-  const randomPrompt = () => {
-    setRandomPromptLoading(true);
-    setTimeout(() => {
-      setPrompt(
-        PROMPTS_FOR_AI[Math.floor(Math.random() * PROMPTS_FOR_AI.length)]
-      );
-      setRandomPromptLoading(false);
-    }, 400);
-  };
+  const isSameHtml = useMemo(() => {
+    return isTheSameHtml(html);
+  }, [html]);
 
   return (
-    <div className="p-3 w-full">
-      <div className="relative bg-neutral-800 border border-neutral-700 rounded-2xl ring-[4px] focus-within:ring-neutral-500/30 focus-within:border-neutral-600 ring-transparent z-20 w-full group">
-        {thinkingContent && (
+    <div className="px-3">
+      <div className="relative bg-neutral-800 border border-neutral-700 rounded-2xl ring-[4px] focus-within:ring-neutral-500/30 focus-within:border-neutral-600 ring-transparent z-10 w-full group">
+        {think && (
           <div className="w-full border-b border-neutral-700 relative overflow-hidden">
             <header
               className="flex items-center justify-between px-5 py-2.5 group hover:bg-neutral-600/20 transition-colors duration-200 cursor-pointer"
@@ -201,18 +325,11 @@ export const AskAi = ({
               )}
             >
               <p className="text-[13px] text-neutral-400 whitespace-pre-line px-5 pb-4 pt-3">
-                {thinkingContent}
+                {think}
               </p>
             </main>
           </div>
         )}
-        <SelectedFiles
-          files={selectedFiles}
-          isAiWorking={isAiWorking}
-          onDelete={(file) =>
-            setSelectedFiles(selectedFiles.filter((f) => f !== file))
-          }
-        />
         {selectedElement && (
           <div className="px-4 pt-3">
             <SelectedHtmlElement
@@ -222,59 +339,37 @@ export const AskAi = ({
             />
           </div>
         )}
-        {redesignData && (
-          <div className="px-4 pt-3">
-            <SelectedRedesignUrl
-              url={redesignData.url}
-              isAiWorking={isAiWorking}
-              onDelete={() => setRedesignData(undefined)}
-            />
-          </div>
-        )}
         <div className="w-full relative flex items-center justify-between">
-          {(isAiWorking || isUploading || isThinking || isLoadingProject) && (
-            <div className="absolute bg-neutral-800 top-0 left-4 w-[calc(100%-30px)] h-full z-1 flex items-start pt-3.5 justify-between max-lg:text-sm">
-              <AiLoading
-                text={
-                  isLoadingProject
-                    ? "Fetching your project..."
-                    : isUploading
-                    ? "Uploading images..."
-                    : isAiWorking && !isSameHtml
-                    ? "DeepSite is working..."
-                    : "DeepSite is working..."
-                }
-              />
-              {isAiWorking && (
-                <Button
-                  size="iconXs"
-                  variant="outline"
-                  className="!rounded-md mr-0.5"
-                  onClick={cancelRequest}
-                >
-                  <CircleStop className="size-4" />
-                </Button>
-              )}
+          {isAiWorking && (
+            <div className="absolute bg-neutral-800 rounded-lg bottom-0 left-4 w-[calc(100%-30px)] h-full z-1 flex items-center justify-between max-lg:text-sm">
+              <div className="flex items-center justify-start gap-2">
+                <Loading overlay={false} className="!size-4" />
+                <p className="text-neutral-400 text-sm">
+                  AI is {isThinking ? "thinking" : "coding"}...{" "}
+                </p>
+              </div>
+              <div
+                className="text-xs text-neutral-400 px-1 py-0.5 rounded-md border border-neutral-600 flex items-center justify-center gap-1.5 bg-neutral-800 hover:brightness-110 transition-all duration-200 cursor-pointer"
+                onClick={stopController}
+              >
+                <FaStopCircle />
+                Stop generation
+              </div>
             </div>
           )}
-          <textarea
-            disabled={
-              isAiWorking || isUploading || isThinking || isLoadingProject
-            }
+          <input
+            type="text"
+            disabled={isAiWorking}
             className={classNames(
-              "w-full bg-transparent text-sm outline-none text-white placeholder:text-neutral-400 p-4 resize-none",
+              "w-full bg-transparent text-sm outline-none text-white placeholder:text-neutral-400 p-4",
               {
-                "!pt-2.5":
-                  selectedElement &&
-                  !(isAiWorking || isUploading || isThinking),
+                "!pt-2.5": selectedElement && !isAiWorking,
               }
             )}
             placeholder={
               selectedElement
                 ? `Ask DeepSite about ${selectedElement.tagName.toLowerCase()}...`
-                : redesignData
-                ? "Ask DeepSite anything about the redesign of your site..."
-                : isFollowUp && (!isSameHtml || pages?.length > 1)
+                : hasAsked
                 ? "Ask DeepSite for edits"
                 : "Ask DeepSite anything..."
             }
@@ -286,68 +381,91 @@ export const AskAi = ({
               }
             }}
           />
-          {isNew && !isAiWorking && isSameHtml && (
-            <Button
-              size="iconXs"
-              variant="outline"
-              className="!rounded-md -translate-y-2 -translate-x-4"
-              onClick={() => randomPrompt()}
-            >
-              <Dice6
-                className={classNames("size-4", {
-                  "animate-spin animation-duration-500": randomPromptLoading,
-                })}
-              />
-            </Button>
-          )}
         </div>
-        <div className="flex items-center justify-between gap-2 px-4 pb-3 mt-2">
-          <div className="flex-1 flex items-center justify-start gap-1.5 flex-wrap">
-            {isNew ? (
-              <PromptBuilder
-                enhancedSettings={enhancedSettings!}
-                setEnhancedSettings={setEnhancedSettings}
-              />
-            ) : (
-              <Context />
+        <div className="flex items-center justify-between gap-2 px-4 pb-3">
+          <div className="flex-1 flex items-center justify-start gap-1.5">
+            <ReImagine onRedesign={(md) => callAi(md)} />
+            {!isSameHtml && (
+              <Tooltip>
+                <TooltipTrigger asChild>
+                  <Button
+                    size="xs"
+                    variant={isEditableModeEnabled ? "default" : "outline"}
+                    onClick={() => {
+                      setIsEditableModeEnabled?.(!isEditableModeEnabled);
+                    }}
+                    className={classNames("h-[28px]", {
+                      "!text-neutral-400 hover:!text-neutral-200 !border-neutral-600 !hover:!border-neutral-500":
+                        !isEditableModeEnabled,
+                    })}
+                  >
+                    <Crosshair className="size-4" />
+                    Edit
+                  </Button>
+                </TooltipTrigger>
+                <TooltipContent
+                  align="start"
+                  className="bg-neutral-950 text-xs text-neutral-200 py-1 px-2 rounded-md -translate-y-0.5"
+                >
+                  Select an element on the page to ask DeepSite edit it
+                  directly.
+                </TooltipContent>
+              </Tooltip>
             )}
+            <InviteFriends />
+          </div>
+          <div className="flex items-center justify-end gap-2">
             <Settings
+              provider={provider as string}
+              model={model as string}
+              onChange={setProvider}
+              onModelChange={setModel}
               open={openProvider}
               error={providerError}
               isFollowUp={!isSameHtml && isFollowUp}
               onClose={setOpenProvider}
             />
-            {!isNew && <Uploader project={project} />}
-            {isNew && (
-              <ReImagine
-                onRedesign={(md, url) =>
-                  setRedesignData({ markdown: md, url: url })
-                }
-              />
-            )}
-            {!isNew && !isSameHtml && <Selector />}
-          </div>
-          <div className="flex items-center justify-end gap-2">
             <Button
               size="iconXs"
-              variant="outline"
-              className="!rounded-md"
-              disabled={
-                !!redesignData?.url?.trim()
-                  ? false
-                  : isAiWorking || isUploading || isThinking || !prompt.trim()
-              }
-              onClick={() => callAi(redesignData?.markdown)}
+              disabled={isAiWorking || !prompt.trim()}
+              onClick={() => callAi()}
             >
               <ArrowUp className="size-4" />
             </Button>
           </div>
         </div>
+        <LoginModal open={open} onClose={() => setOpen(false)} html={html} />
+        <ProModal
+          html={html}
+          open={openProModal}
+          onClose={() => setOpenProModal(false)}
+        />
+        {!isSameHtml && (
+          <div className="absolute top-0 right-0 -translate-y-[calc(100%+8px)] select-none text-xs text-neutral-400 flex items-center justify-center gap-2 bg-neutral-800 border border-neutral-700 rounded-md p-1 pr-2.5">
+            <label
+              htmlFor="diff-patch-checkbox"
+              className="flex items-center gap-1.5 cursor-pointer"
+            >
+              <Checkbox
+                id="diff-patch-checkbox"
+                checked={isFollowUp}
+                onCheckedChange={(e) => {
+                  if (e === true && !isSameHtml && selectedModel?.isThinker) {
+                    setModel(MODELS[0].value);
+                  }
+                  setIsFollowUp(e === true);
+                }}
+              />
+              Diff-Patch Update
+            </label>
+            <FollowUpTooltip />
+          </div>
+        )}
       </div>
-      <audio ref={hookAudio} id="audio" className="hidden">
-        <source src="/deepsite/success.mp3" type="audio/mpeg" />
+      <audio ref={audio} id="audio" className="hidden">
+        <source src="/success.mp3" type="audio/mpeg" />
         Your browser does not support the audio element.
       </audio>
     </div>
   );
-};
+}

components/editor/ask-ai/loading.tsx

@@ -1,68 +0,0 @@
-"use client";
-import Loading from "@/components/loading";
-import { useState, useEffect } from "react";
-import { useInterval } from "react-use";
-
-const TEXTS = [
-  "Teaching pixels to dance with style...",
-  "AI is having a creative breakthrough...",
-  "Channeling digital vibes into pure code...",
-  "Summoning the website spirits...",
-  "Brewing some algorithmic magic...",
-  "Composing a symphony of divs and spans...",
-  "Riding the wave of computational creativity...",
-  "Aligning the stars for perfect design...",
-  "Training circus animals to write CSS...",
-  "Launching ideas into the digital stratosphere...",
-];
-
-export const AiLoading = ({
-  text,
-  className,
-}: {
-  text?: string;
-  className?: string;
-}) => {
-  const [selectedText, setSelectedText] = useState(
-    text ?? TEXTS[0] // Start with first text to avoid hydration issues
-  );
-
-  // Set random text on client-side only to avoid hydration mismatch
-  useEffect(() => {
-    if (!text) {
-      setSelectedText(TEXTS[Math.floor(Math.random() * TEXTS.length)]);
-    }
-  }, [text]);
-
-  useInterval(() => {
-    if (!text) {
-      if (selectedText === TEXTS[TEXTS.length - 1]) {
-        setSelectedText(TEXTS[0]);
-      } else {
-        setSelectedText(TEXTS[TEXTS.indexOf(selectedText) + 1]);
-      }
-    }
-  }, 12000);
-  return (
-    <div className={`flex items-center justify-start gap-2 ${className}`}>
-      <Loading overlay={false} className="!size-5 opacity-50" />
-      <p className="text-neutral-400 text-sm">
-        <span className="inline-flex">
-          {selectedText.split("").map((char, index) => (
-            <span
-              key={index}
-              className="bg-gradient-to-r from-neutral-100 to-neutral-300 bg-clip-text text-transparent animate-pulse"
-              style={{
-                animationDelay: `${index * 0.1}s`,
-                animationDuration: "1.3s",
-                animationIterationCount: "infinite",
-              }}
-            >
-              {char === " " ? "\u00A0" : char}
-            </span>
-          ))}
-        </span>
-      </p>
-    </div>
-  );
-};

components/editor/ask-ai/prompt-builder/content-modal.tsx

@@ -1,196 +0,0 @@
-import classNames from "classnames";
-import { ChevronRight, RefreshCcw } from "lucide-react";
-import { useState } from "react";
-import { TailwindColors } from "./tailwind-colors";
-import { Switch } from "@/components/ui/switch";
-import { Button } from "@/components/ui/button";
-import { Themes } from "./themes";
-import { EnhancedSettings } from "@/types";
-
-export const ContentModal = ({
-  enhancedSettings,
-  setEnhancedSettings,
-}: {
-  enhancedSettings: EnhancedSettings;
-  setEnhancedSettings: (settings: EnhancedSettings) => void;
-}) => {
-  const [collapsed, setCollapsed] = useState(["colors", "theme"]);
-  return (
-    <main className="overflow-x-hidden max-h-[50dvh] overflow-y-auto transparent-scroll">
-      <section className="w-full border-b border-neutral-800/80 px-6 py-3.5 sticky top-0 bg-neutral-900 z-10">
-        <div className="flex items-center justify-between gap-3">
-          <p className="text-base font-semibold text-neutral-200">
-            Allow DeepSite to enhance your prompt
-          </p>
-          <Switch
-            checked={enhancedSettings.isActive}
-            onCheckedChange={() =>
-              setEnhancedSettings({
-                ...enhancedSettings,
-                isActive: !enhancedSettings.isActive,
-              })
-            }
-          />
-        </div>
-        <p className="text-sm text-neutral-500 mt-2">
-          While using DeepSite enhanced prompt, you'll get better results. We'll
-          add more details and features to your request.
-        </p>
-        <div className="text-sm text-sky-500 mt-3 bg-gradient-to-r from-sky-400/15 to-purple-400/15 rounded-md px-3 py-2 border border-white/10">
-          <p className="text-transparent bg-gradient-to-r from-sky-400 to-purple-400 bg-clip-text">
-            You can also use the custom properties below to set specific
-            information.
-          </p>
-        </div>
-      </section>
-      <section className="py-3.5 border-b border-neutral-800/80">
-        <div
-          className={classNames(
-            "flex items-center justify-start gap-3 px-4 cursor-pointer text-neutral-400 hover:text-neutral-200",
-            {
-              "!text-neutral-200": collapsed.includes("colors"),
-            }
-          )}
-          onClick={() =>
-            setCollapsed((prev) => {
-              if (prev.includes("colors")) {
-                return prev.filter((item) => item !== "colors");
-              }
-              return [...prev, "colors"];
-            })
-          }
-        >
-          <ChevronRight className="size-4" />
-          <p className="text-base font-semibold">Colors</p>
-        </div>
-        {collapsed.includes("colors") && (
-          <div className="mt-4 space-y-4">
-            <article className="w-full">
-              <div className="flex items-center justify-start gap-2 px-5">
-                <p className="text-xs font-medium uppercase text-neutral-400">
-                  Primary Color
-                </p>
-                <Button
-                  variant="bordered"
-                  size="xss"
-                  className={`${
-                    enhancedSettings.primaryColor ? "" : "opacity-0"
-                  }`}
-                  onClick={() =>
-                    setEnhancedSettings({
-                      ...enhancedSettings,
-                      primaryColor: undefined,
-                    })
-                  }
-                >
-                  <RefreshCcw className="size-2.5" />
-                  Reset
-                </Button>
-              </div>
-              <div className="text-muted-foreground text-sm mt-4">
-                <TailwindColors
-                  value={enhancedSettings.primaryColor}
-                  onChange={(value) =>
-                    setEnhancedSettings({
-                      ...enhancedSettings,
-                      primaryColor: value,
-                    })
-                  }
-                />
-              </div>
-            </article>
-            <article className="w-full">
-              <div className="flex items-center justify-start gap-2 px-5">
-                <p className="text-xs font-medium uppercase text-neutral-400">
-                  Secondary Color
-                </p>
-                <Button
-                  variant="bordered"
-                  size="xss"
-                  className={`${
-                    enhancedSettings.secondaryColor ? "" : "opacity-0"
-                  }`}
-                  onClick={() =>
-                    setEnhancedSettings({
-                      ...enhancedSettings,
-                      secondaryColor: undefined,
-                    })
-                  }
-                >
-                  <RefreshCcw className="size-2.5" />
-                  Reset
-                </Button>
-              </div>
-              <div className="text-muted-foreground text-sm mt-4">
-                <TailwindColors
-                  value={enhancedSettings.secondaryColor}
-                  onChange={(value) =>
-                    setEnhancedSettings({
-                      ...enhancedSettings,
-                      secondaryColor: value,
-                    })
-                  }
-                />
-              </div>
-            </article>
-          </div>
-        )}
-      </section>
-      <section className="py-3.5 border-b border-neutral-800/80">
-        <div
-          className={classNames(
-            "flex items-center justify-start gap-3 px-4 cursor-pointer text-neutral-400 hover:text-neutral-200",
-            {
-              "!text-neutral-200": collapsed.includes("theme"),
-            }
-          )}
-          onClick={() =>
-            setCollapsed((prev) => {
-              if (prev.includes("theme")) {
-                return prev.filter((item) => item !== "theme");
-              }
-              return [...prev, "theme"];
-            })
-          }
-        >
-          <ChevronRight className="size-4" />
-          <p className="text-base font-semibold">Theme</p>
-        </div>
-        {collapsed.includes("theme") && (
-          <article className="w-full mt-4">
-            <div className="flex items-center justify-start gap-2 px-5">
-              <p className="text-xs font-medium uppercase text-neutral-400">
-                Theme
-              </p>
-              <Button
-                variant="bordered"
-                size="xss"
-                className={`${enhancedSettings.theme ? "" : "opacity-0"}`}
-                onClick={() =>
-                  setEnhancedSettings({
-                    ...enhancedSettings,
-                    theme: undefined,
-                  })
-                }
-              >
-                <RefreshCcw className="size-2.5" />
-                Reset
-              </Button>
-            </div>
-            <div className="text-muted-foreground text-sm mt-4">
-              <Themes
-                value={enhancedSettings.theme}
-                onChange={(value) =>
-                  setEnhancedSettings({
-                    ...enhancedSettings,
-                    theme: value,
-                  })
-                }
-              />
-            </div>
-          </article>
-        )}
-      </section>
-    </main>
-  );
-};

components/editor/ask-ai/prompt-builder/index.tsx

@@ -1,68 +0,0 @@
-import { useState } from "react";
-import { WandSparkles } from "lucide-react";
-
-import { Button } from "@/components/ui/button";
-import { useEditor } from "@/hooks/useEditor";
-import { useAi } from "@/hooks/useAi";
-import {
-  Dialog,
-  DialogContent,
-  DialogFooter,
-  DialogTitle,
-} from "@/components/ui/dialog";
-import { ContentModal } from "./content-modal";
-import { EnhancedSettings } from "@/types";
-
-export const PromptBuilder = ({
-  enhancedSettings,
-  setEnhancedSettings,
-}: {
-  enhancedSettings: EnhancedSettings;
-  setEnhancedSettings: (settings: EnhancedSettings) => void;
-}) => {
-  const { globalAiLoading } = useAi();
-  const { globalEditorLoading } = useEditor();
-
-  const [open, setOpen] = useState(false);
-  return (
-    <>
-      <Button
-        size="xs"
-        variant="outline"
-        className="!rounded-md !border-white/10 !bg-gradient-to-r from-sky-400/15 to-purple-400/15 light-sweep hover:brightness-110"
-        disabled={globalAiLoading || globalEditorLoading}
-        onClick={() => {
-          setOpen(true);
-        }}
-      >
-        <WandSparkles className="size-3.5 text-sky-500 relative z-10" />
-        <span className="text-transparent bg-gradient-to-r from-sky-400 to-purple-400 bg-clip-text relative z-10">
-          Enhance
-        </span>
-      </Button>
-      <Dialog open={open} onOpenChange={() => setOpen(false)}>
-        <DialogContent className="sm:max-w-xl !p-0 !rounded-3xl !bg-neutral-900 !border-neutral-800/80 !gap-0">
-          <DialogTitle className="px-6 py-3.5 border-b border-neutral-800">
-            <div className="flex items-center justify-start gap-2 text-neutral-200 text-base font-medium">
-              <WandSparkles className="size-3.5" />
-              <p>Enhance Prompt</p>
-            </div>
-          </DialogTitle>
-          <ContentModal
-            enhancedSettings={enhancedSettings}
-            setEnhancedSettings={setEnhancedSettings}
-          />
-          <DialogFooter className="px-6 py-3.5 border-t border-neutral-800">
-            <Button
-              variant="bordered"
-              size="default"
-              onClick={() => setOpen(false)}
-            >
-              Close
-            </Button>
-          </DialogFooter>
-        </DialogContent>
-      </Dialog>
-    </>
-  );
-};

components/editor/ask-ai/prompt-builder/tailwind-colors.tsx

@@ -1,58 +0,0 @@
-import classNames from "classnames";
-import { useRef } from "react";
-
-import { TAILWIND_COLORS } from "@/lib/prompt-builder";
-import { useMount } from "react-use";
-
-export const TailwindColors = ({
-  value,
-  onChange,
-}: {
-  value: string | undefined;
-  onChange: (value: string) => void;
-}) => {
-  const ref = useRef<HTMLDivElement>(null);
-
-  useMount(() => {
-    if (ref.current) {
-      if (value) {
-        const color = ref.current.querySelector(`[data-color="${value}"]`);
-        if (color) {
-          color.scrollIntoView({ inline: "center" });
-        }
-      }
-    }
-  });
-  return (
-    <div
-      ref={ref}
-      className="flex items-center justify-start gap-3 overflow-x-auto px-5 transparent-scroll"
-    >
-      {TAILWIND_COLORS.map((color) => (
-        <div
-          key={color}
-          className={classNames(
-            "flex flex-col items-center justify-center p-3 size-16 min-w-16 gap-2 rounded-lg border border-neutral-800 bg-neutral-800/30 hover:brightness-120 cursor-pointer",
-            {
-              "!border-neutral-700 !bg-neutral-800/80 hover:!brightness-100":
-                value === color,
-            }
-          )}
-          data-color={color}
-          onClick={() => onChange(color)}
-        >
-          <div
-            className={`w-4 h-4 min-w-4 min-h-4 rounded-xl ${
-              ["white", "black"].includes(color)
-                ? `bg-${color}`
-                : `bg-${color}-500`
-            }`}
-          />
-          <p className="text-xs capitalize text-neutral-200 truncate">
-            {color}
-          </p>
-        </div>
-      ))}
-    </div>
-  );
-};