GameSentenceMiner 2.7.1__py3-none-any.whl → 2.7.3__py3-none-any.whl

GameSentenceMiner/electron_config.py

@@ -0,0 +1,315 @@
+import json
+import os
+from dataclasses import dataclass, field
+from typing import List, Optional
+from dataclasses_json import dataclass_json
+
+from GameSentenceMiner.configuration import get_app_directory
+
+
+@dataclass_json
+@dataclass
+class SteamGame:
+    id: int
+    name: str
+    processName: str
+    script: str
+
+@dataclass_json
+@dataclass
+class YuzuConfig:
+    emuPath: str = "C:\\Emulation\\Emulators\\yuzu-windows-msvc\\yuzu.exe"
+    romsPath: str = "C:\\Emulation\\Yuzu\\Games"
+    launchGameOnStart: str = ""
+    lastGameLaunched: str = ""
+
+@dataclass_json
+@dataclass
+class VNConfig:
+    vns: List[str] = field(default_factory=list)
+    textractorPath: str = ""
+    launchVNOnStart: str = ""
+    lastVNLaunched: str = ""
+
+@dataclass_json
+@dataclass
+class SteamConfig:
+    steamPath: str = ""
+    steamGames: List[SteamGame] = field(default_factory=list)
+    launchSteamOnStart: int = 0
+    lastGameLaunched: int = 0
+
+@dataclass_json
+@dataclass
+class OCRConfig:
+    twoPassOCR: bool = False
+    ocr1: str = "oneOCR"
+    ocr2: str = "glens"
+    window_name: str = ""
+    requiresOpenWindow: Optional[bool] = None
+    scanRate: Optional[float] = None
+
+@dataclass_json
+@dataclass
+class StoreConfig:
+    yuzu: YuzuConfig = field(default_factory=YuzuConfig)
+    agentScriptsPath: str = "E:\\Japanese Stuff\\agent-v0.1.4-win32-x64\\data\\scripts"
+    textractorPath: str = "E:\\Japanese Stuff\\Textractor\\Textractor.exe"
+    startConsoleMinimized: bool = False
+    autoUpdateElectron: bool = True
+    autoUpdateGSMApp: bool = False
+    pythonPath: str = ""
+    VN: VNConfig = field(default_factory=VNConfig)
+    steam: SteamConfig = field(default_factory=SteamConfig)
+    agentPath: str = ""
+    OCR: OCRConfig = field(default_factory=OCRConfig)
+
+class Store:
+    def __init__(self, config_path=os.path.join(get_app_directory(), "electron", "config.json"), defaults: Optional[StoreConfig] = None):
+        self.config_path = config_path
+        self.defaults = defaults if defaults is not None else StoreConfig()
+        self._load_config()
+
+    def _load_config(self):
+        if os.path.exists(self.config_path):
+            with open(self.config_path, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+                self.data = StoreConfig.from_dict(data)
+        else:
+            self.data = self.defaults
+            self._save_config()
+
+    def _save_config(self):
+        with open(self.config_path, 'w', encoding='utf-8') as f:
+            json.dump(self.data.to_dict(), f, indent=4)
+
+    def get(self, key, default=None):
+        keys = key.split('.')
+        value = self.data
+        for k in keys:
+            if hasattr(value, '__dataclass_fields__') and k in value.__dataclass_fields__:
+                value = getattr(value, k)
+            else:
+                return default
+        return value
+
+    def set(self, key, value):
+        keys = key.split('.')
+        current = self.data
+        for i, k in enumerate(keys):
+            if i == len(keys) - 1:
+                setattr(current, k, value)
+            else:
+                if not hasattr(current, '__dataclass_fields__') or k not in current.__dataclass_fields__:
+                    return  # Key doesn't exist in the dataclass structure
+                if not hasattr(getattr(current, k), '__dataclass_fields__'):
+                    setattr(current, k, object()) # Create a new object if it's not a dataclass instance yet
+                current = getattr(current, k)
+        self._save_config()
+
+    def delete(self, key):
+        keys = key.split('.')
+        if not keys:
+            return False
+        current = self.data
+        for i, k in enumerate(keys[:-1]):
+            if not hasattr(current, '__dataclass_fields__') or k not in current.__dataclass_fields__:
+                return False
+            current = getattr(current, k)
+        if hasattr(current, keys[-1]):
+            delattr(current, keys[-1])
+            self._save_config()
+            return True
+        return False
+
+    def print_store(self):
+        """Prints the entire contents of the store in a readable JSON format."""
+        print(json.dumps(self.data.to_dict(), indent=4))
+
+# Initialize the store
+store = Store(config_path=os.path.join(get_app_directory(), "electron", "config.json"), defaults=StoreConfig())
+
+# --- Convenience functions ---
+
+def get_auto_update_gsm_app() -> bool:
+    return store.get("autoUpdateGSMApp")
+
+def set_auto_update_gsm_app(auto_update: bool):
+    store.set("autoUpdateGSMApp", auto_update)
+
+def get_auto_update_electron() -> bool:
+    return store.get("autoUpdateElectron")
+
+def set_auto_update_electron(auto_update: bool):
+    store.set("autoUpdateElectron", auto_update)
+
+def get_python_path() -> str:
+    return store.get("pythonPath")
+
+def set_python_path(path: str):
+    store.set("pythonPath", path)
+
+# OCR
+
+def get_ocr_config() -> OCRConfig:
+    ocr_data = store.get("OCR")
+    return ocr_data if isinstance(ocr_data, OCRConfig) else OCRConfig.from_dict(ocr_data) if isinstance(ocr_data, dict) else OCRConfig()
+
+def set_ocr_config(config: OCRConfig):
+    store.set("OCR", config)
+
+def get_two_pass_ocr() -> bool:
+    return store.get("OCR.twoPassOCR")
+
+def set_two_pass_ocr(two_pass: bool):
+    store.set("OCR.twoPassOCR", two_pass)
+
+def get_ocr1() -> str:
+    return store.get("OCR.ocr1")
+
+def set_ocr1(ocr: str):
+    store.set("OCR.ocr1", ocr)
+
+def get_ocr2() -> str:
+    return store.get("OCR.ocr2")
+
+def set_ocr2(ocr: str):
+    store.set("OCR.ocr2", ocr)
+
+def get_window_name() -> str:
+    return store.get("OCR.window_name")
+
+def set_window_name(name: str):
+    store.set("OCR.window_name", name)
+
+def get_requires_open_window() -> Optional[bool]:
+    return store.get("OCR.requiresOpenWindow")
+
+def set_requires_open_window(requires_open_window: Optional[bool]):
+    store.set("OCR.requiresOpenWindow", requires_open_window)
+
+def get_ocr_scan_rate() -> Optional[int]:
+    return store.get("OCR.scanRate")
+
+def set_ocr_scan_rate(scan_rate: Optional[int]):
+    store.set("OCR.scanRate", scan_rate)
+
+# Yuzu config getters and setters
+def get_yuzu_config() -> YuzuConfig:
+    yuzu_data = store.get('yuzu')
+    return yuzu_data if isinstance(yuzu_data, YuzuConfig) else YuzuConfig.from_dict(yuzu_data) if isinstance(yuzu_data, dict) else YuzuConfig()
+
+def set_yuzu_config(config: YuzuConfig):
+    store.set('yuzu', config)
+
+
+# Yuzu emulator path getters and setters
+def get_yuzu_emu_path() -> str:
+    return store.get('yuzu.emuPath')
+
+def set_yuzu_emu_path(path: str):
+    store.set('yuzu.emuPath', path)
+
+# Yuzu ROMs path getters and setters
+def get_yuzu_roms_path() -> str:
+    return store.get('yuzu.romsPath')
+
+def set_yuzu_roms_path(path: str):
+    store.set('yuzu.romsPath', path)
+
+def get_launch_yuzu_game_on_start() -> str:
+    return store.get("yuzu.launchGameOnStart")
+
+def set_launch_yuzu_game_on_start(path: str):
+    store.set("yuzu.launchGameOnStart", path)
+
+def get_last_yuzu_game_launched() -> str:
+    return store.get("yuzu.lastGameLaunched")
+
+def set_last_yuzu_game_launched(path: str):
+    store.set("yuzu.lastGameLaunched", path)
+
+# Agent scripts path getters and setters
+def get_agent_scripts_path() -> str:
+    return store.get('agentScriptsPath')
+
+def set_agent_scripts_path(path: str):
+    store.set('agentScriptsPath', path)
+
+def set_agent_path(path: str):
+    store.set('agentPath', path)
+
+def get_agent_path() -> str:
+    return store.get('agentPath')
+
+def get_start_console_minimized() -> bool:
+    return store.get("startConsoleMinimized")
+
+def set_start_console_minimized(should_minimize: bool):
+    store.set("startConsoleMinimized", should_minimize)
+
+def get_vns() -> List[str]:
+    return store.get('VN.vns')
+
+def set_vns(vns: List[str]):
+    store.set('VN.vns', vns)
+
+def get_textractor_path() -> str:
+    return store.get("VN.textractorPath")
+
+def set_textractor_path(path: str):
+    store.set("VN.textractorPath", path)
+
+def get_launch_vn_on_start() -> str:
+    return store.get("VN.launchVNOnStart")
+
+def set_launch_vn_on_start(vn: str):
+    store.set("VN.launchVNOnStart", vn)
+
+def get_last_vn_launched() -> str:
+    return store.get("VN.lastVNLaunched")
+
+def set_last_vn_launched(vn: str):
+    store.set("VN.lastVNLaunched", vn)
+
+def get_steam_path() -> str:
+    return store.get('steam.steamPath')
+
+def set_steam_path(path: str):
+    store.set('steam.steamPath', path)
+
+def get_launch_steam_on_start() -> int:
+    return store.get('steam.launchSteamOnStart')
+
+def set_launch_steam_on_start(game_id: int):
+    store.set('steam.launchSteamOnStart', game_id)
+
+def get_last_steam_game_launched() -> int:
+    return store.get('steam.lastGameLaunched')
+
+def set_last_steam_game_launched(game_id: int):
+    store.set('steam.lastGameLaunched', game_id)
+
+def get_steam_games() -> List[SteamGame]:
+    steam_games_data = store.get('steam.steamGames')
+    return [SteamGame.from_dict(game_data) for game_data in steam_games_data] if isinstance(steam_games_data, list) else []
+
+def set_steam_games(games: List[SteamGame]):
+    store.set('steam.steamGames', [game.to_dict() for game in games])
+
+# if __name__ == "__main__":
+#     # Example usage:
+#     print(f"Initial Yuzu Emulator Path: {get_yuzu_emu_path()}")
+#     set_yuzu_emu_path("D:\\NewEmulators\\yuzu\\yuzu.exe")
+#     print(f"Updated Yuzu Emulator Path: {get_yuzu_emu_path()}")
+#
+#     ocr_config = get_ocr_config()
+#     print(f"Initial Two-Pass OCR: {ocr_config.twoPassOCR}")
+#     set_two_pass_ocr(True)
+#     print(f"Updated Two-Pass OCR: {get_two_pass_ocr()}")
+#
+#     steam_games = get_steam_games()
+#     print(f"Initial Steam Games: {[game.name for game in steam_games]}")
+#     new_games = [SteamGame(123, "Game One"), SteamGame(456, "Game Two")]
+#     set_steam_games(new_games)
+#     print(f"Updated Steam Games: {[game.name for game in get_steam_games()]}")

GameSentenceMiner/ocr/owocr_helper.py

@@ -22,6 +22,7 @@ from GameSentenceMiner.configuration import get_config, get_app_directory
 from GameSentenceMiner.gametext import get_line_history
 from GameSentenceMiner.owocr.owocr import screen_coordinate_picker, run
 from GameSentenceMiner.owocr.owocr.run import TextFiltering
+from GameSentenceMiner.electron_config import store, get_ocr_scan_rate, get_requires_open_window
 
 CONFIG_FILE = Path("ocr_config.json")
 DEFAULT_IMAGE_PATH = r"C:\Users\Beangate\Pictures\msedge_acbl8GL7Ax.jpg"  # CHANGE THIS
@@ -201,8 +202,8 @@ def run_oneocr(ocr_config, i):
     run.run(read_from="screencapture", write_to="callback",
             screen_capture_area=",".join(str(c) for c in ocr_config['rectangles'][i]) if ocr_config['rectangles'] else 'screen_1',
             screen_capture_window=ocr_config.get("window", None),
-            screen_capture_only_active_windows=True if ocr_config.get("window", None) else False,
-            screen_capture_delay_secs=.25, engine=ocr1,
+            screen_capture_only_active_windows=get_requires_open_window(),
+            screen_capture_delay_secs=get_ocr_scan_rate(), engine=ocr1,
             text_callback=text_callback,
             screen_capture_exclusions=ocr_config.get('excluded_rectangles', None),
             rectangle=i)

GameSentenceMiner/owocr/owocr/run.py

@@ -782,6 +782,8 @@ def run(read_from=None,
         else:
             screencapture_mode = 2
             screen_capture_window = screen_capture_area
+        if screen_capture_window:
+            screencapture_mode = 2
 
         if screencapture_mode != 2:
             sct = mss.mss()
@@ -815,6 +817,7 @@ def run(read_from=None,
 
             sct_params = {'top': coord_top, 'left': coord_left, 'width': coord_width, 'height': coord_height}
             logger.opt(ansi=True).info(f'Selected coordinates: {coord_left},{coord_top},{coord_width},{coord_height}')
+        custom_left, custom_top, custom_width, custom_height = [int(c.strip()) for c in screen_capture_area.split(',')]
         if screencapture_mode == 2 or screen_capture_window:
             area_invalid_error = '"screen_capture_area" must be empty, "screen_N" where N is a screen number starting from 1, a valid set of coordinates, or a valid window name'
             if sys.platform == 'darwin':
@@ -1025,8 +1028,14 @@ def run(read_from=None,
                         except pywintypes.error:
                             on_window_closed(False)
                             break
+
+                        rand = random.randint(1, 10)
+
                         img = Image.frombuffer('RGB', (bmpinfo['bmWidth'], bmpinfo['bmHeight']), bmpstr, 'raw', 'BGRX', 0, 1)
 
+                        if custom_left:
+                            img = img.crop((custom_left, custom_top, custom_width, custom_height))
+
                         win32gui.DeleteObject(save_bitmap.GetHandle())
                         save_dc.DeleteDC()
                         mfc_dc.DeleteDC()

{gamesentenceminer-2.7.1.dist-info → gamesentenceminer-2.7.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GameSentenceMiner
-Version: 2.7.1
+Version: 2.7.3
 Summary: A tool for mining sentences from games. Update: Multi-Line Mining! Fixed!
 Author-email: Beangate <bpwhelan95@gmail.com>
 License: MIT License

{gamesentenceminer-2.7.1.dist-info → gamesentenceminer-2.7.3.dist-info}/RECORD

@@ -2,6 +2,7 @@ GameSentenceMiner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 GameSentenceMiner/anki.py,sha256=9E9GRR2zylW3Gp4PNlwYS_Nn-mhojZkjFqfYlTazte8,14068
 GameSentenceMiner/config_gui.py,sha256=-1PanqdtTKiwItxeyt0piXrWo7lGMWwrC4iSo4NiPz4,67521
 GameSentenceMiner/configuration.py,sha256=TIL8yCr-FOScCA4OJt-BAtjEb50wtqFFrpmN-UlaQh4,20405
+GameSentenceMiner/electron_config.py,sha256=dGcPYCISPehXubYSzsDuI2Gl092MYK0u3bTnkL9Jh1Y,9787
 GameSentenceMiner/ffmpeg.py,sha256=iUj-1uLLJla6jjGDKuc1FbcXjMogYpoDhrQqCbQFcWA,13359
 GameSentenceMiner/gametext.py,sha256=brybSe7hRy13JAUwuSs6WqOPFH6to0detbIPVhnkinA,9062
 GameSentenceMiner/gsm.py,sha256=fcczDPDcC0vDLAPZtIY64mU3Di3RZjR40tuLIiwB5lE,24827
@@ -22,21 +23,21 @@ GameSentenceMiner/downloader/download_tools.py,sha256=mI1u_FGBmBqDIpCH3jOv8DOoZ3
 GameSentenceMiner/ocr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/ocr/ocrconfig.py,sha256=hTROOZ3On2HngXKxwQFZvnr5AxlmlMV0mPxv-F3NbMg,6476
 GameSentenceMiner/ocr/owocr_area_selector.py,sha256=6MQPCSAhuyOSEDEgc-w6jVdsELjxstzI2Zhz0JBkG8g,11703
-GameSentenceMiner/ocr/owocr_helper.py,sha256=_ah94hyLIyvHzAbPskkwwBL-abIE-CppOGgm5pjQlPw,11708
+GameSentenceMiner/ocr/owocr_helper.py,sha256=e4WMFAl_tgE6AxofxghfrvisdZO3HmKa_3z-WmkdYYE,11798
 GameSentenceMiner/owocr/owocr/__init__.py,sha256=opjBOyGGyEqZCE6YdZPnyt7nVfiwyELHsXA0jAsjm14,25
 GameSentenceMiner/owocr/owocr/__main__.py,sha256=nkaZGTCZsPDKcej2O0S1SHkyzHJEboTUBc9izyGPvqw,92
 GameSentenceMiner/owocr/owocr/config.py,sha256=738QCJHEWpFhMh966plOcXYWwcshSiRsxjjIwldeTtI,7461
 GameSentenceMiner/owocr/owocr/lens_betterproto.py,sha256=oNoISsPilVVRBBPVDtb4-roJtAhp8ZAuFTci3TGXtMc,39141
 GameSentenceMiner/owocr/owocr/ocr.py,sha256=d0uLAQotZ6xhUSWWEXW07c0-EYjeTmumHxIEsT01Iuo,36604
-GameSentenceMiner/owocr/owocr/run.py,sha256=b6a7AjQBqeNNoHC9CwN2f_KvaSJBfUQvpSBW8z5AcIo,46156
+GameSentenceMiner/owocr/owocr/run.py,sha256=S-eb8wFC-LQLiibHVORTyTKTuPaOM2Uj6awoJxGyInQ,46549
 GameSentenceMiner/owocr/owocr/screen_coordinate_picker.py,sha256=f_YIV4-2OEZ21CAEZ_pbqeTb5D28PS82Tig4ZSnkY4Q,3096
 GameSentenceMiner/vad/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/vad/silero_trim.py,sha256=ULf3zwS-JMsY82cKF7gZxREHw8L6lgpWF2U1YqgE9Oc,1681
 GameSentenceMiner/vad/vosk_helper.py,sha256=125X8C9NxFPlWWpoNsbOnEqKx8RCjXN109zNx_QXhyg,6070
 GameSentenceMiner/vad/whisper_helper.py,sha256=JJ-iltCh813XdjyEw0Wn5DaErf6PDqfH0Efu1Md8cIY,3543
-gamesentenceminer-2.7.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-gamesentenceminer-2.7.1.dist-info/METADATA,sha256=nTXm-64PukVY01SRteJ2ygsZmyuL4881dEvMnkMaP3Y,5839
-gamesentenceminer-2.7.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-gamesentenceminer-2.7.1.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
-gamesentenceminer-2.7.1.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
-gamesentenceminer-2.7.1.dist-info/RECORD,,
+gamesentenceminer-2.7.3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+gamesentenceminer-2.7.3.dist-info/METADATA,sha256=18F6DexdMo-dYWQCtWl_caEy5vo2B7E5Lkfyq4JOdwA,5839
+gamesentenceminer-2.7.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+gamesentenceminer-2.7.3.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
+gamesentenceminer-2.7.3.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
+gamesentenceminer-2.7.3.dist-info/RECORD,,