epub-translator 0.0.6__tar.gz → 0.0.7__tar.gz

{epub_translator-0.0.6 → epub_translator-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: epub-translator
-Version: 0.0.6
+Version: 0.0.7
 Summary: Translate the epub book using LLM. The translated book will retain the original text and list the translated text side by side with the original text.
 License: MIT
 Keywords: epub,llm,translation,translator

epub_translator-0.0.7/epub_translator/__init__.py

@@ -0,0 +1,3 @@
+from .llm import LLM
+from .translator import translate, TranslatedWriteMode
+from .translation import Language, ProgressReporter

{epub_translator-0.0.6 → epub_translator-0.0.7}/epub_translator/epub/html/dom_operator.py

@@ -13,7 +13,7 @@ def read_texts(root: Element) -> Generator[str, None, None]:
     elif position == TextPosition.TAIL:
       yield cast(str, element.tail)
 
-def append_texts(root: Element, texts: Iterable[str | Iterable[str] | None]):
+def write_texts(root: Element, texts: Iterable[str | Iterable[str] | None], append: bool):
   zip_list = list(zip(texts, search_texts(root)))
   for text, (element, position, parent) in reversed(zip_list):
     if text is None:
@@ -23,26 +23,32 @@ def append_texts(root: Element, texts: Iterable[str | Iterable[str] | None]):
       text = "".join(text)
     if position == TextPosition.WHOLE_DOM:
       if parent is not None:
-        _append_dom(parent, element, text)
+        _write_dom(parent, element, text, append)
     elif position == TextPosition.TEXT:
-      element.text = _append_text(element.text, text)
+      element.text = _write_text(element.text, text, append)
     elif position == TextPosition.TAIL:
-      element.tail = _append_text(element.tail, text)
+      element.tail = _write_text(element.tail, text, append)
 
-def _append_dom(parent: Element, origin: Element, text: str):
-  appended = Element(origin.tag, {**origin.attrib})
-  for index, child in enumerate(parent):
-    if child == origin:
-      parent.insert(index + 1, appended)
-      break
-
-  appended.attrib.pop("id", None)
-  appended.text = text
-  appended.tail = origin.tail
-  origin.tail = None
+def _write_dom(parent: Element, origin: Element, text: str, append: bool):
+  if append:
+    appended = Element(origin.tag, {**origin.attrib})
+    for index, child in enumerate(parent):
+      if child == origin:
+        parent.insert(index + 1, appended)
+        break
+    appended.attrib.pop("id", None)
+    appended.text = text
+    appended.tail = origin.tail
+    origin.tail = None
+  else:
+    for child in origin:
+      origin.remove(child)
+    origin.text = text
 
-def _append_text(left: str | None, right: str) -> str:
-  if left is None:
+def _write_text(left: str | None, right: str, append: bool) -> str:
+  if not append:
+    return right
+  elif left is None:
     return right
   else:
     return left + right

{epub_translator-0.0.6 → epub_translator-0.0.7}/epub_translator/epub/html/file.py

@@ -2,7 +2,7 @@ import re
 
 from typing import Iterable
 from xml.etree.ElementTree import fromstring, tostring, Element
-from .dom_operator import read_texts, append_texts
+from .dom_operator import read_texts, write_texts
 from .empty_tags import to_xml, to_html
 
 
@@ -41,8 +41,8 @@ class HTMLFile:
     self._texts_length = len(texts)
     return texts
 
-  def write_texts(self, texts: Iterable[str]):
-    append_texts(self._root, texts)
+  def write_texts(self, texts: Iterable[str], append: bool):
+    write_texts(self._root, texts, append)
 
   @property
   def texts_length(self) -> int:

{epub_translator-0.0.6 → epub_translator-0.0.7}/epub_translator/translator.py

@@ -1,5 +1,6 @@
 from os import PathLike
 from pathlib import Path
+from enum import auto, Enum
 from tempfile import mkdtemp
 from shutil import rmtree
 
@@ -9,11 +10,16 @@ from .zip_context import ZipContext
 from .translation import translate as _translate, Incision, Fragment, Language, ProgressReporter
 
 
+class TranslatedWriteMode(Enum):
+  APPEND = auto()
+  REPLACE = auto()
+
 def translate(
       llm: LLM,
       source_path: PathLike,
       translated_path: PathLike,
       target_language: Language,
+      write_mode: TranslatedWriteMode = TranslatedWriteMode.APPEND,
       user_prompt: str | None = None,
       working_path: PathLike | None = None,
       max_chunk_tokens_count: int = 3000,
@@ -29,6 +35,7 @@ def translate(
   _Translator(
     llm=llm,
     target_language=target_language,
+    write_mode=write_mode,
     user_prompt=user_prompt,
     max_chunk_tokens_count=max_chunk_tokens_count,
     max_threads_count=max_threads_count,
@@ -44,6 +51,7 @@ class _Translator:
         self,
         llm: LLM,
         target_language: Language,
+        write_mode: TranslatedWriteMode,
         user_prompt: str | None,
         max_chunk_tokens_count: int,
         max_threads_count: int,
@@ -52,6 +60,7 @@ class _Translator:
 
     self._llm: LLM = llm
     self._target_language: Language = target_language
+    self._write_mode: TranslatedWriteMode = write_mode
     self._user_prompt: str | None = user_prompt
     self._max_chunk_tokens_count: int = max_chunk_tokens_count
     self._max_threads_count: int = max_threads_count
@@ -110,6 +119,7 @@ class _Translator:
     spine: tuple[Path, HTMLFile] | None = None
     translated_texts: list[str] = []
     translated_count: int = 0
+    append = (self._write_mode == TranslatedWriteMode.APPEND)
 
     for translated_text in _translate(
       llm=self._llm,
@@ -125,7 +135,7 @@ class _Translator:
 
       if spine and translated_count >= len(translated_texts):
         spine_path, spine_file = spine
-        spine_file.write_texts(translated_texts)
+        spine_file.write_texts(translated_texts, append)
         context.write_spine_file(spine_path, spine_file)
         spine = None
 
@@ -152,7 +162,7 @@ class _Translator:
     if spine:
       spine_path, spine_file = spine
       if translated_count > 0:
-        spine_file.write_texts(translated_texts)
+        spine_file.write_texts(translated_texts, append)
       context.write_spine_file(spine_path, spine_file)
 
 def _gen_fragments(context: ZipContext):

{epub_translator-0.0.6 → epub_translator-0.0.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "epub-translator"
-version = "0.0.6"
+version = "0.0.7"
 description = "Translate the epub book using LLM. The translated book will retain the original text and list the translated text side by side with the original text."
 keywords=["epub", "llm", "translation", "translator"]
 authors = [

epub_translator-0.0.6/epub_translator/__init__.py

@@ -1,3 +0,0 @@
-from .llm import LLM
-from .translator import translate
-from .translation import Language, ProgressReporter