deepsite

Sleeping

App Files Files Community

enzostvs HF Staff

victor HF Staff commited on Mar 28

Commit

cf11907

verified ·

1 Parent(s): 08fe4d2

toto (#2)

Browse files

- refactor: Improve AI prompt for better HTML generation and code structure (891e1dcde5faaac21c71dfb80023877f738e1127)

Co-authored-by: Victor Mustar <[email protected]>

Files changed (1) hide show

src/components/ask-ai/ask-ai.tsx +23 -3

src/components/ask-ai/ask-ai.tsx CHANGED Viewed

@@ -25,10 +25,11 @@ function AskAI({
   const [previousPrompt, setPreviousPrompt] = useState("");
   const callAi = async () => {
-    if (isAiWorking) return;
     setisAiWorking(true);
     let contentResponse = "";
     try {
       const request = await fetch("/api/ask-ai", {
         method: "POST",
@@ -64,6 +65,13 @@ function AskAI({
             setPreviousPrompt(prompt);
             setisAiWorking(false);
             setHasAsked(true);
             return;
           }
@@ -71,8 +79,20 @@ function AskAI({
           contentResponse += chunk;
           const newHtml = contentResponse.match(/<!DOCTYPE html>[\s\S]*/)?.[0];
           if (newHtml) {
-            setHtml(newHtml);
-            if (newHtml?.length > 200) {
               onScrollToBottom();
             }
           }

   const [previousPrompt, setPreviousPrompt] = useState("");
   const callAi = async () => {
+    if (isAiWorking || !prompt.trim()) return;
     setisAiWorking(true);
     let contentResponse = "";
+    let lastRenderTime = 0;
     try {
       const request = await fetch("/api/ask-ai", {
         method: "POST",
             setPreviousPrompt(prompt);
             setisAiWorking(false);
             setHasAsked(true);
+            // 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);
+            }
             return;
           }
           contentResponse += chunk;
           const newHtml = contentResponse.match(/<!DOCTYPE html>[\s\S]*/)?.[0];
           if (newHtml) {
+            // Force-close the HTML tag so the iframe doesn't render half-finished markup
+            let partialDoc = newHtml;
+            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();
             }
           }