immosquare-translate 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 812bd63f44d434dc29f98e8b2707c16c68b28a8979c4b098d3f91b4a6b05f618
-  data.tar.gz: c2aaf90c499a6d079eb5d2ce067d4501b532c9502c5590592985d292b811fc73
+  metadata.gz: 326589c34a947bb34891fdef41683bf0a0d2bc5dd649c0598d63d7591eac1341
+  data.tar.gz: e28ead0b001c5c3ffea1a322e35ba3293e7c916dbe71a8c677d9484efb86a5c8
 SHA512:
-  metadata.gz: 8f80c08d77141e8cb522f1274aeea0fa1658aa118b37cf7b8fb89d946b8e643e8d160ab6f93712417cafc175d80932e181f686414597df61638f2a082f78b90b
-  data.tar.gz: f6a7c90d9adc7414bcf170d4b01973c0521b25d3e6b53ffc4db6eb11067790b7349ed8b518c32a2cba3ffb7ed0319a9aa4f5c3f7786104caeb80420192d11a2e
+  metadata.gz: 8d74b478dbaa133200e77b2638e38bf986204ecfc5946b2653cb7687b40a8bfbfb28e6e163c772aac679f1c599169bc44c66c14cdb224be8fba830353ad21f69
+  data.tar.gz: bbf947a36c077cfc1b419936cac7a88e4cee75df414c11133f85172aa431cff10e95b5336e202171a18b34036a9964339be5515096ad554fa14a5d271b20e586

data/lib/immosquare-translate/railtie.rb

@@ -0,0 +1,9 @@
+module ImmosquareTranslate
+  class Railtie < Rails::Railtie
+
+    rake_tasks do
+      load "tasks/immosquare-translate.rake"
+    end
+
+  end
+end

data/lib/immosquare-translate/translator.rb

@@ -20,7 +20,7 @@ module ImmosquareTranslate
           model         = OPEN_AI_MODELS.find {|m| m[:name] == model_name }
           model         = OPEN_AI_MODELS.find {|m| m[:name] == "gpt-4-0125-preview" } if model.nil?
           from_iso      = ISO_639.find_by_code(from).english_name.split(";").first
-          to_iso        = to.map {|l| ISO_639.find_by_code(l).english_name.split(";").first }
+          to_iso        = to.map {|iso| [iso, ISO_639.find_by_code(iso).english_name.split(";").first] }
           headers       = {
             "Content-Type"  => "application/json",
             "Authorization" => "Bearer #{ImmosquareTranslate.configuration.openai_api_key}"
@@ -28,19 +28,21 @@ module ImmosquareTranslate
 
           prompt_system = "As a sophisticated translation AI, your role is to translate sentences from a specified source language to multiple target languages. " \
                           "It is imperative that you return the translations in a single, pure JSON string format. Use ISO 639-1 language codes for specifying languages. " \
-                          "if string is html, you should return the translated html." \
+                          "If string is html, you should return the translated html." \
                           "Ensure that the output does not include markdown (```json) or any other formatting characters. Adhere to the JSON structure meticulously."
 
 
-          prompt = "Translate the following sentences from '#{from_iso}' into the languages #{to_iso.join(", ")}, and format the output as a single, pure JSON string. " \
-                   "Follow the structure: {\"datas\":[{\"en\":\"English Translation\",\"es\":\"Spanish Translation\",\"it\":\"Italian Translation\"}]}, using the correct ISO 639-1 language codes for each translation. " \
-                   "Your response should strictly conform to this JSON structure without any additional characters or formatting. Sentences to translate are:"
+          prompt = "Translate the #{text.size} following  #{text.size == 1 ? "sentence" : "sentences"} from #{from_iso} (with ISO 639-1 code : #{from}) into the languages #{to_iso.map {|iso, l| "#{l} (with ISO 639-1 code : #{iso})" }.join(", ")}.\n" \
+                   "Format the output as a single pure JSON string.\n" \
+                   "Follow this array structure: {\"datas\":[{\"locale_iso\":\"Translated Text\"}]}, using the correct ISO 639-1 language codes for each translation. " \
+                   "Your response should strictly conform to this JSON structure without any additional characters or formatting.\nSentences to translate are:"
 
           text.each_with_index do |sentence, index|
             prompt += "\n#{index + 1}: #{sentence}"
           end
 
 
+
           body = {
             :model       => model[:name],
             :messages    => [
@@ -74,8 +76,14 @@ module ImmosquareTranslate
           puts("Estimate price => #{input_price.round(3)} + #{output_price.round(3)} = #{price.round(3)} USD")
 
 
+          ##============================================================##
+          ## On s'assure de ne renvoyer que les locales demandées
+          ## car l'API peut renvoyer des locales non demandées...
+          ##============================================================##
           p = JSON.parse(choice["message"]["content"])
-          p["datas"]
+          p["datas"].map! do |my_hash|
+            my_hash.select! {|iso, _value| to.include?(iso.to_s) }
+          end
         rescue StandardError => e
           puts(e.message)
           puts(e.backtrace)

data/lib/immosquare-translate/version.rb

@@ -1,3 +1,3 @@
 module ImmosquareTranslate
-  VERSION = "0.1.2".freeze
+  VERSION = "0.1.4".freeze
 end

data/lib/immosquare-translate.rb

@@ -2,7 +2,7 @@ require_relative "immosquare-translate/configuration"
 require_relative "immosquare-translate/shared_methods"
 require_relative "immosquare-translate/yml_translator"
 require_relative "immosquare-translate/translator"
-
+require_relative "immosquare-translate/railtie" if defined?(Rails)
 
 
 ##===========================================================================##

data/lib/tasks/{immosquare-yaml.rake → immosquare-translate.rake}

@@ -4,9 +4,9 @@ namespace :immosquare_translate do
   ## rake immosquare_translate:translate SOURCE_LOCALE=fr
   ##============================================================##
   desc "Translate translation files in rails app"
-  task :translate => :environment do
+  task :translate_rails_locales => :environment do
     begin
-      source_locale      = ENV.fetch("SOURCE_LOCALE", nil) || "fr"
+      source_locale      = ENV.fetch("SOURCE_LOCALE", nil)      || "fr"
       reset_translations = ENV.fetch("RESET_TRANSLATIONS", nil) || false
       reset_translations = reset_translations == "true"
 
@@ -24,14 +24,4 @@ namespace :immosquare_translate do
       puts(e.message)
     end
   end
-
-  ##============================================================##
-  ## Function to clean translation files in rails app
-  ##============================================================##
-  desc "Clean translation files in rails app"
-  task :clean => :environment do
-    Dir.glob("#{Rails.root}/config/locales/**/*.yml").each do |file|
-      ImmosquareYaml.clean(file)
-    end
-  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: immosquare-translate
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - IMMO SQUARE
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-30 00:00:00.000000000 Z
+date: 2024-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -71,11 +71,12 @@ extra_rdoc_files: []
 files:
 - lib/immosquare-translate.rb
 - lib/immosquare-translate/configuration.rb
+- lib/immosquare-translate/railtie.rb
 - lib/immosquare-translate/shared_methods.rb
 - lib/immosquare-translate/translator.rb
 - lib/immosquare-translate/version.rb
 - lib/immosquare-translate/yml_translator.rb
-- lib/tasks/immosquare-yaml.rake
+- lib/tasks/immosquare-translate.rake
 homepage: https://github.com/IMMOSQUARE/immosquare-translate
 licenses:
 - MIT