handle free request

app/api/ask/route.ts

@@ -66,7 +66,8 @@ export async function POST(request: NextRequest) {
     );
   }
 
-  let token = userToken;
+  let token: string | null = null;
+  if (userToken) token = userToken;
   let billTo: string | null = null;
 
   /**

components/contexts/login-context.tsx

@@ -13,6 +13,7 @@ interface LoginContextType {
 interface LoginModalOptions {
   pages?: Page[];
   title?: string;
+  prompt?: string;
   description?: string;
 }
 
@@ -44,8 +45,8 @@ export function LoginProvider({ children }: { children: ReactNode }) {
       <LoginModal
         open={isOpen}
         onClose={setIsOpen}
-        pages={modalOptions.pages}
         title={modalOptions.title}
+        prompt={modalOptions.prompt}
         description={modalOptions.description}
       />
     </LoginContext.Provider>

components/editor/ask-ai/index.tsx

@@ -33,7 +33,6 @@ export const AskAi = ({
 }) => {
   const { user, projects } = useUser();
   const { isSameHtml, isUploading, pages, isLoadingProject } = useEditor();
-  console.log("isSameHtml", isSameHtml);
   const {
     isAiWorking,
     isThinking,
@@ -57,7 +56,7 @@ export const AskAi = ({
 
   const [enhancedSettings, setEnhancedSettings, removeEnhancedSettings] =
     useLocalStorage<EnhancedSettings>("deepsite-enhancedSettings", {
-      isActive: true,
+      isActive: false,
       primaryColor: undefined,
       secondaryColor: undefined,
       theme: undefined,
@@ -65,7 +64,9 @@ export const AskAi = ({
   const [promptStorage, , removePromptStorage] = useLocalStorage("prompt", "");
 
   const [isFollowUp, setIsFollowUp] = useState(true);
-  const [prompt, setPrompt] = useState("");
+  const [prompt, setPrompt] = useState(
+    promptStorage && promptStorage.trim() !== "" ? promptStorage : ""
+  );
   const [think, setThink] = useState("");
   const [openThink, setOpenThink] = useState(false);
 
@@ -77,20 +78,19 @@ export const AskAi = ({
 
   useMount(() => {
     if (promptStorage && promptStorage.trim() !== "") {
-      setPrompt(promptStorage);
       callAi();
     }
   });
 
   const callAi = async (redesignMarkdown?: string) => {
-    if (!user) return openLoginModal();
     removePromptStorage();
-    if (!user.isPro && projects.length >= MAX_FREE_PROJECTS)
+    if (user && !user.isPro && projects.length >= MAX_FREE_PROJECTS)
       return openProModal([]);
     if (isAiWorking) return;
     if (!redesignMarkdown && !prompt.trim()) return;
 
     if (isFollowUp && !redesignMarkdown && !isSameHtml) {
+      if (!user) return openLoginModal({ prompt });
       const result = await callAiFollowUp(prompt, enhancedSettings, isNew);
 
       if (result?.error) {
@@ -106,10 +106,7 @@ export const AskAi = ({
         prompt,
         enhancedSettings,
         redesignMarkdown,
-        handleThink,
-        () => {
-          setIsThinking(false);
-        }
+        !!user
       );
 
       if (result?.error) {

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

@@ -11,8 +11,6 @@ import {
   DialogTitle,
 } from "@/components/ui/dialog";
 import { ContentModal } from "./content-modal";
-import { useLoginModal } from "@/components/contexts/login-context";
-import { useUser } from "@/hooks/useUser";
 import { EnhancedSettings } from "@/types";
 
 export const PromptBuilder = ({
@@ -22,8 +20,6 @@ export const PromptBuilder = ({
   enhancedSettings: EnhancedSettings;
   setEnhancedSettings: (settings: EnhancedSettings) => void;
 }) => {
-  const { user } = useUser();
-  const { openLoginModal } = useLoginModal();
   const { globalAiLoading } = useAi();
   const { globalEditorLoading } = useEditor();
 
@@ -36,7 +32,6 @@ export const PromptBuilder = ({
         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={() => {
-          if (!user) return openLoginModal();
           setOpen(true);
         }}
       >

components/editor/ask-ai/re-imagine.tsx

@@ -13,8 +13,6 @@ import Loading from "@/components/loading";
 import { api } from "@/lib/api";
 import { useAi } from "@/hooks/useAi";
 import { useEditor } from "@/hooks/useEditor";
-import { useUser } from "@/hooks/useUser";
-import { useLoginModal } from "@/components/contexts/login-context";
 
 export function ReImagine({
   onRedesign,
@@ -26,8 +24,6 @@ export function ReImagine({
   const [isLoading, setIsLoading] = useState(false);
   const { globalAiLoading } = useAi();
   const { globalEditorLoading } = useEditor();
-  const { user } = useUser();
-  const { openLoginModal } = useLoginModal();
 
   const checkIfUrlIsValid = (url: string) => {
     const urlPattern = new RegExp(
@@ -62,19 +58,6 @@ export function ReImagine({
     setIsLoading(false);
   };
 
-  if (!user)
-    return (
-      <Button
-        size="xs"
-        variant="outline"
-        className="!rounded-md"
-        onClick={() => openLoginModal()}
-      >
-        <Paintbrush className="size-3.5" />
-        Redesign
-      </Button>
-    );
-
   return (
     <Popover open={open} onOpenChange={setOpen}>
       <form>

components/editor/index.tsx

@@ -1,6 +1,6 @@
 "use client";
 import { useMemo, useRef, useState, useEffect } from "react";
-import { useCopyToClipboard, useMount } from "react-use";
+import { useCopyToClipboard, useLocalStorage, useMount } from "react-use";
 import { CopyIcon } from "lucide-react";
 import { toast } from "sonner";
 import classNames from "classnames";
@@ -18,6 +18,7 @@ import { Preview } from "./preview";
 import { SaveChangesPopup } from "./save-changes-popup";
 import Loading from "../loading";
 import { LivePreviewRef } from "./live-preview";
+import { Page } from "@/types";
 
 export const AppEditor = ({
   namespace,
@@ -43,6 +44,7 @@ export const AppEditor = ({
   const { isAiWorking } = useAi(undefined, livePreviewRef);
   const [, copyToClipboard] = useCopyToClipboard();
   const [showSavePopup, setShowSavePopup] = useState(false);
+  const [pagesStorage, , removePagesStorage] = useLocalStorage<Page[]>("pages");
 
   const monacoRef = useRef<any>(null);
   const editor = useRef<HTMLDivElement>(null);
@@ -50,12 +52,8 @@ export const AppEditor = ({
 
   useMount(() => {
     if (isNew) {
-      setPages([
-        {
-          path: "index.html",
-          html: defaultHTML,
-        },
-      ]);
+      setPages(pagesStorage || []);
+      removePagesStorage();
     }
   });
 

components/login-modal/index.tsx

@@ -4,23 +4,29 @@ import { Dialog, DialogContent, DialogTitle } from "@/components/ui/dialog";
 import { useUser } from "@/hooks/useUser";
 import { isTheSameHtml } from "@/lib/compare-html-diff";
 import { Page } from "@/types";
+import { useEditor } from "@/hooks/useEditor";
 
 export const LoginModal = ({
   open,
-  pages,
   onClose,
   title = "Log In to use DeepSite for free",
   description = "Log In through your Hugging Face account to continue using DeepSite and increase your monthly free limit.",
+  prompt,
 }: {
   open: boolean;
-  pages?: Page[];
   onClose: React.Dispatch<React.SetStateAction<boolean>>;
   title?: string;
   description?: string;
+  prompt?: string;
 }) => {
   const { openLoginWindow } = useUser();
+  const { pages } = useEditor();
   const [, setStorage] = useLocalStorage("pages");
+  const [, setPromptStorage] = useLocalStorage("prompt", "");
   const handleClick = async () => {
+    if (prompt) {
+      setPromptStorage(prompt);
+    }
     if (pages && !isTheSameHtml(pages[0].html)) {
       setStorage(pages);
     }

components/my-projects/index.tsx

@@ -93,7 +93,7 @@ export function MyProjects() {
           )}
           {projects.map((project: ProjectType) => (
             <ProjectCard
-              key={project.id}
+              key={project.name}
               project={project}
               onDelete={() => onDelete(project)}
             />

components/not-logged/not-logged.tsx

@@ -1,13 +1,9 @@
 "use client";
 
-import { useUser } from "@/hooks/useUser";
-import { Button } from "@/components/ui/button";
 import { AnimatedBlobs } from "../animated-blobs";
-import { AnimatedText } from "../animated-text";
 import { FakeAskAi } from "../editor/ask-ai/fake-ask";
 
 export const NotLogged = () => {
-  const { openLoginWindow } = useUser();
   return (
     <section className="relative max-w-[86rem] mx-auto">
       <header className="container mx-auto pt-20 px-6 relative flex flex-col items-center justify-center text-center">

hooks/useAi.ts

@@ -111,27 +111,36 @@ export const useAi = (onScrollToBottom?: () => void, livePreviewRef?: React.RefO
     client.setQueryData(["ai.model"], newModel);
   };
 
-  const createNewProject = async (prompt: string, htmlPages: Page[], projectName: string | undefined) => {
-    const response = await api.post("/me/projects", {
-      title: projectName,
-      pages: htmlPages,
-      prompt,
-    });
-    if (response.data.ok) {
+  const createNewProject = async (prompt: string, htmlPages: Page[], projectName: string | undefined, isLoggedIn?: boolean) => {
+    if (isLoggedIn) {
+      const response = await api.post("/me/projects", {
+        title: projectName,
+        pages: htmlPages,
+        prompt,
+      });
+      if (response.data.ok) {
+        setIsAiWorking(false);
+        // Reset live preview when project is created
+        if (livePreviewRef?.current) {
+          livePreviewRef.current.reset();
+        }
+        router.replace(`/projects/${response.data.space.project.space_id}`);
+        setProject(response.data.space);
+        setProjects([...projects, response.data.space]);
+        toast.success("AI responded successfully");
+        if (audio.current) audio.current.play();
+      }
+    } else {
       setIsAiWorking(false);
-      // Reset live preview when project is created
       if (livePreviewRef?.current) {
         livePreviewRef.current.reset();
       }
-      router.replace(`/projects/${response.data.space.project.space_id}`);
-      setProject(response.data.space);
-      setProjects([...projects, response.data.space]);
       toast.success("AI responded successfully");
       if (audio.current) audio.current.play();
     }
   }
   
-  const callAiNewProject = async (prompt: string, enhancedSettings?: EnhancedSettings, redesignMarkdown?: string, handleThink?: (think: string) => void, onFinishThink?: () => void) => {
+  const callAiNewProject = async (prompt: string, enhancedSettings?: EnhancedSettings, redesignMarkdown?: string, isLoggedIn?: boolean) => {
     if (isAiWorking) return;
     if (!redesignMarkdown && !prompt.trim()) return;
     
@@ -160,16 +169,12 @@ export const useAi = (onScrollToBottom?: () => void, livePreviewRef?: React.RefO
       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 contentResponse = "";
 
         const read = async (): Promise<any> => {
           const { done, value } = await reader.read();
           
           if (done) {
-            // Check if the response is a JSON error message
             const trimmedResponse = contentResponse.trim();
             if (trimmedResponse.startsWith("{") && trimmedResponse.endsWith("}")) {
               try {
@@ -196,7 +201,7 @@ export const useAi = (onScrollToBottom?: () => void, livePreviewRef?: React.RefO
             const projectName = contentResponse.match(/<<<<<<< PROJECT_NAME_START ([\s\S]*?) >>>>>>> PROJECT_NAME_END/)?.[1]?.trim();
             setPages(newPages);
             setLastSavedPages([...newPages]); // Mark initial pages as saved
-            createNewProject(prompt, newPages, projectName);
+            createNewProject(prompt, newPages, projectName, isLoggedIn);
             setPrompts([...prompts, prompt]);
 
             return { success: true, pages: newPages };
@@ -226,17 +231,6 @@ export const useAi = (onScrollToBottom?: () => void, livePreviewRef?: React.RefO
               // Not a complete JSON yet, continue reading
             }
           }
-          
-          // if (selectedModel?.isThinker) {
-          //   const thinkMatch = contentResponse.match(/<think>[\s\S]*/)?.[0];
-          //   if (thinkMatch && !contentResponse?.includes("</think>")) {
-          //     handleThink?.(thinkMatch.replace("<think>", "").trim());
-          //   }
-          // }
-
-          // if (contentResponse.includes("</think>")) {
-          //   onFinishThink?.();
-          // }
 
           formatPages(contentResponse);
           

hooks/useEditor.ts

@@ -1,6 +1,6 @@
 import { useQuery, useQueryClient, useMutation } from "@tanstack/react-query";
 import { useMemo } from "react";
-import { useUpdateEffect } from "react-use";
+import { useLocalStorage, useMount, useUpdateEffect } from "react-use";
 import { toast } from "sonner";
 import { useRouter } from "next/navigation";
 
@@ -12,7 +12,6 @@ import { isTheSameHtml } from "@/lib/compare-html-diff";
 export const useEditor = (namespace?: string, repoId?: string) => {
   const client = useQueryClient();
   const router = useRouter();
-  // const [pagesStorage,, removePagesStorage] = useLocalStorage<Page[]>("pages");
 
   const { data: project, isFetching: isLoadingProject } = useQuery({
     queryKey: ["editor.project"],

lib/prompts.ts

@@ -1,7 +1,7 @@
 export const SEARCH_START = "<<<<<<< SEARCH";
 export const DIVIDER = "=======";
 export const REPLACE_END = ">>>>>>> REPLACE";
-export const MAX_REQUESTS_PER_IP = 2;
+export const MAX_REQUESTS_PER_IP = 4;
 export const TITLE_PAGE_START = "<<<<<<< START_TITLE ";
 export const TITLE_PAGE_END = " >>>>>>> END_TITLE";
 export const NEW_PAGE_START = "<<<<<<< NEW_PAGE_START ";