From 6546d56abe883d8c364417e487b495f5132c8b47 Mon Sep 17 00:00:00 2001
From: DazedAnon <dazedanon@gmail.com>
Date: Thu, 19 Dec 2024 13:19:23 -0600
Subject: [PATCH] Formatting

---
 modules/csv.py            |  2 +-
 modules/json.py           |  2 +-
 modules/nscript.py        |  5 +----
 modules/regex.py          | 35 +++++++++++++++++++++++++++++++++--
 modules/renpy.py          |  2 +-
 modules/rpgmakerace.py    |  2 +-
 modules/rpgmakermvmz.py   |  2 +-
 modules/rpgmakerplugin.py |  2 +-
 modules/text.py           |  2 +-
 modules/tyrano.py         |  2 +-
 modules/wolf.py           |  2 +-
 modules/wolf2.py          |  2 +-
 12 files changed, 44 insertions(+), 16 deletions(-)

diff --git a/modules/csv.py b/modules/csv.py
index b4de1cc..a9a64c2 100644
--- a/modules/csv.py
+++ b/modules/csv.py
@@ -622,4 +622,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/json.py b/modules/json.py
index 30fbd58..e4859b8 100644
--- a/modules/json.py
+++ b/modules/json.py
@@ -563,4 +563,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/nscript.py b/modules/nscript.py
index 1b00918..ee3754e 100644
--- a/modules/nscript.py
+++ b/modules/nscript.py
@@ -622,7 +622,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    if format == "json":
-        return [finalList, totalTokens]
-    else:
-        return [finalList[0], totalTokens]
+    return [finalList, totalTokens]
diff --git a/modules/regex.py b/modules/regex.py
index 4dee434..1ca3bd0 100644
--- a/modules/regex.py
+++ b/modules/regex.py
@@ -189,6 +189,24 @@ def translateRegex(data, translatedList):
         lineRegexText = r"t\s'(.*)'$"
         lineRegexSpeaker = r"n\s'(.*)'$"
         choiceRegex = r"choice:\d+\s'(.*)'"
+        titleRegex = r"title\s'(.*)'$"
+
+        # Title
+        match = re.search(titleRegex, data[i])
+        if match:
+            response = translateGPT(
+                [match.group(1)],
+                f"Reply with the {LANGUAGE} translation of the chapter title",
+                True,
+            )
+            tokens[0] += response[1][0]
+            tokens[1] += response[1][1]
+            title = response[0]
+
+            # Set
+            if translatedList:
+                title = re.sub(r"(?<!\\)'", r"\\'", title[0])
+                data[i] = data[i].replace(match.group(1), title)
 
         # Speaker
         match = re.search(lineRegexSpeaker, data[i])
@@ -198,8 +216,10 @@ def translateRegex(data, translatedList):
                 speaker = response[0]
                 tokens[0] += response[1][0]
                 tokens[1] += response[1][1]
-                
+
                 if translatedList:
+                    # Escape Quotes
+                    speaker = re.sub(r"(?<!\\)'", r"\\'", speaker)
                     data[i] = data[i].replace(match.group(1), speaker)
             else:
                 speaker = None
@@ -252,6 +272,12 @@ def translateRegex(data, translatedList):
                     # Escape Quotes
                     translatedText = re.sub(r"(?<!\\)'", r"\\'", translatedText)
 
+                    # Remove Repeating Characters
+                    pattern = re.compile(r"(.)\s*\1(?:\s*\1){" + str(10 - 1) + r",}")
+                    translatedText = pattern.sub(
+                        lambda match: match.group(0).replace(" ", "")[:10], translatedText
+                    )
+
                     # Textwrap
                     translatedText = textwrap.fill(translatedText, width=WIDTH)
                     translatedTextList = translatedText.split("\n")
@@ -442,6 +468,9 @@ def cleanTranslatedText(translatedText, varResponse):
     for target, replacement in placeholders.items():
         translatedText = translatedText.replace(target, replacement)
 
+    # Remove Repeating Characters
+    translatedText = re.sub(r"^(.){10,}$", "\1\1\1\1\1\1\1\1\1\1", translatedText)
+
     # Elongate Long Dashes (Since GPT Ignores them...)
     translatedText = elongateCharacters(translatedText)
     return translatedText
@@ -559,7 +588,9 @@ def translateGPT(text, history, fullPromptFlag):
                 extractedTranslations = extractTranslation(translatedText, True)
                 if extractedTranslations == None or len(tItem) != len(extractedTranslations):
                     # Mismatch. Try Again
-                    response = translateText(system, user, history, 0.05, format, "gpt-4")
+                    response = translateText(
+                        system, user, history, 0.05, format, "gpt-4-turbo-2024-04-09"
+                    )
                     translatedText = response.choices[0].message.content
                     totalTokens[0] += response.usage.prompt_tokens
                     totalTokens[1] += response.usage.completion_tokens
diff --git a/modules/renpy.py b/modules/renpy.py
index b7894e9..26d1918 100644
--- a/modules/renpy.py
+++ b/modules/renpy.py
@@ -546,4 +546,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/rpgmakerace.py b/modules/rpgmakerace.py
index 1f59ce2..f4484bb 100644
--- a/modules/rpgmakerace.py
+++ b/modules/rpgmakerace.py
@@ -2678,4 +2678,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/rpgmakermvmz.py b/modules/rpgmakermvmz.py
index 948f562..cbc9d09 100644
--- a/modules/rpgmakermvmz.py
+++ b/modules/rpgmakermvmz.py
@@ -2667,4 +2667,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/rpgmakerplugin.py b/modules/rpgmakerplugin.py
index ec28d25..68299ea 100644
--- a/modules/rpgmakerplugin.py
+++ b/modules/rpgmakerplugin.py
@@ -537,4 +537,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/text.py b/modules/text.py
index d2ba578..2434aae 100644
--- a/modules/text.py
+++ b/modules/text.py
@@ -537,4 +537,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/tyrano.py b/modules/tyrano.py
index 176f7c6..972ee7e 100644
--- a/modules/tyrano.py
+++ b/modules/tyrano.py
@@ -613,4 +613,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/wolf.py b/modules/wolf.py
index ed756e5..0aa72e2 100644
--- a/modules/wolf.py
+++ b/modules/wolf.py
@@ -2475,4 +2475,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]
diff --git a/modules/wolf2.py b/modules/wolf2.py
index 62651d2..24ac17d 100644
--- a/modules/wolf2.py
+++ b/modules/wolf2.py
@@ -580,4 +580,4 @@ def translateGPT(text, history, fullPromptFlag):
                 tList[index] = translatedText.replace("Placeholder Text", "")
 
     finalList = combineList(tList, text)
-    return [finalList, totalTokens]
\ No newline at end of file
+    return [finalList, totalTokens]