github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-05 19:15:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: refactoring

Browse Source
This commit is contained in:
mkXultra 2024-08-16 14:03:20 +09:00
parent f412a95707
commit cbe791f4f4
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
book_maker/translator/chatgptapi_translator.py
View File

@ -412,7 +412,7 @@ class ChatGPTAPI(Base):
sanitized_book_name = sanitized_book_name.strip("._") sanitized_book_name = sanitized_book_name.strip("._")
return sanitized_book_name return sanitized_book_name
def batch_result_file_path(self): def batch_metadata_file_path(self):
return f"{os.getcwd()}/batch_files/{self.book_name}_info.json" return f"{os.getcwd()}/batch_files/{self.book_name}_info.json"
def batch_dir(self): def batch_dir(self):
@ -422,13 +422,13 @@ class ChatGPTAPI(Base):
return f"{self.book_name}-{book_index}" return f"{self.book_name}-{book_index}"
def is_completed_batch(self): def is_completed_batch(self):
batch_result_file_path = self.batch_result_file_path() batch_metadata_file_path = self.batch_metadata_file_path()
if not os.path.exists(batch_result_file_path): if not os.path.exists(batch_metadata_file_path):
print("Batch result file does not exist") print("Batch result file does not exist")
raise Exception("Batch result file does not exist") raise Exception("Batch result file does not exist")
with open(batch_result_file_path, "r", encoding="utf-8") as f: with open(batch_metadata_file_path, "r", encoding="utf-8") as f:
batch_info = json.load(f) batch_info = json.load(f)
for batch_file in batch_info["batch_files"]: for batch_file in batch_info["batch_files"]:
@ -440,8 +440,8 @@ class ChatGPTAPI(Base):
def batch_translate(self, book_index): def batch_translate(self, book_index):
if self.batch_info_cache is None: if self.batch_info_cache is None:
batch_result_file_path = self.batch_result_file_path() batch_metadata_file_path = self.batch_metadata_file_path()
with open(batch_result_file_path, "r", encoding="utf-8") as f: with open(batch_metadata_file_path, "r", encoding="utf-8") as f:
self.batch_info_cache = json.load(f) self.batch_info_cache = json.load(f)
batch_info = self.batch_info_cache batch_info = self.batch_info_cache
@ -524,12 +524,12 @@ class ChatGPTAPI(Base):
self.rotate_model() self.rotate_model()
# current working directory # current working directory
batch_dir = self.batch_dir() batch_dir = self.batch_dir()
batch_result_file_path = self.batch_result_file_path() batch_metadata_file_path = self.batch_metadata_file_path()
# cleanup batch dir and result file # cleanup batch dir and result file
if os.path.exists(batch_dir): if os.path.exists(batch_dir):
shutil.rmtree(batch_dir) shutil.rmtree(batch_dir)
if os.path.exists(batch_result_file_path): if os.path.exists(batch_metadata_file_path):
os.remove(batch_result_file_path) os.remove(batch_metadata_file_path)
os.makedirs(batch_dir, exist_ok=True) os.makedirs(batch_dir, exist_ok=True)
# batch execute # batch execute
batch_files = self.create_batch_files(batch_dir) batch_files = self.create_batch_files(batch_dir)
@ -548,7 +548,7 @@ class ChatGPTAPI(Base):
"batch_date": time.strftime("%Y-%m-%d %H:%M:%S"), "batch_date": time.strftime("%Y-%m-%d %H:%M:%S"),
"batch_files": batch_info, "batch_files": batch_info,
} }
with open(batch_result_file_path, "w", encoding="utf-8") as f: with open(batch_metadata_file_path, "w", encoding="utf-8") as f:
json.dump(batch_info_json, f, ensure_ascii=False, indent=2) json.dump(batch_info_json, f, ensure_ascii=False, indent=2)
def create_batch_info(self, file_id, batch, start_index, end_index): def create_batch_info(self, file_id, batch, start_index, end_index):