BabelDOC 0.3.21__tar.gz → 0.3.23__tar.gz

{babeldoc-0.3.21 → babeldoc-0.3.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: BabelDOC
-Version: 0.3.21
+Version: 0.3.23
 Summary: Yet Another Document Translator
 Project-URL: Homepage, https://github.com/funstory-ai/BabelDOC
 Project-URL: Issues, https://github.com/funstory-ai/BabelDOC/issues

babeldoc-0.3.23/babeldoc/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.3.23"

{babeldoc-0.3.21 → babeldoc-0.3.23}/babeldoc/const.py

@@ -3,7 +3,7 @@ import shutil
 import subprocess
 from pathlib import Path
 
-__version__ = "0.3.21"
+__version__ = "0.3.23"
 
 CACHE_FOLDER = Path.home() / ".cache" / "babeldoc"
 

{babeldoc-0.3.21 → babeldoc-0.3.23}/babeldoc/document_il/frontend/il_creater.py

@@ -346,6 +346,7 @@ def parse_font_file(doc, idx, encoding, differences):
     bbox_list = []
     data = doc.xref_stream(idx)
     face = freetype.Face(BytesIO(data))
+    scale = 1000 / face.units_per_EM
     for charmap in face.charmaps:
         if charmap.encoding_name == "FT_ENCODING_ADOBE_CUSTOM":
             face.select_charmap(freetype.FT_ENCODING_ADOBE_CUSTOM)
@@ -354,7 +355,8 @@ def parse_font_file(doc, idx, encoding, differences):
     if differences:
         for code, name in differences:
             bbox_list[code] = get_name_cbox(face, name.encode("U8"))
-    return bbox_list
+    norm_bbox_list = [[v * scale for v in box] for box in bbox_list]
+    return norm_bbox_list
 
 
 def parse_encoding(obj_str):

{babeldoc-0.3.21 → babeldoc-0.3.23}/babeldoc/document_il/translator/translator.py

@@ -215,9 +215,7 @@ class OpenAITranslator(BaseTranslator):
             **self.options,
             messages=self.prompt(text),
         )
-        self.token_count.inc(response.usage.total_tokens)
-        self.prompt_token_count.inc(response.usage.prompt_tokens)
-        self.completion_token_count.inc(response.usage.completion_tokens)
+        self.update_token_count(response)
         return response.choices[0].message.content.strip()
 
     def prompt(self, text):
@@ -255,11 +253,20 @@ class OpenAITranslator(BaseTranslator):
                 },
             ],
         )
-        self.token_count.inc(response.usage.total_tokens)
-        self.prompt_token_count.inc(response.usage.prompt_tokens)
-        self.completion_token_count.inc(response.usage.completion_tokens)
+        self.update_token_count(response)
         return response.choices[0].message.content.strip()
 
+    def update_token_count(self, response):
+        try:
+            if response.usage and response.usage.total_tokens:
+                self.token_count.inc(response.usage.total_tokens)
+            if response.usage and response.usage.prompt_tokens:
+                self.prompt_token_count.inc(response.usage.prompt_tokens)
+            if response.usage and response.usage.completion_tokens:
+                self.completion_token_count.inc(response.usage.completion_tokens)
+        except Exception as e:
+            logger.exception("Error updating token count")
+
     def get_formular_placeholder(self, placeholder_id: int):
         return "{{v" + str(placeholder_id) + "}}"
         return "{{" + str(placeholder_id) + "}}"

{babeldoc-0.3.21 → babeldoc-0.3.23}/babeldoc/high_level.py

@@ -547,7 +547,8 @@ def do_translate(
         return result
 
     except Exception as e:
-        logger.exception(f"translate error: {e}")
+        logger.error(f"translate error: {e}")
+        pm.disable = False
         pm.translate_error(e)
         raise
     finally:
@@ -712,7 +713,9 @@ def _do_translate_single(
     try:
         if translation_config.watermark_output_mode == WatermarkOutputMode.Both:
             mono_watermark_first_page_doc_bytes, dual_watermark_first_page_doc_bytes = (
-                generate_first_page_with_watermark(doc_pdf2zh, translation_config, docs)
+                generate_first_page_with_watermark(
+                    doc_pdf2zh, translation_config, docs, mediabox_data
+                )
             )
     except Exception:
         logger.warning(
@@ -762,6 +765,7 @@ def generate_first_page_with_watermark(
     mupdf: Document,
     translation_config: TranslationConfig,
     doc_il: il_version_1.Document,
+    mediabox_data: dict[int, Any] | None = None,
 ) -> (io.BytesIO, io.BytesIO):
     first_page_doc = Document()
     first_page_doc.insert_pdf(mupdf, from_page=0, to_page=0)
@@ -784,6 +788,7 @@ def generate_first_page_with_watermark(
             watermarked_temp_pdf_path.as_posix(),
             il_only_first_page_doc,
             watermarked_config,
+            mediabox_data,
         )
         result = pdf_creater.write(watermarked_config)
         mono_pdf_bytes = None

{babeldoc-0.3.21 → babeldoc-0.3.23}/babeldoc/main.py

@@ -23,7 +23,7 @@ from babeldoc.translation_config import TranslationConfig
 from babeldoc.translation_config import WatermarkOutputMode
 
 logger = logging.getLogger(__name__)
-__version__ = "0.3.21"
+__version__ = "0.3.23"
 
 
 def create_parser():
@@ -395,6 +395,9 @@ async def main():
                 progress_handler(event)
                 if config.debug:
                     logger.debug(event)
+                if event["type"] == "error":
+                    logger.error(f"Error: {event['error']}")
+                    break
                 if event["type"] == "finish":
                     result = event["translate_result"]
                     logger.info(str(result))

{babeldoc-0.3.21 → babeldoc-0.3.23}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "BabelDOC"
-version = "0.3.21"
+version = "0.3.23"
 description = "Yet Another Document Translator"
 license = "AGPL-3.0"
 readme = "README.md"
@@ -149,7 +149,7 @@ pythonpath = [".", "src"]
 testpaths = ["tests"]
 
 [bumpver]
-current_version = "0.3.21"
+current_version = "0.3.23"
 version_pattern = "MAJOR.MINOR.PATCH[.PYTAGNUM]"
 
 [bumpver.file_patterns]