natsukantou 0.1.3 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b299497adbec2c0446350fc1085c4f17c6a9ebe165246dbba07fef1914348e5
-  data.tar.gz: 8164f3a25f80bc7fde73fd8062563ecbdf34f5abc2f5b1b594e78a219bafe2b8
+  metadata.gz: 1edac0c042ee1ad4c1051d15028d5c6ff9ac397160c5bc83c9cd8c7fe1730d3f
+  data.tar.gz: 9e8950028f71f90f4e1b141a63d74027c021b36a2231151e290b50fcd396ec34
 SHA512:
-  metadata.gz: 34b63359b8441b651ffc071c841eae8cb74a29b2c7ea0fb2a717794a16de5dcc2fa8b94c5abf9e59f600cc49c7c9716a668e456790e01bf709d70a39737ef435
-  data.tar.gz: 951610545756fbe6b1b8384ba358b64b32f471ec6e5574f270b88f166ec70ef575270ac90b4ec2624f904d3d8bfb6ffe7b977a3aa4930ac82fc6ed04f1d90d70
+  metadata.gz: 82d10f0790a4eb1899ec277b90b9fa3c66de282d08915fd5e9d2fe6e53d97a39b12205a9ec8389c6bdbf5587f492fb072ca70d02326bd991be7f94217f144b8e
+  data.tar.gz: 7c81870166e14a41957246cd266a8b0e58e6868421ed7115deb853746e036d3f913fed5880412d73e9b530f3dcab8304a430ca5475c1decbf7ad7f0876c458bb

data/README.md

@@ -36,6 +36,7 @@ If you choose to save the config (`translator_config.rb`), next time you can reu
 
 * [DeepL API](https://www.deepl.com/pro-api)
 * [みんなの自動翻訳＠TexTra](https://mt-auto-minhon-mlt.ucri.jgn-x.jp/)
+* [ChatGPT](https://openai.com/)
 
 ### Middlewares
 

data/lib/natsukantou/chat_gpt.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require 'openai'
+
+# Machine translator
+#
+# ChatGPT
+module Natsukantou
+  class ChatGpt
+    include UtilityBase
+
+    # @param app [Hash]
+    # @param access_token [String] API access token
+    #
+    # @param translate_by_section [String] CSS selector to translate matched elements one by one.
+    #   Specify a css path (e.g. "body>p") to translate in smaller chunks.
+    def initialize(
+      app, access_token:,
+      translate_by_section:
+    )
+      @app = app
+      @access_token = access_token
+
+      ### Non request related setting
+      @translate_by_section = translate_by_section
+    end
+
+    attr_reader :env
+    attr_reader :access_token
+    attr_reader :translate_by_section
+
+    def call(env)
+      @env = env
+
+      if translate_by_section
+        env[:dom].css(translate_by_section).each do |node|
+          next if node.text.empty?
+
+          translated_xml = translate(node.to_xml)
+          node.replace(dom_node(translated_xml))
+        end
+      else
+        # TODO
+      end
+
+      env[:dom].lang = env[:lang_to].code
+
+      @app.call(env)
+    end
+
+    private
+
+    def translate(text)
+      message = "Please help me to translate the XML `#{text}` from #{env.lang_from.code} to #{env.lang_to.code}, maintaining XML structure." # rubocop:disable Layout/LineLength
+
+      response = client.chat(
+        parameters: {
+          model: "gpt-3.5-turbo",
+          messages: [{ role: "user", content: message }],
+          temperature: 0.7,
+        }
+      )
+
+      translated_xml = response.dig("choices", 0, "message", "content")
+
+      translated_xml.strip!
+
+      translated_xml
+    rescue Net::OpenTimeout
+      sleep 10
+      retry
+    end
+
+    def client
+      @client ||= ::OpenAI::Client.new(access_token: access_token)
+    end
+  end
+end

data/lib/natsukantou/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Natsukantou
-  VERSION = "0.1.3"
+  VERSION = "0.2.0"
 end

data/lib/natsukantou.rb

@@ -33,6 +33,7 @@ module Natsukantou
   # Translators
   autoload_and_register :translator, :DeepL, "natsukantou/deep_l"
   autoload_and_register :translator, :Minhon, "natsukantou/minhon"
+  autoload_and_register :translator, :ChatGpt, "natsukantou/chat_gpt"
 
   # Middlewares
   autoload_and_register :middleware, :SubstitudeGlossary, "natsukantou/substitude_glossary"

data/natsukantou.gemspec

@@ -42,6 +42,9 @@ Gem::Specification.new do |spec|
   # Minhon
   spec.add_dependency 'oauth', '~> 0.5.1'
 
+  # ChatGPT
+  spec.add_dependency 'ruby-openai', '~> 3.5.0'
+
   # For more information and examples about making a new gem, check out our
   # guide at: https://bundler.io/guides/creating_gem.html
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: natsukantou
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.2.0
 platform: ruby
 authors:
 - lulalala
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-02-27 00:00:00.000000000 Z
+date: 2023-03-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: middleware
@@ -108,6 +108,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.5.1
+- !ruby/object:Gem::Dependency
+  name: ruby-openai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.5.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.5.0
 description: human language translation library for XML documents, supporting DeepL
 email:
 - mark@goodlife.tw
@@ -128,6 +142,7 @@ files:
 - lib/monkey_patch/oga_document.rb
 - lib/monkey_patch/runner.rb
 - lib/natsukantou.rb
+- lib/natsukantou/chat_gpt.rb
 - lib/natsukantou/deep_l.rb
 - lib/natsukantou/handle_ruby_markup.rb
 - lib/natsukantou/minhon.rb
@@ -167,7 +182,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.6
+rubygems_version: 3.4.7
 signing_key:
 specification_version: 4
 summary: human language translation library for XML documents