chatgpt_assistant 0.1.4 → 0.1.6

data/lib/bots/telegram_bot.rb

@@ -1,14 +1,13 @@
 # frozen_string_literal: true
 
 require_relative "helpers/telegram_helper"
+require_relative "helpers/telegram_voice_helper"
 
 module ChatgptAssistant
   # This class is responsible for the telegram bot features
   class TelegramBot < ApplicationBot
     def start
       bot.listen do |message|
-        next if message.chat.type != "private" # disable group and channel messages, we will enable it later
-
         @msg = message
         @visitor = telegram_visited?(@msg.chat.id)
         next unless telegram_text_or_audio?
@@ -24,6 +23,7 @@ module ChatgptAssistant
     private
 
       include TelegramHelper
+      include TelegramVoiceHelper
 
       attr_accessor :msg, :visitor, :chat, :chat_id
 
@@ -40,123 +40,141 @@ module ChatgptAssistant
         when "/stop"
           stop_event
         when nil
-          nil_event
+          raise NilError
         else
           action_events
         end
+      rescue NilError => e
+        send_message e.message, msg.chat.id
       end
 
       def start_event
-        register_visitor_action("start", visitor.id) unless visitor.tel_user
-        register_user_action("start", visitor.tel_user.id) if visitor.tel_user
         telegram_send_start_message
       end
 
       def help_event
-        register_visitor_action("help", visitor.id) unless visitor.tel_user
-        register_user_action("help", visitor.tel_user.id) if visitor.tel_user
         help_messages.each { |m| send_message m, msg.chat.id }
       end
 
       def hist_event
-        return not_logged_in_message unless user
-        return no_chat_selected_message unless user.current_chat
-        return no_messages_founded_message if user.current_chat.messages.count.zero?
+        raise UserNotLoggedInError if user.nil?
+        raise NoChatSelectedError if user.current_chat.nil?
+        raise NoMessagesFoundedError if user.current_chat.messages.count.zero?
 
-        register_user_action("hist", user.id)
-        telegram_user_history.each do |m|
+        user.chat_history.each do |m|
           send_message m, msg.chat.id
         end
+      rescue NoChatSelectedError, UserNotLoggedInError, NoMessagesFoundedError => e
+        send_message e.message, msg.chat.id
       end
 
       def list_event
-        return unless valid_for_list_action?
+        raise UserNotLoggedInError if user.nil?
+        raise NoChatsFoundedError if user.chats.count.zero?
 
-        register_user_action("list", user.id)
-        send_message commom_messages[:chat_list], msg.chat.id
+        send_message common_messages[:chat_list], msg.chat.id
         chats_str = ""
         user.chats.each_with_index { |c, i| chats_str += "Chat #{i + 1} - #{c.title}\n" }
         send_message chats_str, msg.chat.id
+      rescue NoChatsFoundedError, UserNotLoggedInError => e
+        send_message e.message, msg.chat.id
       end
 
       def action_events
-        return login_event if msg.text.include?("login/")
-        return register_event if msg.text.include?("register/")
+        return auth_events if auth_event?
         return new_chat_event if msg.text.include?("new_chat/")
         return select_chat_event if msg.text.include?("sl_chat/")
         return telegram_chat_event unless telegram_actions?
 
-        invalid_command_error_message
+        raise InvalidCommandError
+      rescue InvalidCommandError => e
+        send_message e.message, msg.chat.id
+      end
+
+      def auth_event?
+        msg.text.include?("login/") || msg.text.include?("register/") || msg.text.include?("sign_out/")
+      end
+
+      def auth_events
+        return login_event if msg.text.include?("login/")
+        return register_event if msg.text.include?("register/")
+        return sign_out_event if msg.text.include?("sign_out/")
       end
 
       def login_event
-        register_visitor_action("login", visitor.id) unless visitor.tel_user
+        raise UserLoggedInError if user
+
         user_info = msg.text.split("/").last
         email, password = user_info.split(":")
         case telegram_user_auth(email, password, msg.chat.id)
         when "user not found"
-          user_not_found_error_message
+          raise UserNotFoundError
         when "wrong password"
-          wrong_password_error_message
+          raise WrongPasswordError
         when email
           user_logged_in_message
         end
+      rescue UserNotFoundError, WrongPasswordError, UserLoggedInError => e
+        send_message e.message, msg.chat.id
       end
 
       def register_event
-        register_visitor_action("register", visitor.id)
         user_info = msg.text.split("/").last
+        raise NoRegisterInfoError if user_info.nil?
+        raise UserLoggedInError if user
+
         email, password = user_info.split(":")
-        registered_email = telegram_user_create visitor.id, email, password
-        registered_email == email ? user_created_message : user_creation_error_message
+        raise NoRegisterInfoError if email.nil? || password.nil?
+
+        RegisterJob.perform_async(email, password, visitor.telegram_id)
+      rescue NoRegisterInfoError, UserLoggedInError => e
+        send_message e.message, msg.chat.id
+      end
+
+      def sign_out_event
+        raise UserNotLoggedInError if user.nil?
+
+        user.update(telegram_id: nil)
+        send_message success_messages[:user_logged_out], msg.chat.id
+      rescue UserNotLoggedInError => e
+        send_message e.message, msg.chat.id
       end
 
       def new_chat_event
-        return not_logged_in_message unless user
+        raise UserNotLoggedInError if user.nil?
 
-        register_user_action("new_chat", user.id)
-        telegram_create_chat
+        NewChatJob.perform_async(msg.text.split("/").last, user.id, msg.chat.id)
+      rescue UserNotLoggedInError => e
+        send_message e.message, msg.chat.id
       end
 
       def select_chat_event
-        return not_logged_in_message unless user
+        raise UserNotLoggedInError if user.nil?
 
-        register_user_action("select_chat", user.id)
         title = msg.text.split("/").last
         chat = user.chat_by_title(title)
-        return chat_not_found_message unless chat
+        raise ChatNotFoundError if chat.nil?
 
-        user.update(current_chat_id: chat.id)
-        send_message success_messages[:chat_selected], msg.chat.id
-      end
+        raise ChatNotFoundError unless user.update(current_chat_id: chat.id)
 
-      def stop_event
-        register_event_action("stop", visitor.id) unless visitor.tel_user
-        register_user_action("stop", visitor.tel_user.id) if visitor.tel_user
-        send_message commom_messages[:stop], msg.chat.id
-        bot.api.leave_chat(chat_id: msg.chat.id)
-      end
-
-      def nil_event
-        register_event_action("nil", visitor.id) unless visitor.tel_user
-        register_user_action("nil", visitor.tel_user.id) if visitor.tel_user
-        send_message error_messages[:nil], msg.chat.id
+        send_message success_messages[:chat_selected], msg.chat.id
+      rescue UserNotLoggedInError, ChatNotFoundError => e
+        send_message e.message, msg.chat.id
       end
 
       def audio_event
-        return not_logged_in_message unless user
-        return no_chat_selected_message if user.current_chat_id.nil?
+        raise UserNotLoggedInError if user.nil?
+        raise NoChatSelectedError if user.current_chat.nil?
 
-        register_user_action("audio", user.id)
         user_audio = transcribe_file(telegram_audio_url)
         message = Message.new(content: user_audio[:text], chat_id: user.current_chat_id, role: "user")
-        if message.save
-          ai_response = telegram_process_ai_voice(user_audio[:file])
-          telegram_send_voice_message(voice: ai_response[:voice], text: ai_response[:text])
-          delete_file ai_response[:voice]
-        else
-          send_message error_messages[:message_creation_error], msg.chat.id
-        end
+        raise MessageNotSavedError unless message.save
+
+        ai_response = telegram_process_ai_voice(user_audio[:file])
+        telegram_send_voice_message(voice: ai_response[:voice], text: ai_response[:text])
+        delete_file ai_response[:voice]
+      rescue UserNotLoggedInError, NoChatSelectedError, MessageNotSavedError => e
+        send_message e.message, msg.chat.id
       end
   end
 end

data/lib/chatgpt_assistant/chatter.rb

@@ -10,27 +10,36 @@ module ChatgptAssistant
       @openai_api_key = openai_api_key
     end
 
-    def chat(message, chat_id)
+    def chat(message, chat_id, error_message)
+      @error_message = error_message
       @chat_id = chat_id
       @message = message
       @response = request(message)
       @json = JSON.parse(response.body)
 
-      return error_log if response.status != 200
+      return no_response_error if json["choices"].empty?
+      return bot_offline_error if response.status != 200
 
       text = json["choices"][0]["message"]["content"]
 
       Message.create(content: text, role: "assistant", chat_id: chat_id)
       text
+    rescue StandardError => e
+      Error.create(message: e.message, backtrace: e.backtrace)
+      error_message
     end
 
     private
 
       attr_reader :openai_api_key, :response, :message
-      attr_accessor :chat_id, :json
+      attr_accessor :chat_id, :json, :error_message
 
-      def error_log
-        "Algo deu errado, tente novamente mais tarde." # TODO: use a DefaultMessage object
+      def no_response_error
+        "I'm sorry, I didn't understand you. Please, try again."
+      end
+
+      def bot_offline_error
+        "I'm sorry, I'm offline. Please, try again later."
       end
 
       def header
@@ -59,7 +68,9 @@ module ChatgptAssistant
                    end
         {
           model: "gpt-3.5-turbo",
-          messages: messages
+          messages: messages,
+          max_tokens: 1000,
+          temperature: 0.1
         }.to_json
       end
 

data/lib/chatgpt_assistant/config.rb

@@ -44,13 +44,27 @@ module ChatgptAssistant
       ActiveRecord::Base.logger = Logger.new($stdout) if ENV["ENV_TYPE"] == "development"
     end
 
+    def create_db
+      db_connection
+      return if database_exists?
+
+      ActiveRecord::Base.establish_connection(
+        adapter: "postgresql",
+        host: database_host,
+        port: 5432,
+        database: "postgres",
+        username: database_username,
+        password: database_password
+      )
+      ActiveRecord::Base.logger = Logger.new($stdout) if ENV["ENV_TYPE"] == "development"
+      ActiveRecord::Base.connection.create_database(database_name)
+    end
+
     def migrate
       db_connection
-      ActiveRecord::Base.logger = Logger.new($stdout)
+      ActiveRecord::Base.logger = Logger.new($stdout) if ENV["ENV_TYPE"] == "development"
       VisitorMigration.new.migrate(:up)
-      VisitorActionsMigration.new.migrate(:up)
       UserMigration.new.migrate(:up)
-      UserActionsMigration.new.migrate(:up)
       ChatMigration.new.migrate(:up)
       MessageMigration.new.migrate(:up)
       ErrorMigration.new.migrate(:up)
@@ -59,5 +73,11 @@ module ChatgptAssistant
     private
 
       attr_reader :database_host, :database_name, :database_username, :database_password
+
+      def database_exists?
+        ActiveRecord::Base.connection
+      rescue ActiveRecord::NoDatabaseError
+        false
+      end
   end
 end

data/lib/chatgpt_assistant/default_messages.rb

@@ -3,15 +3,15 @@
 module ChatgptAssistant
   # This class is responsible for storing the default messages
   class DefaultMessages
-    def initialize(language = "en")
+    def initialize(language = "en", _discord_prefix = "!")
       @language = language
       load_message_context
     end
 
-    attr_reader :language, :commom_messages, :success_messages, :error_messages, :help_messages
+    attr_reader :language, :common_messages, :success_messages, :error_messages, :help_messages
 
     def load_message_context
-      @commom_messages = send("commom_messages_#{language}")
+      @common_messages = send("common_messages_#{language}")
       @success_messages = send("success_messages_#{language}")
       @error_messages = send("error_messages_#{language}")
       @help_messages = send("help_messages_#{language}")
@@ -19,7 +19,7 @@ module ChatgptAssistant
 
     private
 
-      def commom_messages_pt
+      def common_messages_pt
         {
           start: "Olá, eu sou o Chatgpt Assistant, um chatbot que usa a API da OpenAI para responder suas perguntas no Telegram e Discord.",
           stop: "Até mais!",
@@ -42,6 +42,7 @@ module ChatgptAssistant
           chat_created: "Chat criado com sucesso!",
           chat_selected: "Chat selecionado com sucesso!",
           user_logged_in: "Login realizado com sucesso!",
+          user_logged_out: "Logout realizado com sucesso!",
           voice_channel_connected: "O bot entrou no canal de voz com sucesso!"
         }
       end
@@ -49,26 +50,34 @@ module ChatgptAssistant
       def error_messages_pt
         {
           nil: "Não entendi o que você disse. Tente novamente",
-          email: "O email que você digitou não é válido. Tente novamente",
-          password: "A senha que você digitou não é válida. Tente novamente",
+          wrong_email: "O email que você digitou não é válido. Tente novamente",
           wrong_password: "A senha que você digitou não é válida. Tente novamente",
-          user: "O usuário que você digitou não existe. Tente novamente",
-          user_creation: "Erro ao criar usuário. Tente novamente",
+
           user_already_exists: "O usuário que você digitou já existe. Tente novamente",
-          chat_creation: "Erro ao criar chat. Tente novamente",
-          no_messages_founded: "Nenhuma mensagem encontrada",
+          chat_already_exists: "Você já possui um chat com este titulo. Tente novamente",
+
+          no_register_info: "Você não digitou o email e a senha. Tente novamente",
+          sign_up_error: "Erro ao criar usuário. Tente novamente",
+          chat_not_created_error: "Erro ao criar chat. Tente novamente",
+          message_not_created_error: "Erro ao criar mensagem. Tente novamente",
           no_chat_selected: "Nenhum chat selecionado",
-          chat_not_found: "Chat não encontrado",
+
+          no_messages_founded: "Nenhuma mensagem encontrada",
           no_chats_founded: "Nenhum chat encontrado",
+          chat_not_found: "Chat não encontrado",
+          user_not_found: "O usuário que você digitou não existe. Tente novamente",
+          user_not_registered: "O usuário não está registrado no sistema",
+          user_logged_in: "Usuário está logado no sistema, faça logout para continuar",
           user_not_logged_in: "Usuário não logado",
-          user_not_found: "Usuário não encontrado",
-          something_went_wrong: "Algo deu errado. Tente novamente mais tarde.",
-          message_history_too_long: "O histórico mensagem é muito longo.",
-          text_length: "O texto de resposta é muito longo. Tente diminuir a quantidade de respostas na mesma mensagem.",
+
           user_not_in_voice_channel: "Você não está em um canal de voz.",
           bot_not_in_voice_channel: "O bot não está em um canal de voz.",
+
+          message_history_too_long: "O histórico mensagem é muito longo.",
+          text_length_too_long: "O texto de resposta é muito longo. Tente diminuir a quantidade de respostas na mesma mensagem.",
+
           invalid_command: "Comando inválido. Tente novamente.",
-          message_creation_error: "Erro ao criar mensagem. Tente novamente."
+          something_went_wrong: "Algo deu errado. Tente novamente mais tarde."
         }
       end
 
@@ -83,7 +92,18 @@ module ChatgptAssistant
          "Para ver esta mensagem novamente, digite /help"]
       end
 
-      def commom_messages_en
+      def help_message_discord_pt
+        ["Para começar a conversar comigo, digite #{discord_prefix}start",
+         "Para parar de conversar comigo, digite #{discord_prefix}stop",
+         "Para se registrar no sistema, digite #{discord_prefix}register email:senha (a senha deve ser um numero de 4 digitos ex: 1234)",
+         "Para fazer login no sistema, digite #{discord_prefix}login email:senha (a senha deve ser um numero de 4 digitos ex: 1234)",
+         "Para criar um novo chat, digite #{discord_prefix}new_chat nome do chat",
+         "Para selecionar um chat, digite #{discord_prefix}sl_chat nome do chat",
+         "Para listar os chats que você criou, digite #{discord_prefix}list",
+         "Para ver esta mensagem novamente, digite #{discord_prefix}help"]
+      end
+
+      def common_messages_en
         {
           start: "Hello, I'm the Chatgpt Assistant, a chatbot that uses the OpenAI API to answer your questions on Telegram and Discord.",
           stop: "See you later!",
@@ -105,32 +125,43 @@ module ChatgptAssistant
           user_created: "User created successfully!",
           chat_created: "Chat created successfully!",
           chat_selected: "Chat selected successfully!",
-          user_logged_in: "Login successfully!"
+          user_logged_in: "Login successfully!",
+          user_logged_out: "Logout successfully!",
+          voice_channel_connected: "The bot entered the voice channel successfully!"
         }
       end
 
       def error_messages_en
         {
-          nil: "I didn't understand what you said. Try again",
-          email: "The email you typed is not valid. Try again",
-          password: "The password you typed is not valid. Try again",
+          nil: "I don't understand what you said. Try again",
+          wrong_email: "The email you typed is not valid. Try again",
           wrong_password: "The password you typed is not valid. Try again",
-          user: "The user you typed does not exist. Try again",
-          user_creation: "Error creating user. Try again",
-          chat_creation: "Error creating chat. Try again",
-          no_messages_founded: "No messages found",
+
+          user_already_exists: "The user you typed already exists. Try again",
+          chat_already_exists: "You already have a chat with this title. Try again",
+
+          no_register_info: "You did not type the email and password. Try again",
+          sign_up_error: "Error creating user. Try again",
+          chat_not_created_error: "Error creating chat. Try again",
+          message_not_created_error: "Error creating message. Try again",
           no_chat_selected: "No chat selected",
+
+          no_messages_founded: "No messages found",
           no_chats_founded: "No chats found",
           chat_not_found: "Chat not found",
+          user_not_found: "The user you typed does not exist. Try again",
+          user_not_registered: "The user is not registered in the system",
+          user_logged_in: "User is logged in the system, do logout to continue",
           user_not_logged_in: "User not logged in",
-          user_not_found: "User not found",
-          something_went_wrong: "Something went wrong. Try again later.",
-          message_history_too_long: "The message history is too long.",
-          text_length: "The response text is too long. Try to reduce the number of answers in the same message.",
+
           user_not_in_voice_channel: "You are not in a voice channel.",
           bot_not_in_voice_channel: "The bot is not in a voice channel.",
+
+          message_history_too_long: "The message history is too long.",
+          text_length_too_long: "The response text is too long. Try to reduce the number of responses in the same message.",
+
           invalid_command: "Invalid command. Try again.",
-          message_creation_error: "Error creating message. Try again."
+          something_went_wrong: "Something went wrong. Try again later."
         }
       end
 
@@ -144,5 +175,16 @@ module ChatgptAssistant
          "To list the chats you created, type /list",
          "To see this message again, type /help"]
       end
+
+      def help_message_discord_en
+        ["To start talking to me, type #{discord_prefix}start",
+         "To stop talking to me, type #{discord_prefix}stop",
+         "To register in the system, type #{discord_prefix}register email:password (the password must be a 4 digit number ex: 1234)",
+         "To log in to the system, type #{discord_prefix}login email:password (the password must be a 4 digit number ex: 1234)",
+         "To create a new chat, type #{discord_prefix}new_chat chat name",
+         "To select a chat, type #{discord_prefix}sl_chat chat name",
+         "To list the chats you created, type #{discord_prefix}list",
+         "To see this message again, type #{discord_prefix}help"]
+      end
   end
 end