fix: #359

Signed-off-by: yihong0618 <zouzou0208@gmail.com>

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)

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):

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",