dazed/DazedTL
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update vocab for certain files

Browse source
This commit is contained in:
dazedanon 2025-10-13 18:47:43 -05:00
parent 69fd25e355
commit 0750e7d0cb
1 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
modules/srpg.py
View file

@ -813,8 +813,9 @@ def translateGeneric(data, filename, translatedDataList=None, pbar=None):
originalMsgCount = len(msgList)
originalCommandArrayCount = len(commandArrayList)
# Keep a copy of original names for vocab update (for characters.json)
originalNameList = nameList.copy() if "characters" in filename.lower() else []
# Keep a copy of original names for vocab update (for characters/items/skills/classes/weapons)
vocab_name_files = ["characters", "items", "skills", "classes", "weapons"]
originalNameList = nameList.copy() if any(tag in filename.lower() for tag in vocab_name_files) else []
# Batch translate names
if nameList:
@ -829,12 +830,25 @@ def translateGeneric(data, filename, translatedDataList=None, pbar=None):
totalTokens[0] += response[1][0]
totalTokens[1] += response[1][1]
# Update vocab.txt for characters.json
if "characters" in filename.lower() and originalNameList and nameList:
# Update vocab.txt for name-bearing files
if originalNameList and nameList:
try:
# Create pairs of (original, translated) for vocab
vocab_pairs = list(zip(originalNameList, nameList))
update_vocab_section("Speakers", vocab_pairs)
file_lower = filename.lower()
section = None
if "characters" in file_lower:
section = "Speakers"
elif "items" in file_lower:
section = "Items"
elif "skills" in file_lower:
section = "Skills"
elif "classes" in file_lower:
section = "Classes"
elif "weapons" in file_lower:
section = "Weapons"
if section:
vocab_pairs = list(zip(originalNameList, nameList))
update_vocab_section(section, vocab_pairs)
except Exception:
traceback.print_exc()