"""
* Version: 0.1.0.0.0

* Purposes:
    * Translate Markdown files from English to Simplified Chinese.
"""

import sys
import pingcap.pe.flash.util as util
import pingcap.pe.flash.markdown_processor as md_processor

if __name__ == "__main__":

    input_md_folder_path = sys.argv[1]
    output_md_folder_path = sys.argv[2]
    skip_count = 0
    mode = "standard"
    engine = "openai"

    print(util.speak_title("Translate Markdown to Markdown in Simplified Chinese"))

    if len(sys.argv) > 3:
        engine = sys.argv[3]
    if len(sys.argv) > 4:
        skip_count = int(sys.argv[4])
    if len(sys.argv) > 5:
        mode = sys.argv[5]

    if mode != "silent":
        print(
            util.speak_2nd_title(
                "From folder "
                + input_md_folder_path
                + " to folder "
                + output_md_folder_path
            )
        )

    en_1gram_list = util.load_1gram_terms_en("pe")

    translated_chapter = md_processor.translate_chapter_v2(
        input_md_folder_path,
        "en2zh",
        "terms" + "-" + "pe" + "-en-ja",
        en_1gram_list,
        skip_count=skip_count,
        mode=mode,
        engine=engine,
    )
    md_processor.write_markdown_folder(translated_chapter, output_md_folder_path, mode)