GameSentenceMiner 2.5.6__py3-none-any.whl → 2.5.7__py3-none-any.whl

GameSentenceMiner/communication/__init__.py

@@ -0,0 +1,22 @@
+import os.path
+from dataclasses import dataclass, field
+from typing import Dict, Optional, Any
+
+from dataclasses_json import dataclass_json, Undefined
+from websocket import WebSocket
+
+from GameSentenceMiner.configuration import get_app_directory
+
+CONFIG_FILE = os.path.join(get_app_directory(), "shared_config.json")
+websocket: WebSocket = None
+
+@dataclass_json(undefined=Undefined.RAISE)
+@dataclass
+class Message:
+    """
+    Represents a message for inter-process communication.
+    Mimics the structure of IPC or HTTP calls.
+    """
+    function: str
+    data: Dict[str, Any] = field(default_factory=dict)
+    id: Optional[str] = None

GameSentenceMiner/communication/send.py

@@ -0,0 +1,7 @@
+import json
+
+from GameSentenceMiner.communication.websocket import websocket, Message
+
+def send_restart_signal():
+    if websocket and websocket.connected:
+        websocket.send(json.dumps(Message(function="restart").to_json()))

GameSentenceMiner/communication/websocket.py

@@ -0,0 +1,77 @@
+import asyncio
+import os.path
+
+import websockets
+import json
+
+from websocket import WebSocket
+
+from GameSentenceMiner.communication import Message
+from GameSentenceMiner.configuration import get_app_directory, logger
+
+CONFIG_FILE = os.path.join(get_app_directory(), "shared_config.json")
+websocket: WebSocket = None
+handle_websocket_message = None
+
+async def do_websocket_connection(port):
+    """
+    Connects to the WebSocket server running in the Electron app.
+    """
+    global websocket
+
+    uri = f"ws://localhost:{port}"  # Use the port from Electron
+    logger.debug(f"Electron Communication : Connecting to server at {uri}...")
+    try:
+        async with websockets.connect(uri) as websocket:
+            logger.debug(f"Connected to websocket server at {uri}")
+
+            # Send an initial message
+            message = Message(function="on_connect", data={"message": "Hello from Python!"})
+            await websocket.send(message.to_json())
+            logger.debug(f"> Sent: {message}")
+
+            # Receive messages from the server
+            while True:
+                try:
+                    response = await websocket.recv()
+                    if response is None:
+                        break
+                    logger.debug(f"Electron Communication : < Received: {response}")
+                    handle_websocket_message(Message.from_json(response))
+                    await asyncio.sleep(1)  # keep the connection alive
+                except websockets.ConnectionClosedOK:
+                    logger.debug("Electron Communication : Connection closed by server")
+                    break
+                except websockets.ConnectionClosedError as e:
+                    logger.debug(f"Electron Communication : Connection closed with error: {e}")
+                    break
+    except ConnectionRefusedError:
+        logger.debug(f"Electron Communication : Error: Could not connect to server at {uri}.  Electron App not running..")
+    except Exception as e:
+        logger.debug(f"Electron Communication : An error occurred: {e}")
+
+def connect_websocket():
+    """
+    Main function to run the WebSocket client.
+    """
+    # Load the port from the same config.json the Electron app uses
+    try:
+        with open(CONFIG_FILE, "r") as f:
+            config = json.load(f)
+            port = config["port"]
+    except FileNotFoundError:
+        print("Error: shared_config.json not found.  Using default port 8766.  Ensure Electron app creates this file.")
+        port = 8766  # Default port, same as in Electron
+    except json.JSONDecodeError:
+        print("Error: shared_config.json was not valid JSON.  Using default port 8765.")
+        port = 8766
+
+    asyncio.run(do_websocket_connection(port))
+
+def register_websocket_message_handler(handler):
+    global handle_websocket_message
+    handle_websocket_message = handler
+
+
+if __name__ == "__main__":
+    connect_websocket()

{gamesentenceminer-2.5.6.dist-info → gamesentenceminer-2.5.7.dist-info}/METADATA

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

{gamesentenceminer-2.5.6.dist-info → gamesentenceminer-2.5.7.dist-info}/RECORD

@@ -11,6 +11,9 @@ GameSentenceMiner/obs.py,sha256=N7XoPSzLk9rHi4sgsG_LS2dN06MrqwN2mpSHfjONTjE,7368
 GameSentenceMiner/package.py,sha256=YlS6QRMuVlm6mdXx0rlXv9_3erTGS21jaP3PNNWfAH0,1250
 GameSentenceMiner/util.py,sha256=tkaoU1bj8iPMTNwUCWUzFLAnT44Ot92D1tYwQMEnARw,7336
 GameSentenceMiner/utility_gui.py,sha256=aVdI9zVXADS53g7QtTgmkVK1LumBsXF4Lou3qJzgHN8,7487
+GameSentenceMiner/communication/__init__.py,sha256=_jGn9PJxtOAOPtJ2rI-Qu9hEHVZVpIvWlxKvqk91_zI,638
+GameSentenceMiner/communication/send.py,sha256=oOJdCS6-LNX90amkRn5FL2xqx6THGm56zHR2ntVIFTE,229
+GameSentenceMiner/communication/websocket.py,sha256=vrZ9KwRUyZOepjayJkxZZTsNIbHGcDLgDRO9dNDwizM,2914
 GameSentenceMiner/downloader/Untitled_json.py,sha256=RUUl2bbbCpUDUUS0fP0tdvf5FngZ7ILdA_J5TFYAXUQ,15272
 GameSentenceMiner/downloader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/downloader/download_tools.py,sha256=mI1u_FGBmBqDIpCH3jOv8DOoZ3obgP5pIf9o9SVfX2Q,8131
@@ -18,9 +21,9 @@ GameSentenceMiner/vad/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 GameSentenceMiner/vad/silero_trim.py,sha256=-thDIZLuTLra3YBj7WR16Z6JeDgSpge2YuahprBvD8I,1585
 GameSentenceMiner/vad/vosk_helper.py,sha256=BI_mg_qyrjNbuEJjXSUDoV0FWEtQtEOAPmrrNixnZ_8,5974
 GameSentenceMiner/vad/whisper_helper.py,sha256=OF4J8TPPoKPJR1uFwrWAZ2Q7v0HJkVvNGmF8l1tACX0,3447
-gamesentenceminer-2.5.6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-gamesentenceminer-2.5.6.dist-info/METADATA,sha256=7mOtW_I06CEVBOLuzCqUvR9h7h_RwrS3OBJoIwsUOqg,5435
-gamesentenceminer-2.5.6.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
-gamesentenceminer-2.5.6.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
-gamesentenceminer-2.5.6.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
-gamesentenceminer-2.5.6.dist-info/RECORD,,
+gamesentenceminer-2.5.7.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+gamesentenceminer-2.5.7.dist-info/METADATA,sha256=TVU0qDawh6UwnB9CD2y_zKATbvuVSVhTDKGjgPCRBLI,5435
+gamesentenceminer-2.5.7.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
+gamesentenceminer-2.5.7.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
+gamesentenceminer-2.5.7.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
+gamesentenceminer-2.5.7.dist-info/RECORD,,