GameSentenceMiner 2.8.6__py3-none-any.whl → 2.8.7__py3-none-any.whl

GameSentenceMiner/web/texthooking_page.py

@@ -1,19 +1,25 @@
+import asyncio
 import datetime
 import json
 import os
+import queue
+import sqlite3
+import threading
 from dataclasses import dataclass
 
 import flask
+import websockets
 
-from GameSentenceMiner.text_log import GameLine, get_line_by_id
+from GameSentenceMiner.text_log import GameLine, get_line_by_id, initial_time
 from flask import request, jsonify, send_from_directory
 import webbrowser
 from GameSentenceMiner import obs
-from GameSentenceMiner.configuration import logger, get_config
+from GameSentenceMiner.configuration import logger, get_config, DB_PATH
 from GameSentenceMiner.util import TEXT_REPLACEMENTS_FILE
 
 port = get_config().general.texthooker_port
 url = f"http://localhost:{port}"
+websocket_port = 55001
 
 
 @dataclass
@@ -22,16 +28,25 @@ class EventItem:
     id: str
     text: str
     time: datetime.datetime
-    timestamp: float
     checked: bool = False
+    history: bool = False
 
     def to_dict(self):
         return {
             'id': self.id,
             'text': self.text,
             'time': self.time,
-            'timestamp': self.timestamp,
-            'checked': self.checked
+            'checked': self.checked,
+            'history': self.history,
+        }
+
+    def to_serializable(self):
+        return {
+            'id': self.id,
+            'text': self.text,
+            'time': self.time.isoformat(),
+            'checked': self.checked,
+            'history': self.history,
         }
 
 class EventManager:
@@ -43,6 +58,36 @@ class EventManager:
     def __init__(self):
         self.events = []
         self.events_dict = {}
+        self._connect()
+        self._create_table()
+        self._load_events_from_db()
+        # self.close_connection()
+
+    def _connect(self):
+        self.conn = sqlite3.connect(DB_PATH)
+        self.cursor = self.conn.cursor()
+
+    def _create_table(self):
+        self.cursor.execute("""
+            CREATE TABLE IF NOT EXISTS events (
+                event_id TEXT PRIMARY KEY,
+                line_id TEXT,
+                text TEXT,
+                time TEXT
+            )
+        """)
+        self.conn.commit()
+
+    def _load_events_from_db(self):
+        self.cursor.execute("SELECT * FROM events")
+        rows = self.cursor.fetchall()
+        for row in rows:
+            event_id, line_id, text, timestamp = row
+            timestamp = datetime.datetime.fromisoformat(timestamp)
+            line = GameLine(line_id, text, timestamp, None, None, 0)
+            event = EventItem(line, event_id, text, timestamp, False, timestamp < initial_time)
+            self.events.append(event)
+            self.events_dict[event_id] = event
 
     def __iter__(self):
         return iter(self.events)
@@ -51,20 +96,69 @@ class EventManager:
         self.events = new_events
 
     def add_gameline(self, line: GameLine):
-        new_event = EventItem(line, line.id, line.text, line.time, line.time.timestamp(), False)
+        new_event = EventItem(line, line.id, line.text, line.time, False, False)
         self.events_dict[line.id] = new_event
         self.events.append(new_event)
+        # self.store_to_db(new_event)
+        event_queue.put(new_event)
+        return new_event
 
     def get_events(self):
         return self.events
 
     def add_event(self, event):
         self.events.append(event)
+        event_queue.put(event)
 
     def get(self, event_id):
         return self.events_dict.get(event_id)
 
+    def close_connection(self):
+        if self.conn:
+            self.conn.close()
+
+class EventProcessor(threading.Thread):
+    def __init__(self, event_queue, db_path):
+        super().__init__()
+        self.event_queue = event_queue
+        self.db_path = db_path
+        self.conn = None
+        self.cursor = None
+        self.daemon = True
+
+    def _connect(self):
+        self.conn = sqlite3.connect(self.db_path)
+        self.cursor = self.conn.cursor()
+
+    def run(self):
+        self._connect()
+        while True:
+            try:
+                event = self.event_queue.get()
+                if event is None:  # Exit signal
+                    break
+                self._store_to_db(event)
+            except Exception as e:
+                logger.error(f"Error processing event: {e}")
+        self._close_connection()
+
+    def _store_to_db(self, event):
+        self.cursor.execute("""
+            INSERT INTO events (event_id, line_id, text, time)
+            VALUES (?, ?, ?, ?)
+        """, (event.id, event.line.id, event.text, event.time.isoformat()))
+        self.conn.commit()
+
+    def _close_connection(self):
+        if self.conn:
+            self.conn.close()
+
 event_manager = EventManager()
+event_queue = queue.Queue()
+
+# Initialize the EventProcessor with the queue and event manager
+event_processor = EventProcessor(event_queue, DB_PATH)
+event_processor.start()
 
 server_start_time = datetime.datetime.now().timestamp()
 
@@ -119,27 +213,28 @@ def serve_static(filename):
 
 @app.route('/')
 def index():
-    with open(os.path.join(app.root_path, 'static', 'utility.html'), encoding='utf-8') as file:
-        return file.read()
+    return flask.render_template('utility.html', websocket_port=websocket_port)
 
 @app.route('/texthooker')
 def texthooker():
-    with open(os.path.join(app.root_path, 'static', 'utility.html'), encoding='utf-8') as file:
-        return file.read()
+    return flask.render_template('utility.html', websocket_port=websocket_port)
 
 @app.route('/textreplacements')
 def textreplacements():
-    with open(os.path.join(app.root_path, 'static', 'text_replacements.html'), encoding='utf-8') as file:
-        return file.read()
+    return flask.render_template('text_replacements.html')
 
 @app.route('/data', methods=['GET'])
 def get_data():
     return jsonify([event.to_dict() for event in event_manager])
 
 
-def add_event_to_texthooker(line: GameLine):
+async def add_event_to_texthooker(line: GameLine):
     logger.info("Adding event to web server: %s", line.text)
-    event_manager.add_gameline(line)
+    new_event = event_manager.add_gameline(line)
+    await broadcast_message({
+        'event': 'text_received',
+        'data': new_event.to_serializable()
+    })
 
 
 @app.route('/update', methods=['POST'])
@@ -178,6 +273,42 @@ def play_audio():
     return jsonify({}), 200
 
 
+connected_clients = set()
+
+async def websocket_handler(websocket):
+    logger.debug(f"Client connected: {websocket.remote_address}")
+    connected_clients.add(websocket)
+    try:
+        async for message in websocket:
+            try:
+                data = json.loads(message)
+                if 'type' in data and data['type'] == 'get_events':
+                    initial_events = [{'id': 1, 'text': 'Initial event from WebSocket'}, {'id': 2, 'text': 'Another initial event'}]
+                    await websocket.send(json.dumps({'event': 'initial_events', 'payload': initial_events}))
+                elif 'update_checkbox' in data:
+                    print(f"Received checkbox update: {data}")
+                    # Handle checkbox update logic
+                    pass
+                await websocket.send(json.dumps({'response': f'Server received: {message}'}))
+            except json.JSONDecodeError:
+                await websocket.send(json.dumps({'error': 'Invalid JSON format'}))
+    except websockets.exceptions.ConnectionClosedError:
+        print(f"Client disconnected abruptly: {websocket.remote_address}")
+    except websockets.exceptions.ConnectionClosedOK:
+        print(f"Client disconnected gracefully: {websocket.remote_address}")
+    finally:
+        connected_clients.discard(websocket)
+
+async def broadcast_message(message):
+    if connected_clients:
+        tasks = [client.send(json.dumps(message)) for client in connected_clients]
+        await asyncio.gather(*tasks)
+
+# async def main():
+#     async with websockets.serve(websocket_handler, "localhost", 8765): # Choose a port for WebSocket
+#         print("WebSocket server started on ws://localhost:8765/ws (adjust as needed)")
+#         await asyncio.Future()  # Keep the server running
+
 # @app.route('/store-events', methods=['POST'])
 # def store_events():
 #     data = request.get_json()
@@ -225,5 +356,31 @@ def start_web_server():
 
     app.run(port=port, debug=False) # debug=True provides helpful error messages during development
 
+async def run_websocket_server(host="0.0.0.0", port=55001):
+    global websocket_port
+    while True:
+        websocket_port = port
+        try:
+            async with websockets.serve(websocket_handler, host, port):
+                logger.debug(f"WebSocket server started at ws://{host}:{port}/")
+                await asyncio.Future()  # Keep the WebSocket server running
+        except OSError as e:
+            logger.debug(f"Port {port} is in use. Trying the next port...")
+            port += 1
+
+
+
+async def texthooker_page_coro():
+    # Run the WebSocket server in the asyncio event loop
+    flask_thread = threading.Thread(target=start_web_server)
+    flask_thread.daemon = True
+    flask_thread.start()
+
+    # Keep the main asyncio event loop running (for the WebSocket server)
+    await run_websocket_server()
+
+def run_text_hooker_page():
+    asyncio.run(texthooker_page_coro())
+
 if __name__ == '__main__':
-    start_web_server()
+    asyncio.run(run_text_hooker_page())

{gamesentenceminer-2.8.6.dist-info → gamesentenceminer-2.8.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GameSentenceMiner
-Version: 2.8.6
+Version: 2.8.7
 Summary: A tool for mining sentences from games. Update: Multi-Line Mining! Fixed!
 Author-email: Beangate <bpwhelan95@gmail.com>
 License: MIT License
@@ -36,6 +36,7 @@ Requires-Dist: pywin32; sys_platform == "win32"
 Requires-Dist: google-generativeai
 Requires-Dist: pygetwindow; sys_platform == "win32"
 Requires-Dist: flask
+Requires-Dist: groq
 Dynamic: license-file
 
 # Game Sentence Miner

{gamesentenceminer-2.8.6.dist-info → gamesentenceminer-2.8.7.dist-info}/RECORD

@@ -1,19 +1,19 @@
 GameSentenceMiner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-GameSentenceMiner/anki.py,sha256=2Oy3a8Od8DC3vs1-ktS6KF1iUj74Ptd0SCe-U0cZcfc,14206
-GameSentenceMiner/config_gui.py,sha256=8Wc5YdAyaYT09T4pvHiL5Q4NiKRaL19CS3A0JhnSu6k,68822
-GameSentenceMiner/configuration.py,sha256=NQMsM23p4_keTFXEaKMTMacIx0TCdk4-TIl4usZkJy4,20801
+GameSentenceMiner/anki.py,sha256=pSkf-il17vKZIBt1ZHHDMfvfO9M-GdF1zox-c--KkAY,14208
+GameSentenceMiner/config_gui.py,sha256=mOsk_9rqQ09VUdUNuSMx-gHTAS2o41T3kD7qpYq3TJA,70928
+GameSentenceMiner/configuration.py,sha256=nV2FLrT-XbedPakHDKo8TddtnWj9Q2RCPFGcVs_S4RY,21673
 GameSentenceMiner/electron_config.py,sha256=dGcPYCISPehXubYSzsDuI2Gl092MYK0u3bTnkL9Jh1Y,9787
 GameSentenceMiner/ffmpeg.py,sha256=mcEcJnYl06oJGbLaymFUfqClFiHf6Hhf2SXo3UV9tvM,13378
-GameSentenceMiner/gametext.py,sha256=pPJO-NYW15SqAG-IQ0ZHpPt8py4g0yL1Lx0UlR-en-4,5431
-GameSentenceMiner/gsm.py,sha256=6BcCYNV9tGQ8FgDh6Y8bPbZ66KWP57oBUqIoVF_nnXg,24667
+GameSentenceMiner/gametext.py,sha256=ClSpOeohBWG17MRVIbhXfNDnkUdxU9mTspGv9975uEc,5422
+GameSentenceMiner/gsm.py,sha256=vCEPg7rE68AUD2u6DesjLxuZSy1gJzbiA0fig673WRM,25254
 GameSentenceMiner/model.py,sha256=JdnkT4VoPOXmOpRgFdvERZ09c9wLN6tUJxdrKlGZcqo,5305
 GameSentenceMiner/notification.py,sha256=FY39ChSRK0Y8TQ6lBGsLnpZUFPtFpSy2tweeXVoV7kc,2809
-GameSentenceMiner/obs.py,sha256=DLTcK48Dqcg-5BYC2Fp-tPCO4dVI9FuVYbwIawECkwQ,8977
+GameSentenceMiner/obs.py,sha256=GPlsFrcv1eYelXyJfpspGK0iZK5AXPkoFsIGdB7eJrk,10002
 GameSentenceMiner/package.py,sha256=YlS6QRMuVlm6mdXx0rlXv9_3erTGS21jaP3PNNWfAH0,1250
 GameSentenceMiner/text_log.py,sha256=tBuZ8ElUgPtiQV8U6U90kmRxposwIkL3fjOYejdzikc,5153
 GameSentenceMiner/util.py,sha256=bb75EQdk4Nf0i9t3XMznjd6NxTWfkOVeE5bZFstbCbU,8910
 GameSentenceMiner/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-GameSentenceMiner/ai/gemini.py,sha256=6kTQPuRH16D-1srhrWa5uPGIy-jBqNm9eRdKBSbpiXA,5423
+GameSentenceMiner/ai/ai_prompting.py,sha256=tu7MxItzuJ5iSyieWIt1YHrMDEUKYAAlur6WWLZ61D4,9137
 GameSentenceMiner/communication/__init__.py,sha256=_jGn9PJxtOAOPtJ2rI-Qu9hEHVZVpIvWlxKvqk91_zI,638
 GameSentenceMiner/communication/send.py,sha256=oOJdCS6-LNX90amkRn5FL2xqx6THGm56zHR2ntVIFTE,229
 GameSentenceMiner/communication/websocket.py,sha256=pTcUe_ZZRp9REdSU4qalhPmbT_1DKa7w18j6RfFLELA,3074
@@ -24,21 +24,21 @@ GameSentenceMiner/downloader/oneocr_dl.py,sha256=o3ANp5IodEQoQ8GPcJdg9Y8JzA_lict
 GameSentenceMiner/ocr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/ocr/gsm_ocr_config.py,sha256=zagsB4UD9mmZX_r6dFBCXZqdDa0XGk-RvIqbKoPB9lQ,1932
 GameSentenceMiner/ocr/ocrconfig.py,sha256=_tY8mjnzHMJrLS8E5pHqYXZjMuLoGKYgJwdhYgN-ny4,6466
-GameSentenceMiner/ocr/owocr_area_selector.py,sha256=a2i4kW9g9_DJKt6OJ5dlSiqJGiXkKyqAQaAiXBdrq8U,46680
-GameSentenceMiner/ocr/owocr_helper.py,sha256=J6FrcxikTbfoHnnvQfp5uvGIragASrB4I3oepBCcziE,16076
+GameSentenceMiner/ocr/owocr_area_selector.py,sha256=gwYOz-fA5qoL63wh77eyGJtBtO7YVvWyO5cHb3D0Oz4,46738
+GameSentenceMiner/ocr/owocr_helper.py,sha256=PCsPbqrECSbi4u8NIq3PJotStaYVBBbfccHu-DrqpwU,17269
 GameSentenceMiner/owocr/owocr/__init__.py,sha256=opjBOyGGyEqZCE6YdZPnyt7nVfiwyELHsXA0jAsjm14,25
 GameSentenceMiner/owocr/owocr/__main__.py,sha256=r8MI6RAmbkTWqOJ59uvXoDS7CSw5jX5war9ULGWELrA,128
-GameSentenceMiner/owocr/owocr/config.py,sha256=738QCJHEWpFhMh966plOcXYWwcshSiRsxjjIwldeTtI,7461
+GameSentenceMiner/owocr/owocr/config.py,sha256=n-xtVylb2Q_H84jb1ZsIGxPQjTNnyvnRny1RhtaLJM8,7550
 GameSentenceMiner/owocr/owocr/lens_betterproto.py,sha256=oNoISsPilVVRBBPVDtb4-roJtAhp8ZAuFTci3TGXtMc,39141
 GameSentenceMiner/owocr/owocr/ocr.py,sha256=n24Xg8Z8dbcgLpq1u4d22z3tLV1evmf0dK3-Xocv3vs,39878
-GameSentenceMiner/owocr/owocr/run.py,sha256=WP3YTocRnZh4AHysIgLokemcaowatMjwfxS0IcxIHps,47859
+GameSentenceMiner/owocr/owocr/run.py,sha256=fzRUo1iYDZffolYHffqCaIYpvBLO2r_FbXgoQ35Eiqk,51151
 GameSentenceMiner/owocr/owocr/screen_coordinate_picker.py,sha256=fjJ3CSXLti3WboGPpmsa7MWOwIXsfpHC8N4zKahGGY0,3346
 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/web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-GameSentenceMiner/web/texthooking_page.py,sha256=INCBzEUk0I_452OpDnfDMeGMaMCAJWSmq14S4V1ToHA,7227
+GameSentenceMiner/web/texthooking_page.py,sha256=bI1itAIY4iPjGlFTmtrLbcjo6pvvp_dFlSOYrEDWPJ0,12760
 GameSentenceMiner/web/static/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/web/static/apple-touch-icon.png,sha256=OcMI8af_68DA_tweOsQ5LytTyMwm7-hPW07IfrOVgEs,46132
 GameSentenceMiner/web/static/favicon-96x96.png,sha256=lOePzjiKl1JY2J1kT_PMdyEnrlJmi5GWbmXJunM12B4,16502
@@ -46,13 +46,11 @@ GameSentenceMiner/web/static/favicon.ico,sha256=7d25r_FBqRSNsAoEHpSzNoT7zyVt2DJR
 GameSentenceMiner/web/static/favicon.svg,sha256=x305AP6WlXGtrXIZlaQspdLmwteoFYUoe5FyJ9MYlJ8,11517
 GameSentenceMiner/web/static/site.webmanifest,sha256=kaeNT-FjFt-T7JGzOhXH7YSqsrDeiplZ2kDxCN_CFU4,436
 GameSentenceMiner/web/static/style.css,sha256=bPZK0NVMuyRl5NNDuT7ZTzVLKlvSsdmeVHmAW4y5FM0,7001
-GameSentenceMiner/web/static/text_replacements.html,sha256=tV5c8mCaWSt_vKuUpbdbLAzXZ3ATZeDvQ9PnnAfqY0M,8598
-GameSentenceMiner/web/static/utility.html,sha256=58br1whazdfrevWCHEzYNa1_VdBvebDDX2tWlX2QrFs,9964
 GameSentenceMiner/web/static/web-app-manifest-192x192.png,sha256=EfSNnBmsSaLfESbkGfYwbKzcjKOdzuWo18ABADfN974,51117
 GameSentenceMiner/web/static/web-app-manifest-512x512.png,sha256=wyqgCWCrLEUxSRXmaA3iJEESd-vM-ZmlTtZFBY4V8Pk,230819
-gamesentenceminer-2.8.6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-gamesentenceminer-2.8.6.dist-info/METADATA,sha256=-y0KqwQGu8y7OVhrpmzL0JSPusDvd4YtcxSip82DND4,5912
-gamesentenceminer-2.8.6.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-gamesentenceminer-2.8.6.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
-gamesentenceminer-2.8.6.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
-gamesentenceminer-2.8.6.dist-info/RECORD,,
+gamesentenceminer-2.8.7.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+gamesentenceminer-2.8.7.dist-info/METADATA,sha256=D_BRnBo7BO5pHItwjMraVgSCcZAfuJdjhhId2ouPwaA,5932
+gamesentenceminer-2.8.7.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+gamesentenceminer-2.8.7.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
+gamesentenceminer-2.8.7.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
+gamesentenceminer-2.8.7.dist-info/RECORD,,

{gamesentenceminer-2.8.6.dist-info → gamesentenceminer-2.8.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.1)
+Generator: setuptools (80.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

GameSentenceMiner/ai/gemini.py

@@ -1,143 +0,0 @@
-import google.generativeai as genai
-
-from GameSentenceMiner.configuration import get_config, logger
-
-MODEL = "gemini-2.0-flash"  # or "gemini-pro-vision" if you need image support
-
-genai.configure(api_key=get_config().ai.api_key)
-model = genai.GenerativeModel(MODEL)
-
-def translate_with_context(lines, sentence, current_line_index, game_title=""):
-    """
-    Translates a line of dialogue with context from surrounding lines.
-
-    Args:
-        lines: A list of strings representing the dialogue lines.
-        sentence: Sentence to get translation for
-        current_line_index: The index of the line to translate.
-        game_title: Optional title of the game for added context.
-
-    Returns:
-        A string containing the translated sentence with context.
-    """
-
-    if not lines or current_line_index < 0 or current_line_index >= len(lines):
-        return "Invalid input."
-
-    context_lines = []
-
-    # Get the previous 10 lines (or fewer if at the beginning)
-    for i in range(max(0, current_line_index - 10), current_line_index):
-        context_lines.append(lines[i].text)
-
-    # Get the current line
-    current_line = lines[current_line_index]
-    context_lines.append(current_line.text)
-
-    #Get the next 10 lines (or fewer if at the end)
-    for i in range(current_line_index + 1, min(current_line_index + 11, len(lines))):
-        context_lines.append(lines[i].text)
-
-    ai_config = get_config().ai
-
-#this is ugly, but prettier in the output... so idk
-    if ai_config.use_canned_translation_prompt:
-        prompt_to_use = \
-f"""
-Translate the following Japanese dialogue from the game {game_title} into natural, context-aware English. Focus on preserving the tone, intent, and emotional nuance of the original text, paying close attention to the context provided by surrounding lines. The dialogue may include slang, idioms, implied meanings, or game-specific terminology that should be adapted naturally for English-speaking players. Ensure the translation feels immersive and aligns with the game's narrative style and character voices.
-Translate only the specified line below, providing a single result. Do not include additional text, explanations, or other lines unless explicitly requested. Allow expletives if more natural. Allow HTML tags for emphasis, italics, and other formatting as needed. Please also try to preserve existing HTML tags from the specified sentence if appropriate.
-
-Line to Translate:
-"""
-    elif ai_config.use_canned_context_prompt:
-        prompt_to_use = \
-f"""
-Provide a very brief summary of the scene in English based on the provided Japanese dialogue and context. Focus on the characters' actions and the immediate situation being described.
-
-Current Sentence:
-"""
-    else:
-        prompt_to_use = ai_config.custom_prompt
-
-
-    prompt = \
-f"""
-Dialogue Context:
-
-{chr(10).join(context_lines)}
-
-I am playing the game {game_title}. With that, and the above dialogue context in mind, answer the following prompt.
-
-{prompt_to_use}
-
-{sentence}
-"""
-
-    logger.debug(prompt)
-    try:
-        response = model.generate_content(prompt)
-        return response.text.strip()
-    except Exception as e:
-        return f"Translation failed: {e}"
-
-# Example Usage: Zero Escape: 999 examples
-
-# zero_escape_dialogue1 = [
-#     "扉は開いた…？",
-#     "まさか、こんな仕掛けが…",
-#     "一体、何が起こっているんだ？",
-#     "この数字の意味は…？",
-#     "落ち着いて、考えるんだ。",
-#     "でも、時間が…",
-#     "まだ、諦めるな！",
-#     "一体、誰が…？",
-#     "まさか、あの人が…？",
-#     "もう、ダメだ…",
-#     "まだ、希望はある！",
-#     "この部屋から、脱出するんだ！",
-#     "でも、どうやって…？",
-#     "何か、手がかりがあるはずだ！",
-#     "早く、見つけないと…",
-# ]
-#
-#
-# current_line_index = 3
-# translation = translate_with_context(zero_escape_dialogue1, current_line_index, "Zero Escape: 999")
-# print(f"Original: {zero_escape_dialogue1[current_line_index]}")
-# print(f"Translation: {translation}")
-#
-# # Example with fewer context lines at the beginning.
-# zero_escape_dialogue2 = [
-#     "このアミュレット…",
-#     "何かを感じる…",
-#     "この数字は…",
-#     "９…？",
-#     "まさか、これは…",
-#     "何かの手がかり…？",
-#     "急がないと…",
-#     "時間がない…",
-#     "早く、脱出を…",
-# ]
-#
-# current_line_index = 3
-# translation = translate_with_context(zero_escape_dialogue2, current_line_index, "Zero Escape: 999")
-# print(f"Original: {zero_escape_dialogue2[current_line_index]}")
-# print(f"Translation: {translation}")
-#
-# #example with fewer context lines at the end.
-# zero_escape_dialogue3 = [
-#     "この状況、理解できない。",
-#     "誰かが、私たちを閉じ込めたのか？",
-#     "なぜ、こんなことを…？",
-#     "このゲームの目的は…？",
-#     "一体、何が真実なんだ？",
-#     "信じられるのは、誰…？",
-#     "疑心暗鬼になるな。",
-#     "でも、どうすれば…？",
-#     "とにかく、進むしかない。"
-# ]
-#
-# current_line_index = 4
-# translation = translate_with_context(zero_escape_dialogue3, current_line_index, "Zero Escape: 999")
-# print(f"Original: {zero_escape_dialogue3[current_line_index]}")
-# print(f"Translation: {translation}")