From e5bfee9cec0c1971f0c9f8d51ef159796aced9a2 Mon Sep 17 00:00:00 2001 From: yihong0618 Date: Wed, 3 Jan 2024 22:35:17 +0800 Subject: [PATCH] fix: #359 Signed-off-by: yihong0618 --- book_maker/loader/epub_loader.py | 7 ++----- book_maker/loader/helper.py | 2 +- setup.py | 2 +- 3 files changed, 4 insertions(+), 7 deletions(-) diff --git a/book_maker/loader/epub_loader.py b/book_maker/loader/epub_loader.py index fcce66a..bcd8721 100644 --- a/book_maker/loader/epub_loader.py +++ b/book_maker/loader/epub_loader.py @@ -86,7 +86,7 @@ class EPUBBookLoader(BaseBookLoader): pass epub.EpubWriter._write_items = _write_items_patch - epub.EpubReader._check_deprecated = _check_deprecated # drop the warning + epub.EpubReader._check_deprecated = _check_deprecated try: self.origin_book = epub.read_epub(self.epub_name) @@ -164,6 +164,7 @@ class EPUBBookLoader(BaseBookLoader): wait_p_list = [] for i in range(len(p_list)): p = p_list[i] + print(f"translating {i}/{len(p_list)}") temp_p = copy(p) for p_exclude in self.exclude_translate_tags.split(","): @@ -193,10 +194,6 @@ class EPUBBookLoader(BaseBookLoader): if count + length < send_num: count += length wait_p_list.append(p) - # This is because the more paragraphs, the easier it is possible to translate different numbers of paragraphs, maybe you should find better values than 15 and 2 - # if len(wait_p_list) > 15 and count > send_num / 2: - # self.helper.deal_old(wait_p_list) - # count = 0 else: self.helper.deal_old(wait_p_list, self.single_translate) wait_p_list.append(p) diff --git a/book_maker/loader/helper.py b/book_maker/loader/helper.py index f3beaa9..5f35901 100644 --- a/book_maker/loader/helper.py +++ b/book_maker/loader/helper.py @@ -40,7 +40,7 @@ class EPUBBookLoaderHelper: if not wait_p_list: return - result_txt_list = self.translate_model.translate_list(wait_p_list, context_flag) + result_txt_list = self.translate_model.translate_list(wait_p_list) for i in range(len(wait_p_list)): if i < len(result_txt_list): diff --git a/setup.py b/setup.py index 47ca539..ce4ee06 100644 --- a/setup.py +++ b/setup.py @@ -17,7 +17,7 @@ packages = [ setup( name="bbook_maker", description="The bilingual_book_maker is an AI translation tool that uses ChatGPT to assist users in creating multi-language versions of epub/txt files and books.", - version="0.6.1", + version="0.6.2", license="MIT", author="yihong0618", author_email="zouzou0208@gmail.com",