yptools 1.0.17 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 05c1d548417627949ac94ebf616c8fa43f617effa7abe15c6696f0fd7a4ebc98
-  data.tar.gz: 77b047b1c9e4a24dbe2ca490884459a715a4f522d25b6ade0d117de59495745f
+  metadata.gz: 98dd6d1d9b7cc1c72d121fce68e65a6d035ebef4bab15b3fd18c0b550c2e23ec
+  data.tar.gz: 3ddc62e82485cf62bf760a091b73d96769616e45d52de26cf98de7ca2eeb45e2
 SHA512:
-  metadata.gz: 30bb630cb7d84c1e80dec1bf59f0c667b53f21ba96ee0284c92013af3421eed6bea1290ddccd6fd85cf3d9d0f9ac2ddf34a4a2ebba9c323b8a7e34a4dc2c8380
-  data.tar.gz: fa7e14ba1b2a495849921bb73b64d71446a0bed3b4c09a50bafa928c415e325f994c8ee1208810f04fb54b73f2fd0f53ef065dcbc06e679e5710dfe18ff301f3
+  metadata.gz: b3138ef32c117ca0c53f553fe0b1a95500369fcc186e1a60a569980e1efcaeae041de2c95f4324906162e8d6f780f450df3358f998261aa0f158f004ee740a58
+  data.tar.gz: 0cc0a0b57eb2c8ee090c1f7501342948457397fbc72f91a14d9919dcd09a0a9c80d23a7b64007b026c354db6ff7e76f17d0f5f4d3fe882afce98edf1208f4a7e

data/lib/yptools/chatai/yp_chatai.rb

@@ -115,4 +115,63 @@ class YPChatAI
         yp_log_success "chatGPT：" + yp_answer.gsub(/\n+/, "\n")
     end
 
+    def self.openaiimg(message) 
+        yp_chataiURL = 'https://api.openai.com/v1/images/generations'
+        # 开始发送网络请求
+        url = URI.parse(yp_chataiURL)
+        http = Net::HTTP.new(url.host, url.port)
+        http.use_ssl = true 
+        # 简单了加了下密，免费的apikey，就不要扒去用了（我的代码是开源放到GitHub，加密的目的是ChatGPT会检测秘钥是否在网络泄露）
+        yp_token = "c2stTVRVMVZqeUdpNWxzYlU1TmNlU1pUM0JsYmtGSmNYam5iUk5ROENVYUd2QVR4WXpp"
+        # 将Base64字符串解码为原始二进制数据
+        decoded_data = Base64.decode64(yp_token).force_encoding("UTF-8")
+        # 设置请求头
+        headers = {
+            'Content-Type' => 'application/json',
+            'Authorization' => 'Bearer ' + decoded_data
+        }
+        # 设置ai根据上下文
+        question = message
+
+        # 设置请body
+        data = {
+            # 'model' => 'text-davinci-003', # 然后GPT-3模型会根据您的输入文本自动生成文本补全或摘要。
+            'prompt' => question, # 问的问题
+            "n": 3, #几个回答
+            "size": "512x512"
+        }
+        
+        request = Net::HTTP::Post.new(url.path, headers)
+        request.body = data.to_json
+
+        begin
+            response = http.request(request)
+            # 处理响应数据
+            yp_message_response = JSON(response.body)
+            if !yp_message_response["error"]
+                created = yp_message_response["created"]
+                data = yp_message_response["data"]
+                puts ""
+                if !data.empty?
+                    index = 1
+                    for item in data
+                        yp_log_success "图#{index} 【复制下面链接到浏览器打开或者 command + 鼠标左键快速打开 】"
+                        yp_log_msg item["url"]
+                        index = index + 1
+                        puts ""
+                    end
+                else 
+                    yp_log_fail "请求失败，" + yp_message_response
+                end
+            else 
+                message = yp_message_response["error"]["message"]
+                yp_log_fail  "请求失败，请稍后再试！！！\n" + message
+            end
+        rescue StandardError => e
+            # 处理异常
+            yp_log_fail "请求的次数太多了，请稍后再试！！！"
+            yp_log_fail "发生异常：#{e.message}"
+        end
+    end
+
 end

data/lib/yptools/help/yp_help.rb

@@ -3,8 +3,10 @@ class YPHelp
   def self.message
     puts %q{
       
-      chatai: use [yptools chatai] 创建会话列表与 chatgpt 聊天，会记录上下内容（科学上网）
-                  [yptools chatai ...] 快速与 chatgpt 沟通，不会记录上下内容
+      chatgpt: use [yptools chatgpt] 创建会话列表与 chatgpt 聊天，会记录上下内容（科学上网）
+                   [yptools chatgpt ...] 快速与 chatgpt 沟通，不会记录上下内容
+
+      openai: use [yptools openaiimg ...] 根据文本描述生成图像（eg: yptools openaiimg '老虎和狮子大战' ）
 
       autocre: use [yptools autocre ...] 自动化工具命令
                use [yptools autocre -objc ...] 根据 json 自动创建 Objective-C 数据库操作文件 .h|.m 文件。（依赖三方库 FMDB ）

data/lib/yptools.rb

@@ -18,13 +18,22 @@ class YPTools
         cmd = argvs[0]
         case cmd
 
-        when 'chatai'
+        when 'chatgpt'
             if argvs.size > 1
                 name = argvs[1]
                 self.chatai name
             else
                 self.startChat
             end
+        when 'openaiimg'
+            if argvs.size > 1
+                name = argvs[1..-1]
+                name = name.join(' ')
+                self.openaiimg name
+            else
+                yp_log_fail "'yptools openaiimg ..' 参数缺失"
+                self.help
+            end
         when 'autocre'
             if argvs.size > 1
                 name = argvs[1]
@@ -144,6 +153,10 @@ class YPTools
     def self.startChat 
         YPChatAI.startChatAI()
     end
+
+    def self.openaiimg(message)
+        YPChatAI.openaiimg(message)
+    end
     
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: yptools
 version: !ruby/object:Gem::Version
-  version: 1.0.17
+  version: 1.1.0
 platform: ruby
 authors:
 - chenghengsheng
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-07 00:00:00.000000000 Z
+date: 2023-03-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler