Spaces:
Running
Running
File size: 5,676 Bytes
ea6c2a8 bcb0fad a2c35f4 ea6c2a8 cf11907 ea6c2a8 cf11907 ea6c2a8 a2c35f4 ea6c2a8 a582fa0 ea6c2a8 a2c35f4 ea6c2a8 bcb0fad cf11907 ea6c2a8 cf11907 ea6c2a8 a582fa0 ea6c2a8 a582fa0 ea6c2a8 a582fa0 bcb0fad ea6c2a8 18847f3 ea6c2a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 |
import { useState } from "react";
import { RiSparkling2Fill } from "react-icons/ri";
import { GrSend } from "react-icons/gr";
import classNames from "classnames";
import { toast } from "react-toastify";
import Login from "../login/login";
import { defaultHTML } from "../../utils/consts";
function AskAI({
html,
setHtml,
onScrollToBottom,
isAiWorking,
setisAiWorking,
}: {
html: string;
setHtml: (html: string) => void;
onScrollToBottom: () => void;
isAiWorking: boolean;
setisAiWorking: React.Dispatch<React.SetStateAction<boolean>>;
}) {
const [open, setOpen] = useState(false);
const [prompt, setPrompt] = useState("");
const [hasAsked, setHasAsked] = useState(false);
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",
body: JSON.stringify({
prompt,
...(html === defaultHTML ? {} : { html }),
...(previousPrompt ? { previousPrompt } : {}),
}),
headers: {
"Content-Type": "application/json",
},
});
if (request && request.body) {
if (!request.ok) {
const res = await request.json();
if (res.openLogin) {
setOpen(true);
} else {
// don't show toast if it's a login error
toast.error(res.message);
}
setisAiWorking(false);
return;
}
const reader = request.body.getReader();
const decoder = new TextDecoder("utf-8");
const read = async () => {
const { done, value } = await reader.read();
if (done) {
toast.success("AI responded successfully");
setPrompt("");
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;
}
const chunk = decoder.decode(value, { stream: true });
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();
}
}
read();
};
read();
}
// eslint-disable-next-line @typescript-eslint/no-explicit-any
} catch (error: any) {
setisAiWorking(false);
toast.error(error.message);
if (error.openLogin) {
setOpen(true);
}
}
};
return (
<div
className={`bg-gray-950 rounded-xl py-2 lg:py-2.5 pl-3.5 lg:pl-4 pr-2 lg:pr-2.5 absolute lg:sticky bottom-3 left-3 lg:bottom-4 lg:left-4 w-[calc(100%-1.5rem)] lg:w-[calc(100%-2rem)] z-10 group ${
isAiWorking ? "animate-pulse" : ""
}`}
>
<div className="w-full relative flex items-center justify-between">
<RiSparkling2Fill className="text-lg lg:text-xl text-gray-500 group-focus-within:text-pink-500" />
<input
type="text"
disabled={isAiWorking}
className="w-full bg-transparent max-lg:text-sm outline-none pl-3 text-white placeholder:text-gray-500 font-code"
placeholder={
hasAsked ? "What do you want to ask AI next?" : "Ask AI anything..."
}
value={prompt}
onChange={(e) => setPrompt(e.target.value)}
onKeyDown={(e) => {
if (e.key === "Enter") {
callAi();
}
}}
/>
<button
disabled={isAiWorking}
className="relative overflow-hidden cursor-pointer flex-none flex items-center justify-center rounded-full text-sm font-semibold size-8 text-center bg-pink-500 hover:bg-pink-400 text-white shadow-sm dark:shadow-highlight/20 disabled:bg-gray-300 disabled:text-gray-500 disabled:cursor-not-allowed disabled:hover:bg-gray-300"
onClick={callAi}
>
<GrSend className="-translate-x-[1px]" />
</button>
</div>
<div
className={classNames(
"h-screen w-screen bg-black/20 fixed left-0 top-0 z-10",
{
"opacity-0 pointer-events-none": !open,
}
)}
onClick={() => setOpen(false)}
></div>
<div
className={classNames(
"absolute top-0 -translate-y-[calc(100%+8px)] right-0 z-10 w-80 bg-white border border-gray-200 rounded-lg shadow-lg transition-all duration-75 overflow-hidden",
{
"opacity-0 pointer-events-none": !open,
}
)}
>
<Login html={html}>
<p className="text-gray-500 text-sm mb-3">
You reached the limit of free AI usage. Please login to continue.
</p>
</Login>
</div>
</div>
);
}
export default AskAI;
|