epub-translator 0.0.4__py3-none-any.whl → 0.0.5__py3-none-any.whl

epub_translator/translation/translation.py

@@ -1,3 +1,4 @@
+from math import ceil
 from typing import Callable, Iterator, Generator
 from pathlib import Path
 from concurrent.futures import as_completed, ThreadPoolExecutor
@@ -158,7 +159,7 @@ def _translate_texts(
     text_tag="TXT",
     user_data=user_data,
     parser=lambda r: r,
-    max_tokens=texts_tokens * _PLAIN_TEXT_SCALE,
+    max_tokens=ceil(texts_tokens * _PLAIN_TEXT_SCALE),
     params={
       "target_language": language_chinese_name(target_language),
       "user_prompt": user_prompt,
@@ -179,7 +180,7 @@ def _translate_texts(
   return llm.request_xml(
     template_name="format",
     user_data=request_text,
-    max_tokens=texts_tokens * _XML_TEXT_SCALE,
+    max_tokens=ceil(texts_tokens * _XML_TEXT_SCALE),
     parser=lambda r: _parse_translated_response(r, len(texts)),
     params={
       "target_language": language_chinese_name(target_language),

epub_translator/translation/types.py

@@ -22,28 +22,19 @@ class Language(Enum):
   JAPANESE = "ja"
   KOREAN = "ko"
 
+_LANGUAGE_NAMES = {
+  Language.SIMPLIFIED_CHINESE: "简体中文",
+  Language.TRADITIONAL_CHINESE: "繁体中文",
+  Language.ENGLISH: "英语",
+  Language.FRENCH: "法语",
+  Language.GERMAN: "德语",
+  Language.SPANISH: "西班牙语",
+  Language.RUSSIAN: "俄语",
+  Language.ITALIAN: "意大利语",
+  Language.PORTUGUESE: "葡萄牙语",
+  Language.JAPANESE: "日语",
+  Language.KOREAN: "韩语",
+}
+
 def language_chinese_name(language: Language) -> str:
-  if language == Language.SIMPLIFIED_CHINESE:
-    return "简体中文"
-  elif language == Language.TRADITIONAL_CHINESE:
-    return "繁体中文"
-  elif language == Language.ENGLISH:
-    return "英语"
-  elif language == Language.FRENCH:
-    return "法语"
-  elif language == Language.GERMAN:
-    return "德语"
-  elif language == Language.SPANISH:
-    return "西班牙语"
-  elif language == Language.RUSSIAN:
-    return "俄语"
-  elif language == Language.ITALIAN:
-    return "意大利语"
-  elif language == Language.PORTUGUESE:
-    return "葡萄牙语"
-  elif language == Language.JAPANESE:
-    return "日语"
-  elif language == Language.KOREAN:
-    return "韩语"
-  else:
-    raise ValueError(f"Unknown language: {language}")
+  return _LANGUAGE_NAMES[language]

{epub_translator-0.0.4.dist-info → epub_translator-0.0.5.dist-info}/METADATA

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

@@ -18,8 +18,8 @@ epub_translator/translation/__init__.py,sha256=QvHULHbBc7NKDlV6d0hyXAFXcAWI1tcts
 epub_translator/translation/chunk.py,sha256=ERIDvhM7kZB_ZqWGw4UQfKgjIHn7TDWsxa-RuTpxRs0,3639
 epub_translator/translation/splitter.py,sha256=xOaP1p3lqY95CR0vDXdeGUMHYObiqs3y093EUAxJ-jI,2676
 epub_translator/translation/store.py,sha256=4sR3DYZuU56IaNcw3xIMf8ZQP19kcoP5MZUaM8j4gxw,1067
-epub_translator/translation/translation.py,sha256=1ETNgP1zDEbJnUEPaMIWbw1vlh2jDVHmFEJ4fqDe8Ic,7078
-epub_translator/translation/types.py,sha256=OUxqgdyvrDEUFz21b0tv28D_oIqrP77Yv_xKAoXI-7c,1231
+epub_translator/translation/translation.py,sha256=xRZtoLrAxIwZPxcf6ltyq6tzAF38C4Q20VD_O-2A-OE,7112
+epub_translator/translation/types.py,sha256=Oj86wC2Go5SvqJ5Hsv-9ZsRqdwbImLs9c3vAXycsoPQ,932
 epub_translator/translation/utils.py,sha256=G6Gqq6mot3lgFA-jqUD0UqtDS0GC1wrb9DnK7rTxJNs,223
 epub_translator/translator.py,sha256=U0ZBBpyRNdDEvrAM7AA3b3TatdUwlhM_MSTOt5Bclsw,5520
 epub_translator/xml/__init__.py,sha256=o2_qwUYU_MUcyfmfKkiOQ-cKUQyl4PiRL8YHVzCTAZU,106
@@ -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.4.dist-info/LICENSE,sha256=5RF32sL3LtMOJIErdDKp1ZEYPGXS8WPpsiSz_jMBnGI,1066
-epub_translator-0.0.4.dist-info/METADATA,sha256=rM-6yy977tIiJvpAXHCJYFGYTgjTuOxKLJhbSbYwJNs,2342
-epub_translator-0.0.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-epub_translator-0.0.4.dist-info/RECORD,,
+epub_translator-0.0.5.dist-info/LICENSE,sha256=5RF32sL3LtMOJIErdDKp1ZEYPGXS8WPpsiSz_jMBnGI,1066
+epub_translator-0.0.5.dist-info/METADATA,sha256=bSlB0b1lwQb6zWFruz5RwcIT80sfFCEZLPoqQhceYs8,2342
+epub_translator-0.0.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+epub_translator-0.0.5.dist-info/RECORD,,