mirror of
https://github.com/yihong0618/bilingual_book_maker.git
synced 2025-06-02 09:30:24 +00:00
128 lines
3.5 KiB
Python
128 lines
3.5 KiB
Python
import argparse
|
|
import os
|
|
from os import environ as env
|
|
|
|
from book_maker.loader import BOOK_LOADER_DICT
|
|
from book_maker.translator import MODEL_DICT
|
|
from book_maker.utils import LANGUAGES, TO_LANGUAGE_CODE
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--book_name",
|
|
dest="book_name",
|
|
type=str,
|
|
help="your epub book file path",
|
|
)
|
|
parser.add_argument(
|
|
"--openai_key",
|
|
dest="openai_key",
|
|
type=str,
|
|
default="",
|
|
help="openai api key,if you have more than one key,you can use comma"
|
|
" to split them and you can break through the limitation",
|
|
)
|
|
parser.add_argument(
|
|
"--no_limit",
|
|
dest="no_limit",
|
|
action="store_true",
|
|
help="If you are a paying customer you can add it",
|
|
)
|
|
parser.add_argument(
|
|
"--test",
|
|
dest="test",
|
|
action="store_true",
|
|
help="if test we only translat 10 contents you can easily check",
|
|
)
|
|
parser.add_argument(
|
|
"--test_num",
|
|
dest="test_num",
|
|
type=int,
|
|
default=10,
|
|
help="test num for the test",
|
|
)
|
|
parser.add_argument(
|
|
"-m",
|
|
"--model",
|
|
dest="model",
|
|
type=str,
|
|
default="chatgptapi",
|
|
choices=["chatgptapi", "gpt3"], # support DeepL later
|
|
metavar="MODEL",
|
|
help="Which model to use, available: {%(choices)s}",
|
|
)
|
|
parser.add_argument(
|
|
"--language",
|
|
type=str,
|
|
choices=sorted(LANGUAGES.keys())
|
|
+ sorted([k.title() for k in TO_LANGUAGE_CODE.keys()]),
|
|
default="zh-hans",
|
|
metavar="LANGUAGE",
|
|
help="language to translate to, available: {%(choices)s}",
|
|
)
|
|
parser.add_argument(
|
|
"--resume",
|
|
dest="resume",
|
|
action="store_true",
|
|
help="if program accidentally stop you can use this to resume",
|
|
)
|
|
parser.add_argument(
|
|
"-p",
|
|
"--proxy",
|
|
dest="proxy",
|
|
type=str,
|
|
default="",
|
|
help="use proxy like http://127.0.0.1:7890",
|
|
)
|
|
# args to change api_base
|
|
parser.add_argument(
|
|
"--api_base",
|
|
dest="api_base",
|
|
type=str,
|
|
help="replace base url from openapi",
|
|
)
|
|
|
|
options = parser.parse_args()
|
|
PROXY = options.proxy
|
|
if PROXY != "":
|
|
os.environ["http_proxy"] = PROXY
|
|
os.environ["https_proxy"] = PROXY
|
|
|
|
OPENAI_API_KEY = options.openai_key or env.get("OPENAI_API_KEY")
|
|
if not OPENAI_API_KEY:
|
|
raise Exception("Need openai API key, please google how to")
|
|
|
|
book_type = options.book_name.split(".")[-1]
|
|
support_type_list = list(BOOK_LOADER_DICT.keys())
|
|
if book_type not in support_type_list:
|
|
raise Exception(f"now only support {','.join(support_type_list)} files")
|
|
translate_model = MODEL_DICT.get(options.model)
|
|
assert translate_model is not None, "Not support model"
|
|
|
|
book_loader = BOOK_LOADER_DICT.get(book_type)
|
|
assert book_loader is not None, "Not support loader"
|
|
language = options.language
|
|
if options.language in LANGUAGES:
|
|
# use the value for prompt
|
|
language = LANGUAGES.get(language, language)
|
|
|
|
# change api_base for issue #42
|
|
model_api_base = options.api_base
|
|
|
|
e = book_loader(
|
|
options.book_name,
|
|
translate_model,
|
|
OPENAI_API_KEY,
|
|
options.resume,
|
|
language=language,
|
|
model_api_base=model_api_base,
|
|
is_test=options.test,
|
|
test_num=options.test_num,
|
|
)
|
|
e.make_bilingual_book()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|