130 lines
4.6 KiB
Python
130 lines
4.6 KiB
Python
import os
|
|
import re
|
|
import textwrap
|
|
import json
|
|
from dotenv import load_dotenv
|
|
import openai
|
|
|
|
load_dotenv()
|
|
openai.organization = os.getenv('org')
|
|
openai.api_key = os.getenv('key')
|
|
pipe = '###'
|
|
|
|
def main():
|
|
# Load JSON file into a Python dictionary
|
|
with open('Map013.json', 'r', encoding='utf-8') as f:
|
|
data = json.load(f)
|
|
parse401(data)
|
|
|
|
def count401Groups(data):
|
|
"""
|
|
Counts the number of groups of 401 in a dictionary of events.
|
|
|
|
Args:
|
|
data (list): A list of events.
|
|
|
|
Returns:
|
|
int: The number of groups of 401.
|
|
"""
|
|
# Count 401 Groups
|
|
events = data['events']
|
|
groupCount = 0
|
|
currentGroupCount = 0
|
|
for event in events:
|
|
if event is not None:
|
|
for page in event['pages']:
|
|
for command in page['list']:
|
|
if command['code'] == 401:
|
|
currentGroupCount += 1
|
|
else:
|
|
if currentGroupCount > 0:
|
|
groupCount += 1
|
|
currentGroupCount = 0
|
|
if currentGroupCount > 1:
|
|
groupCount += 1
|
|
currentGroupCount = 0
|
|
return groupCount
|
|
|
|
def parse401(data):
|
|
"""
|
|
Extracts the first parameter of all commands with code 401 from a dictionary of events.
|
|
|
|
Args:
|
|
data (list): A list of events.
|
|
|
|
Returns:
|
|
list: A list of strings containing the first parameter of all commands with code 401.
|
|
"""
|
|
# Extract 401 Text (Needed for Context)
|
|
translatedText = ''
|
|
untranslatedTextList = []
|
|
currentGroup = []
|
|
textHistory = []
|
|
|
|
events = data['events']
|
|
for event in events:
|
|
if event is not None:
|
|
for page in event['pages']:
|
|
try:
|
|
for i in range(len(page['list'])):
|
|
if page['list'][i]['code'] == 401:
|
|
currentGroup.append(page['list'][i]['parameters'][0])
|
|
textHistory.append(page['list'][i]['parameters'][0])
|
|
|
|
while page['list'][i+1]['code'] == 401:
|
|
del page['list'][i]
|
|
currentGroup.append(page['list'][i]['parameters'][0])
|
|
textHistory.append(page['list'][i]['parameters'][0])
|
|
else:
|
|
if len(currentGroup) > 0:
|
|
text = ''.join(currentGroup)
|
|
print('Translating' + text)
|
|
translatedText = translateGPT(text, ''.join(textHistory))
|
|
page['list'][i-1]['parameters'][0] = translatedText
|
|
if len(textHistory) > 50:
|
|
for i in range(len(currentGroup)):
|
|
textHistory.pop(0)
|
|
currentGroup = []
|
|
except IndexError:
|
|
print('End of List')
|
|
|
|
# Append leftover groups
|
|
if len(currentGroup) > 0:
|
|
translatedText = translateGPT(''.join(currentGroup), ' '.join(textHistory))
|
|
page['list'][i]['parameters'][0] = translatedText
|
|
currentGroup = []
|
|
|
|
with open('file.json', 'w', encoding='utf-8') as f:
|
|
json.dump(data, f)
|
|
|
|
def translateGPT(t, history):
|
|
|
|
pattern = r'[一-龠]+|[ぁ-ゔ]+|[ァ-ヴー]+'
|
|
if not re.search(pattern, t):
|
|
return t
|
|
|
|
"""Translate text using GPT"""
|
|
|
|
system = "Context: " + history + "\n\n###\n\n You are a professional Japanese visual novel translator, editor, and localizer, \
|
|
with a writing style could only be described as being a mixture of \
|
|
John Steinbeck's, Oscar Wilde's, and Vladimir Nabokov's writing styles. \
|
|
You always manage to carry all of the little nuances of the original Japanese text to your output, \
|
|
while still making it a prose masterpiece, and localizing it in a way that an average American would understand. \
|
|
You read and understand the 'Context' \
|
|
You translate only what I give you. \
|
|
You include line breaks in your translation. \
|
|
You translate sound effects and Onomatopoeia in their romanji form. \
|
|
You translate anything inside '<>' as a name"
|
|
|
|
response = openai.ChatCompletion.create(
|
|
temperature=0,
|
|
max_tokens = 3000,
|
|
model="gpt-3.5-turbo",
|
|
messages=[
|
|
{"role": "system", "content": system},
|
|
{"role": "user", "content": t}
|
|
]
|
|
)
|
|
return response.choices[0].message.content
|
|
|
|
main()
|