Update Dockerfile

Dockerfile

@@ -41,7 +41,6 @@ const generateRandomUser = () => {
     const email = `${emailUsername}@gmail.com`;
     const id = Array.from({ length: 21 }, () => Math.floor(Math.random() * 10)).join('');
     const image = `https://robohash.org/${id}.png?set=set2&bgset=bg1&size=96x96`;
-
     return { name: fullName, email, image, id };
 };
 
@@ -66,114 +65,150 @@ const parseApiResponse = (data) => {
     return { answer: data, sources: [] };
 };
 
+const createPayload = (prompt, system, web, asyncMode = false) => {
+    const newUserMessage = { role: 'user', content: prompt, id: generateId() };
+    const messagesForPayload = [newUserMessage];
+
+    let webSearchModeOption = { autoMode: true, webMode: false, offlineMode: false };
+    if (web) {
+        const webMode = String(web).toLowerCase();
+        if (webMode === 'true') webSearchModeOption = { autoMode: false, webMode: true, offlineMode: false };
+        else if (webMode === 'false') webSearchModeOption = { autoMode: false, webMode: false, offlineMode: true };
+    }
+
+    const randomUser = generateRandomUser();
+    const expiresDate = new Date();
+    expiresDate.setFullYear(expiresDate.getFullYear() + 1);
+
+    return {
+        messages: messagesForPayload,
+        id: newUserMessage.id,
+        userSystemPrompt: system || null,
+        validated: VALIDATED_TOKEN,
+        previewToken: null,
+        userId: null,
+        codeModelMode: true,
+        trendingAgentMode: {},
+        isMicMode: false,
+        maxTokens: 1024,
+        playgroundTopP: null,
+        playgroundTemperature: null,
+        isChromeExt: false,
+        githubToken: '',
+        clickedAnswer2: false,
+        clickedAnswer3: false,
+        clickedForceWebSearch: false,
+        visitFromDelta: false,
+        isMemoryEnabled: false,
+        mobileClient: false,
+        userSelectedModel: null,
+        userSelectedAgent: 'VscodeAgent',
+        imageGenerationMode: false,
+        imageGenMode: 'autoMode',
+        webSearchModePrompt: false,
+        deepSearchMode: false,
+        domains: null,
+        vscodeClient: false,
+        codeInterpreterMode: false,
+        customProfile: {
+            name: '',
+            occupation: '',
+            traits: [],
+            additionalInfo: '',
+            enableNewChats: false,
+        },
+        webSearchModeOption,
+        session: {
+            user: randomUser,
+            expires: expiresDate.toISOString(),
+            isNewUser: Math.random() < 0.1,
+        },
+        isPremium: false,
+        subscriptionCache: {
+            status: 'FREE',
+            expiryTimestamp: null,
+            lastChecked: Date.now(),
+            isTrialSubscription: false,
+        },
+        beastMode: false,
+        reasoningMode: false,
+        designerMode: false,
+        workspaceId: '',
+        asyncMode,
+        integrations: {},
+        isTaskPersistent: false,
+        selectedElement: null,
+    };
+};
+
 app.use(cors());
 app.use(express.json());
 
+const chatApiUrl = 'https://www.blackbox.ai/api/chat';
+const headers = {
+    Accept: '*/*',
+    'Content-Type': 'application/json',
+    Origin: 'https://www.blackbox.ai',
+    Referer: 'https://www.blackbox.ai/',
+    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36',
+};
+
 app.post('/api/generate', async (req, res) => {
     const { prompt, system, web } = req.body;
-
     if (!prompt) {
         return res.status(400).json({ error: 'Request body harus menyertakan "prompt"' });
     }
-
     try {
-        const newUserMessage = { role: 'user', content: prompt, id: generateId() };
-        const messagesForPayload = [newUserMessage];
-
-        let webSearchModeOption = { autoMode: true, webMode: false, offlineMode: false };
-        if (web) {
-            const webMode = String(web).toLowerCase();
-            if (webMode === 'true') webSearchModeOption = { autoMode: false, webMode: true, offlineMode: false };
-            else if (webMode === 'false') webSearchModeOption = { autoMode: false, webMode: false, offlineMode: true };
-        }
-
-        const randomUser = generateRandomUser();
-        const expiresDate = new Date();
-        expiresDate.setFullYear(expiresDate.getFullYear() + 1);
-
-        const payload = {
-            messages: messagesForPayload,
-            id: newUserMessage.id,
-            userSystemPrompt: system || null,
-            validated: VALIDATED_TOKEN,
-            previewToken: null,
-            userId: null,
-            codeModelMode: true,
-            trendingAgentMode: {},
-            isMicMode: false,
-            maxTokens: 1024,
-            playgroundTopP: null,
-            playgroundTemperature: null,
-            isChromeExt: false,
-            githubToken: '',
-            clickedAnswer2: false,
-            clickedAnswer3: false,
-            clickedForceWebSearch: false,
-            visitFromDelta: false,
-            isMemoryEnabled: false,
-            mobileClient: false,
-            userSelectedModel: null,
-            userSelectedAgent: 'VscodeAgent',
-            imageGenerationMode: false,
-            imageGenMode: 'autoMode',
-            webSearchModePrompt: false,
-            deepSearchMode: false,
-            domains: null,
-            vscodeClient: false,
-            codeInterpreterMode: false,
-            customProfile: {
-                name: '',
-                occupation: '',
-                traits: [],
-                additionalInfo: '',
-                enableNewChats: false,
-            },
-            webSearchModeOption,
-            session: {
-                user: randomUser,
-                expires: expiresDate.toISOString(),
-                isNewUser: Math.random() < 0.1,
-            },
-            isPremium: false,
-            subscriptionCache: {
-                status: 'FREE',
-                expiryTimestamp: null,
-                lastChecked: Date.now(),
-                isTrialSubscription: false,
-            },
-            beastMode: false,
-            reasoningMode: false,
-            designerMode: false,
-            workspaceId: '',
-            asyncMode: false,
-            integrations: {},
-            isTaskPersistent: false,
-            selectedElement: null,
-        };
-
-        const chatApiUrl = 'https://www.blackbox.ai/api/chat';
-        const headers = {
-            Accept: '*/*',
-            'Content-Type': 'application/json',
-            Origin: 'https://www.blackbox.ai',
-            Referer: 'https://www.blackbox.ai/',
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36',
-        };
-
+        const payload = createPayload(prompt, system, web, false);
         const chatResponse = await axios.post(chatApiUrl, payload, { headers });
         const assistantRawResponse = chatResponse.data;
         const parsedResult = parseApiResponse(assistantRawResponse);
-
         res.send(parsedResult.answer);
+    } catch (error) {
+        console.error("Error pada /api/generate:", error.message);
+        res.status(500).send("Terjadi kesalahan pada server.");
+    }
+});
+
+app.post('/api/stream', async (req, res) => {
+    const { prompt, system, web } = req.body;
+    if (!prompt) {
+        return res.status(400).json({ error: 'Request body harus menyertakan "prompt"' });
+    }
+    try {
+        const payload = createPayload(prompt, system, web, true);
+        const chatResponse = await axios.post(chatApiUrl, payload, {
+            headers,
+            responseType: 'stream'
+        });
+
+        res.setHeader('Content-Type', 'text/plain; charset=utf-8');
+        res.setHeader('Cache-Control', 'no-cache');
+        res.setHeader('Connection', 'keep-alive');
+        
+        chatResponse.data.pipe(res);
+
+        chatResponse.data.on('error', (err) => {
+            console.error("Stream error:", err.message);
+            if (!res.headersSent) {
+                res.status(500).send("Terjadi kesalahan pada stream.");
+            }
+            res.end();
+        });
 
     } catch (error) {
-        console.error("Error saat scraping Blackbox AI:", error);
-        res.status(500).send("Terjadi kesalahan pada server saat memproses permintaan Anda.");
+        console.error("Error pada /api/stream:", error.message);
+        if (!res.headersSent) {
+            res.status(500).send("Terjadi kesalahan pada server saat memulai stream.");
+        }
     }
 });
 
 app.listen(PORT, () => {
     console.log(`Server berjalan di http://localhost:${PORT}`);
+    console.log('Endpoints tersedia:');
+    console.log('  - POST /api/generate (non-streaming)');
+    console.log('  - POST /api/stream (streaming)');
 });
 EOF