From cc64495d900d6b1cf0a3a43f718b71a4e77f8aea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Will=20=E4=BF=9D=E5=93=A5?= Date: Sun, 2 Apr 2023 21:05:10 +0800 Subject: [PATCH] Add some color on the translated text output (#226) --- book_maker/translator/chatgptapi_translator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/book_maker/translator/chatgptapi_translator.py b/book_maker/translator/chatgptapi_translator.py index 19f3456..1afb205 100644 --- a/book_maker/translator/chatgptapi_translator.py +++ b/book_maker/translator/chatgptapi_translator.py @@ -2,6 +2,7 @@ import re import time from copy import copy from os import environ +from rich import print import openai @@ -133,7 +134,7 @@ The total token is too long and cannot be completely translated\n # todo: Determine whether to print according to the cli option if needprint: - print(re.sub("\n{3,}", "\n\n", t_text)) + print("[bold green]" + re.sub("\n{3,}", "\n\n", t_text) + "[/bold green]") time.time() - start_time # print(f"translation time: {elapsed_time:.1f}s")