Added better devider

pescheckit · Dec 22, 2023 · bcf6628 · bcf6628
1 parent d9e3f58
commit bcf6628
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/python_gpt_po/po_translator.py b/python_gpt_po/po_translator.py
@@ -18,6 +18,8 @@
 load_dotenv()
 logging.basicConfig(level=logging.INFO)
 
+UNIQUE_DIVIDER = "\n\n--- Translation Divider ---\n\n"
+
 
 class TranslationConfig:
     """ Class to hold configuration parameters for the translation service. """
@@ -44,11 +46,10 @@ def translate_bulk(self, texts, target_language, po_file_path, current_batch):
             batch_texts = texts[i:i + self.batch_size]
             batch_info = f"File: {po_file_path}, Batch {i}/{self.total_batches}"
             batch_info += f" (texts {i + 1}-{min(i + self.batch_size, len(texts))})"
-            translation_request = (f"Translate the following texts into {target_language} as a single, "
-                                   "continuous text. "
-                                   "Do not repeat the original texts, just provide the "
-                                   "translations:\n\n")
-            translation_request += "\n\n".join(batch_texts)
+            translation_request = (f"Translate the following texts into {target_language} as separate segments. "
+                                   f"Use the unique divider '{UNIQUE_DIVIDER.strip()}' between each translation. "
+                                   "Do not repeat the original texts, just provide the translations:\n\n")
+            translation_request += f"{UNIQUE_DIVIDER}".join(batch_texts)
             retries = 3
 
             while retries:
@@ -72,7 +73,7 @@ def perform_translation(self, translation_request, translated_texts):
         """Performs the translation and updates the results."""
         message = {"role": "user", "content": translation_request}
         completion = self.config.client.chat.completions.create(model=self.config.model, messages=[message])
-        batch_translations = completion.choices[0].message.content.strip().split('\n\n')
+        batch_translations = completion.choices[0].message.content.strip().split(UNIQUE_DIVIDER)
         translated_texts.extend(batch_translations)
 
     def scan_and_process_po_files(self, input_folder, languages):

diff --git a/python_gpt_po/version.py b/python_gpt_po/version.py
@@ -19,4 +19,4 @@
     using the '--version' argument.
 """
 
-__version__ = "0.1.6"
+__version__ = "0.1.7"