Update index.js

index.js

@@ -12,26 +12,18 @@ const port = 7860;
 
 app.use(bodyParser.json());
 
-
-
-
-// Fungsi untuk mengonversi teks menjadi Base64
 function convertTextToBase64(text) {
   return Buffer.from(text).toString('base64');
 }
 
-// Fungsi untuk mengembalikan teks dari Base64
 function convertBase64ToText(base64) {
   return Buffer.from(base64, 'base64').toString('utf-8');
 }
 
-// Inisialisasi GoogleGenerativeAI
 const genAI = new GoogleGenerativeAI(convertBase64ToText("QUl6YVN5QXYyeHlBblQ5dXZqVTcwdlN1YVBXNGRFa25vWHpnMDVN"));
 
-// Data sesi chat untuk versi 2
 const db_chatSessions = {};
 
-// Fungsi untuk menghasilkan UID
 function generateUID(length) {
   const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
   let uid = '';
@@ -41,43 +33,54 @@ function generateUID(length) {
   return uid;
 }
 
-// Fungsi untuk mengirim pesan versi 2
 async function sendChatMessageV2(data_chat, msg) {
   try {
     const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
-
     const chat = model.startChat(data_chat);
     const result = await chat.sendMessage(msg);
     const response = await result.response;
     const text = response.text();
-
     return text;
   } catch (error) {
     console.error(error);
+    throw error;
   }
 }
 
-// Fungsi untuk membersihkan pengguna tidak aktif versi 2
 async function clearInactiveUsersV2() {
   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) {
-      // Hapus pengguna yang tidak aktif lebih dari 20 menit dari db_chatSessions
       delete db_chatSessions[userId];
     }
   });
-
-  // Jadwalkan eksekusi berikutnya setelah 3 menit
   setTimeout(clearInactiveUsersV2, 180000);
 }
 
-// Mulai eksekusi terus-menerus untuk membersihkan pengguna tidak aktif
 clearInactiveUsersV2();
 
-// Endpoint untuk mengirim pesan versi 2
+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();
+}
+
+async function pingEvery5Hours() {
+  await pingWebsite();
+  setInterval(async () => {
+    await pingWebsite();
+  }, 5 * 60 * 60 * 1000);
+}
+
+pingEvery5Hours();
+
 app.get('/v2/send', async (req, res) => {
   try {
     const { id, message } = req.query;
@@ -91,42 +94,35 @@ app.get('/v2/send', async (req, res) => {
     let chatData = db_chatSessions[id];
 
     if (!chatData) {
-      // Jika data riwayat obrolan tidak ada, buat data baru
       chatData = {
         lastChat: DateTime.local(),
         data: {
-          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,
-          },
+          history: []
         }
       };
     }
 
+    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 }]
+        });
+      }
+    }
+
+    addToChatHistory(chatData, 'user', message);
+
     const botResponse = await sendChatMessageV2(chatData.data, message);
 
-    // 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
+    addToChatHistory(chatData, 'model', botResponse);
+
     chatData.lastChat = DateTime.local();
     db_chatSessions[id] = chatData;
+
     res.status(200).json({
       success: true,
       response: botResponse
@@ -139,13 +135,11 @@ app.get('/v2/send', async (req, res) => {
   }
 });
 
-// Endpoint untuk daftar pengguna versi 2
 app.get('/v2/listuser', (req, res) => {
   const userList = Object.keys(db_chatSessions);
   res.status(200).json({ userList });
 });
 
-// Endpoint untuk mendapatkan pesan versi 2
 app.get('/v2/get/message', (req, res) => {
   const id = req.query.id;
   if (!id) return res.status(400).json({ success: false, response: "input id" });
@@ -156,30 +150,6 @@ app.get('/v2/get/message', (req, res) => {
   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}`);
-});
+});