GameSentenceMiner 2.13.9__py3-none-any.whl → 2.13.10__py3-none-any.whl

GameSentenceMiner/ocr/gsm_ocr_config.py

@@ -33,6 +33,7 @@ class Rectangle:
     monitor: Monitor
     coordinates: List[Union[float, int]]
     is_excluded: bool
+    is_secondary: bool = False
 
 @dataclass_json
 @dataclass
@@ -41,7 +42,8 @@ class WindowGeometry:
     top: int
     width: int
     height: int
-
+    
+    
 @dataclass_json
 @dataclass
 class OCRConfig:

GameSentenceMiner/ocr/owocr_area_selector.py

@@ -34,7 +34,7 @@ class ScreenSelector:
     def __init__(self, result, window_name, use_window_as_config, use_obs_screenshot=False):
         if not selector_available:
             raise RuntimeError("tkinter is not available.")
-        if not window_name:
+        if not window_name and not use_obs_screenshot:
             raise ValueError("A target window name is required for configuration.")
 
         obs.connect_to_obs_sync()
@@ -185,7 +185,7 @@ class ScreenSelector:
                     monitor_index = rect_data["monitor"]['index']
                     target_monitor = next((m for m in self.monitors if m['index'] == monitor_index), None)
                     if target_monitor:
-                        self.rectangles.append((target_monitor, abs_coords, rect_data["is_excluded"]))
+                        self.rectangles.append((target_monitor, abs_coords, rect_data["is_excluded"], rect_data.get("is_secondary", False)))
                         loaded_count += 1
                 except (KeyError, ValueError, TypeError) as e:
                     print(f"Skipping malformed rectangle data: {rect_data}, Error: {e}")
@@ -204,7 +204,7 @@ class ScreenSelector:
         print(f"Saving rectangles to: {config_path} relative to window: {win_geom}")
 
         serializable_rects = []
-        for monitor_dict, abs_coords, is_excluded in self.rectangles:
+        for monitor_dict, abs_coords, is_excluded, is_secondary in self.rectangles:
             x_abs, y_abs, w_abs, h_abs = abs_coords
 
             # Convert absolute pixel coordinates to percentages
@@ -217,7 +217,8 @@ class ScreenSelector:
             serializable_rects.append({
                 "monitor": {'index': monitor_dict['index']},
                 "coordinates": coords_to_save,
-                "is_excluded": is_excluded
+                "is_excluded": is_excluded,
+                "is_secondary": is_secondary
             })
 
         save_data = {
@@ -254,13 +255,14 @@ class ScreenSelector:
 
     def redo_last_rect(self, event=None):
         if not self.redo_stack: return
-        monitor, abs_coords, is_excluded, old_rect_id = self.redo_stack.pop()
+        monitor, abs_coords, is_excluded, is_secondary, old_rect_id = self.redo_stack.pop()
         canvas = event.widget.winfo_toplevel().winfo_children()[0]
         x_abs, y_abs, w_abs, h_abs = abs_coords
         canvas_x, canvas_y = x_abs - self.bounding_box['left'], y_abs - self.bounding_box['top']
+        outline_color = 'purple' if is_secondary else ('orange' if is_excluded else 'green')
         new_rect_id = canvas.create_rectangle(canvas_x, canvas_y, canvas_x + w_abs, canvas_y + h_abs,
-                                              outline='orange' if is_excluded else 'green', width=2)
-        self.rectangles.append((monitor, abs_coords, is_excluded))
+                                              outline=outline_color, width=2)
+        self.rectangles.append((monitor, abs_coords, is_excluded, is_secondary))
         self.drawn_rect_ids.append(new_rect_id)
         print("Redo: Restored rectangle.")
 
@@ -291,6 +293,7 @@ class ScreenSelector:
             "How to Use:\n"
             "• Left Click + Drag: Create a capture area (green).\n"
             "• Shift + Left Click + Drag: Create an exclusion area (orange).\n"
+            "• Ctrl + Left Click + Drag: Create a secondary (menu) area (purple).\n"
             "• Right-Click on a box: Delete it."
         )
         tk.Label(main_frame, text=instructions_text, justify=tk.LEFT, anchor="w").pack(pady=(0, 10), fill=tk.X)
@@ -426,17 +429,25 @@ class ScreenSelector:
         # --- END MODIFICATION ---
 
         # Draw existing rectangles (which were converted to absolute pixels on load)
-        for _, abs_coords, is_excluded in self.rectangles:
+        for _, abs_coords, is_excluded, is_secondary in self.rectangles:
             x_abs, y_abs, w_abs, h_abs = abs_coords
             canvas_x = x_abs - self.bounding_box['left']
             canvas_y = y_abs - self.bounding_box['top']
+            outline_color = 'purple' if is_secondary else ('orange' if is_excluded else 'green')
             rect_id = self.canvas.create_rectangle(canvas_x, canvas_y, canvas_x + w_abs, canvas_y + h_abs,
-                                              outline='orange' if is_excluded else 'green', width=2)
+                                              outline=outline_color, width=2)
             self.drawn_rect_ids.append(rect_id)
 
         def on_click(event):
             self.start_x, self.start_y = event.x, event.y
-            outline = 'purple' if bool(event.state & 0x0001) else 'red'
+            ctrl_held = bool(event.state & 0x0004)
+            shift_held = bool(event.state & 0x0001)
+            if ctrl_held:
+                outline = 'purple'
+            elif shift_held:
+                outline = 'orange'
+            else:
+                outline = 'green'
             self.current_rect_id = self.canvas.create_rectangle(self.start_x, self.start_y, self.start_x, self.start_y,
                                                            outline=outline, width=2)
 
@@ -451,8 +462,12 @@ class ScreenSelector:
             w, h = int(abs(coords[2] - coords[0])), int(abs(coords[3] - coords[1]))
 
             if w >= MIN_RECT_WIDTH and h >= MIN_RECT_HEIGHT:
-                is_excl = bool(event.state & 0x0001)
-                self.canvas.itemconfig(self.current_rect_id, outline='orange' if is_excl else 'green')
+                ctrl_held = bool(event.state & 0x0004)
+                shift_held = bool(event.state & 0x0001)
+                is_excl = shift_held
+                is_secondary = ctrl_held
+                outline_color = 'purple' if is_secondary else ('orange' if is_excl else 'green')
+                self.canvas.itemconfig(self.current_rect_id, outline=outline_color)
 
                 center_x, center_y = x_abs + w / 2, y_abs + h / 2
                 target_mon = self.monitors[0]
@@ -462,7 +477,7 @@ class ScreenSelector:
                         target_mon = mon
                         break
 
-                self.rectangles.append((target_mon, (x_abs, y_abs, w, h), is_excl))
+                self.rectangles.append((target_mon, (x_abs, y_abs, w, h), is_excl, is_secondary))
                 self.drawn_rect_ids.append(self.current_rect_id)
                 self.redo_stack.clear()
             else:
@@ -472,7 +487,7 @@ class ScreenSelector:
         def on_right_click(event):
             # Iterate through our rectangles in reverse to find the topmost one.
             for i in range(len(self.rectangles) - 1, -1, -1):
-                _monitor, abs_coords, _is_excluded = self.rectangles[i]
+                _monitor, abs_coords, _is_excluded, _is_secondary = self.rectangles[i]
                 x_abs, y_abs, w_abs, h_abs = abs_coords
                 canvas_x1 = x_abs - self.bounding_box['left']
                 canvas_y1 = y_abs - self.bounding_box['top']
@@ -540,7 +555,7 @@ def run_screen_selector(result_dict, window_name, use_window_as_config, use_obs_
 
 def get_screen_selection(window_name, use_window_as_config=False, use_obs_screenshot=False):
     if not selector_available: return None
-    if not window_name:
+    if not window_name and not use_obs_screenshot:
         print("Error: A target window name must be provided.", file=sys.stderr)
         return None
 
@@ -567,7 +582,7 @@ if __name__ == "__main__":
 
     parser = argparse.ArgumentParser(description="Screen Selector Arguments")
     parser.add_argument("window_title", nargs="?", default="", help="Target window title")
-    parser.add_argument("--obs_ocr", action="store_true", help="Use OBS screenshot")
+    parser.add_argument("--obs_ocr", action="store_true", default=True, help="Use OBS screenshot")
     parser.add_argument("--use_window_for_config", action="store_true", help="Use window for config")
     args = parser.parse_args()
 

GameSentenceMiner/ocr/owocr_helper.py

@@ -52,7 +52,6 @@ logger.addHandler(console_handler)
 def get_ocr_config(window=None, use_window_for_config=False) -> OCRConfig:
     """Loads and updates screen capture areas from the corresponding JSON file."""
     ocr_config_dir = get_ocr_config_path()
-    obs.connect_to_obs_sync(retry=0)
     obs.update_current_game()
     if use_window_for_config and window:
         scene = sanitize_filename(window)
@@ -202,7 +201,7 @@ def do_second_ocr(ocr1_text, time, img, filtering, pre_crop_image=None, ignore_f
     try:
         orig_text, text = run.process_and_write_results(img, None, last_ocr2_result if not ignore_previous_result else None, filtering, None,
                                                         engine=get_ocr_ocr2(), furigana_filter_sensitivity=furigana_filter_sensitivity if not ignore_furigana_filter else 0)
-
+        
         if compare_ocr_results(last_sent_result, text, threshold=80):
             if text:
                 logger.info("Seems like Text we already sent, not doing anything.")
@@ -447,6 +446,17 @@ def run_oneocr(ocr_config: OCRConfig, rectangles, config_check_thread):
 
 def add_ss_hotkey(ss_hotkey="ctrl+shift+g"):
     import keyboard
+
+    def ocr_secondary_rectangles():
+        logger.info("Running secondary OCR rectangles...")
+        ocr_config = get_ocr_config()
+        img = obs.get_screenshot_PIL(compression=80, img_format="jpg")
+        ocr_config.scale_to_custom_size(img.width, img.height)
+        img = run.apply_ocr_config_to_image(img, ocr_config, is_secondary=True)
+        do_second_ocr("", datetime.now(), img, TextFiltering(lang=get_ocr_language()), ignore_furigana_filter=True, ignore_previous_result=True)
+        
+    if not manual:
+        keyboard.add_hotkey(get_ocr_manual_ocr_hotkey().lower(), ocr_secondary_rectangles)
     secret_ss_hotkey = "F14"
     filtering = TextFiltering(lang=get_ocr_language())
     cropper = ScreenCropper()
@@ -544,6 +554,8 @@ if __name__ == "__main__":
         use_window_for_config = args.use_window_for_config
         keep_newline = args.keep_newline
         obs_ocr = args.obs_ocr
+        
+        obs.connect_to_obs_sync(retry=0)
     
         # Start config change checker thread
         config_check_thread = ConfigChangeCheckThread()

GameSentenceMiner/owocr/owocr/run.py

@@ -832,7 +832,6 @@ class OBSScreenshotThread(threading.Thread):
         super().__init__(daemon=True)
         self.ocr_config = ocr_config
         self.interval = interval
-        self.obs_client = None
         self.websocket = None
         self.current_source = None
         self.current_source_name = None
@@ -930,16 +929,6 @@ class OBSScreenshotThread(threading.Thread):
         init_config()
         start = time.time()
         while not terminated:
-            if time.time() - start > 5:
-                if not self.obs_client:
-                    self.connect_obs()
-                else:
-                    try:
-                        self.obs_client.get_version()
-                    except Exception as e:
-                        logger.error(f"Lost connection to OBS: {e}")
-                        self.obs_client = None
-                        self.connect_obs()
             if not screenshot_event.wait(timeout=0.1):
                 continue
 
@@ -990,15 +979,14 @@ class OBSScreenshotThread(threading.Thread):
                 continue
 
 
-def apply_ocr_config_to_image(img, ocr_config):
+def apply_ocr_config_to_image(img, ocr_config, is_secondary=False):
     for rectangle in ocr_config.rectangles:
         if rectangle.is_excluded:
             left, top, width, height = rectangle.coordinates
             draw = ImageDraw.Draw(img)
-            draw.rectangle((left, top, left + width, top +
-                           height), fill=(0, 0, 0, 0))
+            draw.rectangle((left, top, left + width, top + height), fill=(0, 0, 0, 0))
             
-    rectangles = [r for r in ocr_config.rectangles if not r.is_excluded]
+    rectangles = [r for r in ocr_config.rectangles if not r.is_excluded and r.is_secondary == is_secondary]
     
     # Sort top to bottom
     if rectangles:

{gamesentenceminer-2.13.9.dist-info → gamesentenceminer-2.13.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GameSentenceMiner
-Version: 2.13.9
+Version: 2.13.10
 Summary: A tool for mining sentences from games. Update: Overlay?
 Author-email: Beangate <bpwhelan95@gmail.com>
 License: MIT License

{gamesentenceminer-2.13.9.dist-info → gamesentenceminer-2.13.10.dist-info}/RECORD

@@ -19,17 +19,17 @@ GameSentenceMiner/locales/en_us.json,sha256=9aFJw4vls269_ufnVx03uRLXG8-E-hdG9hJ9
 GameSentenceMiner/locales/ja_jp.json,sha256=zx5R_DLsdUw9Szg7RrOkRprMcnumM5Bnl4QhsfOj-qM,27190
 GameSentenceMiner/locales/zh_cn.json,sha256=fwP5GQh0WkftdpXuyyt9IiaEIPK_ZXtEa_3kmoeQMAk,24198
 GameSentenceMiner/ocr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-GameSentenceMiner/ocr/gsm_ocr_config.py,sha256=Ezj-0k6Wo-una91FvYhMp6KGkRhWYihXzLAoh_Wu2xY,5329
+GameSentenceMiner/ocr/gsm_ocr_config.py,sha256=7h_nfC7TRQQXW-zkYT0YLvupGBCxlIp62NDOwAASxiI,5369
 GameSentenceMiner/ocr/ocrconfig.py,sha256=_tY8mjnzHMJrLS8E5pHqYXZjMuLoGKYgJwdhYgN-ny4,6466
-GameSentenceMiner/ocr/owocr_area_selector.py,sha256=tlzQp0uVT-OdM99IqycA58hBbVb4c8LdTAope2hUXWg,27319
-GameSentenceMiner/ocr/owocr_helper.py,sha256=EDKDhmZ0kF1xNmw2obZwAdXj-L-zDK09JGZPTJBja-c,26523
+GameSentenceMiner/ocr/owocr_area_selector.py,sha256=Y505tm7KT18flaIyFUe5DfBcBo99e9P3nQDgF4NTmbI,28209
+GameSentenceMiner/ocr/owocr_helper.py,sha256=_FHwiSNsKvRGWwu9huAepbr2Hpij1ewNxSQC4Gwrd7o,27156
 GameSentenceMiner/ocr/ss_picker.py,sha256=0IhxUdaKruFpZyBL-8SpxWg7bPrlGpy3lhTcMMZ5rwo,5224
 GameSentenceMiner/owocr/owocr/__init__.py,sha256=87hfN5u_PbL_onLfMACbc0F5j4KyIK9lKnRCj6oZgR0,49
 GameSentenceMiner/owocr/owocr/__main__.py,sha256=XQaqZY99EKoCpU-gWQjNbTs7Kg17HvBVE7JY8LqIE0o,157
 GameSentenceMiner/owocr/owocr/config.py,sha256=qM7kISHdUhuygGXOxmgU6Ef2nwBShrZtdqu4InDCViE,8103
 GameSentenceMiner/owocr/owocr/lens_betterproto.py,sha256=oNoISsPilVVRBBPVDtb4-roJtAhp8ZAuFTci3TGXtMc,39141
 GameSentenceMiner/owocr/owocr/ocr.py,sha256=L_Dhy5qoSmVNtzBGVHM8aCJtfFsgqbSdTqXd1Obh1EM,62531
-GameSentenceMiner/owocr/owocr/run.py,sha256=asd5RsYRlsN7FhnMgbjDcN_m3QtVCWzysb6P9_rplMY,68523
+GameSentenceMiner/owocr/owocr/run.py,sha256=0rsxeIlUys_bpwKDTfBwjER2FpgXKbfzM6lOxgFzzWU,68086
 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
@@ -67,9 +67,9 @@ GameSentenceMiner/web/templates/index.html,sha256=Gv3CJvNnhAzIVV_QxhNq4OD-pXDt1v
 GameSentenceMiner/web/templates/text_replacements.html,sha256=tV5c8mCaWSt_vKuUpbdbLAzXZ3ATZeDvQ9PnnAfqY0M,8598
 GameSentenceMiner/web/templates/utility.html,sha256=3flZinKNqUJ7pvrZk6xu__v67z44rXnaK7UTZ303R-8,16946
 GameSentenceMiner/wip/get_overlay_coords.py,sha256=nJRytHJwUBToXeAIkf45HP7Yv42YO-ILbP5h8GVeE2Q,19791
-gamesentenceminer-2.13.9.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-gamesentenceminer-2.13.9.dist-info/METADATA,sha256=XDe-B_AOtfP-Kno8LypRFVvCfU80qZZylxnB1N9eWKM,1463
-gamesentenceminer-2.13.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gamesentenceminer-2.13.9.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
-gamesentenceminer-2.13.9.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
-gamesentenceminer-2.13.9.dist-info/RECORD,,
+gamesentenceminer-2.13.10.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+gamesentenceminer-2.13.10.dist-info/METADATA,sha256=iKam6S744mbWWZ_s-slGjGGAzhy88vTsVQO3uDZCEGY,1464
+gamesentenceminer-2.13.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gamesentenceminer-2.13.10.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
+gamesentenceminer-2.13.10.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
+gamesentenceminer-2.13.10.dist-info/RECORD,,