immosquare-translate 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2cf94a139c881e315445fb0df7f24699166c1833b2b5f0746b86c91f5025ba5c
-  data.tar.gz: b518012e6deafacb308f178c1c21ab0191c2e491d47f0bda97e976605155a37f
+  metadata.gz: 9daf467e21100501ea1833692e80742d341950fb05f1a451b25134a83e7ba7e1
+  data.tar.gz: 7fe077c183fb8beee8d917abad7d014eafd811ef300574cfc776427efad51f0e
 SHA512:
-  metadata.gz: 1f96fdafd54a457e86ee7d8b0475807e4f8308ff5e8080a08ca6df369a1c4d515dc45ec90a240d4a4143a531bdaaaf51881532313252054e133b5fd0d33281ae
-  data.tar.gz: 715d03787f074bff87e69039ddeb466f9cbe449e1e57021ea9a3587fc272cc71d8f431e6f74aa75c33cb9accab2976e559d4ad885989b6d6c2ceab21637208e3
+  metadata.gz: 5758dc40a24ab4721d4ec72184d181567e96949b03aeb71e8bbdd14ffc107f0def5197719bf069ab65486fb65876223029d67f5f82de12a5ee96391c5dd61b97
+  data.tar.gz: 8ade76c458bb26cec34a1c369703b6e1141a10fc99fbb56774e423ea3ccb95658c82863c7c86cb9169d3fd82e2434cafeb200390f2b7d283a391de44c574e868

data/lib/immosquare-translate/translator.rb

@@ -20,27 +20,34 @@ 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}"
           }
 
-          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." \
+          prompt_system = "As a sophisticated translation AI, your role is to translate sentences from a specified source language to multiple target languages.\n" \
+                          "It is imperative that you return the translations in a single, pure JSON string format. Use ISO 639-1 language codes for specifying languages.\n" \
+                          "If string is html, you should return the translated html.\n" \
                           "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} #{text.size == 1 ? "sentence" : "sentences"} provided below from the source language (ISO 639-1 code: #{from}) to the target languages specified: #{to_iso.map {|iso, language| "#{language} (ISO 639-1 code: #{iso})" }.join(", ")}. " \
+                   "Before translating, correct any spelling or grammatical errors in each sentence. " \
+                   "If the source language is also a target language, include the corrected version of the sentence for that language as well. " \
+                   "Format the translation output as a JSON string adhering to the following structure: " \
+                   "{\"datas\":[{\"locale_iso\": \"Translated Text\"}]}, where 'locale_iso' is replaced with the appropriate ISO 639-1 code for each translation. " \
+                   "Ensure the output strictly follows this JSON format, without any extraneous characters or formatting. " \
+                   "Sentences to translate:"
+
 
           text.each_with_index do |sentence, index|
             prompt += "\n#{index + 1}: #{sentence}"
           end
 
 
+
           body = {
             :model       => model[:name],
             :messages    => [
@@ -74,8 +81,15 @@ module ImmosquareTranslate
           puts("Estimate price => #{input_price.round(3)} + #{output_price.round(3)} = #{price.round(3)} USD")
 
 
-          p = JSON.parse(choice["message"]["content"])
-          p["datas"]
+          ##============================================================##
+          ## On s'assure de ne renvoyer que les locales demandées
+          ## car l'API peut renvoyer des locales non demandées...
+          ##============================================================##
+          content = JSON.parse(choice["message"]["content"])
+          datas   = content["datas"]
+          datas.map do |hash|
+            hash.select {|key, _| to.include?(key) }
+          end.reject(&:empty?)
         rescue StandardError => e
           puts(e.message)
           puts(e.backtrace)

data/lib/immosquare-translate/version.rb

@@ -1,3 +1,3 @@
 module ImmosquareTranslate
-  VERSION = "0.1.3".freeze
+  VERSION = "0.1.5".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: immosquare-translate
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - IMMO SQUARE
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-31 00:00:00.000000000 Z
+date: 2024-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty