keep translation only

yihong0618 · Mar 6, 2023 · 12d42be · 12d42be
1 parent 1a55633
commit 12d42be
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/make_book.py b/make_book.py
@@ -193,6 +193,8 @@ def make_bilingual_book(self):
                             new_p.string = self.translate_model.translate(p.text)
                             self.p_to_save.append(new_p.text)
                         p.insert_after(new_p)
+                        if (TRANSLATION_ONLY): 
+                            p.extract()
                         index += 1
                         if IS_TEST and index > TEST_NUM:
                             break
@@ -275,6 +277,12 @@ def save_progress(self):
         default="zh-hans",
         help="language to translate to",
     )
+    parser.add_argument(
+        "--translation-only",
+        dest="translation_only",
+        action="store_false",
+        help="remove original paragraphs, and keep translation only",
+    )
     parser.add_argument(
         "--resume",
         dest="resume",
@@ -308,6 +316,7 @@ def save_progress(self):
 
     OPENAI_API_KEY = options.openai_key or env.get("OPENAI_API_KEY")
     RESUME = options.resume
+    TRANSLATION_ONLY = options.translation_only
     if not OPENAI_API_KEY:
         raise Exception("Need openai API key, please google how to")
     if not options.book_name.endswith(".epub"):