feat(tasks): support the write_mode field

flows/playground/flow.oo.yaml

@@ -23,5 +23,7 @@ nodes:
         value: 5
       - handle: retry_interval_seconds
         value: 6.5
+      - handle: write_mode
+        value: append
     task: self::translator
 title: Playground

package.oo.yaml

@@ -8,8 +8,8 @@ scripts:
 dependencies:
   {}
 name: books-translator
-version: 0.1.10
+version: 0.1.11
 displayName: Books Translator
 description: Translate the epub book using LLM side by side.
-releaseNotes: add groups & descriptions
+releaseNotes: support the write_mode field
 repository: https://github.com/oomol-flows/books-translater

pyproject.toml

@@ -9,7 +9,7 @@ readme = "README.md"
 requires-python = ">=3.10,<3.13"
 dependencies = [
     "requests (>=2.32.4,<3.0.0)",
-    "epub-translator (==0.0.6)",
+    "epub-translator (==0.0.7)",
 ]
 
 

tasks/translator/__init__.py

@@ -3,7 +3,7 @@
 
 from pathlib import Path
 from oocana import Context
-from epub_translator import translate, LLM, Language
+from epub_translator import translate, LLM, Language, TranslatedWriteMode
 
 
 #region generated meta
@@ -13,6 +13,7 @@ class Inputs(typing.TypedDict):
   source_file: str
   translated_file: str | None
   language: typing.Literal["zh-Hans", "zh-Hant", "en", "fr", "de", "es", "ru", "it", "pt", "ja", "ko"]
+  write_mode: typing.Literal["append", "replace"]
   prompt: str | None
   max_chunk_tokens: int
   threads: int
@@ -55,6 +56,7 @@ def main(params: Inputs, context: Context) -> Outputs:
     target_language=_parse_language_code(params["language"]),
     user_prompt=params["prompt"],
     working_path=working_path,
+    write_mode=_transform_write_mode(params["write_mode"]),
     max_chunk_tokens_count=params["max_chunk_tokens"],
     max_threads_count=params["threads"],
     report_progress=lambda p: context.report_progress(100.0 * p),
@@ -63,6 +65,14 @@ def main(params: Inputs, context: Context) -> Outputs:
     "translated_file": str(translated_file)
   }
 
+def _transform_write_mode(write_mode: str) -> TranslatedWriteMode:
+  if write_mode == "append":
+    return TranslatedWriteMode.APPEND
+  elif write_mode == "replace":
+    return TranslatedWriteMode.REPLACE
+  else:
+    raise ValueError(f"Invalid write mode: {write_mode}")
+
 def _prepare_working_path(source_file: Path, context: Context) -> Path:
   st_mtime = source_file.stat().st_mtime
   hash = hashlib.new(name="sha512")

tasks/translator/task.oo.yaml

@@ -28,6 +28,8 @@ inputs_def:
               - .epub
     value:
     nullable: true
+  - group: Translate
+    collapsed: false
   - handle: language
     description: Translation target language
     json_schema:
@@ -57,6 +59,18 @@ inputs_def:
           - Japanese
           - Korean
     value: zh-Hans
+  - handle: write_mode
+    json_schema:
+      enum:
+        - append
+        - replace
+      ui:options:
+        labels:
+          - Append
+          - Replace
+    description: 'How is the translation written into an EPUB file: Is it "append"
+      to the original text? Or does it "replace" the original text?'
+    value: append
   - handle: prompt
     description: Provide LLM with necessary information for translating books. For
       example, you can fill in the glossary and character name specifications.