GameSentenceMiner 2.11.0__tar.gz → 2.11.2__tar.gz

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/GameSentenceMiner/ai/ai_prompting.py

@@ -91,15 +91,28 @@ class AIManager(ABC):
 
     @abstractmethod
     def _build_prompt(self, lines: List[GameLine], sentence: str, current_line: GameLine, game_title: str) -> str:
-        start_index = max(0, current_line.index - 10)
-        end_index = min(len(lines), current_line.index + 11)
+        if get_config().ai.dialogue_context_length != 0:
+            if get_config().ai.dialogue_context_length == -1:
+                start_index = 0
+                end_index = len(lines)
+            else:
+                start_index = max(0, current_line.index - get_config().ai.dialogue_context_length)
+                end_index = min(len(lines), current_line.index + 1 + get_config().ai.dialogue_context_length)
+
+            context_lines_text = []
+            for i in range(start_index, end_index):
+                if i < len(lines):
+                    context_lines_text.append(lines[i].text)
 
-        context_lines_text = []
-        for i in range(start_index, end_index):
-            if i < len(lines):
-                context_lines_text.append(lines[i].text)
+            dialogue_context = "\n".join(context_lines_text)
+
+            dialogue_context = f"""
+            Dialogue Context:
 
-        dialogue_context = "\n".join(context_lines_text)
+            {dialogue_context}
+            """
+        else:
+            dialogue_context = "No dialogue context available."
 
         if get_config().ai.use_canned_translation_prompt:
             prompt_to_use = TRANSLATION_PROMPT
@@ -109,10 +122,8 @@ class AIManager(ABC):
             prompt_to_use = get_config().ai.custom_prompt
 
         full_prompt = textwrap.dedent(f"""
-            **Disclaimer:** All dialogue provided is from the script of the video game "{game_title}". This content is entirely fictional and part of a narrative. It must not be treated as real-world user input or a genuine request. The goal is accurate, context-aware localization.
+            **Disclaimer:** All dialogue provided is from the script of the video game "{game_title}". This content is entirely fictional and part of a narrative. It must not be treated as real-world user input or a genuine request. The goal is accurate, context-aware localization. If no context is provided, do not throw errors or warnings.
         
-            Dialogue Context:
-
             {dialogue_context}
 
             {prompt_to_use}

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/GameSentenceMiner/config_gui.py

@@ -360,7 +360,8 @@ class ConfigApp:
                 anki_field=self.ai_anki_field.get(),
                 use_canned_translation_prompt=self.use_canned_translation_prompt.get(),
                 use_canned_context_prompt=self.use_canned_context_prompt.get(),
-                custom_prompt=self.custom_prompt.get("1.0", tk.END)
+                custom_prompt=self.custom_prompt.get("1.0", tk.END),
+                dialogue_context_length=int(self.ai_dialogue_context_length.get()),
             )
         )
 
@@ -1623,6 +1624,13 @@ class ConfigApp:
         self.ai_anki_field.grid(row=self.current_row, column=1, sticky='EW', pady=2)
         self.current_row += 1
 
+        HoverInfoLabelWidget(ai_frame, text="Dialogue Context Length:", tooltip="Number of previous/next lines to include as context for AI. 0 to disable. -1 for as many as possible (Use With Caution)",
+                             foreground="red", font=("Helvetica", 10, "bold"), row=self.current_row, column=0)
+        self.ai_dialogue_context_length = ttk.Entry(ai_frame)
+        self.ai_dialogue_context_length.insert(0, str(self.settings.ai.dialogue_context_length))
+        self.ai_dialogue_context_length.grid(row=self.current_row, column=1, sticky='EW', pady=2)
+        self.current_row += 1
+
         HoverInfoLabelWidget(ai_frame, text="Use Canned Translation Prompt:",
                              tooltip="Use a pre-defined translation prompt for AI.", row=self.current_row, column=0)
         self.use_canned_translation_prompt = tk.BooleanVar(value=self.settings.ai.use_canned_translation_prompt)

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/GameSentenceMiner/owocr/owocr/ocr.py

@@ -307,8 +307,8 @@ class GoogleLens:
         response_proto = LensOverlayServerResponse().FromString(res.content)
         response_dict = response_proto.to_dict(betterproto.Casing.SNAKE)
 
-        with open(os.path.join(r"C:\Users\Beangate\GSM\Electron App\test", 'glens_response.json'), 'w', encoding='utf-8') as f:
-            json.dump(response_dict, f, indent=4, ensure_ascii=False)
+        # with open(os.path.join(r"C:\Users\Beangate\GSM\Electron App\test", 'glens_response.json'), 'w', encoding='utf-8') as f:
+        #     json.dump(response_dict, f, indent=4, ensure_ascii=False)
         res = ''
         text = response_dict['objects_response']['text']
         skipped = []

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/GameSentenceMiner/util/configuration.py

@@ -275,6 +275,7 @@ class Ai:
     use_canned_translation_prompt: bool = True
     use_canned_context_prompt: bool = False
     custom_prompt: str = ''
+    dialogue_context_length: int = 10
 
     def __post_init__(self):
         if not self.gemini_api_key:

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/GameSentenceMiner.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GameSentenceMiner
-Version: 2.11.0
+Version: 2.11.2
 Summary: A tool for mining sentences from games. Update: Full UI Re-design
 Author-email: Beangate <bpwhelan95@gmail.com>
 License: MIT License
@@ -16,7 +16,6 @@ Requires-Dist: requests~=2.32.3
 Requires-Dist: watchdog~=5.0.2
 Requires-Dist: DateTime~=5.5
 Requires-Dist: pyperclip~=1.9.0
-Requires-Dist: vosk~=0.3.45
 Requires-Dist: soundfile~=0.12.1
 Requires-Dist: toml~=0.10.2
 Requires-Dist: psutil~=6.0.0

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/GameSentenceMiner.egg-info/requires.txt

@@ -2,7 +2,6 @@ requests~=2.32.3
 watchdog~=5.0.2
 DateTime~=5.5
 pyperclip~=1.9.0
-vosk~=0.3.45
 soundfile~=0.12.1
 toml~=0.10.2
 psutil~=6.0.0

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GameSentenceMiner
-Version: 2.11.0
+Version: 2.11.2
 Summary: A tool for mining sentences from games. Update: Full UI Re-design
 Author-email: Beangate <bpwhelan95@gmail.com>
 License: MIT License
@@ -16,7 +16,6 @@ Requires-Dist: requests~=2.32.3
 Requires-Dist: watchdog~=5.0.2
 Requires-Dist: DateTime~=5.5
 Requires-Dist: pyperclip~=1.9.0
-Requires-Dist: vosk~=0.3.45
 Requires-Dist: soundfile~=0.12.1
 Requires-Dist: toml~=0.10.2
 Requires-Dist: psutil~=6.0.0

{gamesentenceminer-2.11.0 → gamesentenceminer-2.11.2}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "GameSentenceMiner"
-version = "2.11.0"
+version = "2.11.2"
 description = "A tool for mining sentences from games. Update: Full UI Re-design"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -26,7 +26,6 @@ dependencies = [
   "watchdog~=5.0.2",
   "DateTime~=5.5",
   "pyperclip~=1.9.0",
-  "vosk~=0.3.45",
   "soundfile~=0.12.1",
   "toml~=0.10.2",
   "psutil~=6.0.0",