improved commands & added direct communication with AI

Author: axbuglak

lib/chatbot.js

@@ -35,9 +35,10 @@ class ChatBot {
       const prediction = await this.api.ai.predict({
         form: ctx.session.form,
         language: this.lang.name, // Use the ChatBot's prop for language
+        token: ctx.from.id,
       });
       await ctx.reply(prediction.answer);
-      return ctx.scene.leave();
+      return await ctx.reply(ctx.session.lang.furtherQuestion);
     }
     const question = ctx.session.actualField;
     if (question.type === 'text' || question.type === 'input') {
@@ -56,17 +57,16 @@ class ChatBot {
   }
 
   async handleField(ctx) {
-    // think about how to remove it from here -> cohesion
-    if (ctx.message.text === '/stop') return ctx.scene.leave();
-    if (ctx.message.text === '/language') {
-      await ctx.scene.leave();
-      // cohesion
-      const languagesMarkups = Object.keys(languages).map((name) =>
-        Markup.button.callback(name, name),
-      );
-      ctx.reply('Choose', Markup.inlineKeyboard(languagesMarkups).oneTime());
-      return true;
+    if (ctx.session.test.length === 0) {
+      const prediction = await this.api.ai.ask({
+        question: ctx.message.text,
+        language: this.lang.name, // Use the ChatBot's prop for language
+        token: ctx.from.id,
+      });
+      await ctx.reply(prediction.answer);
+      return;
     }
+
     const answer = ctx.message.text;
     const field = ctx.session.actualField;
     const question = ctx.session.test[0];
@@ -76,20 +76,23 @@ class ChatBot {
       ctx.session.test.shift();
       ctx.session.actualField = ctx.session.test[0];
       await this.showQuestion(ctx);
-      return true;
+      return;
     }
     // only 2 types of questions
     if (field.type === 'choice') {
       const choice = Object.values(field.choices).find(
         (c) => c.name === answer,
       );
-      if (!choice) return await ctx.reply(this.lang.choose_options);
+      if (!choice) {
+        await ctx.reply(this.lang.choose_options);
+        return;
+      }
       if (choice.type === 'text') {
         ctx.session.form += question.name + ': ' + choice.name + '\n';
         ctx.session.test.shift();
         ctx.session.actualField = ctx.session.test[0];
         await this.showQuestion(ctx);
-        return true;
+        return;
       }
       // for input and choice types
       await ctx.reply(choice.description);
@@ -101,9 +104,9 @@ class ChatBot {
       ctx.session.test.shift();
       ctx.session.actualField = ctx.session.test[0];
       await this.showQuestion(ctx);
-      return true;
+      return;
     }
-    return false;
+    return;
   }
 }
 module.exports = ChatBot;

main.js

@@ -10,6 +10,8 @@ const API_URL = process.env.API_URL;
 const endpoints = {
   ai: {
     predict: 'post',
+    ask: 'post',
+    leave: 'post',
   },
 };
 
@@ -30,15 +32,25 @@ const endpoints = {
   // eslint-disable-next-line camelcase
   const options = { reply_markup: { remove_keyboard: true } };
 
+  const stage = new Scenes.Stage([formScene]);
   formScene.enter(async (ctx) => await chatBot.start(ctx));
   const listener = async (ctx) => await chatBot.handleField(ctx);
   formScene.on('text', listener);
   formScene.on('message', listener);
   formScene.leave(async (ctx) => {
+    ctx.session.test = [];
+    ctx.session.form = '';
+    await api.ai.leave({ token: ctx.from.id });
     await ctx.reply(ctx.session.lang.leave, options);
     await ctx.reply(ctx.session.lang.help);
   });
-  const stage = new Scenes.Stage([formScene]);
+  stage.command('stop', async (ctx) => await ctx.scene.leave());
+  stage.command('language', async (ctx) => {
+    const languagesMarkups = Object.keys(languages).map((name) =>
+      Markup.button.callback(name, name),
+    );
+    ctx.reply('Choose', Markup.inlineKeyboard(languagesMarkups).oneTime());
+  });
 
   const logStatics = () => {
     console.clear();
@@ -95,6 +107,7 @@ const endpoints = {
     bot.action(lang, async (ctx) => {
       ctx.session.lang = languages[lang];
       await ctx.reply(ctx.session.lang.done);
+      if (ctx.session.test && ctx.session.test.length) await chatBot.start(ctx);
     });
   });
 

static/languages.json

@@ -1,5 +1,6 @@
 {
   "english": {
+    "furtherQuestion": "Ask me whatever you want if you would like to know more about your future or have some tips! 😉 \n Or you can leave this dialog using /stop",
     "name": "english",
     "publicity": "Your amazing ad could be here! 🌟",
     "welcome": "Hi there! We're thrilled to have you with us! 😊",
@@ -14,6 +15,7 @@
     "getHelp": "Here is how we can help you: \n\n1. /start - Start the conversation \n2. /stop - Exit the conversation \n3. /language - Change the language \n4. /help - Get help"
   },
   "russian": {
+    "furtherQuestion": "Ты так же можешь задать мне любой вопрос о твоем будущем, а так получить советы или мотивацию! 😉 \n Или ты можешь просто покинуть этот диалог /stop",
     "name": "russian",
     "publicity": "Ваша реклама могла бы быть здесь! 🌟",
     "welcome": "Привет! Мы рады, что вы с нами! 😊",
@@ -28,6 +30,7 @@
     "getHelp": "Здесь все, что может тебе помочь: \n\n1. /start - Начать разговор \n2. /stop - Завершить разговор \n3. /language - Изменить язык \n4. /help - Получить помощь"
   },
   "ukrainian": {
+    "furtherQuestion": "Ти так само можеш поставити мені будь-яке питання про твоє майбутнє, а так отримати поради чи мотивацію! 😉 \n Або ти можеш просто покинути цей діалог /stop",
     "name": "ukrainian",
     "publicity": "Ваша реклама може бути тут! 🌟",
     "welcome": "Вітаємо! Ми раді бачити вас тут! 😊",