github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-06 11:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

update: refine the text of messages to the end user (#100)

Browse Source 
* update: refine the text of messages to the end user

* update: refine message text

* fix: lint

---------

Co-authored-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
Yurii 2023-03-08 11:24:05 +01:00 committed by GitHub
parent 6c72c29943
commit 3472f3e673
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
book_maker/cli.py
View File

@ -13,34 +13,34 @@ def main():
"--book_name",
dest="book_name",
type=str,
help="your epub book file path",
help="path of the epub file to be translated",
)
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",
help="OpenAI api key,if you have more than one key, please use comma"
" to split them to go beyond the rate limits",
)
parser.add_argument(
"--no_limit",
dest="no_limit",
action="store_true",
help="If you are a paying customer you can add it",
help="with a paid account, you can specify this option",
)
parser.add_argument(
"--test",
dest="test",
action="store_true",
help="if test we only translat 10 contents you can easily check",
help="only the first 10 paragraphs will be translated, for testing",
)
parser.add_argument(
"--test_num",
dest="test_num",
type=int,
default=10,
help="test num for the test",
help="how many paragraphs will be translated for testing",
)
parser.add_argument(
"-m",
@ -50,7 +50,7 @@ def main():
default="chatgptapi",
choices=["chatgptapi", "gpt3"], # support DeepL later
metavar="MODEL",
help="Which model to use, available: {%(choices)s}",
help="model to use, available: {%(choices)s}",
)
parser.add_argument(
"--language",
@ -65,7 +65,7 @@ def main():
"--resume",
dest="resume",
action="store_true",
help="if program accidentally stop you can use this to resume",
help="if program stop unexpected you can use this to resume",
)
parser.add_argument(
"-p",
@ -80,7 +80,7 @@ def main():
"--api_base",
dest="api_base",
type=str,
help="replace base url from openapi",
help="specify base url other than the OpenAI's official API address",
)
options = parser.parse_args()
@ -91,17 +91,19 @@ def main():
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")
raise Exception("OpenAI API key not provided, please google how to obtain it")
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")
raise Exception(
f"now only support files of these formats: {','.join(support_type_list)}"
)
translate_model = MODEL_DICT.get(options.model)
assert translate_model is not None, "Not support model"
assert translate_model is not None, "unsupported model"
book_loader = BOOK_LOADER_DICT.get(book_type)
assert book_loader is not None, "Not support loader"
assert book_loader is not None, "unsupported loader"
language = options.language
if options.language in LANGUAGES:
# use the value for prompt