Update index.js

index.js

@@ -1,85 +1,70 @@
 const express = require('express');
 const bodyParser = require('body-parser');
+const app = express();
 const axios = require('axios');
 const { DateTime, Duration } = require('luxon');
-const { format } = require("util");
-const puppeteer = require("puppeteer");
-const { GoogleGenerativeAI } = require("@google/generative-ai");
-const GPT4js = require('gpt4js');
-
-const app = express();
+const { format } = require("util")
 const port = 7860;
-
+const puppeteer = require("puppeteer");
 app.use(bodyParser.json());
 
-function convertTextToBase64(text) {
+function textToBase64(text) {
   return Buffer.from(text).toString('base64');
 }
 
-function convertBase64ToText(base64) {
+// Fungsi untuk mengembalikan teks dari Base64
+function base64ToText(base64) {
   return Buffer.from(base64, 'base64').toString('utf-8');
 }
 
-const genAI = new GoogleGenerativeAI(convertBase64ToText("QUl6YVN5QXYyeHlBblQ5dXZqVTcwdlN1YVBXNGRFa25vWHpnMDVN"));
+const { GoogleGenerativeAI } = require("@google/generative-ai");
+const genAI = new GoogleGenerativeAI(base64ToText("QUl6YVN5QXYyeHlBblQ5dXZqVTcwdlN1YVBXNGRFa25vWHpnMDVN"));
 
-const db_chatSessions = {};
+const db_chatHistory = {};
 
 function generateUID(length) {
   const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
+  const charactersLength = characters.length;
   let uid = '';
   for (let i = 0; i < length; i++) {
-    uid += characters.charAt(Math.floor(Math.random() * characters.length));
+    uid += characters.charAt(Math.floor(Math.random() * charactersLength));
   }
   return uid;
 }
 
-async function sendChatMessageV2(data_chat, msg) {
+async function sendMessageV2(data_chat, msg) {
   try {
-    const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
-    const chat = model.startChat(data_chat);
+    const model = genAI.getGenerativeModel({ model: "gemini-pro"});
+    
+    const chat = model.startChat(data_chat)
     const result = await chat.sendMessage(msg);
     const response = await result.response;
-    const text = response.text();
+	const text = response.text();
+	
     return text;
   } catch (error) {
     console.error(error);
-    throw error;
   }
 }
 
-async function clearInactiveUsersV2() {
+async function cleanInactiveUsersV2() {
   const currentTime = DateTime.local();
-  Object.keys(db_chatSessions).forEach((userId) => {
-    const user = db_chatSessions[userId];
-    if (user && user.lastChat && user.lastChat.plus(Duration.fromObject({ minutes: 20 })) < currentTime) {
-      delete db_chatSessions[userId];
-    }
-  });
-  setTimeout(clearInactiveUsersV2, 180000);
-}
 
-clearInactiveUsersV2();
+  Object.keys(db_chatHistory).forEach((userId) => {
+    const user = db_chatHistory[userId];
 
-async function pingWebsite() {
-  const browser = await puppeteer.launch({
-    headless: true,
-    args: ['--no-sandbox', '--disable-setuid-sandbox']
+    if (user && user.lastChat && user.lastChat.plus(Duration.fromObject({ minutes: 20 })) < currentTime) {
+      // User's last message was more than 10 minutes ago, remove the user from db_chatHistory
+      delete db_chatHistory[userId];
+    }
   });
-  const page = await browser.newPage();
-  await page.setUserAgent("Mozilla/5.0 (Linux; Android 10; SM-G965U Build/QP1A.190711.020; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/114.0.5735.141 Mobile Safari/537.36 [FB_IAB/FB4A;FBAV/420.0.0.32.61;]");
-  await page.goto('https://huggingface.co/spaces/Nexchan/gptdemo');
-  console.log("Ping");
-  await browser.close();
-}
 
-async function pingEvery5Hours() {
-  await pingWebsite();
-  setInterval(async () => {
-    await pingWebsite();
-  }, 5 * 60 * 60 * 1000);
+  // Schedule the next execution after 1 minute
+  setTimeout(cleanInactiveUsersV2, 180000);
 }
 
-pingEvery5Hours();
+// Start the continuous execution of cleanInactiveUsers
+cleanInactiveUsersV2();
 
 app.get('/v2/send', async (req, res) => {
   try {
@@ -91,41 +76,45 @@ app.get('/v2/send', async (req, res) => {
       });
     }
 
-    let chatData = db_chatSessions[id];
+    let chatData = db_chatHistory[id];
 
     if (!chatData) {
+      // Jika data riwayat obrolan tidak ada, buat data baru
       chatData = {
         lastChat: DateTime.local(),
         data: {
-          history: []
+		    history: [
+		      {
+		        role: "user",
+		        parts: [{text: "hello, can you answer my question with a cute kaomoji like this ｡⁠◕⁠‿⁠◕⁠｡"}]
+		      },
+		      {
+		        role: "model",
+		        parts: [{text: "sure (⁠◠⁠‿⁠◕⁠) , so.... (⁠ ⁠/⁠^⁠ω⁠^⁠)⁠/⁠♪⁠♪ how can I help dear user (⁠ ⁠╹⁠▽⁠╹⁠ ⁠)?"}],
+		      },
+		    ],
+		    generationConfig: {
+            maxOutputTokens: 500,
+          },
         }
-      };
-    }
-
-    function addToChatHistory(chatData, role, text) {
-      const lastEntry = chatData.data.history[chatData.data.history.length - 1];
-      if (lastEntry && lastEntry.role === role) {
-        lastEntry.parts.push({ text });
-      } else {
-        chatData.data.history.push({
-          role,
-          parts: [{ text }]
-        });
       }
     }
+    
+    const botResponse = await sendMessageV2(chatData.data, message);
 
-    if (!chatData.lastRole || chatData.lastRole === 'model') {
-      addToChatHistory(chatData, 'user', message);
-    }
-
-    const botResponse = await sendChatMessageV2(chatData.data, message);
-
-    addToChatHistory(chatData, 'model', botResponse);
-
+    // Simpan pesan user ke riwayat obrolan
+    chatData.data.history.push({
+      role: 'user',
+      parts: [{text: message}]
+    });
+    // Simpan bot response ke riwayat obrolan
+    chatData.data.history.push({
+    	role: 'model',
+        parts: [{text: botResponse}]
+    });
+    // Update waktu terakhir obrolan
     chatData.lastChat = DateTime.local();
-    chatData.lastRole = 'model'; // Set lastRole to model after bot response
-    db_chatSessions[id] = chatData;
-
+    db_chatHistory[id] = chatData;
     res.status(200).json({
       success: true,
       response: botResponse
@@ -138,21 +127,46 @@ app.get('/v2/send', async (req, res) => {
   }
 });
 
+
 app.get('/v2/listuser', (req, res) => {
-  const userList = Object.keys(db_chatSessions);
+  const userList = Object.keys(db_chatHistory);
   res.status(200).json({ userList });
 });
 
 app.get('/v2/get/message', (req, res) => {
   const id = req.query.id;
   if (!id) return res.status(400).json({ success: false, response: "input id" });
-  if (!db_chatSessions[id]) {
+  if (!db_chatHistory[id]) {
     return res.status(404).json({ error: 'User not found' });
   }
-  const userMessages = db_chatSessions[id].data.history;
+  const userMessages = db_chatHistory[id].data.history;
   res.status(200).json({ messages: userMessages });
 });
 
+// Fungsi untuk ping website
+async function pingWebsite() {
+  const browser = await puppeteer.launch({
+    headless: true,
+    args: ['--no-sandbox', '--disable-setuid-sandbox']
+  });
+  const page = await browser.newPage();
+  await page.setUserAgent("Mozilla/5.0 (Linux; Android 10; SM-G965U Build/QP1A.190711.020; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/114.0.5735.141 Mobile Safari/537.36 [FB_IAB/FB4A;FBAV/420.0.0.32.61;]");
+  await page.goto('https://huggingface.co/spaces/Nexchan/gptdemo');
+  console.log("Ping");
+  await browser.close();
+}
+
+// Ping website setiap 5 jam
+async function pingEvery5Hours() {
+  await pingWebsite();
+  setInterval(async () => {
+    await pingWebsite();
+  }, 5 * 60 * 60 * 1000); // 5 jam dalam milidetik
+}
+
+// Mulai ping
+pingEvery5Hours();
+
 app.listen(port, () => {
-  console.log(`Server is running on port ${port}`);
-});
+	console.log(`Server is running on port ${port}`);
+});