epub-translator 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

epub_translator/translator.py

@@ -63,6 +63,7 @@ class _Translator:
     try:
       temp_dir = _clean_path(working_path / "temp")
       temp_dir.mkdir(parents=True, exist_ok=True)
+      cache_path = working_path / "cache"
 
       context = ZipContext(
         epub_path=Path(source_path),
@@ -70,11 +71,12 @@ class _Translator:
       )
       context.replace_ncx(lambda texts: self._translate_ncx(
         texts=texts,
+        cache_path=cache_path,
         report_progress=lambda p: self._report_progress(p * 0.1)),
       )
       self._translate_spine(
         context=context,
-        working_path=working_path,
+        cache_path=cache_path,
         report_progress=lambda p: self._report_progress(0.1 + p * 0.8),
       )
       context.archive(translated_path)
@@ -84,10 +86,10 @@ class _Translator:
       if is_temp_workspace:
         rmtree(working_path, ignore_errors=True)
 
-  def _translate_ncx(self, texts: list[str], report_progress: ProgressReporter) -> list[str]:
+  def _translate_ncx(self, texts: list[str], cache_path: Path, report_progress: ProgressReporter) -> list[str]:
     return list(_translate(
       llm=self._llm,
-      cache_path=None,
+      cache_path=cache_path,
       max_chunk_tokens_count=self._max_chunk_tokens_count,
       max_threads_count=1,
       target_language=self._target_language,
@@ -103,7 +105,7 @@ class _Translator:
       ),
     ))
 
-  def _translate_spine(self, context: ZipContext, working_path: Path, report_progress: ProgressReporter):
+  def _translate_spine(self, context: ZipContext, cache_path: Path, report_progress: ProgressReporter):
     spine_paths_iter = iter(list(context.search_spine_paths()))
     spine_file: HTMLFile | None = None
     translated_texts: list[str] = []
@@ -112,7 +114,7 @@ class _Translator:
     for translated_text in _translate(
       llm=self._llm,
       gen_fragments_iter=lambda: _gen_fragments(context),
-      cache_path=working_path / "cache",
+      cache_path=cache_path,
       max_chunk_tokens_count=self._max_chunk_tokens_count,
       max_threads_count=self._max_threads_count,
       target_language=self._target_language,

{epub_translator-0.0.2.dist-info → epub_translator-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: epub-translator
-Version: 0.0.2
+Version: 0.0.3
 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.2.dist-info → epub_translator-0.0.3.dist-info}/RECORD

@@ -21,7 +21,7 @@ epub_translator/translation/store.py,sha256=1FmksPAUj0mt3tN8Jdb_L1ovaI1p_5OhTWgx
 epub_translator/translation/translation.py,sha256=_qiw6s_z_Tv4VmIP1U-_YhDYNiKEDNfigHIIGKa41fU,5734
 epub_translator/translation/types.py,sha256=vDW5bVqYwngW_YUgf0SgfZ5zIFWUxcbBGO1U9Dsxc0o,499
 epub_translator/translation/utils.py,sha256=G6Gqq6mot3lgFA-jqUD0UqtDS0GC1wrb9DnK7rTxJNs,223
-epub_translator/translator.py,sha256=k4YmmCh0W5Zyz8CZJTXO86SShuwg8ydm38CuIB_mlTw,5253
+epub_translator/translator.py,sha256=oh7PdCijUSGU6f72hzc6doJdMWnAqg6zHRFIj3aeTjc,5332
 epub_translator/xml/__init__.py,sha256=o2_qwUYU_MUcyfmfKkiOQ-cKUQyl4PiRL8YHVzCTAZU,106
 epub_translator/xml/decoder.py,sha256=UlqgmEKQDzxt3lvBeNGHgZP6jznmnq_1HLJuAe5X0C4,2181
 epub_translator/xml/encoder.py,sha256=p4A7GRSOM2i0WOh1lLtEdTTg2gXSQrxDdzMgUqbiV18,2428
@@ -30,7 +30,7 @@ epub_translator/xml/tag.py,sha256=QLZImF0PtYyiASI7swrB8DL_qUwcYpU6cL68jEXDnvg,23
 epub_translator/xml/transform.py,sha256=vS_a4d_o2Qqf9B6k2CovQVLUknp6TyUi3FyLOu21Vio,1126
 epub_translator/xml/utils.py,sha256=KDNGWHwaIiFKS27sjZF0e-bBSjeTxzceae_aeuj4wzI,384
 epub_translator/zip_context.py,sha256=7_05kycmADb4-vxHkw_DX__vkKOxT4zo9pr2a8F4L_U,2409
-epub_translator-0.0.2.dist-info/LICENSE,sha256=5RF32sL3LtMOJIErdDKp1ZEYPGXS8WPpsiSz_jMBnGI,1066
-epub_translator-0.0.2.dist-info/METADATA,sha256=L9JKe5KAg-_naR8vLIGR043p1m_LvbGsvlSTqGzVGaw,2342
-epub_translator-0.0.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-epub_translator-0.0.2.dist-info/RECORD,,
+epub_translator-0.0.3.dist-info/LICENSE,sha256=5RF32sL3LtMOJIErdDKp1ZEYPGXS8WPpsiSz_jMBnGI,1066
+epub_translator-0.0.3.dist-info/METADATA,sha256=HuD7ogzPi96e8ZKjAJyTARhSeR4xmu3vZt0Q3Vmg0Hk,2342
+epub_translator-0.0.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+epub_translator-0.0.3.dist-info/RECORD,,