Update server.js

server.js

@@ -3,15 +3,13 @@ import path from "path";
 import { fileURLToPath } from "url";
 import dotenv from "dotenv";
 import bodyParser from "body-parser";
-import { InferenceClient } from "@huggingface/inference";
 
 dotenv.config();
 
 const app = express();
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
-const PORT = process.env.APP_PORT || 3000;
-const MODEL_ID = "deepseek-ai/DeepSeek-V3-0324";
+const PORT = process.env.APP_PORT || 7860;
 const OPENROUTER_API_KEY = process.env.OPENROUTER_API_KEY;
 
 app.use(bodyParser.json());
@@ -34,48 +32,77 @@ app.post("/api/ask-ai", async (req, res) => {
   res.setHeader("Cache-Control", "no-cache");
   res.setHeader("Connection", "keep-alive");
 
-  const client = new InferenceClient(OPENROUTER_API_KEY);
-  let completeResponse = "";
+  const messages = [
+    {
+      role: "system",
+      content:
+        "Você é um especialista em landing pages. Gere um HTML bonito e persuasivo para a copy enviada. Responda apenas com código HTML válido e completo.",
+    },
+    ...(previousPrompt ? [{ role: "user", content: previousPrompt }] : []),
+    ...(html ? [{ role: "assistant", content: `The current code is: ${html}` }] : []),
+    { role: "user", content: prompt },
+  ];
 
   try {
-    const chatCompletion = client.chatCompletionStream({
-      model: MODEL_ID,
-      provider: "openrouter",
-      messages: [
-        {
-          role: "system",
-          content:
-            "ONLY USE HTML, CSS AND JAVASCRIPT. Use TailwindCSS whenever possible. Return the full code as a single HTML file.",
-        },
-        ...(previousPrompt
-          ? [{ role: "user", content: previousPrompt }]
-          : []),
-        ...(html
-          ? [{ role: "assistant", content: `The current code is: ${html}.` }]
-          : []),
-        { role: "user", content: prompt },
-      ],
-      max_tokens: 4000,
+    const response = await fetch("https://openrouter.ai/api/v1/chat/completions", {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${OPENROUTER_API_KEY}`,
+        "Content-Type": "application/json",
+        "HTTP-Referer": "https://raypages.com", // personalize aqui
+        "X-Title": "RayPages AI",
+      },
+      body: JSON.stringify({
+        model: "deepseek-chat-v3.0",
+        stream: true,
+        messages,
+      }),
     });
 
-    while (true) {
-      const { done, value } = await chatCompletion.next();
-      if (done) break;
+    if (!response.ok || !response.body) {
+      return res.status(500).send({
+        ok: false,
+        message: "Failed to fetch OpenRouter response",
+      });
+    }
+
+    const reader = response.body.getReader();
+    const decoder = new TextDecoder("utf-8");
+    let completeResponse = "";
+
+    const stream = async () => {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
 
-      const chunk = value.choices[0]?.delta?.content;
-      if (chunk) {
-        completeResponse += chunk;
-        res.write(chunk);
+        const chunk = decoder.decode(value, { stream: true });
+        const matches = chunk.match(/"content":"([^"]*)"/g);
 
-        if (completeResponse.includes("</html>")) break;
+        if (matches) {
+          for (const match of matches) {
+            const text = match
+              .replace(/"content":"|"/g, "")
+              .replace(/\\n/g, "\n")
+              .replace(/\\t/g, "\t");
+
+            completeResponse += text;
+            res.write(text);
+
+            if (completeResponse.includes("</html>")) {
+              res.end();
+              return;
+            }
+          }
+        }
       }
-    }
+      res.end();
+    };
 
-    res.end();
-  } catch (error) {
+    stream();
+  } catch (err) {
     res.status(500).send({
       ok: false,
-      message: error.message || "AI request failed",
+      message: err.message || "AI request failed",
     });
   }
 });
@@ -85,5 +112,5 @@ app.get("*", (_req, res) => {
 });
 
 app.listen(PORT, () => {
-  console.log(`Server running on port ${PORT}`);
+  console.log(`✅ Server running on port ${PORT}`);
 });