GameSentenceMiner 2.11.1__py3-none-any.whl → 2.11.2__py3-none-any.whl

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/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/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.1.dist-info → gamesentenceminer-2.11.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GameSentenceMiner
-Version: 2.11.1
+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

{gamesentenceminer-2.11.1.dist-info → gamesentenceminer-2.11.2.dist-info}/RECORD

@@ -1,12 +1,12 @@
 GameSentenceMiner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/anki.py,sha256=3BVFXAM7tpJAxHMbsMpnMHUoDfyqHQ1JSYJThW18QWA,16846
-GameSentenceMiner/config_gui.py,sha256=B0VjddeKcnc8w9nJMLK0I0XlGVOo-6KQRtISZ5K6j08,98511
+GameSentenceMiner/config_gui.py,sha256=QTK1yBDcfHaIUR_JyekkRQY9CVI_rh3Cae0bi7lviIo,99198
 GameSentenceMiner/gametext.py,sha256=6VkjmBeiuZfPk8T6PHFdIAElBH2Y_oLVYvmcafqN7RM,6747
 GameSentenceMiner/gsm.py,sha256=wTERcvG37SeDel51TCFusoQqk5B_b11YY4QZMTF0a6s,24954
 GameSentenceMiner/obs.py,sha256=o_I6213VZvXqYkZDdUBgUg2KWi9SbnNZZjjUnKnQkK4,15190
 GameSentenceMiner/vad.py,sha256=A3CvBQ67w3c7L8s7mTMxo6U_9ZQXlCToIpGUbePotfA,18321
 GameSentenceMiner/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-GameSentenceMiner/ai/ai_prompting.py,sha256=l8JKyL_4zO5x3QKExN4c3HUPF2mQfB4B-VKclO5ofmM,25348
+GameSentenceMiner/ai/ai_prompting.py,sha256=ojp7i_xg2YB1zALgFbivwtXPMVkThnSbPoUiAs-nz_g,25892
 GameSentenceMiner/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/assets/icon.png,sha256=9GRL8uXUAgkUSlvbm9Pv9o2poFVRGdW6s2ub_DeUD9M,937624
 GameSentenceMiner/assets/icon128.png,sha256=l90j7biwdz5ahwOd5wZ-406ryEV9Pan93dquJQ3e1CI,18395
@@ -30,7 +30,7 @@ GameSentenceMiner/owocr/owocr/run.py,sha256=lwZOj5nmQWgXlQOA_b2FYrMjhGWCPA7ZZCBg
 GameSentenceMiner/owocr/owocr/screen_coordinate_picker.py,sha256=Na6XStbQBtpQUSdbN3QhEswtKuU1JjReFk_K8t5ezQE,3395
 GameSentenceMiner/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/util/audio_offset_selector.py,sha256=8Stk3BP-XVIuzRv9nl9Eqd2D-1yD3JrgU-CamBywJmY,8542
-GameSentenceMiner/util/configuration.py,sha256=VXed6OKZPjwo1sK9CbpCerEsB3Dro82sEpFiQrWh97w,28961
+GameSentenceMiner/util/configuration.py,sha256=vwBehuEP2QVvbsyJx5x1EPXZEOtGsqeIQjlEES0vVdQ,28999
 GameSentenceMiner/util/electron_config.py,sha256=3VmIrcXhC-wIMMc4uqV85NrNenRl4ZUbnQfSjWEwuig,9852
 GameSentenceMiner/util/ffmpeg.py,sha256=t0tflxq170n8PZKkdw8fTZIUQfXD0p_qARa9JTdhBTc,21530
 GameSentenceMiner/util/gsm_utils.py,sha256=iRyLVcodMptRhkCzLf3hyqc6_RCktXnwApi6mLju6oQ,11565
@@ -62,9 +62,9 @@ GameSentenceMiner/web/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm
 GameSentenceMiner/web/templates/index.html,sha256=Gv3CJvNnhAzIVV_QxhNq4OD-pXDt1vKCu9k6WdHSXuA,215343
 GameSentenceMiner/web/templates/text_replacements.html,sha256=tV5c8mCaWSt_vKuUpbdbLAzXZ3ATZeDvQ9PnnAfqY0M,8598
 GameSentenceMiner/web/templates/utility.html,sha256=3flZinKNqUJ7pvrZk6xu__v67z44rXnaK7UTZ303R-8,16946
-gamesentenceminer-2.11.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-gamesentenceminer-2.11.1.dist-info/METADATA,sha256=3N_GqO3KWzYe3PeyDywk4J80KJPHUFDOq4svUTwC7t8,7319
-gamesentenceminer-2.11.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gamesentenceminer-2.11.1.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
-gamesentenceminer-2.11.1.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
-gamesentenceminer-2.11.1.dist-info/RECORD,,
+gamesentenceminer-2.11.2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+gamesentenceminer-2.11.2.dist-info/METADATA,sha256=zpe9LIoyiK5AyBRjnGH7Blp8z5rksCm3Nrxng5CKhyg,7319
+gamesentenceminer-2.11.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gamesentenceminer-2.11.2.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
+gamesentenceminer-2.11.2.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
+gamesentenceminer-2.11.2.dist-info/RECORD,,