Rename app.py to index.js

app.py

@@ -1,63 +0,0 @@
-import gradio as gr
-from huggingface_hub import InferenceClient
-
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-
-
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    messages = [{"role": "system", "content": system_message}]
-
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
-    messages.append({"role": "user", "content": message})
-
-    response = ""
-
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-
-        response += token
-        yield response
-
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
-
-
-if __name__ == "__main__":
-    demo.launch()

index.js

@@ -0,0 +1,174 @@
+const express = require('express');
+const axios = require('axios');
+const bodyParser = require('body-parser');
+const fs = require('fs').promises;
+const path = require('path');
+
+const app = express();
+const port = process.env.PORT || 3000;
+
+app.use(bodyParser.json());
+
+const sessionFilePath = path.join(__dirname, 'session.json');
+const SESSION_TIMEOUT = 5 * 60 * 1000;
+let sessions = {};
+
+// Data untuk statistik
+let totalRequests = 0;
+let requestTimestamps = [];
+
+async function loadSessions() {
+    try {
+        const data = await fs.readFile(sessionFilePath, 'utf-8');
+        sessions = JSON.parse(data);
+    } catch (error) {
+        if (error.code === 'ENOENT') {
+            sessions = {};
+        } else {
+            console.error('Error reading session file:', error);
+        }
+    }
+}
+
+async function saveSessions() {
+    try {
+        await fs.writeFile(sessionFilePath, JSON.stringify(sessions, null, 2), 'utf-8');
+    } catch (error) {
+        console.error('Error writing to session file:', error);
+    }
+}
+
+function cleanUpSessions() {
+    const now = Date.now();
+    for (const username in sessions) {
+        if (sessions[username].lastAccess && (now - sessions[username].lastAccess > SESSION_TIMEOUT)) {
+            delete sessions[username];
+        }
+    }
+}
+
+async function blackboxChat(content, username = null, prompt = null, webSearchMode = true) {
+    const url = "https://www.blackbox.ai/api/chat";
+    const headers = {
+        "Accept": "*/*",
+        "Accept-Language": "id-ID,en;q=0.5",
+        "Referer": "https://www.blackbox.ai/",
+        "Content-Type": "application/json",
+        "Origin": "https://www.blackbox.ai",
+        "Alt-Used": "www.blackbox.ai"
+    };
+
+    let messages;
+    if (username) {
+        cleanUpSessions();
+        if (!sessions[username]) {
+            sessions[username] = { messages: [], lastAccess: Date.now() };
+        } else {
+            sessions[username].lastAccess = Date.now();
+        }
+        sessions[username].messages.push({ role: "user", content });
+        messages = sessions[username].messages;
+    } else {
+        messages = [{ role: "user", content }];
+    }
+
+    const data = {
+        messages,
+        previewToken: null,
+        codeModelMode: true,
+        agentMode: {},
+        trendingAgentMode: {},
+        isMicMode: false,
+        userSystemPrompt: prompt || " and answers questions effectively. Keep a cool but supportive tone so users feel comfortable and appreciated. Be patient, empathetic and attentive to user needs, providing thoughtful and helpful responses. Make sure your way of speaking is indistinguishable from a human's, so that your presence feels natural and authentic. Use simple language, avoid technical jargon, remain consistent in providing solutions, and always be supportive and attentive. Show a humane attitude in every interaction and use Indonesian that is easy to understand.",
+        maxTokens: 1024,
+        webSearchMode: webSearchMode,
+        promptUrls: "",
+        isChromeExt: false,
+        githubToken: null
+    };
+
+    try {
+        const response = await axios.post(url, data, { headers });
+        let result = response.data;
+
+        // Existing replacement
+        result = result.replace(/\$@\$.*?\$@\$|v=.*?(\s|$)/g, '').trim(); // Remove content between $~~~$ markers
+        result = result.replace(/\$~~~\$.*?\$~~~\$/gs, '').trim();
+
+        const sourcesRegex = /Sources:.*?(\n|$)/s;
+        const sourcesMatch = result.match(sourcesRegex);
+        let sourcesText = "";
+
+        if (sourcesMatch) {
+            sourcesText = sourcesMatch[0].trim();
+            result = result.replace(sourcesRegex, '').trim();
+        }
+
+        if (sourcesText) {
+            result = `${result}\n\n${sourcesText}`;
+        }
+
+        if (username) {
+            sessions[username].messages.push({ role: "assistant", content: result });
+            await saveSessions();
+        }
+
+        return result;
+    } catch (error) {
+        console.error("Error fetching data:", error);
+        return "Maaf, terjadi kesalahan saat memproses permintaan Anda.";
+    }
+}
+
+loadSessions().catch(error => console.error("Failed to load sessions:", error));
+setInterval(cleanUpSessions, SESSION_TIMEOUT);
+
+// Middleware untuk melacak total permintaan dan RPS
+app.use((req, res, next) => {
+    totalRequests++;
+    const now = Date.now();
+    requestTimestamps.push(now);
+    requestTimestamps = requestTimestamps.filter(timestamp => now - timestamp <= 1000);
+    next();
+});
+
+app.post('/', async (req, res) => {
+    const { content, user, prompt, webSearchMode } = req.body;
+    try {
+        const result = await blackboxChat(content, user, prompt, webSearchMode);
+        res.json({ result });
+    } catch (error) {
+        console.error("Error:", error.message);
+        res.status(400).json({ error: error.message });
+    }
+});
+
+app.get('/api/stats', (req, res) => {
+    const seconds = Number(process.uptime());
+    const d = Math.floor(seconds / (3600 * 24));
+    const h = Math.floor(seconds % (3600 * 24) / 3600);
+    const m = Math.floor(seconds % 3600 / 60);
+    const s = Math.floor(seconds % 60);
+
+    const dDisplay = d > 0 ? d + (d == 1 ? " Hari, " : " Hari, ") : "";
+    const hDisplay = h > 0 ? h + (h == 1 ? " Jam, " : " Jam, ") : "";
+    const mDisplay = m > 0 ? m + (m == 1 ? " Menit, " : " Menit, ") : "";
+    const sDisplay = s > 0 ? s + (s == 1 ? " Detik" : " Detik") : "";
+
+    const runtime = dDisplay + hDisplay + mDisplay + sDisplay;
+    const rps = requestTimestamps.length;
+
+    res.json({
+        rps: rps, // Requests Per Second
+        totalRequests: totalRequests, // Total Requests
+        runtime: runtime // Runtime in the desired format
+    });
+});
+
+app.use((req, res) => {
+    res.sendFile(path.join(__dirname, 'index.html'));
+});
+
+app.listen(7860, () => {
+    console.log(`Server is listening at http://localhost:${port}`);
+});