slack-smart-bot 1.13.2 → 1.14.0

data/lib/slack/smart-bot/commands/general/ai/open_ai/open_ai_generate_image.rb

@@ -0,0 +1,65 @@
+class SlackSmartBot
+  module Commands
+    module General
+      module AI
+        module OpenAI
+          def open_ai_generate_image(message, delete_history = false, repeat: false)
+            save_stats(__method__)
+            get_personal_settings()
+            @ai_open_ai, message_connect = SlackSmartBot::AI::OpenAI.connect(@ai_open_ai, config, @personal_settings, reconnect: delete_history)
+            respond message_connect if message_connect
+            user = Thread.current[:user]
+            if !@ai_open_ai[user.name].nil? and !@ai_open_ai[user.name][:client].nil?
+              @ai_open_ai_image ||= {}
+              @ai_open_ai_image[user.name] ||= []
+              react :art
+              begin
+                @ai_open_ai_image[user.name] = [] if delete_history
+                if delete_history and message == ""
+                  respond "*OpenAI*: Let's start a new image generation. Use `?i PROMPT` to generate an image."
+                elsif repeat and @ai_open_ai_image[user.name].empty? and message == ""
+                  respond "*OpenAI*: Sorry, I need to generate an image first. Use `?i PROMPT` to generate an image."
+                else
+                  @ai_open_ai_image[user.name] << message unless repeat
+                  success, res = SlackSmartBot::AI::OpenAI.send_image_generation(@ai_open_ai[user.name][:client], @ai_open_ai_image[user.name].join("\n"), @ai_open_ai[user.name][:image_size])
+                  if success
+                    urls = res
+                    urls = [urls] if urls.is_a?(String)
+                    if urls.nil? or urls.empty?
+                      respond "*OpenAI*: Sorry, I'm having some problems. OpenAI was not able to generate an image."
+                    else
+                      session_name = @ai_open_ai_image[user.name].first[0..29]
+                      messagersp = "OpenAI Session: _<#{session_name}...>_ (id:#{@ai_open_ai_image[user.name].object_id})"
+                      message = "Repeat" if repeat
+                      require "uri"
+                      urls.each do |url|
+                        uri = URI.parse(url)
+                        require "nice_http"
+                        http = NiceHttp.new(host: "https://#{uri.host}")
+                        Dir.mkdir("#{config.path}/tmp") unless Dir.exist?("#{config.path}/tmp")
+                        file_path_name = "#{config.path}/tmp/#{user.name}_#{@ai_open_ai_image[user.name].object_id}.png"
+                        res = http.get(uri.path + "?#{uri.query}", save_data: file_path_name)
+                        if config.simulate
+                          respond "file: #{file_path_name}, #{messagersp}, #{message}, image/png, png"
+                        else
+                          send_file(Thread.current[:dest], messagersp, file_path_name, message, "image/png", "png")
+                        end
+                        http.close unless http.nil?
+                      end
+                    end
+                  else
+                    respond res
+                  end
+                end
+              rescue => exception
+                respond "*OpenAI*: Sorry, I'm having some problems. OpenAI probably is not available. Please try again later."
+                @logger.warn exception
+              end
+              unreact :art
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/commands/general/ai/open_ai/open_ai_models.rb

@@ -0,0 +1,37 @@
+class SlackSmartBot
+  module Commands
+    module General
+      module AI
+        module OpenAI
+          def open_ai_models(model='')
+            save_stats(__method__)
+            get_personal_settings()
+            @ai_open_ai, message_connect = SlackSmartBot::AI::OpenAI.connect(@ai_open_ai, config, @personal_settings)
+            respond message_connect if message_connect
+            user = Thread.current[:user]
+            if !@ai_open_ai[user.name].nil? and !@ai_open_ai[user.name][:client].nil?
+                react :running
+                begin
+                  res = SlackSmartBot::AI::OpenAI.models(@ai_open_ai[user.name][:client], model)
+                  if model == ''
+                    message = ["*OpenAI*"]
+                    message << "If you want to use a model, you can use on a DM with the SmartBot the command"
+                    message << "`set personal settings ai.open_ai.gpt_model MODEL_NAME`"
+                    message << "`set personal settings ai.open_ai.whisper_model MODEL_NAME`"
+                    message << "Here are the models available for use:"
+                    message << "```#{res.strip}```"
+                    respond message.join("\n")
+                  else
+                    respond "*OpenAI* Info about #{model} model:\n```#{res.strip}```"
+                  end
+                rescue => exception
+                  respond "*OpenAI*: Sorry, I'm having some problems. OpenAI probably is not available. Please try again later."
+                end
+                unreact :running
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/commands/general/ai/open_ai/open_ai_variations_image.rb

@@ -0,0 +1,84 @@
+class SlackSmartBot
+  module Commands
+    module General
+      module AI
+        module OpenAI
+          def open_ai_variations_image(message, variations, files)
+            save_stats(__method__)
+            get_personal_settings()
+            @ai_open_ai, message_connect = SlackSmartBot::AI::OpenAI.connect(@ai_open_ai, config, @personal_settings)
+            respond message_connect if message_connect
+            user = Thread.current[:user]
+            if !@ai_open_ai[user.name].nil? and !@ai_open_ai[user.name][:client].nil?
+              if variations > 9
+                respond "*OpenAI*: I'm sorry, I can only generate up to 9 variations at a time. Please try again."
+              else
+                variations = 1 if variations == 0
+                @ai_open_ai_image ||= {}
+                @ai_open_ai_image[user.name] ||= []
+                react :art
+                begin
+                  @ai_open_ai_image[user.name] = [] if !files.nil? and files.size == 1
+
+                  if (!File.exist?("#{config.path}/tmp/#{user.name}_#{@ai_open_ai_image[user.name].object_id}.png") and (files.nil? or files.size != 1))
+                    respond "*OpenAI*: Sorry, I need to generate an image first. Use `?i PROMPT` to generate an image or upload an image to generate variations."
+                  else
+                    image = "#{config.path}/tmp/#{user.name}_#{@ai_open_ai_image[user.name].object_id}.png"
+                    if !files.nil? and files.size == 1
+                      require "nice_http"
+                      http = NiceHttp.new(host: "https://files.slack.com", headers: { "Authorization" => "Bearer #{config.token}" })
+                      res = http.get(files[0].url_private_download, save_data: image)
+                      success, res = SlackSmartBot::AI::OpenAI.send_image_variation(@ai_open_ai[user.name].client, image, variations, size: @ai_open_ai[user.name][:image_size])
+                    else
+                      success, res = SlackSmartBot::AI::OpenAI.send_image_variation(@ai_open_ai[user.name].client, image, variations, size: @ai_open_ai[user.name][:image_size])
+                    end
+                    if success
+                      urls = res
+                      urls = [urls] if urls.is_a?(String)
+                      if urls.nil? or urls.empty?
+                        respond "*OpenAI*: Sorry, I'm having some problems. OpenAI was not able to generate an image."
+                      else
+                        if @ai_open_ai_image[user.name].empty?
+                          session_name = "Temporary Variation"
+                        else
+                          session_name = @ai_open_ai_image[user.name].first[0..29]
+                        end
+                        messagersp = "OpenAI Session: _<#{session_name}...>_ (id:#{@ai_open_ai_image[user.name].object_id})"
+                        require "uri"
+                        urls.each do |url|
+                          uri = URI.parse(url)
+                          require "nice_http"
+                          http = NiceHttp.new(host: "https://#{uri.host}")
+                          file_path_name = "#{config.path}/tmp/#{user.name}_#{@ai_open_ai_image[user.name].object_id}.png"
+                          res = http.get(uri.path + "?#{uri.query}", save_data: file_path_name)
+                          message = "Variation #{urls.index(url)+1} of #{urls.size}"
+                          if config.simulate
+                            respond "file: #{file_path_name}, #{messagersp}, #{message}, image/png, png"
+                          else
+                            send_file(Thread.current[:dest], messagersp, file_path_name, message, "image/png", "png")
+                          end
+                          http.close unless http.nil?
+                        end
+                      end
+                    else
+                      respond res
+                    end
+                  end
+                rescue => exception
+                  respond "*OpenAI*: Sorry, I'm having some problems. OpenAI probably is not available. Please try again later."
+                  @logger.warn exception
+                end
+                unreact :art
+              end
+            end
+
+            if variations > 9
+              respond "*OpenAI*: I'm sorry, I can only generate up to 9 variations at a time. Please try again."
+            else
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/commands/general/ai/open_ai/open_ai_whisper.rb

@@ -0,0 +1,51 @@
+class SlackSmartBot
+  module Commands
+    module General
+      module AI
+        module OpenAI
+          def open_ai_whisper(message, files)
+            save_stats(__method__)
+            get_personal_settings()
+            @ai_open_ai, message_connect = SlackSmartBot::AI::OpenAI.connect(@ai_open_ai, config, @personal_settings)
+            respond message_connect if message_connect
+            user = Thread.current[:user]
+            if !@ai_open_ai[user.name].nil? and !@ai_open_ai[user.name][:client].nil?
+              react :speech_balloon 
+              begin
+                if files.nil? or files.size != 1
+                  respond "*OpenAI Whisper*: Sorry, I need an audio file to transcribe. Please upload an audio file and try again."
+                else
+                  require "nice_http"
+                  audio = "#{config.path}/tmp/#{user.name}_audio.wav"
+                  http = NiceHttp.new(host: "https://files.slack.com", headers: { "Authorization" => "Bearer #{config.token}" })
+                  res = http.get(files[0].url_private_download, save_data: audio)
+                  success, res = SlackSmartBot::AI::OpenAI.whisper_transcribe(@ai_open_ai[user.name][:client], @ai_open_ai[user.name].whisper_model, audio)
+                  if success
+                    if message.to_s != ''
+                      success, res = SlackSmartBot::AI::OpenAI.send_gpt_chat(@ai_open_ai[user.name][:client], @ai_open_ai[user.name].gpt_model, "#{message}:\n#{res}")
+                      type_whisper = message
+                    else
+                      type_whisper = "Transcribe"
+                    end
+                    if res.size > 3000
+                      send_file(Thread.current[:dest], "*OpenAI Whisper*", '', type_whisper, "text/plain", "text", content: res)
+                    else
+                      respond "*OpenAI Whisper* \n#{res.strip}"
+                    end
+                  else
+                    respond "*OpenAI Whisper* \n#{res.strip}"
+                  end
+                  File.delete(audio) if File.exist?(audio)
+                end
+              rescue => exception
+                respond "*OpenAI Whisper*: Sorry, I'm having some problems. OpenAI probably is not available. Please try again later."
+                @logger.warn exception
+              end
+              unreact :speech_balloon
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/commands/general/bot_help.rb

@@ -20,6 +20,7 @@ class SlackSmartBot
       commands = []
       commands_search = []
       if help_command.to_s != ""
+        help_command.gsub!("?", "\\?") # for open ai commands
         help_message.gsub(/====+/,'-'*30).split(/^\s*-------*$/).each do |h|
           if strict
             if h.match?(/`#{help_command}`/i) or h.match?(/^\s*command_id:\s+:#{help_command.gsub(' ', '_')}\s*$/)

data/lib/slack/smart-bot/commands/general/personal_settings.rb

@@ -0,0 +1,38 @@
+class SlackSmartBot
+  def personal_settings(user, type, settings_id, settings_value)
+    save_stats "#{type}_personal_settings"
+    if Thread.current[:typem] == :on_dm
+      get_personal_settings()
+      @personal_settings[user.name] ||= {}
+      if type == :get
+        if settings_id.to_s == ''
+          personal_settings_txt = ""
+          @personal_settings[user.name].each do |key, value|
+            personal_settings_txt << "`#{key}`:  #{value}\n"
+          end
+          respond "Personal settings for *#{user.name}* are:\n#{personal_settings_txt}"
+        else
+          if @personal_settings[user.name].key?(settings_id)
+            respond "Personal settings for *#{settings_id}* is: *#{@personal_settings[user.name][settings_id]}*."
+          else
+            respond "Personal settings for *#{settings_id}* not found."
+          end
+        end
+      elsif type == :delete
+        if @personal_settings[user.name].key?(settings_id)
+          @personal_settings[user.name].delete(settings_id)
+          update_personal_settings()
+          respond "Personal settings deleted for *#{settings_id}*."
+        else
+          respond "Personal settings for *#{settings_id}* not found."
+        end
+      else #set
+        @personal_settings[user.name][settings_id] = settings_value
+        update_personal_settings()
+        respond "Personal settings set for *#{settings_id}*."
+      end
+    else
+      respond "This command can only be called on a DM with the SmartBot."
+    end
+  end
+end

data/lib/slack/smart-bot/commands/general/poster.rb

@@ -1,116 +1,119 @@
-def poster(permanent, emoticon_text, emoticon_bg, string, minutes)
-  chars = { a: ["0000", "1111", "1001", "1111", "1001", "1001", "0000"],
-            b: ["0000", "1111", "1001", "1110", "1001", "1111", "0000"],
-            c: ["0000", "0111", "1000", "1000", "1000", "0111", "0000"],
-            d: ["0000", "1110", "1001", "1001", "1001", "1110", "0000"],
-            e: ["0000", "1111", "1000", "1110", "1000", "1111", "0000"],
-            f: ["0000", "1111", "1000", "1110", "1000", "1000", "0000"],
-            g: ["0000", "1111", "1000", "1011", "1001", "1111", "0000"],
-            h: ["0000", "1001", "1001", "1111", "1001", "1001", "0000"],
-            i: ["000", "111", "010", "010", "010", "111", "000"],
-            j: ["0000", "0111", "0010", "0010", "1010", "0110", "0000"],
-            k: ["0000", "1001", "1010", "1100", "1010", "1001", "0000"],
-            l: ["000", "100", "100", "100", "100", "111", "000"],
-            m: ["00000", "10001", "11011", "10101", "10001", "10001", "00000"],
-            n: ["00000", "10001", "11001", "10101", "10011", "10001", "00000"],
-            o: ["0000", "0110", "1001", "1001", "1001", "0110", "0000"],
-            p: ["0000", "1110", "1001", "1110", "1000", "1000", "0000"],
-            q: ["00000", "01100", "10010", "10010", "10010", "01111", "00000"],
-            r: ["0000", "1110", "1001", "1110", "1010", "1001", "0000"],
-            s: ["0000", "0111", "1000", "0110", "0001", "1110", "0000"],
-            t: ["00000", "11111", "00100", "00100", "00100", "00100", "00000"],
-            u: ["00000", "10001", "10001", "10001", "10001", "01110", "00000"],
-            v: ["00000", "10001", "10001", "10001", "01010", "00100", "00000"],
-            w: ["0000000", "1000001", "1000001", "1001001", "1010101", "0100010", "0000000"],
-            x: ["00000", "10001", "01010", "00100", "01010", "10001", "00000"],
-            y: ["00000", "10001", "01010", "00100", "00100", "00100", "00000"],
-            z: ["00000", "11111", "00010", "00100", "01000", "11111", "00000"],
-            '!': ["0", "1", "1", "1", "0", "1", "0"],
-            '+': ["000", "000", "010", "111", "010", "000", "000"],
-            '-': ["000", "000", "000", "111", "000", "000", "000"],
-            '=': ["000", "000", "111", "000", "111", "000", "000"],
-            '?': ["000", "111", "001", "010", "000", "010", "000"],
-            '0': ["000", "111", "101", "101", "101", "111", "000"],
-            '1': ["00", "01", "11", "01", "01", "01", "00"],
-            '2': ["000", "111", "001", "010", "100", "111", "000"],
-            '3': ["000", "111", "001", "111", "001", "111", "000"],
-            '4': ["000", "101", "101", "111", "001", "001", "000"],
-            '5': ["000", "111", "100", "111", "001", "111", "000"],
-            '6': ["000", "111", "100", "111", "101", "111", "000"],
-            '7': ["000", "111", "001", "010", "100", "100", "000"],
-            '8': ["000", "111", "101", "111", "101", "111", "000"],
-            '9': ["000", "111", "101", "111", "001", "111", "000"],
-            ' ': ["0", "0", "0", "0", "0", "0", "0"] }
+class SlackSmartBot
 
-  emoticon_bg = ":white_square:" if emoticon_bg.to_s == ""
-  emoticon_text = ":black_square:" if emoticon_text.to_s == ""
-  string.downcase!
-  messages = []
+  def poster(permanent, emoticon_text, emoticon_bg, string, minutes)
+    chars = { a: ["0000", "1111", "1001", "1111", "1001", "1001", "0000"],
+              b: ["0000", "1111", "1001", "1110", "1001", "1111", "0000"],
+              c: ["0000", "0111", "1000", "1000", "1000", "0111", "0000"],
+              d: ["0000", "1110", "1001", "1001", "1001", "1110", "0000"],
+              e: ["0000", "1111", "1000", "1110", "1000", "1111", "0000"],
+              f: ["0000", "1111", "1000", "1110", "1000", "1000", "0000"],
+              g: ["0000", "1111", "1000", "1011", "1001", "1111", "0000"],
+              h: ["0000", "1001", "1001", "1111", "1001", "1001", "0000"],
+              i: ["000", "111", "010", "010", "010", "111", "000"],
+              j: ["0000", "0111", "0010", "0010", "1010", "0110", "0000"],
+              k: ["0000", "1001", "1010", "1100", "1010", "1001", "0000"],
+              l: ["000", "100", "100", "100", "100", "111", "000"],
+              m: ["00000", "10001", "11011", "10101", "10001", "10001", "00000"],
+              n: ["00000", "10001", "11001", "10101", "10011", "10001", "00000"],
+              o: ["0000", "0110", "1001", "1001", "1001", "0110", "0000"],
+              p: ["0000", "1110", "1001", "1110", "1000", "1000", "0000"],
+              q: ["00000", "01100", "10010", "10010", "10010", "01111", "00000"],
+              r: ["0000", "1110", "1001", "1110", "1010", "1001", "0000"],
+              s: ["0000", "0111", "1000", "0110", "0001", "1110", "0000"],
+              t: ["00000", "11111", "00100", "00100", "00100", "00100", "00000"],
+              u: ["00000", "10001", "10001", "10001", "10001", "01110", "00000"],
+              v: ["00000", "10001", "10001", "10001", "01010", "00100", "00000"],
+              w: ["0000000", "1000001", "1000001", "1001001", "1010101", "0100010", "0000000"],
+              x: ["00000", "10001", "01010", "00100", "01010", "10001", "00000"],
+              y: ["00000", "10001", "01010", "00100", "00100", "00100", "00000"],
+              z: ["00000", "11111", "00010", "00100", "01000", "11111", "00000"],
+              '!': ["0", "1", "1", "1", "0", "1", "0"],
+              '+': ["000", "000", "010", "111", "010", "000", "000"],
+              '-': ["000", "000", "000", "111", "000", "000", "000"],
+              '=': ["000", "000", "111", "000", "111", "000", "000"],
+              '?': ["000", "111", "001", "010", "000", "010", "000"],
+              '0': ["000", "111", "101", "101", "101", "111", "000"],
+              '1': ["00", "01", "11", "01", "01", "01", "00"],
+              '2': ["000", "111", "001", "010", "100", "111", "000"],
+              '3': ["000", "111", "001", "111", "001", "111", "000"],
+              '4': ["000", "101", "101", "111", "001", "001", "000"],
+              '5': ["000", "111", "100", "111", "001", "111", "000"],
+              '6': ["000", "111", "100", "111", "101", "111", "000"],
+              '7': ["000", "111", "001", "010", "100", "100", "000"],
+              '8': ["000", "111", "101", "111", "101", "111", "000"],
+              '9': ["000", "111", "101", "111", "001", "111", "000"],
+              ' ': ["0", "0", "0", "0", "0", "0", "0"] }
 
-  string = string.strip.split(" ").join(" ")
-  lines = [""]
-  string.split(" ").each do |t|
-    if ("#{lines[-1]}#{t}").size <= 6
-      lines[-1] = "#{lines[-1]}#{t} "
-    else
-      lines[-1].strip!
-      lines.pop if lines[-1] == ""
-      if t.size > 6
-        t.chars.each_slice(6).map(&:join).each do |tt|
-          lines << tt.strip
-        end
-        lines[-1] += " " if lines[-1].size < 6
-      else
-        lines << t.strip + " "
-      end
-    end
-  end
-  lines[-1].strip!
-  messages = []
-  lines.each do |line|
-    results = []
-    all_spaces = true
-    line.each_char do |char|
-      if chars.key?(char.to_sym)
-        results << chars[char.to_sym]
-        all_spaces = false
+    emoticon_bg = ":white_square:" if emoticon_bg.to_s == ""
+    emoticon_text = ":black_square:" if emoticon_text.to_s == ""
+    string.downcase!
+    messages = []
+
+    string = string.strip.split(" ").join(" ")
+    lines = [""]
+    string.split(" ").each do |t|
+      if ("#{lines[-1]}#{t}").size <= 6
+        lines[-1] = "#{lines[-1]}#{t} "
       else
-        results << chars[:' ']
+        lines[-1].strip!
+        lines.pop if lines[-1] == ""
+        if t.size > 6
+          t.chars.each_slice(6).map(&:join).each do |tt|
+            lines << tt.strip
+          end
+          lines[-1] += " " if lines[-1].size < 6
+        else
+          lines << t.strip + " "
+        end
       end
     end
-    unless all_spaces
-      rtxt = ""
-      7.times do |n|
-        results.size.times do |i|
-          rtxt += "0#{results[i][n]}0"
+    lines[-1].strip!
+    messages = []
+    lines.each do |line|
+      results = []
+      all_spaces = true
+      line.each_char do |char|
+        if chars.key?(char.to_sym)
+          results << chars[char.to_sym]
+          all_spaces = false
+        else
+          results << chars[:' ']
         end
-        rtxt += "\n"
       end
-      rtxt.gsub!("1", emoticon_text)
-      rtxt.gsub!("0", emoticon_bg)
-      txt = ""
-      msgs = []
-      rtxt.split("\n").each do |m|
-        if (m + txt).size > 4000
-          msgs << txt unless txt == ""
-          txt = ""
+      unless all_spaces
+        rtxt = ""
+        7.times do |n|
+          results.size.times do |i|
+            rtxt += "0#{results[i][n]}0"
+          end
+          rtxt += "\n"
+        end
+        rtxt.gsub!("1", emoticon_text)
+        rtxt.gsub!("0", emoticon_bg)
+        txt = ""
+        msgs = []
+        rtxt.split("\n").each do |m|
+          if (m + txt).size > 4000
+            msgs << txt unless txt == ""
+            txt = ""
+          end
+          txt += (m + "\n")
+        end
+        msgs << txt
+        msgs.flatten!
+        msgs.each do |msg|
+          messages << respond(msg, return_message: true)
         end
-        txt += (m + "\n")
-      end
-      msgs << txt
-      msgs.flatten!
-      msgs.each do |msg|
-        messages << respond(msg, return_message: true)
       end
     end
-  end
-  unless permanent
-    react :heavy_check_mark
-    sleep (minutes.to_i * 60)
-    messages.delete(nil)
-    messages.each do |message|
-      delete(message.channel, message.ts)
+    unless permanent
+      react :heavy_check_mark
+      sleep (minutes.to_i * 60)
+      messages.delete(nil)
+      messages.each do |message|
+        delete(message.channel, message.ts)
+      end
+      react :heavy_minus_sign
     end
-    react :heavy_minus_sign
   end
-end
+end