test

server.js

@@ -102,7 +102,14 @@ app.post("/api/deploy", checkUser, async (req, res) => {
     });
   }
 
-  const file = new Blob([html], { type: "text/html" });
+  // add add the end of the body (before all the  <script></script>) a paragraph with the mention this has been generated by DeepSite
+  // and add a link to the website (https://enzostvs-deepsite.hf.space) + the logo "/logo.svg"
+  const newHtml = html.replace(
+    /<\/body>/,
+    `<p style="text-align: center; font-size: 12px; color: #888; margin-top: 16px;">This website has been generated by <a href="https://enzostvs-deepsite.hf.space" target="_blank">DeepSite</a> <img src="https://enzostvs-deepsite.hf.space/logo.svg" alt="DeepSite Logo" style="width: 16px; height: 16px; vertical-align: middle;"></p></body>`
+  );
+
+  const file = new Blob([newHtml], { type: "text/html" });
   file.name = "index.html"; // Add name property to the Blob
 
   const { hf_token } = req.cookies;
@@ -143,7 +150,7 @@ app.post("/api/deploy", checkUser, async (req, res) => {
 });
 
 app.post("/api/ask-ai", async (req, res) => {
-  const { prompt, html } = req.body;
+  const { prompt, html, previousPrompt } = req.body;
   if (!prompt) {
     return res.status(400).send({
       ok: false,
@@ -182,20 +189,30 @@ app.post("/api/ask-ai", async (req, res) => {
   const client = new InferenceClient(token);
   let completeResponse = "";
 
+  const useNebius = html?.length > 2_500;
+
   try {
     const chatCompletion = client.chatCompletionStream({
       model: MODEL_ID,
-      provider: html ? "fireworks-ai" : "sambanova",
+      provider: useNebius ? "nebius" : "sambanova",
       messages: [
         {
-          role: "user",
+          role: "system",
           content:
             "ONLY USE HTML, CSS AND JAVASCRIPT. If you want to use ICON make sure to import the library first. Try to create the best UI possible by using only HTML, CSS and JAVASCRIPT. Also, try to ellaborate as much as you can, to create something unique. ALWAYS GIVE THE RESPONSE INTO A SINGLE HTML FILE",
         },
+        ...(previousPrompt
+          ? [
+              {
+                role: "user",
+                content: previousPrompt,
+              },
+            ]
+          : []),
         ...(html
           ? [
               {
-                role: "system",
+                role: "assistant",
                 content: `The current code is: ${html}.`,
               },
             ]
@@ -205,7 +222,7 @@ app.post("/api/ask-ai", async (req, res) => {
           content: prompt,
         },
       ],
-      max_tokens: 8_192,
+      max_tokens: useNebius ? 8_192 : 6_000,
     });
 
     while (true) {
@@ -215,16 +232,25 @@ app.post("/api/ask-ai", async (req, res) => {
       }
       const chunk = value.choices[0]?.delta?.content;
       if (chunk) {
-        // Stream chunk to client
-        let newChunk = chunk;
-        if (chunk.includes("</html>")) {
-          // Replace everything after the last </html> tag with an empty string
-          newChunk = newChunk.replace(/<\/html>[\s\S]*/, "</html>");
-        }
-        completeResponse += newChunk;
-        res.write(newChunk);
-        if (newChunk.includes("</html>")) {
-          break;
+        if (useNebius) {
+          res.write(chunk);
+          completeResponse += chunk;
+
+          // Break when HTML is complete
+          if (completeResponse.includes("</html>")) {
+            break;
+          }
+        } else {
+          let newChunk = chunk;
+          if (chunk.includes("</html>")) {
+            // Replace everything after the last </html> tag with an empty string
+            newChunk = newChunk.replace(/<\/html>[\s\S]*/, "</html>");
+          }
+          completeResponse += newChunk;
+          res.write(newChunk);
+          if (newChunk.includes("</html>")) {
+            break;
+          }
         }
       }
     }

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

@@ -22,6 +22,7 @@ function AskAI({
   const [open, setOpen] = useState(false);
   const [prompt, setPrompt] = useState("");
   const [hasAsked, setHasAsked] = useState(false);
+  const [previousPrompt, setPreviousPrompt] = useState("");
 
   const callAi = async () => {
     if (isAiWorking) return;
@@ -34,6 +35,7 @@ function AskAI({
         body: JSON.stringify({
           prompt,
           ...(html === defaultHTML ? {} : { html }),
+          ...(previousPrompt ? { previousPrompt } : {}),
         }),
         headers: {
           "Content-Type": "application/json",
@@ -59,6 +61,7 @@ function AskAI({
           if (done) {
             toast.success("AI responded successfully");
             setPrompt("");
+            setPreviousPrompt(prompt);
             setisAiWorking(false);
             setHasAsked(true);
             return;