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

Adjust tyrano and rpgmakermvmz script

Browse source
This commit is contained in:
Dazed 2023-12-31 17:20:07 -06:00
parent a8b5cf5fa8
commit 156e3f7d78
2 changed files with 21 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
modules/rpgmakermvmz.py
View file

@ -1877,8 +1877,12 @@ def batchList(input_list, batch_size):
def createContext(fullPromptFlag, subbedT):
characters = 'Game Characters:\n\
リオ (Rio) - Female\n\
ラビ (Rabi) - Female\n\
瞳 (Hitomi) - Female\n\
リン (Rin) - Female\n\
アイ (Ai) - Female\n\
玫瑰 (Meigui) - Male\n\
創 (So) - Male\n\
小虎 (Kotora) - Female\n\
'
system = PROMPT if fullPromptFlag else \
@ -1933,7 +1937,7 @@ def cleanTranslatedText(translatedText, varResponse):
return [line for line in translatedText.split('\n') if line]
def extractTranslation(translatedTextList, is_list):
pattern = r'<Line(\d+)>[\\]*`?(.*?)[\\]*?`?</?Line\d+>'
pattern = r'`?<Line(\d+)>[\\]*(.*?)[\\]*?<\/?Line\d+>`?'
# If it's a batch (i.e., list), extract with tags; otherwise, return the single item.
if is_list:
return [re.findall(pattern, line)[0][1] for line in translatedTextList if re.search(pattern, line)]
@ -1977,7 +1981,7 @@ def translateGPT(text, history, fullPromptFlag):
for index, tItem in enumerate(tList):
# Before sending to translation, if we have a list of items, add the formatting
if isinstance(tItem, list):
payload = '\n'.join([f'<Line{i}>`{item}`</Line{i}>' for i, item in enumerate(tItem)])
payload = '\n'.join([f'`<Line{i}>{item}</Line{i}>`' for i, item in enumerate(tItem)])
payload = payload.replace('``', '`Placeholder Text`')
varResponse = subVars(payload)
subbedT = varResponse[0]

22
modules/tyrano.py
View file

@ -48,7 +48,7 @@ if 'gpt-3.5' in MODEL:
elif 'gpt-4' in MODEL:
INPUTAPICOST = .01
OUTPUTAPICOST = .03
BATCHSIZE = 40
BATCHSIZE = 1
def handleTyrano(filename, estimate):
global ESTIMATE
@ -177,8 +177,8 @@ def translateTyrano(data, pbar, totalLines):
speaker = ''
# Choices
elif '[eval exp="f.seltext' in data[i]:
matchList = re.findall(r'\[eval exp=.+?\'(.+)\'', data[i])
elif 'glink' in data[i]:
matchList = re.findall(r'\[glink.+text=\"(.+?)\".+', data[i])
if len(matchList) != 0:
originalText = matchList[0]
if len(textHistory) > 0:
@ -197,6 +197,7 @@ def translateTyrano(data, pbar, totalLines):
# Escape all '
translatedText = translatedText.replace('\\', '')
translatedText = translatedText.replace("'", "\\\'")
translatedText = translatedText.replace(' ', '\u00A0')
# Set Data
translatedText = data[i].replace(originalText, translatedText)
@ -360,7 +361,7 @@ def getSpeaker(speaker):
case '小虎':
return ['Kotora', [0,0]]
case '玫瑰 ':
return ['Meigui', [0,0]]
return ['Rose', [0,0]]
case '創 ':
return ['So', [0,0]]
case '葛生':
@ -518,7 +519,11 @@ def createContext(fullPromptFlag, subbedT):
'
system = PROMPT if fullPromptFlag else \
f'Output ONLY the {LANGUAGE} translation in the following format: `Translation: <{LANGUAGE.upper()}_TRANSLATION>`'
f'\
You are an expert Eroge Game translator who translates Japanese text to English.\n\
You are going to be translating text from a videogame.\n\
I will give you lines of text, and you must translate each line to the best of your ability.\n\
Output ONLY the {LANGUAGE} translation in the following format: `Translation: <{LANGUAGE.upper()}_TRANSLATION>`'
user = f'{subbedT}'
return characters, system, user
@ -540,6 +545,7 @@ def translateText(characters, system, user, history):
response = openai.chat.completions.create(
temperature=0.1,
frequency_penalty=0.1,
presence_penalty=0.1,
model=MODEL,
messages=msg,
)
@ -561,10 +567,7 @@ def cleanTranslatedText(translatedText, varResponse):
translatedText = translatedText.replace(target, replacement)
translatedText = resubVars(translatedText, varResponse[1])
if '\n' in translatedText:
return [line for line in translatedText.split('\n') if line]
else:
return [line for line in translatedText.split('\\n') if line]
return [line for line in translatedText.split('\n') if line]
def extractTranslation(translatedTextList, is_list):
pattern = r'<Line(\d+)>[\\]*`?(.*?)[\\]*?`?</?Line\d+>'
@ -654,3 +657,4 @@ def translateGPT(text, history, fullPromptFlag):
finalList = combineList(tList, text)
return [finalList, totalTokens]