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 = ''
    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)
                                text = text.replace('\\n', '')
                                print('Translating' + text)
                                translatedText = translateGPT(text, ' '.join(textHistory))
                                textHistory.append(translatedText)
                                
                                translatedText = textwrap.fill(translatedText, width=50)
                                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))
                    translatedText = textwrap.fill(translatedText, width=50)
                    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. You always manage to convey the original meaning of the Japanese text to your output, \
    and localize it in a way that an average American would understand. \
    The 'Context' at the top is previously translated text for the work.\
    You translate Onomatopoeia literally. \
    When I give you something to translate, answer with just the translation. \
    Translation Examples: \
    \\n<ルイ>そう、私はルイよ。= \\n<Rui> Yes, I'm Rui. \
    \\nそう、私はルイよ。= \\nYes, I'm Rui." \

    response = openai.ChatCompletion.create(
        temperature=0,
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": system},
            {"role": "user", "content": t}
        ]
    )
    return response.choices[0].message.content
    
main()