rewrite first prompt

app/actions/rewrite-prompt.ts

@@ -0,0 +1,35 @@
+import { InferenceClient } from "@huggingface/inference";
+
+const START_REWRITE_PROMPT = ">>>>>>> START PROMPT >>>>>>";
+const END_REWRITE_PROMPT = ">>>>>>> END PROMPT >>>>>>";
+
+export const callAiRewritePrompt = async (prompt: string, { token, billTo }: { token: string, billTo?: string | null }) => {
+  const client = new InferenceClient(token);
+  const response = await client.chatCompletion(
+    {
+      model: "deepseek-ai/DeepSeek-V3.1",
+      provider: "novita",
+      messages: [{
+        role: "system",
+        content: `You are a helpful assistant that rewrites prompts to make them better. All the prompts will be about creating a website or app.
+Try to make the prompt more detailed and specific to create a good UI/UX Design and good code.
+Format the result by following this format:
+${START_REWRITE_PROMPT}
+new prompt here
+${END_REWRITE_PROMPT}
+If you don't rewrite the prompt, return the original prompt.
+Make sure to return the prompt in the same language as the prompt you are given. Also IMPORTANT: Make sure to keep the original intent of the prompt. Improve it it needed, but don't change the original intent.
+`
+      },{ role: "user", content: prompt }],
+    },
+    billTo ? { billTo } : {}
+  );
+  
+  const responseContent = response.choices[0]?.message?.content;
+  if (!responseContent) {
+    return prompt;
+  }
+  const startIndex = responseContent.indexOf(START_REWRITE_PROMPT);
+  const endIndex = responseContent.indexOf(END_REWRITE_PROMPT);
+  return responseContent.substring(startIndex + START_REWRITE_PROMPT.length, endIndex);
+};

app/api/ask-ai/route.ts

@@ -19,6 +19,7 @@ import {
 } from "@/lib/prompts";
 import MY_TOKEN_KEY from "@/lib/get-cookie-name";
 import { Page } from "@/types";
+import { callAiRewritePrompt } from "@/app/actions/rewrite-prompt";
 
 const ipAddresses = new Map();
 
@@ -97,6 +98,13 @@ export async function POST(request: NextRequest) {
       ? PROVIDERS[selectedModel.autoProvider as keyof typeof PROVIDERS]
       : PROVIDERS[provider as keyof typeof PROVIDERS] ?? DEFAULT_PROVIDER;
 
+  let rewrittenPrompt = prompt;
+
+  if (prompt?.length < 240) {
+    
+    rewrittenPrompt = await callAiRewritePrompt(prompt, { token, billTo });
+  }
+
   try {
     const encoder = new TextEncoder();
     const stream = new TransformStream();
@@ -131,7 +139,7 @@ export async function POST(request: NextRequest) {
                 role: "user",
                 content: redesignMarkdown
                   ? `Here is my current design as a markdown:\n\n${redesignMarkdown}\n\nNow, please create a new design based on this markdown.`
-                  : prompt,
+                  : rewrittenPrompt,
               },
             ],
             max_tokens: selectedProvider.max_tokens,