From 4d1b01c87016198189ac13e99149796119877286 Mon Sep 17 00:00:00 2001 From: DazedAnon Date: Wed, 22 Jul 2026 12:27:22 -0500 Subject: [PATCH] refactor!: rebrand product from DazedMTLTool to DazedTL - Rename desktop launcher, window titles, About, and User-Agents - Point QSettings at DazedTL with one-time migration from legacy app key - Update README, Guide, skills, and launcher/script copy BREAKING CHANGE: QSettings and desktop id now use DazedTL (legacy keys migrated once) --- DazedMTLTool.desktop => DazedTL.desktop | 6 +- README.md | 12 +- START.bat | 4 +- START.sh | 6 +- data/help/00-welcome.md | 4 +- data/help/01-requirements.md | 2 +- data/help/06-examples.md | 2 +- data/skills/project_setup.md | 8 +- gameupdate/GameUpdate.bat | 2 +- gameupdate/README.md | 316 +- .../gameupdate/patch-config.example.txt | 2 +- gui/__init__.py | 46 +- gui/file_manager.py | 854 ++- gui/main.py | 55 +- gui/translation_tab.py | 6825 +++++++++-------- gui/wolf_workflow_tab.py | 6 +- gui/workflow_tab.py | 4 +- scripts/launch.sh | 2 +- scripts/start_gui.py | 4 +- tests/fixtures/README.md | 2 +- tests/run_tests.sh | 2 +- tests/test_bundled_updates.py | 2 +- util/ace/update_tools.py | 8 +- util/forge/update_tools.py | 434 +- util/gameupdate_config.py | 2 +- util/linux_desktop.py | 15 +- util/paths.py | 25 + util/translation.py | 2 +- util/wolfdawn/__init__.py | 8 +- util/wolfdawn/update_tools.py | 4 +- 30 files changed, 4353 insertions(+), 4311 deletions(-) rename DazedMTLTool.desktop => DazedTL.desktop (75%) diff --git a/DazedMTLTool.desktop b/DazedTL.desktop similarity index 75% rename from DazedMTLTool.desktop rename to DazedTL.desktop index 14da745..e46110f 100644 --- a/DazedMTLTool.desktop +++ b/DazedTL.desktop @@ -1,10 +1,10 @@ [Desktop Entry] Type=Application -Name=DazedMTLTool -GenericName=Translation Tool +Name=DazedTL +GenericName=AI Translation Tool Comment=AI translation tool for visual novels and RPG games Exec=bash -c 'exec "$(cd "$(dirname "%k")" && pwd)/scripts/launch.sh"' Icon=%k/assets/icon.png Terminal=false Categories=Development;Utility; -StartupWMClass=DazedMTLTool +StartupWMClass=DazedTL diff --git a/README.md b/README.md index 682bbab..28a0fa6 100644 --- a/README.md +++ b/README.md @@ -1,10 +1,10 @@ -# DazedMTLTool +# DazedTL An AI-powered game translation tool with a GUI. Translate RPG Maker, Ren'Py, Tyrano, Wolf RPG, Kirikiri, and other game engines from Japanese to English using GPT, Gemini, [Mistral](https://docs.mistral.ai/api), or other compatible AI models. ## Credits -- **[Sinflower](https://github.com/Sinflower)** — [RV2JSON](https://github.com/Sinflower/RV2JSON) — enables RPGMaker Ace games to be translated the same way as MV/MZ by converting rvdata2 files to JSON and back. A curated copy is bundled offline in `util/ace/offline/` and updates with DazedMTLTool. +- **[Sinflower](https://github.com/Sinflower)** — [RV2JSON](https://github.com/Sinflower/RV2JSON) — enables RPGMaker Ace games to be translated the same way as MV/MZ by converting rvdata2 files to JSON and back. A curated copy is bundled offline in `util/ace/offline/` and updates with DazedTL. - **Sakura & Kao_SSS** — TL Inspector (`util/tl_inspector/`) — in-game translation source inspector and live-edit plugin for RPG Maker MV/MZ playtesting. - **Len** — [Forge](https://gitgud.io/zero64801/forge-mvmz) MV/MZ playtest plugin (`util/forge/`), Mistral API support (provider integration and adaptive rate limiting), and batch translation mode (Anthropic Message Batches API). @@ -108,7 +108,7 @@ This means Python wasn't added to your PATH. You have two options: ### 1. Download the Tool 1. Click the green **Code** button at the top of this page and select **Download ZIP**. -2. Extract the ZIP to a folder of your choice (e.g., `C:\DazedMTLTool`). +2. Extract the ZIP to a folder of your choice (e.g., `C:\DazedTL`). ### 2. Set Up Your API Key @@ -127,7 +127,7 @@ This means Python wasn't added to your PATH. You have two options: **Windows:** Double-click `START.bat`. It will create a virtual environment, install dependencies, and open the GUI. -**Linux/macOS:** Run `./START.sh`, or double-click `DazedMTLTool.desktop` (choose **Allow Launching** when your file manager asks). From then on, either method works. +**Linux/macOS:** Run `./START.sh`, or double-click `DazedTL.desktop` (choose **Allow Launching** when your file manager asks). From then on, either method works. That's it! Use the same launcher each time you want to open the tool. @@ -350,7 +350,7 @@ After Step 0 extract, the **Database** tab discovery report classifies your game | **Classic RPG** (most dialogue in maps/common events) | Names → foundation DB → maps/events (skip narrative DB if none) | | **Hybrid** | Names → foundation DB → narrative DB → maps/events | -> **`wolf` binary:** Prebuilt `wolf` CLIs for Windows and Linux are bundled offline under `util/wolfdawn/bin//`, so no toolchain or build step is needed. They update when you update DazedMTLTool. If your platform's binary is missing, update the tool or ask the maintainer to refresh the bundled copy. +> **`wolf` binary:** Prebuilt `wolf` CLIs for Windows and Linux are bundled offline under `util/wolfdawn/bin//`, so no toolchain or build step is needed. They update when you update DazedTL. If your platform's binary is missing, update the tool or ask the maintainer to refresh the bundled copy. > **Legacy modules:** The older `Wolf RPG` / `Wolf RPG 2` modules (configured in the Engine Config tab) still exist for edge cases, but the WolfDawn workflow above is the recommended path. @@ -378,7 +378,7 @@ You can also select a block of text, right-click, and choose **Copilot → Fix / ### Modifying Tool Modules -Open the DazedMTLTool folder in VSCode and ask Copilot to make changes to the translation modules. Examples: +Open the DazedTL folder in VSCode and ask Copilot to make changes to the translation modules. Examples: - *"Add a new regex pattern to skip lines that start with //"* - *"Change the wordwrap logic to break on full-width punctuation"* diff --git a/START.bat b/START.bat index b910a60..b96d7f7 100644 --- a/START.bat +++ b/START.bat @@ -2,7 +2,7 @@ setlocal EnableDelayedExpansion echo ========================================== -echo DazedMTLTool Startup Script +echo DazedTL Startup Script echo ========================================== echo. @@ -180,7 +180,7 @@ echo. :: Launch the GUI echo ========================================== -echo Launching DazedMTLTool GUI... +echo Launching DazedTL GUI... echo ========================================== echo. diff --git a/START.sh b/START.sh index f087cf1..a7a9a38 100755 --- a/START.sh +++ b/START.sh @@ -1,5 +1,5 @@ #!/usr/bin/env bash -# DazedMTLTool startup script for Linux/macOS +# DazedTL startup script for Linux/macOS set -o pipefail @@ -150,7 +150,7 @@ ensure_vocab_file() { } echo "==========================================" -echo " DazedMTLTool Startup Script" +echo " DazedTL Startup Script" echo "==========================================" echo @@ -222,7 +222,7 @@ fi echo echo "==========================================" -echo " Launching DazedMTLTool GUI..." +echo " Launching DazedTL GUI..." echo "==========================================" echo diff --git a/data/help/00-welcome.md b/data/help/00-welcome.md index 9d26bba..70fd061 100644 --- a/data/help/00-welcome.md +++ b/data/help/00-welcome.md @@ -1,6 +1,6 @@ -# Welcome to DazedMTLTool +# Welcome to DazedTL -DazedMTLTool translates Japanese games into English with an AI API. +DazedTL translates Japanese games into English with an AI API. It has the strongest support for **RPG Maker** (MV / MZ / Ace) and **Wolf RPG** (WolfDawn), plus modules for Ren'Py, Tyrano, Kirikiri, CSV, and more. diff --git a/data/help/01-requirements.md b/data/help/01-requirements.md index 6262fd1..00681ac 100644 --- a/data/help/01-requirements.md +++ b/data/help/01-requirements.md @@ -69,7 +69,7 @@ Use either: ### Example: open the game in Cursor 1. In Cursor: **File → Open Folder** and select the game root (where `Game.exe` or Wolf `Data` lives). -2. In DazedMTLTool Workflow Step 2, click **Copy Project Setup**. +2. In DazedTL Workflow Step 2, click **Copy Project Setup**. 3. Paste into Cursor chat with the game files available (or `@`-mention key JSON files). 4. Paste the agent's labeled blocks back into the Workflow editors. diff --git a/data/help/06-examples.md b/data/help/06-examples.md index fb39de5..96a4e5f 100644 --- a/data/help/06-examples.md +++ b/data/help/06-examples.md @@ -14,7 +14,7 @@ You just downloaded a small MV game. You want English dialogue, menus that make ### 1. Get the tool ready -1. Launch DazedMTLTool (opens on this Guide). +1. Launch DazedTL (opens on this Guide). 2. Open **Configuration**, set your API key and model, save. 3. Open the game folder in Cursor so the agent can read JSON later. diff --git a/data/skills/project_setup.md b/data/skills/project_setup.md index 6d9937d..61b45a0 100644 --- a/data/skills/project_setup.md +++ b/data/skills/project_setup.md @@ -1,6 +1,6 @@ -# DazedMTLTool — Project Setup +# DazedTL — Project Setup -You are analysing a Japanese game project to produce configuration artifacts for DazedMTLTool. +You are analysing a Japanese game project to produce configuration artifacts for DazedTL. Work in the game repository. Scan files; do not invent content you did not see. --- @@ -99,7 +99,7 @@ Output as short imperative bullets suitable to paste into `skills/quirks.md`. ### Game skill rules (for `game_skill` block) Produce the per-game translation skill saved at `skills/game.md`. -DazedMTLTool **merges this file into the translation system prompt** (before quirks). +DazedTL **merges this file into the translation system prompt** (before quirks). **Translation Frame only** (one compact line per field; evidence-based): - `世界観 (Theme / setting)` - genre, world type, core atmosphere @@ -195,7 +195,7 @@ Output as short imperative bullets suitable to paste into `skills/quirks.md`. ### Game skill rules (for `game_skill` block) Produce the per-game translation skill saved at `skills/game.md`. -DazedMTLTool **merges this file into the translation system prompt** (before quirks). +DazedTL **merges this file into the translation system prompt** (before quirks). **Translation Frame only** (one compact line per field; evidence-based): - `世界観 (Theme / setting)` - genre, world type, core atmosphere diff --git a/gameupdate/GameUpdate.bat b/gameupdate/GameUpdate.bat index 8b6180e..a8705d5 100644 --- a/gameupdate/GameUpdate.bat +++ b/gameupdate/GameUpdate.bat @@ -2,7 +2,7 @@ setlocal EnableExtensions EnableDelayedExpansion REM When deployed on a game: this file sits in the game root; patch scripts live in .\gameupdate\ -REM In this repo the same layout is kept for neatness: this bat is under DazedMTLTool\gameupdate\ next to a nested gameupdate\ folder with patch.ps1. +REM In this repo the same layout is kept for neatness: this bat is under gameupdate\ next to a nested gameupdate\ folder with patch.ps1. set "GU_ROOT=%~dp0" REM Game root is this batch file's folder (not %%CD%%, so full-path launches still work). set "GAME_ROOT=!GU_ROOT!" diff --git a/gameupdate/README.md b/gameupdate/README.md index 0283b56..31b3282 100644 --- a/gameupdate/README.md +++ b/gameupdate/README.md @@ -1,159 +1,159 @@ -# Apply Patch -1. Click Code -2. Click Download ZIP -3. Extract to game folder and Replace All. - -## Future Patching -1. Run **`GameUpdate.bat`** to auto patch (Windows). - -### Folder layout - -**In this translation-tool repo:** Patch payloads stay tidy inside **`gameupdate/gameupdate/`** (`patch.ps1`, `patch-config.txt`, etc.). **`GameUpdate.bat`** and **`GameUpdate_linux.sh`** sit one level up, under **`gameupdate/`**. - -**On an installed game (what gets copied over):** Put **`GameUpdate.bat`** in the **game root**, next to the game exe. Put **`patch.ps1`** and friends inside **`gameupdate\`** under that same root (mirror names—still **`gameupdate\`**). **`GameUpdate.bat`** finds **`gameupdate\patch.ps1`** from its own folder, so **`GameRoot`** is correct even if the console cwd is somewhere else. - -**Copy checklist:** From **`DazedMTLTool/gameupdate/`**, copy **`GameUpdate.bat`** to `\`; copy everything inside **`gameupdate/gameupdate/`** into `\gameupdate\`. - -2. Optional: set `GAMEUPDATE_PROMPT_PWSH=1` before running `GameUpdate.bat` if you want users to be prompted to install PowerShell 7 via winget. -3. Optional: set `GAMEUPDATE_DL_ATTEMPTS` (default `2`) to control retries for API checks/downloads. Lower values fail faster; higher values tolerate flaky networks. - -### patch-config.txt - -Create `gameupdate/patch-config.txt` next to `patch.ps1` (see `patch-config.example.txt`). - -In DazedMTLTool, set **Config → Game Update Defaults** (forge, host, org/username, branch) once. Step 1 **Copy gameupdate/** writes those into each game's `patch-config.txt`. You still set `repo=` per game. - -```txt -forge=gitlab -host=gitgud.io -username=YOUR_ORG_OR_USER -repo=YOUR_PATCH_REPO -branch=main -``` - -| Key | Meaning | -|-----|---------| -| `forge` | `gitlab` (default), `github`, or `forgejo` (`gitea` also accepted) | -| `host` | Hostname only. Defaults: `gitgud.io` / `github.com` / `codeberg.org` | -| `username` | Owner / org / group (`owner=` / `org=` aliases work) | -| `repo` | Repository name | -| `branch` | Branch to track | - -Examples: - -```txt -# GitLab / gitgud -forge=gitlab -host=gitgud.io -username=myorg -repo=cool-game-en -branch=main - -# GitHub -forge=github -host=github.com -username=myorg -repo=cool-game-en -branch=main - -# Forgejo / Codeberg -forge=forgejo -host=codeberg.org -username=myorg -repo=cool-game-en -branch=main -``` - -Older configs with only `username` / `repo` / `branch` still work (assumes GitLab on `gitgud.io`). - -# Troubleshooting -**GAMEUPDATE.bat doesn't update and closes immediately** -1. Make sure your path doesn't contain any Japanese characters or lots of whitespace. -2. Make sure you actually have permissions in the folder -3. Auto-update calls the forge's public HTTP API (GitLab `/api/v4`, GitHub `/repos/...`, Forgejo `/api/v1`), not the web “Download ZIP” button - no account is required for public patch repos. - -# Wolf Games -WOLF RPG installs from DLSite ship a master `Data.wolf` archive that takes priority over loose English patch files. `GameUpdate.bat` detects that and unpacks it automatically with the bundled `UberWolfCli.exe` (MIT, [Sinflower/UberWolf](https://github.com/Sinflower/UberWolf)), then renames `Data.wolf` to `Data.wolf.bak` so the patched loose `Data/` files load. - -Just run **`GameUpdate.bat`**. No manual UberWolf download is required. - -If unpack fails (rare Pro / protected builds), unpack once with [UberWolf](https://github.com/Sinflower/UberWolf/releases), ensure a loose `Data/` folder exists, rename or remove `Data.wolf`, delete `gameupdate/previous_patch_sha.txt`, and run `GameUpdate.bat` again. - -# Edit/Contribute -TLDR 3 steps. - - Fork the repository. - Make the changes. - Submit a merge request. - -If everything looks good and doesn't break things I'll merge it in. - -Longer Version: - -# Required Software: -* [VSCode](https://code.visualstudio.com/) Make sure you check all the boxes for context menus. -* [Git](https://git-scm.com/downloads) (Use the default for everything. Just keep clicking Next) - -# Guide to contributing - -### 1. Fork the Repository -- Go to the repository you want to fork. -- Click the "Fork" button. - -### 2. Clone Your Fork -- Clone your forked repository to your local machine. - ```sh - git clone https://gitgud.io/YOUR_USERNAME/REPO_NAME.git - ``` - -### 3. Make Your Changes (In VSCode) -- Edit the files locally on your new branch using VSCode. -- Add and commit your changes. - ```sh - git add . - git commit -m "Description of your changes" - ``` - -### 4. Push Your Changes -- Push your changes to your fork on GitGud.io. - ```sh - git push origin your-feature-branch - ``` - -### 5. Create a Merge Request -- Go to your fork on GitGud.io. -- Click on "Merge Requests" in the sidebar. -- Click the "New merge request" button. -- Select the branch you made changes to and the target project (the original repo). -- Provide a title and description for your merge request and submit it. - ---- - -## Example - -Assuming you want to fork a repository named `example-project`: - -### 1. Fork the Repo -- Navigate to `https://gitgud.io/original_user/example-project` and click "Fork". - -### 2. Clone Your Fork - ```sh - git clone https://gitgud.io/YOUR_USERNAME/example-project.git - ``` - -### 3. Make Changes and Commit - ```sh - # Make changes to the files - git add . - git commit -m "Add new feature to example project" - ``` - -### 4. Push Changes - ```sh - git push origin add-new-feature - ``` - -### 5. Create a Merge Request -- Go to `https://gitgud.io/YOUR_USERNAME/example-project/merge_requests` and click on "New merge request" -- Choose the source branch `add-new-feature` and target branch (default: `main` or `master`) +# Apply Patch +1. Click Code +2. Click Download ZIP +3. Extract to game folder and Replace All. + +## Future Patching +1. Run **`GameUpdate.bat`** to auto patch (Windows). + +### Folder layout + +**In this translation-tool repo:** Patch payloads stay tidy inside **`gameupdate/gameupdate/`** (`patch.ps1`, `patch-config.txt`, etc.). **`GameUpdate.bat`** and **`GameUpdate_linux.sh`** sit one level up, under **`gameupdate/`**. + +**On an installed game (what gets copied over):** Put **`GameUpdate.bat`** in the **game root**, next to the game exe. Put **`patch.ps1`** and friends inside **`gameupdate\`** under that same root (mirror names—still **`gameupdate\`**). **`GameUpdate.bat`** finds **`gameupdate\patch.ps1`** from its own folder, so **`GameRoot`** is correct even if the console cwd is somewhere else. + +**Copy checklist:** From the tool repo's **`gameupdate/`**, copy **`GameUpdate.bat`** to `\`; copy everything inside **`gameupdate/gameupdate/`** into `\gameupdate\`. + +2. Optional: set `GAMEUPDATE_PROMPT_PWSH=1` before running `GameUpdate.bat` if you want users to be prompted to install PowerShell 7 via winget. +3. Optional: set `GAMEUPDATE_DL_ATTEMPTS` (default `2`) to control retries for API checks/downloads. Lower values fail faster; higher values tolerate flaky networks. + +### patch-config.txt + +Create `gameupdate/patch-config.txt` next to `patch.ps1` (see `patch-config.example.txt`). + +In DazedTL, set **Config → Game Update Defaults** (forge, host, org/username, branch) once. Step 1 **Copy gameupdate/** writes those into each game's `patch-config.txt`. You still set `repo=` per game. + +```txt +forge=gitlab +host=gitgud.io +username=YOUR_ORG_OR_USER +repo=YOUR_PATCH_REPO +branch=main +``` + +| Key | Meaning | +|-----|---------| +| `forge` | `gitlab` (default), `github`, or `forgejo` (`gitea` also accepted) | +| `host` | Hostname only. Defaults: `gitgud.io` / `github.com` / `codeberg.org` | +| `username` | Owner / org / group (`owner=` / `org=` aliases work) | +| `repo` | Repository name | +| `branch` | Branch to track | + +Examples: + +```txt +# GitLab / gitgud +forge=gitlab +host=gitgud.io +username=myorg +repo=cool-game-en +branch=main + +# GitHub +forge=github +host=github.com +username=myorg +repo=cool-game-en +branch=main + +# Forgejo / Codeberg +forge=forgejo +host=codeberg.org +username=myorg +repo=cool-game-en +branch=main +``` + +Older configs with only `username` / `repo` / `branch` still work (assumes GitLab on `gitgud.io`). + +# Troubleshooting +**GAMEUPDATE.bat doesn't update and closes immediately** +1. Make sure your path doesn't contain any Japanese characters or lots of whitespace. +2. Make sure you actually have permissions in the folder +3. Auto-update calls the forge's public HTTP API (GitLab `/api/v4`, GitHub `/repos/...`, Forgejo `/api/v1`), not the web “Download ZIP” button - no account is required for public patch repos. + +# Wolf Games +WOLF RPG installs from DLSite ship a master `Data.wolf` archive that takes priority over loose English patch files. `GameUpdate.bat` detects that and unpacks it automatically with the bundled `UberWolfCli.exe` (MIT, [Sinflower/UberWolf](https://github.com/Sinflower/UberWolf)), then renames `Data.wolf` to `Data.wolf.bak` so the patched loose `Data/` files load. + +Just run **`GameUpdate.bat`**. No manual UberWolf download is required. + +If unpack fails (rare Pro / protected builds), unpack once with [UberWolf](https://github.com/Sinflower/UberWolf/releases), ensure a loose `Data/` folder exists, rename or remove `Data.wolf`, delete `gameupdate/previous_patch_sha.txt`, and run `GameUpdate.bat` again. + +# Edit/Contribute +TLDR 3 steps. + + Fork the repository. + Make the changes. + Submit a merge request. + +If everything looks good and doesn't break things I'll merge it in. + +Longer Version: + +# Required Software: +* [VSCode](https://code.visualstudio.com/) Make sure you check all the boxes for context menus. +* [Git](https://git-scm.com/downloads) (Use the default for everything. Just keep clicking Next) + +# Guide to contributing + +### 1. Fork the Repository +- Go to the repository you want to fork. +- Click the "Fork" button. + +### 2. Clone Your Fork +- Clone your forked repository to your local machine. + ```sh + git clone https://gitgud.io/YOUR_USERNAME/REPO_NAME.git + ``` + +### 3. Make Your Changes (In VSCode) +- Edit the files locally on your new branch using VSCode. +- Add and commit your changes. + ```sh + git add . + git commit -m "Description of your changes" + ``` + +### 4. Push Your Changes +- Push your changes to your fork on GitGud.io. + ```sh + git push origin your-feature-branch + ``` + +### 5. Create a Merge Request +- Go to your fork on GitGud.io. +- Click on "Merge Requests" in the sidebar. +- Click the "New merge request" button. +- Select the branch you made changes to and the target project (the original repo). +- Provide a title and description for your merge request and submit it. + +--- + +## Example + +Assuming you want to fork a repository named `example-project`: + +### 1. Fork the Repo +- Navigate to `https://gitgud.io/original_user/example-project` and click "Fork". + +### 2. Clone Your Fork + ```sh + git clone https://gitgud.io/YOUR_USERNAME/example-project.git + ``` + +### 3. Make Changes and Commit + ```sh + # Make changes to the files + git add . + git commit -m "Add new feature to example project" + ``` + +### 4. Push Changes + ```sh + git push origin add-new-feature + ``` + +### 5. Create a Merge Request +- Go to `https://gitgud.io/YOUR_USERNAME/example-project/merge_requests` and click on "New merge request" +- Choose the source branch `add-new-feature` and target branch (default: `main` or `master`) - Fill in the details and submit the merge request \ No newline at end of file diff --git a/gameupdate/gameupdate/patch-config.example.txt b/gameupdate/gameupdate/patch-config.example.txt index 379c363..cacdf5a 100644 --- a/gameupdate/gameupdate/patch-config.example.txt +++ b/gameupdate/gameupdate/patch-config.example.txt @@ -1,7 +1,7 @@ # GameUpdate patch source (copy to patch-config.txt and edit). # Lines starting with # are comments. Keys are case-insensitive. # -# Tip: set forge / host / username / branch once in DazedMTLTool +# Tip: set forge / host / username / branch once in DazedTL # Config → Game Update Defaults. Step 1 "Copy gameupdate/" writes # those into this file automatically (you still set repo= per game). diff --git a/gui/__init__.py b/gui/__init__.py index a77871a..733002f 100644 --- a/gui/__init__.py +++ b/gui/__init__.py @@ -1,23 +1,23 @@ -""" -DazedMTLTool GUI Package -""" - -__version__ = "1.0.0" -__author__ = "DazedTranslations" - -# Package imports -from .main import DazedMTLGUI -from .config_tab import ConfigTab -from .rpgmaker_tab import RPGMakerTab -from .log_viewer import LogViewer -from .file_manager import FileManager -from .workflow_tab import WorkflowTab - -__all__ = [ - "DazedMTLGUI", - "ConfigTab", - "RPGMakerTab", - "LogViewer", - "FileManager", - "WorkflowTab", -] +""" +DazedTL GUI Package +""" + +__version__ = "1.0.0" +__author__ = "DazedTranslations" + +# Package imports +from .main import DazedMTLGUI +from .config_tab import ConfigTab +from .rpgmaker_tab import RPGMakerTab +from .log_viewer import LogViewer +from .file_manager import FileManager +from .workflow_tab import WorkflowTab + +__all__ = [ + "DazedMTLGUI", + "ConfigTab", + "RPGMakerTab", + "LogViewer", + "FileManager", + "WorkflowTab", +] diff --git a/gui/file_manager.py b/gui/file_manager.py index 932886d..2bdf6b4 100644 --- a/gui/file_manager.py +++ b/gui/file_manager.py @@ -1,426 +1,428 @@ -""" -File Manager - Handle input and output files -""" - -from pathlib import Path -from PyQt5.QtWidgets import ( - QWidget, QVBoxLayout, QHBoxLayout, QTreeWidget, QTreeWidgetItem, - QPushButton, QLabel, QFileDialog, QMessageBox, QGroupBox, - QSplitter, QTextEdit, QProgressBar, QCheckBox -) -from PyQt5.QtCore import Qt, QThread, pyqtSignal, QSettings -import json - - -class FileManager(QWidget): - """Widget for managing input and output files.""" - - files_changed = pyqtSignal() - - def __init__(self): - super().__init__() - # Persistent settings for remembering last-open directory - try: - self.settings = QSettings("DazedTranslations", "DazedMTLTool") - except Exception: - self.settings = None - self.input_files = [] - self.output_directory = Path("translated") - self.init_ui() - self.refresh_files() - - def init_ui(self): - """Initialize the user interface.""" - layout = QVBoxLayout() - - # Create splitter for input and output sections - splitter = QSplitter(Qt.Horizontal) - - # Input files section - input_group = self.create_input_section() - splitter.addWidget(input_group) - - # Output files section - output_group = self.create_output_section() - splitter.addWidget(output_group) - - splitter.setSizes([400, 400]) - layout.addWidget(splitter) - - # Progress section - progress_group = self.create_progress_section() - layout.addWidget(progress_group) - - self.setLayout(layout) - - def create_input_section(self): - """Create the input files section.""" - group = QGroupBox("Input Files") - layout = QVBoxLayout() - - # Buttons - button_layout = QHBoxLayout() - - self.add_files_button = QPushButton("Add Files") - self.add_files_button.clicked.connect(self.add_input_files) - - self.add_folder_button = QPushButton("Add Folder") - self.add_folder_button.clicked.connect(self.add_input_folder) - - self.remove_files_button = QPushButton("Remove Selected") - self.remove_files_button.clicked.connect(self.remove_selected_files) - - self.clear_files_button = QPushButton("Clear All") - self.clear_files_button.clicked.connect(self.clear_input_files) - - button_layout.addWidget(self.add_files_button) - button_layout.addWidget(self.add_folder_button) - button_layout.addWidget(self.remove_files_button) - button_layout.addWidget(self.clear_files_button) - - layout.addLayout(button_layout) - - # File tree - self.input_tree = QTreeWidget() - self.input_tree.setHeaderLabels(["File", "Size", "Type"]) - self.input_tree.itemSelectionChanged.connect(self.on_input_selection_changed) - layout.addWidget(self.input_tree) - - # File info - self.input_info = QTextEdit() - self.input_info.setMaximumHeight(100) - self.input_info.setReadOnly(True) - layout.addWidget(self.input_info) - - group.setLayout(layout) - return group - - def create_output_section(self): - """Create the output files section.""" - group = QGroupBox("Output Files") - layout = QVBoxLayout() - - # Output directory selection - dir_layout = QHBoxLayout() - - self.output_dir_label = QLabel(str(self.output_directory)) - self.output_dir_label.setStyleSheet("border: 1px solid #555; padding: 5px;") - - self.browse_output_button = QPushButton("Browse") - self.browse_output_button.clicked.connect(self.browse_output_directory) - - dir_layout.addWidget(QLabel("Output Directory:")) - dir_layout.addWidget(self.output_dir_label) - dir_layout.addWidget(self.browse_output_button) - - layout.addLayout(dir_layout) - - # Output file tree - self.output_tree = QTreeWidget() - self.output_tree.setHeaderLabels(["File", "Size", "Modified"]) - self.output_tree.itemSelectionChanged.connect(self.on_output_selection_changed) - layout.addWidget(self.output_tree) - - # Output file preview - self.output_preview = QTextEdit() - self.output_preview.setMaximumHeight(100) - self.output_preview.setReadOnly(True) - layout.addWidget(self.output_preview) - - # Output buttons - output_button_layout = QHBoxLayout() - - self.open_output_button = QPushButton("Open in Editor") - self.open_output_button.clicked.connect(self.open_output_file) - - self.delete_output_button = QPushButton("Delete Selected") - self.delete_output_button.clicked.connect(self.delete_output_file) - - output_button_layout.addWidget(self.open_output_button) - output_button_layout.addWidget(self.delete_output_button) - output_button_layout.addStretch() - - layout.addLayout(output_button_layout) - - group.setLayout(layout) - return group - - def create_progress_section(self): - """Create the progress monitoring section.""" - group = QGroupBox("Translation Progress") - layout = QVBoxLayout() - - # Progress bar - self.progress_bar = QProgressBar() - self.progress_bar.setVisible(False) - layout.addWidget(self.progress_bar) - - # Status and options - status_layout = QHBoxLayout() - - self.status_label = QLabel("Ready") - self.backup_cb = QCheckBox("Create Backups") - self.backup_cb.setChecked(True) - - status_layout.addWidget(self.status_label) - status_layout.addStretch() - status_layout.addWidget(self.backup_cb) - - layout.addLayout(status_layout) - - group.setLayout(layout) - return group - - def add_input_files(self): - """Add input files via dialog.""" - # Try to restore last used directory - start_dir = "" - try: - if self.settings: - start_dir = self.settings.value("last_open_dir", "") or "" - except Exception: - start_dir = "" - - files, _ = QFileDialog.getOpenFileNames( - self, - "Select Input Files", - start_dir, - "JSON Files (*.json);;All Files (*)" - ) - - for file_path in files: - if file_path not in self.input_files: - self.input_files.append(file_path) - - # Persist directory used - try: - if self.settings and len(files) > 0: - import os - self.settings.setValue("last_open_dir", os.path.dirname(files[0])) - except Exception: - pass - - self.refresh_input_files() - - def add_input_folder(self): - """Add all files from a folder.""" - folder = QFileDialog.getExistingDirectory(self, "Select Input Folder") - - if folder: - folder_path = Path(folder) - for file_path in folder_path.rglob("*.json"): - file_str = str(file_path) - if file_str not in self.input_files: - self.input_files.append(file_str) - - self.refresh_input_files() - - def remove_selected_files(self): - """Remove selected input files.""" - selected_items = self.input_tree.selectedItems() - for item in selected_items: - file_path = item.text(0) - if file_path in self.input_files: - self.input_files.remove(file_path) - - self.refresh_input_files() - - def clear_input_files(self): - """Clear all input files.""" - self.input_files.clear() - self.refresh_input_files() - - def refresh_input_files(self): - """Refresh the input files display.""" - self.input_tree.clear() - - for file_path in self.input_files: - path_obj = Path(file_path) - if path_obj.exists(): - size = path_obj.stat().st_size - file_type = path_obj.suffix - - item = QTreeWidgetItem([ - path_obj.name, - f"{size:,} bytes", - file_type - ]) - item.setToolTip(0, str(path_obj)) - self.input_tree.addTopLevelItem(item) - else: - # File doesn't exist - show in red - item = QTreeWidgetItem([ - f"{path_obj.name} (Missing)", - "0 bytes", - "N/A" - ]) - item.setBackground(0, Qt.red) - self.input_tree.addTopLevelItem(item) - - # Update info - self.update_input_info() - - def refresh_output_files(self): - """Refresh the output files display.""" - self.output_tree.clear() - - if self.output_directory.exists(): - for file_path in self.output_directory.iterdir(): - if file_path.is_file(): - size = file_path.stat().st_size - modified = file_path.stat().st_mtime - - import datetime - modified_str = datetime.datetime.fromtimestamp(modified).strftime("%Y-%m-%d %H:%M") - - item = QTreeWidgetItem([ - file_path.name, - f"{size:,} bytes", - modified_str - ]) - item.setToolTip(0, str(file_path)) - self.output_tree.addTopLevelItem(item) - - def browse_output_directory(self): - """Browse for output directory.""" - directory = QFileDialog.getExistingDirectory( - self, "Select Output Directory", str(self.output_directory) - ) - - if directory: - self.output_directory = Path(directory) - self.output_dir_label.setText(str(self.output_directory)) - self.refresh_output_files() - - def refresh_files(self): - """Refresh both input and output file displays.""" - self.refresh_input_files() - self.refresh_output_files() - - def on_input_selection_changed(self): - """Handle input file selection change.""" - selected_items = self.input_tree.selectedItems() - if selected_items: - item = selected_items[0] - file_name = item.text(0) - - # Find the full path - full_path = None - for file_path in self.input_files: - if Path(file_path).name == file_name: - full_path = file_path - break - - if full_path and Path(full_path).exists(): - try: - with open(full_path, 'r', encoding='utf-8') as f: - content = f.read() - - # Show preview of file structure - if full_path.endswith('.json'): - data = json.loads(content) - if isinstance(data, dict): - keys = list(data.keys())[:10] # Show first 10 keys - preview = f"JSON Object with {len(data)} keys:\n" - preview += "\n".join(keys) - if len(data) > 10: - preview += f"\n... and {len(data) - 10} more" - elif isinstance(data, list): - preview = f"JSON Array with {len(data)} items" - else: - preview = f"JSON: {type(data).__name__}" - else: - preview = content[:500] + "..." if len(content) > 500 else content - - self.input_info.setPlainText(preview) - except Exception as e: - self.input_info.setPlainText(f"Error reading file: {str(e)}") - - def on_output_selection_changed(self): - """Handle output file selection change.""" - selected_items = self.output_tree.selectedItems() - if selected_items: - item = selected_items[0] - file_name = item.text(0) - file_path = self.output_directory / file_name - - if file_path.exists(): - try: - with open(file_path, 'r', encoding='utf-8') as f: - content = f.read() - - preview = content[:500] + "..." if len(content) > 500 else content - self.output_preview.setPlainText(preview) - except Exception as e: - self.output_preview.setPlainText(f"Error reading file: {str(e)}") - - def open_output_file(self): - """Open selected output file in external editor.""" - selected_items = self.output_tree.selectedItems() - if selected_items: - item = selected_items[0] - file_name = item.text(0) - file_path = self.output_directory / file_name - - if file_path.exists(): - import subprocess - try: - subprocess.run(['notepad.exe', str(file_path)], check=True) - except subprocess.CalledProcessError: - QMessageBox.warning(self, "Warning", "Could not open file in external editor") - - def delete_output_file(self): - """Delete selected output file.""" - selected_items = self.output_tree.selectedItems() - if selected_items: - item = selected_items[0] - file_name = item.text(0) - file_path = self.output_directory / file_name - - reply = QMessageBox.question( - self, - "Confirm Deletion", - f"Are you sure you want to delete '{file_name}'?", - QMessageBox.Yes | QMessageBox.No, - QMessageBox.No - ) - - if reply == QMessageBox.Yes and file_path.exists(): - try: - file_path.unlink() - self.refresh_output_files() - except Exception as e: - QMessageBox.critical(self, "Error", f"Failed to delete file:\n{str(e)}") - - def update_input_info(self): - """Update the input files information.""" - total_files = len(self.input_files) - total_size = 0 - - for file_path in self.input_files: - path_obj = Path(file_path) - if path_obj.exists(): - total_size += path_obj.stat().st_size - - info = f"Total Files: {total_files}\nTotal Size: {total_size:,} bytes" - - # Count by file type - extensions = {} - for file_path in self.input_files: - ext = Path(file_path).suffix - extensions[ext] = extensions.get(ext, 0) + 1 - - if extensions: - info += "\n\nFile Types:" - for ext, count in sorted(extensions.items()): - info += f"\n{ext or 'No extension'}: {count}" - - self.input_info.setPlainText(info) - - def get_input_files(self): - """Get list of input files.""" - return self.input_files.copy() - - def get_output_directory(self): - """Get output directory path.""" - return self.output_directory +""" +File Manager - Handle input and output files +""" + +from pathlib import Path +from PyQt5.QtWidgets import ( + QWidget, QVBoxLayout, QHBoxLayout, QTreeWidget, QTreeWidgetItem, + QPushButton, QLabel, QFileDialog, QMessageBox, QGroupBox, + QSplitter, QTextEdit, QProgressBar, QCheckBox +) +from PyQt5.QtCore import Qt, QThread, pyqtSignal, QSettings +import json + +from util.paths import APP_NAME, ORG_NAME + + +class FileManager(QWidget): + """Widget for managing input and output files.""" + + files_changed = pyqtSignal() + + def __init__(self): + super().__init__() + # Persistent settings for remembering last-open directory + try: + self.settings = QSettings(ORG_NAME, APP_NAME) + except Exception: + self.settings = None + self.input_files = [] + self.output_directory = Path("translated") + self.init_ui() + self.refresh_files() + + def init_ui(self): + """Initialize the user interface.""" + layout = QVBoxLayout() + + # Create splitter for input and output sections + splitter = QSplitter(Qt.Horizontal) + + # Input files section + input_group = self.create_input_section() + splitter.addWidget(input_group) + + # Output files section + output_group = self.create_output_section() + splitter.addWidget(output_group) + + splitter.setSizes([400, 400]) + layout.addWidget(splitter) + + # Progress section + progress_group = self.create_progress_section() + layout.addWidget(progress_group) + + self.setLayout(layout) + + def create_input_section(self): + """Create the input files section.""" + group = QGroupBox("Input Files") + layout = QVBoxLayout() + + # Buttons + button_layout = QHBoxLayout() + + self.add_files_button = QPushButton("Add Files") + self.add_files_button.clicked.connect(self.add_input_files) + + self.add_folder_button = QPushButton("Add Folder") + self.add_folder_button.clicked.connect(self.add_input_folder) + + self.remove_files_button = QPushButton("Remove Selected") + self.remove_files_button.clicked.connect(self.remove_selected_files) + + self.clear_files_button = QPushButton("Clear All") + self.clear_files_button.clicked.connect(self.clear_input_files) + + button_layout.addWidget(self.add_files_button) + button_layout.addWidget(self.add_folder_button) + button_layout.addWidget(self.remove_files_button) + button_layout.addWidget(self.clear_files_button) + + layout.addLayout(button_layout) + + # File tree + self.input_tree = QTreeWidget() + self.input_tree.setHeaderLabels(["File", "Size", "Type"]) + self.input_tree.itemSelectionChanged.connect(self.on_input_selection_changed) + layout.addWidget(self.input_tree) + + # File info + self.input_info = QTextEdit() + self.input_info.setMaximumHeight(100) + self.input_info.setReadOnly(True) + layout.addWidget(self.input_info) + + group.setLayout(layout) + return group + + def create_output_section(self): + """Create the output files section.""" + group = QGroupBox("Output Files") + layout = QVBoxLayout() + + # Output directory selection + dir_layout = QHBoxLayout() + + self.output_dir_label = QLabel(str(self.output_directory)) + self.output_dir_label.setStyleSheet("border: 1px solid #555; padding: 5px;") + + self.browse_output_button = QPushButton("Browse") + self.browse_output_button.clicked.connect(self.browse_output_directory) + + dir_layout.addWidget(QLabel("Output Directory:")) + dir_layout.addWidget(self.output_dir_label) + dir_layout.addWidget(self.browse_output_button) + + layout.addLayout(dir_layout) + + # Output file tree + self.output_tree = QTreeWidget() + self.output_tree.setHeaderLabels(["File", "Size", "Modified"]) + self.output_tree.itemSelectionChanged.connect(self.on_output_selection_changed) + layout.addWidget(self.output_tree) + + # Output file preview + self.output_preview = QTextEdit() + self.output_preview.setMaximumHeight(100) + self.output_preview.setReadOnly(True) + layout.addWidget(self.output_preview) + + # Output buttons + output_button_layout = QHBoxLayout() + + self.open_output_button = QPushButton("Open in Editor") + self.open_output_button.clicked.connect(self.open_output_file) + + self.delete_output_button = QPushButton("Delete Selected") + self.delete_output_button.clicked.connect(self.delete_output_file) + + output_button_layout.addWidget(self.open_output_button) + output_button_layout.addWidget(self.delete_output_button) + output_button_layout.addStretch() + + layout.addLayout(output_button_layout) + + group.setLayout(layout) + return group + + def create_progress_section(self): + """Create the progress monitoring section.""" + group = QGroupBox("Translation Progress") + layout = QVBoxLayout() + + # Progress bar + self.progress_bar = QProgressBar() + self.progress_bar.setVisible(False) + layout.addWidget(self.progress_bar) + + # Status and options + status_layout = QHBoxLayout() + + self.status_label = QLabel("Ready") + self.backup_cb = QCheckBox("Create Backups") + self.backup_cb.setChecked(True) + + status_layout.addWidget(self.status_label) + status_layout.addStretch() + status_layout.addWidget(self.backup_cb) + + layout.addLayout(status_layout) + + group.setLayout(layout) + return group + + def add_input_files(self): + """Add input files via dialog.""" + # Try to restore last used directory + start_dir = "" + try: + if self.settings: + start_dir = self.settings.value("last_open_dir", "") or "" + except Exception: + start_dir = "" + + files, _ = QFileDialog.getOpenFileNames( + self, + "Select Input Files", + start_dir, + "JSON Files (*.json);;All Files (*)" + ) + + for file_path in files: + if file_path not in self.input_files: + self.input_files.append(file_path) + + # Persist directory used + try: + if self.settings and len(files) > 0: + import os + self.settings.setValue("last_open_dir", os.path.dirname(files[0])) + except Exception: + pass + + self.refresh_input_files() + + def add_input_folder(self): + """Add all files from a folder.""" + folder = QFileDialog.getExistingDirectory(self, "Select Input Folder") + + if folder: + folder_path = Path(folder) + for file_path in folder_path.rglob("*.json"): + file_str = str(file_path) + if file_str not in self.input_files: + self.input_files.append(file_str) + + self.refresh_input_files() + + def remove_selected_files(self): + """Remove selected input files.""" + selected_items = self.input_tree.selectedItems() + for item in selected_items: + file_path = item.text(0) + if file_path in self.input_files: + self.input_files.remove(file_path) + + self.refresh_input_files() + + def clear_input_files(self): + """Clear all input files.""" + self.input_files.clear() + self.refresh_input_files() + + def refresh_input_files(self): + """Refresh the input files display.""" + self.input_tree.clear() + + for file_path in self.input_files: + path_obj = Path(file_path) + if path_obj.exists(): + size = path_obj.stat().st_size + file_type = path_obj.suffix + + item = QTreeWidgetItem([ + path_obj.name, + f"{size:,} bytes", + file_type + ]) + item.setToolTip(0, str(path_obj)) + self.input_tree.addTopLevelItem(item) + else: + # File doesn't exist - show in red + item = QTreeWidgetItem([ + f"{path_obj.name} (Missing)", + "0 bytes", + "N/A" + ]) + item.setBackground(0, Qt.red) + self.input_tree.addTopLevelItem(item) + + # Update info + self.update_input_info() + + def refresh_output_files(self): + """Refresh the output files display.""" + self.output_tree.clear() + + if self.output_directory.exists(): + for file_path in self.output_directory.iterdir(): + if file_path.is_file(): + size = file_path.stat().st_size + modified = file_path.stat().st_mtime + + import datetime + modified_str = datetime.datetime.fromtimestamp(modified).strftime("%Y-%m-%d %H:%M") + + item = QTreeWidgetItem([ + file_path.name, + f"{size:,} bytes", + modified_str + ]) + item.setToolTip(0, str(file_path)) + self.output_tree.addTopLevelItem(item) + + def browse_output_directory(self): + """Browse for output directory.""" + directory = QFileDialog.getExistingDirectory( + self, "Select Output Directory", str(self.output_directory) + ) + + if directory: + self.output_directory = Path(directory) + self.output_dir_label.setText(str(self.output_directory)) + self.refresh_output_files() + + def refresh_files(self): + """Refresh both input and output file displays.""" + self.refresh_input_files() + self.refresh_output_files() + + def on_input_selection_changed(self): + """Handle input file selection change.""" + selected_items = self.input_tree.selectedItems() + if selected_items: + item = selected_items[0] + file_name = item.text(0) + + # Find the full path + full_path = None + for file_path in self.input_files: + if Path(file_path).name == file_name: + full_path = file_path + break + + if full_path and Path(full_path).exists(): + try: + with open(full_path, 'r', encoding='utf-8') as f: + content = f.read() + + # Show preview of file structure + if full_path.endswith('.json'): + data = json.loads(content) + if isinstance(data, dict): + keys = list(data.keys())[:10] # Show first 10 keys + preview = f"JSON Object with {len(data)} keys:\n" + preview += "\n".join(keys) + if len(data) > 10: + preview += f"\n... and {len(data) - 10} more" + elif isinstance(data, list): + preview = f"JSON Array with {len(data)} items" + else: + preview = f"JSON: {type(data).__name__}" + else: + preview = content[:500] + "..." if len(content) > 500 else content + + self.input_info.setPlainText(preview) + except Exception as e: + self.input_info.setPlainText(f"Error reading file: {str(e)}") + + def on_output_selection_changed(self): + """Handle output file selection change.""" + selected_items = self.output_tree.selectedItems() + if selected_items: + item = selected_items[0] + file_name = item.text(0) + file_path = self.output_directory / file_name + + if file_path.exists(): + try: + with open(file_path, 'r', encoding='utf-8') as f: + content = f.read() + + preview = content[:500] + "..." if len(content) > 500 else content + self.output_preview.setPlainText(preview) + except Exception as e: + self.output_preview.setPlainText(f"Error reading file: {str(e)}") + + def open_output_file(self): + """Open selected output file in external editor.""" + selected_items = self.output_tree.selectedItems() + if selected_items: + item = selected_items[0] + file_name = item.text(0) + file_path = self.output_directory / file_name + + if file_path.exists(): + import subprocess + try: + subprocess.run(['notepad.exe', str(file_path)], check=True) + except subprocess.CalledProcessError: + QMessageBox.warning(self, "Warning", "Could not open file in external editor") + + def delete_output_file(self): + """Delete selected output file.""" + selected_items = self.output_tree.selectedItems() + if selected_items: + item = selected_items[0] + file_name = item.text(0) + file_path = self.output_directory / file_name + + reply = QMessageBox.question( + self, + "Confirm Deletion", + f"Are you sure you want to delete '{file_name}'?", + QMessageBox.Yes | QMessageBox.No, + QMessageBox.No + ) + + if reply == QMessageBox.Yes and file_path.exists(): + try: + file_path.unlink() + self.refresh_output_files() + except Exception as e: + QMessageBox.critical(self, "Error", f"Failed to delete file:\n{str(e)}") + + def update_input_info(self): + """Update the input files information.""" + total_files = len(self.input_files) + total_size = 0 + + for file_path in self.input_files: + path_obj = Path(file_path) + if path_obj.exists(): + total_size += path_obj.stat().st_size + + info = f"Total Files: {total_files}\nTotal Size: {total_size:,} bytes" + + # Count by file type + extensions = {} + for file_path in self.input_files: + ext = Path(file_path).suffix + extensions[ext] = extensions.get(ext, 0) + 1 + + if extensions: + info += "\n\nFile Types:" + for ext, count in sorted(extensions.items()): + info += f"\n{ext or 'No extension'}: {count}" + + self.input_info.setPlainText(info) + + def get_input_files(self): + """Get list of input files.""" + return self.input_files.copy() + + def get_output_directory(self): + """Get output directory path.""" + return self.output_directory diff --git a/gui/main.py b/gui/main.py index 6723bad..ab43ec8 100644 --- a/gui/main.py +++ b/gui/main.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 """ -DazedMTLTool GUI - Main Application -A PyQt-based graphical user interface for the DazedMTLTool translation system. +DazedTL GUI - Main Application +A PyQt-based graphical user interface for the DazedTL translation system. """ import re @@ -29,7 +29,14 @@ from PyQt5.QtCore import Qt, QThread, pyqtSignal, QTimer, QSettings, QCoreApplic from PyQt5.QtGui import QIcon, QFont, QPixmap, QScreen, QGuiApplication -from util.paths import PROJECT_ROOT, ICON_PATH, LAST_UPDATE_SHA_PATH +from util.paths import ( + APP_NAME, + ICON_PATH, + LAST_UPDATE_SHA_PATH, + ORG_NAME, + PROJECT_ROOT, + migrate_app_settings, +) def load_application_icon() -> QIcon: @@ -56,7 +63,7 @@ def check_tool_update() -> str | None: class BackgroundUpdateCheckThread(QThread): - """Checks for DazedMTLTool updates without blocking the UI.""" + """Checks for DazedTL updates without blocking the UI.""" finished = pyqtSignal(object) # str | None — pending tool SHA when available @@ -151,7 +158,7 @@ class UpdateThread(QThread): @classmethod def fetch_latest_sha(cls) -> str: req = urllib.request.Request( - cls.branch_api_url(), headers={"User-Agent": "DazedMTLTool"} + cls.branch_api_url(), headers={"User-Agent": APP_NAME} ) with urllib.request.urlopen(req, timeout=15) as resp: return json.loads(resp.read())["commit"]["id"] @@ -167,7 +174,7 @@ class UpdateThread(QThread): def _download_archive(self, zip_path: Path): req = urllib.request.Request( - self.archive_zip_url(), headers={"User-Agent": "DazedMTLTool"} + self.archive_zip_url(), headers={"User-Agent": APP_NAME} ) with urllib.request.urlopen(req, timeout=120) as resp, open(zip_path, "wb") as fh: total = int(resp.headers.get("Content-Length", 0) or 0) @@ -367,7 +374,7 @@ class UpdateDialog(QDialog): title_col = QVBoxLayout() title_col.setSpacing(2) - self.title_label = QLabel("DazedMTLTool Update") + self.title_label = QLabel(f"{APP_NAME} Update") self.title_label.setObjectName("updateTitle") self.subtitle_label = QLabel("Stay current with the latest fixes and features.") self.subtitle_label.setObjectName("updateDetail") @@ -577,7 +584,7 @@ class UpdateDialog(QDialog): self.current_version_label.setText(sha) self.headline_label.setText("Update installed successfully") self.detail_label.setText( - "Restart DazedMTLTool to load the new version." + f"Restart {APP_NAME} to load the new version." ) self.subtitle_label.setText("Update complete.") if isinstance(self.parent(), DazedMTLGUI): @@ -620,7 +627,7 @@ from gui.skills_tab import SkillsTab from gui.batch_tab import BatchTab class DazedMTLGUI(QMainWindow): - """Main GUI window for the DazedMTLTool.""" + """Main GUI window for DazedTL.""" PAGE_GUIDE = 0 PAGE_WORKFLOW = 1 @@ -631,7 +638,7 @@ class DazedMTLGUI(QMainWindow): def __init__(self): super().__init__() - self.settings = QSettings("DazedTranslations", "DazedMTLTool") + self.settings = QSettings(ORG_NAME, APP_NAME) self._pending_tool_sha: str | None = None self._update_check_thread = None self._shutdown_started = False @@ -791,16 +798,16 @@ class DazedMTLGUI(QMainWindow): # Update window title only when non-default scale is active if scale_factor != 1.0: - self.setWindowTitle(f"DazedMTLTool - Visual Translation Interface (Font: {scale_factor:.1f}x)") + self.setWindowTitle(f"{APP_NAME} - Visual Translation Interface (Font: {scale_factor:.1f}x)") else: - self.setWindowTitle("DazedMTLTool - Visual Translation Interface") + self.setWindowTitle(f"{APP_NAME} - Visual Translation Interface") except Exception as e: print(f"Warning: Could not apply font scaling: {e}") def init_ui(self): """Initialize the user interface.""" - self.setWindowTitle("DazedMTLTool - Visual Translation Interface") + self.setWindowTitle(f"{APP_NAME} - Visual Translation Interface") # Get screen geometry and set window size more responsively screen = QApplication.primaryScreen() @@ -1012,7 +1019,7 @@ class DazedMTLGUI(QMainWindow): return container def start_background_update_check(self): - """Check for DazedMTLTool updates after the GUI is visible.""" + """Check for DazedTL updates after the GUI is visible.""" if self._update_check_thread and self._update_check_thread.isRunning(): return self._update_check_thread = BackgroundUpdateCheckThread(self) @@ -1221,15 +1228,14 @@ class DazedMTLGUI(QMainWindow): """Show the about dialog.""" QMessageBox.about( self, - "About DazedMTLTool GUI", - """ -

DazedMTLTool GUI

-

A visual interface for the DazedMTLTool translation system.

-

This tool helps translate visual novels, RPG games, and other text-based content using AI translation services.

+ f"About {APP_NAME}", + f""" +

{APP_NAME}

+

An AI translation tool for visual novels, RPG games, and other text-based content.

Features:

    +
  • Guided Workflow for RPG Maker and WolfDawn
  • Visual configuration management
    • -
  • Module-specific settings
  • Real-time translation monitoring
  • File management and organization
@@ -1269,10 +1275,11 @@ def main(): install_qt_message_filter() ensure_linux_desktop_entry() - QGuiApplication.setDesktopFileName("DazedMTLTool") + QGuiApplication.setDesktopFileName(APP_NAME) - QCoreApplication.setOrganizationName("DazedTranslations") - QCoreApplication.setApplicationName("DazedMTLTool") + QCoreApplication.setOrganizationName(ORG_NAME) + QCoreApplication.setApplicationName(APP_NAME) + migrate_app_settings() app = QApplication(sys.argv) @@ -1283,7 +1290,7 @@ def main(): try: import ctypes ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID( - "DazedTranslations.DazedMTLTool.1" + f"{ORG_NAME}.{APP_NAME}.1" ) except Exception: pass diff --git a/gui/translation_tab.py b/gui/translation_tab.py index b10ac04..26588ed 100644 --- a/gui/translation_tab.py +++ b/gui/translation_tab.py @@ -1,3412 +1,3413 @@ -#!/usr/bin/env python3 -""" -Simple Translation Tab for DazedMTLTool GUI - -Simple file management and translation execution with console log display. -""" - -import os -import datetime -import subprocess -import threading -import sys -import io -import time -from contextlib import redirect_stdout -from pathlib import Path -from concurrent.futures import ThreadPoolExecutor, as_completed, Future -import traceback -import signal -import multiprocessing -import re -from colorama import Fore -from tqdm import tqdm -from dotenv import load_dotenv -from PyQt5.QtWidgets import ( - QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QGroupBox, - QTextEdit, QMessageBox, QListWidget, QListWidgetItem, - QSplitter, QFileDialog, QComboBox, QCheckBox, QProgressBar, QFrame, QFormLayout, QStackedWidget, - QTableWidget, QTableWidgetItem, QHeaderView, QAbstractItemView, -) -from PyQt5.QtWidgets import QSizePolicy -from PyQt5.QtCore import Qt, QTimer, pyqtSignal, QThread, QMutex, QProcess, QEvent, QRect, QSettings, QSize -from PyQt5.QtGui import QFont, QColor, QBrush -from gui.log_viewer import LogViewer -from gui import qt_icons - - -def _strip_ansi(text): - if not isinstance(text, str) or not text: - return text - return re.sub(r"\x1b\[[0-?]*[ -/]*[@-~]", "", text) - - -def create_section_header(title): - """Create a clean section header without boxes.""" - return qt_icons.make_section_header( - title, - "QLabel {" - "font-size: 13px;" - "font-weight: bold;" - "color: #007acc;" - "padding: 8px 0px 5px 0px;" - "background-color: transparent;" - "}", - ) - -def create_horizontal_line(): - """Create a horizontal separator line.""" - line = QFrame() - line.setFrameShape(QFrame.HLine) - line.setFrameShadow(QFrame.Sunken) - line.setStyleSheet("QFrame { color: #555555; margin: 5px 0px; }") - return line - - -BATCH_MODE_LABEL = "Batch Translate" -BATCH_MODE_BENEFIT_NOTE = ( - "Anthropic Batches API — 50% or more cheaper than live translate (Claude only)." -) -BATCH_COLLECT_LIVE_CHARGE_NOTE = ( - "During Pass 1, speaker names and similar short strings are translated at live " - "API rates right away (not batched). Dialogue is queued for the batch and billed " - "only after you confirm the estimate." -) - - -class _ShimLabel: - """Plain stand-in for QLabel used by Files-tab row helpers (no real Qt widget).""" - - def __init__(self, text=""): - self._text = text - self._style = "" - self._visible = False - self._tooltip = "" - - def setText(self, text): - self._text = text or "" - - def text(self): - return self._text - - def setStyleSheet(self, style): - self._style = style or "" - - def setVisible(self, visible): - self._visible = bool(visible) - - def setToolTip(self, tip): - self._tooltip = tip or "" - - def toolTip(self): - return self._tooltip - - -class _ShimCheckBox: - def __init__(self): - self._checked = False - self._enabled = False - self._visible = False - - def setChecked(self, checked): - self._checked = bool(checked) - - def isChecked(self): - return self._checked - - def setEnabled(self, enabled): - self._enabled = bool(enabled) - - def setVisible(self, visible): - self._visible = bool(visible) - - -class _ShimProgressBar: - def __init__(self): - self._value = 0 - self._maximum = 100 - self._visible = False - self._text_visible = True - self._style = "" - - def setValue(self, value): - self._value = int(value) - - def setMaximum(self, maximum): - self._maximum = int(maximum) - - def setVisible(self, visible): - self._visible = bool(visible) - - def setTextVisible(self, visible): - self._text_visible = bool(visible) - - def setStyleSheet(self, style): - self._style = style or "" - - -class _ShimWidget: - def __init__(self): - self._visible = False - self._tooltip = "" - - def setVisible(self, visible): - self._visible = bool(visible) - - def setToolTip(self, tip): - self._tooltip = tip or "" - - -class TranslationWorker(QThread): - """Worker thread for running translations without blocking the UI.""" - - log_signal = pyqtSignal(str) - progress_signal = pyqtSignal(int, int, str) # current_file, total_files, filename - item_progress_signal = pyqtSignal(str, int, int) # filename, current_item, total_items (for tqdm within file) - file_error_signal = pyqtSignal(str, str) # filename, error_message - finished_signal = pyqtSignal(bool, str) - batch_phase_signal = pyqtSignal(str, object) # phase name, optional payload - - def __init__(self, project_root, module_info, estimate_only=False, selected_files=None, - parse_speakers=False, batch_mode=False, batch_resume_state=None): - super().__init__() - self.project_root = project_root - self.module_info = module_info # [name, extensions, handler_function] - self.estimate_only = estimate_only - self.selected_files = selected_files # List of files to process - # Whether we should run in speaker-parse mode (special-case for MV/MZ) - self.parse_speakers = parse_speakers - self.batch_mode = batch_mode - self.batch_resume_state = batch_resume_state - self._batch_submit_event = threading.Event() - self._batch_submit_approved = False - self._batch_pending_estimate = None - self.should_stop = False - self.mutex = QMutex() # For thread safety - self.executor = None # Store reference to executor for proper shutdown - self.running_processes = [] # Track running processes for termination - - def set_batch_submit_response(self, approved): - """Called from the UI thread after the submit-batch confirmation dialog.""" - self._batch_submit_approved = approved - self._batch_submit_event.set() - - def _wait_batch_submit(self, estimate): - self._batch_pending_estimate = estimate - self._batch_submit_event.clear() - self.batch_phase_signal.emit("submit", estimate) - self._batch_submit_event.wait() - return self._batch_submit_approved - - def _emit_batch_phase(self, phase, payload=None): - self.batch_phase_signal.emit(phase, payload) - - def _emit_batch_output(self, fn, *args, **kwargs): - """Run a batch helper that prints status lines; forward them to the log.""" - buf = io.StringIO() - with redirect_stdout(buf): - result = fn(*args, **kwargs) - for line in buf.getvalue().splitlines(): - if line.strip(): - self.emit_log(line) - return result - - def _run_batch_poll_fetch(self): - """Submit (if needed), poll until ended, fetch results. None if stopped while polling.""" - from util.translation import ( - submitTranslationBatches, - fetchTranslationBatches, - _read_batch_file, - BATCH_STATE_FILE, - _batch_file_lock, - ) - - with _batch_file_lock(): - state = _read_batch_file(BATCH_STATE_FILE) - if not state.get("batches"): - est = self._batch_pending_estimate - file_set = list(self.selected_files or []) - if not self._emit_batch_output( - submitTranslationBatches, - file_set=file_set, - cost_estimate=est, - ): - return 0, 0 - - poll = int(os.getenv("batchPollInterval", "60") or 60) - self._emit_batch_phase("polling") - self.emit_log( - f"[BATCH] polling every {poll}s (stop is safe - resume later with Batch Translate mode)..." - ) - from util.translation import checkTranslationBatchStatuses - while True: - if self.should_stop: - self.emit_log("[BATCH] Stopped while polling. Batch keeps processing - resume later.") - return None - buf = io.StringIO() - with redirect_stdout(buf): - ended, statuses = checkTranslationBatchStatuses(print_status=True) - for line in buf.getvalue().splitlines(): - if line.strip(): - self.emit_log(line) - if statuses: - self._emit_batch_phase("poll_status", statuses) - if ended: - break - for _ in range(poll * 10): - if self.should_stop: - self.emit_log("[BATCH] Stopped while polling. Batch keeps processing - resume later.") - return None - time.sleep(0.1) - - fetched, errored = self._emit_batch_output(fetchTranslationBatches) - return fetched, errored - - def stop(self): - """Stop the translation process.""" - self.mutex.lock() - try: - if self.should_stop: - # Already stopping, don't log again - return - - self.should_stop = True - self.emit_log("🛑 Stopping translation worker and canceling pending tasks...") - - # Shutdown the executor if it exists - if self.executor: - # For older Python versions compatibility, use shutdown(wait=False) - # and manually cancel futures - try: - # Try to use cancel_futures parameter (Python 3.9+) - self.executor.shutdown(wait=False, cancel_futures=True) - except TypeError: - # Fallback for older Python versions - self.executor.shutdown(wait=False) - - # Terminate any running processes - if self.running_processes: - self.emit_log("🛑 Terminating running translation processes...") - for process in self.running_processes: - try: - if process.poll() is None: # Process is still running - process.terminate() - # Give it a moment to terminate gracefully - try: - process.wait(timeout=2) - except subprocess.TimeoutExpired: - # Force kill if it doesn't terminate - process.kill() - process.wait() - except Exception as e: - self.emit_log(f"⚠️ Warning: Could not terminate process: {e}") - self.running_processes.clear() - finally: - self.mutex.unlock() - - def emit_log(self, message): - """Thread-safe log emission. - - Also mirrors into TRANSLATION_RUN_LOG so the LogViewer file-tail shows - batch/status lines (those never go through translateAI's Input/Output writer). - """ - self.log_signal.emit(message) - try: - run_log = os.getenv("TRANSLATION_RUN_LOG") - if not run_log or not message: - return - text = _strip_ansi(str(message)).rstrip() - if not text: - return - path = Path(run_log) - path.parent.mkdir(parents=True, exist_ok=True) - with open(path, "a", encoding="utf-8") as f: - f.write(text + "\n") - f.flush() - except Exception: - pass - - def emit_progress(self, current, total, filename): - """Thread-safe progress emission.""" - self.progress_signal.emit(current, total, filename) - - def run_module_in_process(self, filename, estimate_only, batch_phase=None): - """Run a module handler in a separate process for better control.""" - try: - # Use the external subprocess runner script - runner_script = self.project_root / "util" / "subprocess_runner.py" - if not runner_script.exists(): - self.emit_log(f"❌ Subprocess runner script not found: {runner_script}") - return "Fail" - - # Run the script in a separate process - env = os.environ.copy() - env['PYTHONIOENCODING'] = 'utf-8' # Force UTF-8 encoding - if batch_phase in ("collect", "consume"): - env["BATCH_PHASE"] = batch_phase - else: - env.pop("BATCH_PHASE", None) - - process = subprocess.Popen( - [ - sys.executable, - str(runner_script), - str(self.project_root), - self.module_info[0], # module name - filename, - str(estimate_only) - ], - stdout=subprocess.PIPE, - stderr=subprocess.PIPE, - text=True, - encoding='utf-8', - errors='replace', - cwd=str(self.project_root), - env=env, - bufsize=1 # Line buffered - ) - - # Track the process for potential termination - self.running_processes.append(process) - - # Read output in real-time to capture progress - stdout_lines = [] - stderr_lines = [] - - def read_stdout(): - """Read stdout line by line.""" - for line in iter(process.stdout.readline, ''): - if not line: - break - line = line.strip() - if line.startswith('PROGRESS:'): - # Parse progress: PROGRESS:filename:current:total - try: - parts = line.split(':', 3) - if len(parts) == 4: - _, desc, current, total = parts - # Emit with filename included - self.item_progress_signal.emit(desc, int(current), int(total)) - except Exception: - pass # Ignore malformed progress lines - else: - stdout_lines.append(line) - process.stdout.close() - - def read_stderr(): - """Read stderr line by line.""" - for line in iter(process.stderr.readline, ''): - if not line: - break - stderr_lines.append(line.strip()) - process.stderr.close() - - # Start reader threads - stdout_thread = threading.Thread(target=read_stdout, daemon=True) - stderr_thread = threading.Thread(target=read_stderr, daemon=True) - stdout_thread.start() - stderr_thread.start() - - # Wait for process completion - process.wait() - - # Wait for reader threads to finish - stdout_thread.join(timeout=1.0) - stderr_thread.join(timeout=1.0) - - # Combine output - stdout = '\n'.join(stdout_lines) - stderr = '\n'.join(stderr_lines) - - # Remove from tracking - if process in self.running_processes: - self.running_processes.remove(process) - - # Check if process was terminated by stop signal - if self.should_stop: - return "Stopped" - - # Forward all stdout output to log (this includes cost information) - for line in stdout.strip().split('\n'): - if line.strip() and not line.startswith('RESULT:'): - self.emit_log(line) - - # Parse result - if process.returncode == 0: - for line in stdout.strip().split('\n'): - if line.startswith('RESULT:'): - result_text = line[7:] # Remove 'RESULT:' prefix - # Clean up any Unicode issues in the result - try: - return result_text - except UnicodeError: - return result_text.encode('ascii', 'ignore').decode('ascii') - return "Success" - else: - # Extract error message from stderr - error_msg = stderr.strip() if stderr.strip() else "Unknown error" - # Handle potential Unicode errors in error messages - try: - clean_error = error_msg.encode('ascii', 'ignore').decode('ascii') - except: - clean_error = "Unicode encoding error in process output" - - # Check if stderr contains the actual exception message - # Format from subprocess_runner.py: "ERROR:actual error message" - actual_error = clean_error - for line in clean_error.split('\n'): - if line.startswith('ERROR:'): - actual_error = line[6:] # Remove 'ERROR:' prefix - break - # Check for exception lines in traceback - if 'NameError:' in line or 'Error:' in line: - # Extract just the error message part - if ':' in line: - actual_error = line.split(':', 1)[1].strip() - break - - self.emit_log(f"❌ Process error: {actual_error}") - # Return the actual error so it can be used to determine if file is unsupported - return ("SUBPROCESS_ERROR", actual_error) - - except Exception as e: - self.emit_log(f"❌ Failed to run module process: {str(e)}") - return "Fail" - - def _run_files(self, matching_files, estimate_only, batch_phase=None): - """Process matching files; return last cost string or 'Fail'.""" - threads = int(os.getenv("fileThreads", "1")) - total_cost = "Fail" - module_name_lower = self.module_info[0].lower() if isinstance(self.module_info[0], str) else "" - is_mvmz = "mv/mz" in module_name_lower - - if self.parse_speakers and is_mvmz: - try: - from modules.rpgmakermvmz import ( - handleMVMZ as handler, setSpeakerParseMode, finalizeSpeakerParse, - resetSpeakerState, TOKENS, calculateCost, MODEL, - ) - except Exception as e: - self.emit_log(f"❌ Could not import rpgmakermvmz for speaker-parse: {e}") - return "Fail" - - try: - resetSpeakerState() - except Exception: - pass - try: - setSpeakerParseMode(True) - except Exception: - pass - - completed_count = 0 - total_files = len(matching_files) - for filename in matching_files: - if self.should_stop: - break - try: - handler(filename, estimate_only) - completed_count += 1 - self.emit_progress(completed_count, total_files, filename) - except Exception as e: - tb_line = str(traceback.extract_tb(sys.exc_info()[2])[-1].lineno) - self.emit_log(f"❌ Error processing {filename}: {str(e)} | Line: {tb_line}") - self.file_error_signal.emit(filename, str(e)) - completed_count += 1 - self.emit_progress(completed_count, total_files, filename) - - try: - before_in, before_out = (0, 0) - try: - before_in, before_out = (int(TOKENS[0]), int(TOKENS[1])) - except Exception: - pass - finalizeSpeakerParse() - after_in, after_out = (0, 0) - try: - after_in, after_out = (int(TOKENS[0]), int(TOKENS[1])) - except Exception: - pass - delta_in = max(0, after_in - before_in) - delta_out = max(0, after_out - before_out) - if delta_in or delta_out: - try: - cost = calculateCost(delta_in, delta_out, MODEL) - self.emit_log(f"Speakers: [Input: {delta_in}][Output: {delta_out}][Cost: ${cost:.4f}] ✓") - except Exception: - pass - except Exception as e: - self.emit_log(f"❌ Failed to finalize speaker parse: {e}") - - try: - setSpeakerParseMode(False) - except Exception: - pass - return "Success" - - max_workers = 1 if estimate_only else threads - self.executor = ThreadPoolExecutor(max_workers=max_workers) - future_to_filename = { - self.executor.submit( - self.run_module_in_process, filename, estimate_only, batch_phase - ): filename - for filename in matching_files - } - - completed_count = 0 - total_files = len(matching_files) - for future in as_completed(future_to_filename): - if self.should_stop: - for remaining_future in future_to_filename: - if not remaining_future.done(): - remaining_future.cancel() - break - - filename = future_to_filename[future] - # Resolve the future before emitting progress so cost lines from the - # subprocess are queued ahead of the file-complete progress event. - try: - result = future.result() - if isinstance(result, tuple) and len(result) == 2 and result[0] == "SUBPROCESS_ERROR": - self.file_error_signal.emit(filename, result[1]) - elif result and result not in ("Fail", "Stopped"): - total_cost = result - elif result == "Stopped": - completed_count += 1 - self.emit_progress(completed_count, total_files, filename) - break - else: - self.emit_log(f"❌ Failed processing {filename}") - self.file_error_signal.emit(filename, "Translation failed") - except Exception as e: - tb_line = str(traceback.extract_tb(sys.exc_info()[2])[-1].lineno) - self.emit_log(f"❌ Error processing {filename}: {str(e)} | Line: {tb_line}") - self.file_error_signal.emit(filename, str(e)) - - completed_count += 1 - self.emit_progress(completed_count, total_files, filename) - - if self.executor: - try: - self.executor.shutdown(wait=False, cancel_futures=True) - except TypeError: - self.executor.shutdown(wait=False) - self.executor = None - - return total_cost - - def run(self): - """Run the translation process.""" - try: - load_dotenv() - sys.path.insert(0, str(self.project_root)) - - from util.translation import clear_cache - - if not (self.batch_mode and self.batch_resume_state): - clear_cache() - - required_envs = ["api", "key", "model", "language", "timeout", "fileThreads", "threads", "width", "listWidth"] - missing_envs = [ - env for env in required_envs - if os.getenv(env) is None or str(os.getenv(env))[:1] == "<" - ] - if missing_envs: - names = ", ".join(missing_envs) - self.emit_log(f"❌ Missing required environment variable(s): {names}") - self.emit_log(" Check your .env file (see .env.example).") - self.finished_signal.emit(False, f"Missing env: {names}") - return - - if self.batch_mode and self.parse_speakers: - self.emit_log("❌ Batch Translate does not support Parse Speakers mode.") - self.finished_signal.emit(False, "Batch + Parse Speakers unsupported") - return - - files_dir = self.project_root / "files" - if not files_dir.exists(): - self.emit_log("❌ Files directory does not exist!") - self.finished_signal.emit(False, "Files directory missing") - return - - if self.selected_files: - matching_files = self.selected_files - else: - matching_files = [] - for file_path in files_dir.iterdir(): - if file_path.is_file() and file_path.name != '.gitkeep': - for ext in self.module_info[1]: - if file_path.name.endswith(ext): - matching_files.append(file_path.name) - break - - if not matching_files: - self.emit_log(f"❌ No files found matching extensions: {', '.join(self.module_info[1])}") - self.finished_signal.emit(False, "No matching files") - return - - self.emit_log(f"📁 Found {len(matching_files)} files to process:") - for filename in matching_files: - self.emit_log(f" • {filename}") - self.emit_log(f"🔧 Using module: {self.module_info[0]}") - if self.batch_mode: - self.emit_log("📦 Batch mode: Anthropic Batches API (50% off)") - else: - self.emit_log(f"📊 Estimate only: {'Yes' if self.estimate_only else 'No'}") - self.emit_log("") - - total_cost = "Fail" - old_cwd = os.getcwd() - os.chdir(str(self.project_root)) - - if self.estimate_only: - try: - from util.translation import clear_estimate_written_sizes - clear_estimate_written_sizes() - except Exception: - pass - - try: - if self.batch_mode: - from util.translation import ( - clearBatchFiles, - pendingBatchRequests, - estimateBatchCost, - ) - - run_consume = True - if self.batch_resume_state is None: - clearBatchFiles() - self._emit_batch_phase("collect") - self.emit_log("[BATCH] Pass 1/2: collecting requests...") - self.emit_log( - "[BATCH] Note: speaker names and similar short strings translate at " - "live API rates during collect (dialogue is batched after you confirm)." - ) - total_cost = self._run_files(matching_files, False, batch_phase="collect") - if self.should_stop: - self.finished_signal.emit(False, "Translation stopped") - return - - if pendingBatchRequests() == 0: - self.emit_log("[BATCH] No requests queued — nothing needed the API.") - run_consume = False - else: - n_requests = pendingBatchRequests() - self._emit_batch_phase("collect_done", { - "files": len(matching_files), - "requests": n_requests, - }) - est = self._emit_batch_output(estimateBatchCost) - if est is not None: - est = dict(est) - est["files"] = len(matching_files) - if not self._wait_batch_submit(est): - self.emit_log( - "[BATCH] Not submitted. Queue kept in log/batch_requests.json " - "(resume with Batch Translate to submit without re-collecting)." - ) - self.finished_signal.emit(True, "Batch not submitted") - return - poll_result = self._run_batch_poll_fetch() - if poll_result is None: - self.finished_signal.emit(False, "Batch polling stopped") - return - elif self.batch_resume_state == "queued": - # Resume a declined/interrupted collect: estimate + submit only. - self.emit_log( - "[BATCH] Resuming queued requests (skipping re-collect to avoid " - "duplicate live charges and a second batch submission)..." - ) - n_requests = pendingBatchRequests() - if n_requests == 0: - self.emit_log("[BATCH] Queue is empty - nothing to submit.") - run_consume = False - else: - self._emit_batch_phase("collect_done", { - "files": len(matching_files), - "requests": n_requests, - }) - est = self._emit_batch_output(estimateBatchCost) - if est is not None: - est = dict(est) - est["files"] = len(matching_files) - if not self._wait_batch_submit(est): - self.emit_log( - "[BATCH] Not submitted. Queue kept in log/batch_requests.json." - ) - self.finished_signal.emit(True, "Batch not submitted") - return - poll_result = self._run_batch_poll_fetch() - if poll_result is None: - self.finished_signal.emit(False, "Batch polling stopped") - return - elif self.batch_resume_state == "submitted": - self._emit_batch_phase("polling") - self.emit_log("[BATCH] Resuming submitted batch...") - poll_result = self._run_batch_poll_fetch() - if poll_result is None: - self.finished_signal.emit(False, "Batch polling stopped") - return - else: - self.emit_log("[BATCH] Resuming from fetched results...") - - if run_consume and not self.should_stop: - try: - from util.batch_history import missing_result_count - present, expected = missing_result_count() - if expected and present < expected: - self.emit_log( - f"[BATCH] WARNING: only {present}/{expected} results present. " - "Missing keys will fall back to the live API (full price)." - ) - except Exception: - pass - self._emit_batch_phase("consume") - self.emit_log("[BATCH] Pass 2/2: writing translated files...") - total_cost = self._run_files(matching_files, False, batch_phase="consume") - if not self.should_stop: - self._emit_batch_phase("done") - else: - total_cost = self._run_files(matching_files, self.estimate_only) - finally: - os.chdir(old_cwd) - - # Clean up temporary files - tmp_file = self.project_root / "csv.tmp" - if tmp_file.exists(): - tmp_file.unlink() - - # Clean up any remaining temporary scripts - temp_script = self.project_root / "temp_translation_script.py" - if temp_script.exists(): - temp_script.unlink() - - # Ensure all processes are terminated - if self.running_processes: - for process in self.running_processes: - try: - if process.poll() is None: - process.terminate() - process.wait(timeout=1) - except: - pass - self.running_processes.clear() - - # Report results - if total_cost != "Fail" and not self.should_stop: - if self.batch_mode: - try: - from util.translation import clearBatchFiles - clearBatchFiles() - except Exception: - pass - self.emit_log("") - self.emit_log(f"💰 {total_cost}") - if self.batch_mode: - self.emit_log("✅ Batch translation completed!") - elif not self.estimate_only: - self.emit_log("✅ Translation completed successfully!") - else: - self.emit_log("✅ Estimation completed!") - try: - from util.translation import clear_estimate_written_sizes - clear_estimate_written_sizes() - except Exception: - pass - self.finished_signal.emit(True, str(total_cost)) - else: - if not self.should_stop: - self.emit_log("❌ Translation failed!") - self.finished_signal.emit(False, "Translation failed") - else: - # Only log the final stop message here - self.emit_log("🛑 Translation stopped by user") - self.finished_signal.emit(False, "Translation stopped") - - except Exception as e: - error_msg = f"❌ Unexpected error: {str(e)}" - self.emit_log(error_msg) - self.finished_signal.emit(False, error_msg) - - -class TranslationTab(QWidget): - """Simple translation tab with file management and console log. - - Emits engine_changed(str) when the selected module implies a different - engine configuration tab should be displayed. - """ - engine_changed = pyqtSignal(str) - - def __init__(self, parent=None): - super().__init__(parent) - self.parent_window = parent - # Persistent settings (remember last directory used in file dialogs) - try: - self.settings = QSettings("DazedTranslations", "DazedMTLTool") - except Exception: - self.settings = None - # If the worker signals finished before all file progress updates - # have been received, we queue the finalization until the last - # file progress update arrives. - self._finish_pending = None - self.translation_process = None - self.log_buffer = [] # Buffer for batching log messages - self.log_timer = QTimer() # Timer for flushing log buffer - self.log_timer.timeout.connect(self.flush_log_buffer) - - # Set up directories - self.project_root = Path(__file__).parent.parent - self.files_dir = self.project_root / "files" - self.translated_dir = self.project_root / "translated" - - # Ensure directories exist - self.files_dir.mkdir(exist_ok=True) - self.translated_dir.mkdir(exist_ok=True) - - # Initialize tracking variables - self.files_completed = 0 - self.files_total = 0 - self.file_progress_items = {} # filename -> {widget, label, progress_bar, checkbox} - self.current_translating_file = None - # Totals tracking - self.totals_input_tokens = 0 - self.totals_output_tokens = 0 - self.totals_cost = 0.0 - self.totals_time = 0.0 - # Track which filenames' totals have been applied (prevents double-counting) - self._applied_file_totals = set() - # Filenames from the most recently completed translation run (used by post-run export) - self._last_run_files: list = [] - # Totals widget reference - self.totals_widget = None - self._batch_active = False - self._batch_ui_phase = None - self._batch_consume_started = False - self._batch_tab_index = -1 - - self.setup_ui() - self.setup_module_list() - self.refresh_file_lists() - - def setup_ui(self): - """Set up the user interface.""" - # Create a fixed horizontal layout to separate translation controls from log viewer. - # Using a layout instead of QSplitter prevents the user from resizing panes. - main_container = QWidget() - main_hbox = QHBoxLayout() - # Match left side padding so headers align at the top of the boxes - main_hbox.setContentsMargins(15, 15, 15, 15) - main_hbox.setSpacing(8) - # Align child widgets individually when needed; avoid setting a - # global AlignTop on the HBox so children with Expanding size - # policies can grow vertically to fill available space. - - # Left side - translation controls - left_widget = QWidget() - layout = QVBoxLayout() - layout.setSpacing(8) - # Remove the top internal margin so the left header lines up with the - # right header (main_hbox already provides top padding). - layout.setContentsMargins(15, 0, 15, 15) - - # Files Section (at the top) - layout.addWidget(create_section_header("📁 Input Files")) - - # Create stacked widget to switch between file list and progress view - self.file_stack = QStackedWidget() - - # Page 0: Normal file list with buttons - file_list_page = QWidget() - file_list_layout = QVBoxLayout() - file_list_layout.setContentsMargins(0, 0, 0, 0) - - # Files Section with side buttons - files_container = QHBoxLayout() - files_container.setSpacing(5) # Add spacing between list and buttons - - # File list with checkboxes - self.file_list = QListWidget() - # Allow the file list to expand vertically to fill available space - # (remove fixed minimum height so it can stretch). - self.file_list.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) - # No max height - let it expand - self.file_list.setSelectionMode(QListWidget.NoSelection) # Disable selection highlighting - # Use an event filter installed on the viewport so we reliably - # intercept mouse events that occur over the item rows and - # checkbox indicator. Installing on the viewport is more - # reliable cross-platform than installing on the list itself. - self.file_list.viewport().installEventFilter(self) - self._file_list_filter_installed = True - self.file_list.setFocusPolicy(Qt.NoFocus) # Remove focus outline - self.file_list.setStyleSheet(""" - QListWidget { - outline: none; - border: 1px solid #555555; - } - QListWidget::item { - border: none; - outline: none; - } - QListWidget::item:hover { - background-color: #3e3e42; - } - """) - # Place the two main file buttons on the left and totals on the right - # File management buttons (icon-based, vertical on the side) - file_buttons = QVBoxLayout() - file_buttons.setSpacing(0) - file_buttons.setContentsMargins(0, 0, 0, 0) - - # Button style for all icon buttons - all same size - icon_button_style = """ - QPushButton { - font-size: 13px; - padding: 0px; - min-width: 32px; - max-width: 32px; - min-height: 32px; - max-height: 32px; - border: 1px solid #555555; - border-top: none; - border-radius: 0px; - background-color: #2d2d30; - } - QPushButton:hover { - background-color: #3e3e42; - border-left-color: #007acc; - } - QPushButton:pressed { - background-color: #007acc; - } - """ - - # First button style - same size but with top border - first_button_style = """ - QPushButton { - font-size: 13px; - padding: 0px; - min-width: 32px; - max-width: 32px; - min-height: 32px; - max-height: 32px; - border: 1px solid #555555; - border-radius: 0px; - background-color: #2d2d30; - } - QPushButton:hover { - background-color: #3e3e42; - border-left-color: #007acc; - } - QPushButton:pressed { - background-color: #007acc; - } - """ - - _icon_size = QSize(18, 18) - - def _icon_button(glyph, tooltip, slot, style): - btn = QPushButton() - qt_icons.apply_button_icon(btn, glyph, color="#dddddd") - btn.setIconSize(_icon_size) - btn.setToolTip(tooltip) - btn.clicked.connect(slot) - btn.setStyleSheet(style) - file_buttons.addWidget(btn) - return btn - - select_all_btn = _icon_button("✓", "Select all files", self.select_all_files, first_button_style) - deselect_all_btn = _icon_button("✗", "Deselect all files", self.deselect_all_files, icon_button_style) - add_files_btn = _icon_button("➕", "Add files to translate", self.add_input_files, icon_button_style) - remove_files_btn = _icon_button("🗑️", "Remove selected files", self.remove_selected_files, icon_button_style) - open_folder_btn = _icon_button("📁", "Open files folder in explorer", self.open_input_folder, icon_button_style) - refresh_btn = _icon_button("🔄", "Refresh file list", self.refresh_file_lists, icon_button_style) - self.sidebar_export_btn = _icon_button( - "📤", - "Export selected files → Game Folder\nCopy translated files for the checked items into your game's data directory", - self._export_selected_files, - icon_button_style, - ) - pricing_test_btn = _icon_button("💰", "Check live pricing for the current model", self._check_model_pricing, icon_button_style) - - # Add stretch to push buttons to top - file_buttons.addStretch() - - # Add button column to the container on the LEFT - files_container.addLayout(file_buttons) - # Then add the file list (center) - files_container.addWidget(self.file_list) - - # (Totals footer will be created below and shown only when translation starts) - - # Add the container to file list page and allow it to expand so - # the file list can grow and push settings to the bottom. - file_list_layout.addLayout(files_container, 1) - # (Totals footer removed here; totals will be shown next to the - # back/open buttons in the progress view as requested.) - file_list_page.setLayout(file_list_layout) - self.file_stack.addWidget(file_list_page) # Index 0 - - # Page 1: Progress view (shown during translation) - progress_view_page = QWidget() - progress_view_layout = QVBoxLayout() - progress_view_layout.setContentsMargins(0, 0, 0, 0) - progress_view_layout.setSpacing(8) - - # Batch pipeline panel (shown only during batch runs) - self.batch_pipeline_widget = QWidget() - self.batch_pipeline_widget.setVisible(False) - batch_pipe_layout = QVBoxLayout() - batch_pipe_layout.setContentsMargins(12, 12, 12, 12) - batch_pipe_layout.setSpacing(10) - - self.batch_phase_title = QLabel("Batch Translate") - self.batch_phase_title.setStyleSheet("color:#4ec9b0;font-weight:bold;font-size:13px;") - batch_pipe_layout.addWidget(self.batch_phase_title) - - # Step strip: Collect → Submit → Process → Write - self.batch_steps_row = QHBoxLayout() - self.batch_steps_row.setSpacing(6) - self._batch_step_labels = [] - for i, name in enumerate(("1. Collect", "2. Submit", "3. Process", "4. Write")): - lab = QLabel(name) - lab.setAlignment(Qt.AlignCenter) - lab.setStyleSheet(self._batch_step_style("idle")) - self.batch_steps_row.addWidget(lab, 1) - self._batch_step_labels.append(lab) - if i < 3: - arrow = QLabel("→") - arrow.setStyleSheet("color:#666666;font-size:12px;") - arrow.setAlignment(Qt.AlignCenter) - self.batch_steps_row.addWidget(arrow, 0) - batch_pipe_layout.addLayout(self.batch_steps_row) - - self.batch_overall_bar = QProgressBar() - self.batch_overall_bar.setRange(0, 100) - self.batch_overall_bar.setValue(0) - self.batch_overall_bar.setFixedHeight(20) - self.batch_overall_bar.setTextVisible(True) - self.batch_overall_bar.setFormat("%p%") - self.batch_overall_bar.setStyleSheet(""" - QProgressBar { - border: none; - border-radius: 3px; - text-align: center; - background-color: #2b2b2b; - color: #cccccc; - } - QProgressBar::chunk { - background-color: #007acc; - border-radius: 3px; - } - """) - batch_pipe_layout.addWidget(self.batch_overall_bar) - - self.batch_pipeline_stack = QStackedWidget() - - collect_page = QWidget() - collect_layout = QVBoxLayout(collect_page) - collect_layout.setContentsMargins(0, 0, 0, 0) - collect_layout.setSpacing(8) - self.batch_collect_warning = QLabel(BATCH_COLLECT_LIVE_CHARGE_NOTE) - self.batch_collect_warning.setWordWrap(True) - self.batch_collect_warning.setStyleSheet("color:#f0ad4e;font-size:11px;") - collect_layout.addWidget(self.batch_collect_warning) - self.batch_collect_status = QLabel("Pass 1/2: collecting API requests from selected files…") - self.batch_collect_status.setWordWrap(True) - self.batch_collect_status.setStyleSheet("color:#cccccc;font-size:12px;") - collect_layout.addWidget(self.batch_collect_status) - self.batch_pipeline_stack.addWidget(collect_page) - - submit_page = QWidget() - submit_layout = QVBoxLayout(submit_page) - submit_layout.setContentsMargins(0, 0, 0, 0) - submit_layout.setSpacing(8) - self.batch_submit_summary = QLabel("") - self.batch_submit_summary.setWordWrap(True) - self.batch_submit_summary.setAlignment(Qt.AlignTop) - self.batch_submit_summary.setStyleSheet("color:#cccccc;font-size:12px;") - submit_layout.addWidget(self.batch_submit_summary) - # Cost chips row - self.batch_cost_row = QHBoxLayout() - self.batch_cost_row.setSpacing(8) - self.batch_cost_cached = QLabel("Batch+cache: —") - self.batch_cost_nocache = QLabel("Batch worst: —") - self.batch_cost_live = QLabel("Live: —") - for lab in (self.batch_cost_cached, self.batch_cost_nocache, self.batch_cost_live): - lab.setStyleSheet( - "color:#cccccc;background:#1e1e1e;border:1px solid #3e3e42;" - "border-radius:4px;padding:8px;font-size:12px;" - ) - lab.setAlignment(Qt.AlignCenter) - self.batch_cost_row.addWidget(lab, 1) - submit_layout.addLayout(self.batch_cost_row) - submit_btn_row = QHBoxLayout() - submit_btn_row.addStretch() - self.batch_submit_yes_btn = QPushButton("Submit Batch") - self.batch_submit_yes_btn.setStyleSheet( - "QPushButton{background-color:#007acc;color:white;font-weight:bold;padding:6px 16px;border-radius:4px;}" - "QPushButton:hover{background-color:#106ebe;}" - ) - self.batch_submit_yes_btn.clicked.connect(self._on_batch_submit_yes) - self.batch_submit_no_btn = QPushButton("Cancel") - self.batch_submit_no_btn.setStyleSheet( - "QPushButton{background-color:#555;color:white;padding:6px 16px;border-radius:4px;}" - "QPushButton:hover{background-color:#666;}" - ) - self.batch_submit_no_btn.clicked.connect(self._on_batch_submit_no) - submit_btn_row.addWidget(self.batch_submit_no_btn) - submit_btn_row.addWidget(self.batch_submit_yes_btn) - submit_layout.addLayout(submit_btn_row) - submit_layout.addStretch(1) - self.batch_pipeline_stack.addWidget(submit_page) - - poll_page = QWidget() - poll_layout = QVBoxLayout(poll_page) - poll_layout.setContentsMargins(0, 0, 0, 0) - poll_layout.setSpacing(8) - self.batch_poll_id = QLabel("Batch: —") - self.batch_poll_id.setStyleSheet("color:#9cdcfe;font-size:12px;font-family:monospace;") - self.batch_poll_id.setTextInteractionFlags(Qt.TextSelectableByMouse) - poll_layout.addWidget(self.batch_poll_id) - self.batch_poll_status = QLabel("Waiting for Anthropic to finish processing the batch…") - self.batch_poll_status.setWordWrap(True) - self.batch_poll_status.setStyleSheet("color:#cccccc;font-size:12px;") - poll_layout.addWidget(self.batch_poll_status) - # Request count chips - self.batch_count_row = QHBoxLayout() - self.batch_count_row.setSpacing(6) - self._batch_count_labels = {} - for key, color in ( - ("succeeded", "#4ec9b0"), - ("processing", "#007acc"), - ("errored", "#f44747"), - ("canceled", "#ce9178"), - ("expired", "#dcdcaa"), - ): - lab = QLabel(f"{key}: 0") - lab.setAlignment(Qt.AlignCenter) - lab.setStyleSheet( - f"color:{color};background:#1e1e1e;border:1px solid #3e3e42;" - f"border-radius:4px;padding:6px 4px;font-size:11px;font-weight:bold;" - ) - self.batch_count_row.addWidget(lab, 1) - self._batch_count_labels[key] = lab - poll_layout.addLayout(self.batch_count_row) - self.batch_poll_bar = QProgressBar() - self.batch_poll_bar.setRange(0, 100) - self.batch_poll_bar.setValue(0) - self.batch_poll_bar.setFixedHeight(16) - self.batch_poll_bar.setTextVisible(True) - self.batch_poll_bar.setFormat("Requests complete: %v / %m") - self.batch_poll_bar.setStyleSheet(""" - QProgressBar { border: none; background-color: #2b2b2b; border-radius: 3px; color:#ccc; text-align:center; } - QProgressBar::chunk { background-color: #4ec9b0; border-radius: 3px; } - """) - poll_layout.addWidget(self.batch_poll_bar) - self.batch_poll_hint = QLabel("Stop is safe - the batch keeps running on Anthropic. Resume later from Batches or Batch Translate.") - self.batch_poll_hint.setWordWrap(True) - self.batch_poll_hint.setStyleSheet("color:#888888;font-size:11px;") - poll_layout.addWidget(self.batch_poll_hint) - poll_layout.addStretch(1) - self.batch_pipeline_stack.addWidget(poll_page) - - consume_page = QWidget() - consume_layout = QVBoxLayout(consume_page) - consume_layout.setContentsMargins(0, 0, 0, 0) - consume_layout.setSpacing(8) - self.batch_consume_status = QLabel("Pass 2/2: writing translated files from batch results…") - self.batch_consume_status.setWordWrap(True) - self.batch_consume_status.setStyleSheet("color:#cccccc;font-size:12px;") - consume_layout.addWidget(self.batch_consume_status) - self.batch_consume_hint = QLabel("Watch the Translation Log for [BATCH]/[CACHE] Input/Output pairs as each chunk is applied.") - self.batch_consume_hint.setWordWrap(True) - self.batch_consume_hint.setStyleSheet("color:#888888;font-size:11px;") - consume_layout.addWidget(self.batch_consume_hint) - consume_layout.addStretch(1) - self.batch_pipeline_stack.addWidget(consume_page) - - batch_pipe_layout.addWidget(self.batch_pipeline_stack, 1) - self.batch_live_status = QLabel("") - self.batch_live_status.setWordWrap(True) - self.batch_live_status.setStyleSheet("color:#666666;font-size:11px;") - batch_pipe_layout.addWidget(self.batch_live_status) - - self.batch_pipeline_widget.setLayout(batch_pipe_layout) - self.batch_pipeline_widget.setObjectName("batchPipeline") - self.batch_pipeline_widget.setStyleSheet(""" - #batchPipeline { - background-color: #252526; - border: 1px solid #3e3e42; - border-radius: 4px; - } - """) - self.batch_pipeline_widget.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) - self.batch_pipeline_stack.setStyleSheet("background: transparent;") - self.batch_pipeline_stack.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) - - # Prefer a Batch-history-style table for per-file status. Keep a hidden, - # parented list as a compatibility shim so clear()/legacy refs stay safe - # without creating an orphan top-level window (minimize/restore glitches). - self.progress_list = QListWidget(self) - self.progress_list.setVisible(False) - self.progress_list.setMaximumSize(0, 0) - - self.progress_files_summary = QLabel("No files in this run.") - self.progress_files_summary.setStyleSheet("color:#9d9d9d;font-size:12px;padding:2px 0;") - - self.progress_table = QTableWidget(0, 6) - self.progress_table.setHorizontalHeaderLabels( - ["File", "Status", "Progress", "Tokens", "Cost", "Time"] - ) - self.progress_table.setSelectionBehavior(QAbstractItemView.SelectRows) - self.progress_table.setSelectionMode(QAbstractItemView.SingleSelection) - self.progress_table.setEditTriggers(QAbstractItemView.NoEditTriggers) - self.progress_table.setAlternatingRowColors(True) - self.progress_table.verticalHeader().setVisible(False) - self.progress_table.setShowGrid(True) - self.progress_table.setStyleSheet( - "QTableWidget{background-color:#1e1e1e;color:#cccccc;gridline-color:#3a3a3a;" - "alternate-background-color:#252526;border:1px solid #555555;}" - "QHeaderView::section{background-color:#2d2d30;color:#cccccc;padding:4px;" - "border:1px solid #3a3a3a;font-weight:bold;}" - ) - hdr = self.progress_table.horizontalHeader() - hdr.setSectionResizeMode(0, QHeaderView.Stretch) - for col in range(1, 6): - hdr.setSectionResizeMode(col, QHeaderView.ResizeToContents) - self.progress_table.verticalHeader().setDefaultSectionSize(28) - - self.progress_overview_page = QWidget() - overview_layout = QVBoxLayout(self.progress_overview_page) - overview_layout.setContentsMargins(0, 0, 0, 0) - overview_layout.setSpacing(0) - overview_layout.addWidget(self.batch_pipeline_widget, 1) - - self.progress_files_page = QWidget() - files_page_layout = QVBoxLayout(self.progress_files_page) - files_page_layout.setContentsMargins(0, 0, 0, 0) - files_page_layout.setSpacing(8) - files_page_layout.addWidget(self.progress_files_summary) - files_page_layout.addWidget(self.progress_table, 1) - - progress_tab_btn_style = """ - QPushButton { - background: #2d2d30; - color: #aaa; - border: 1px solid #3e3e42; - border-radius: 4px; - padding: 0px; - } - QPushButton:hover { - background: #353538; - color: #ddd; - } - """ - progress_tab_btn_active_style = """ - QPushButton { - background: #252526; - color: #4ec9b0; - border: 1px solid #4a4a4f; - border-radius: 4px; - padding: 0px; - } - """ - self._progress_tab_btn_style = progress_tab_btn_style - self._progress_tab_btn_active_style = progress_tab_btn_active_style - - self.progress_tab_row = QWidget() - progress_tab_row_layout = QHBoxLayout(self.progress_tab_row) - progress_tab_row_layout.setContentsMargins(0, 0, 0, 0) - progress_tab_row_layout.setSpacing(6) - self.batch_tab_btn = QPushButton("Batch") - self.batch_tab_btn.setFixedSize(118, 34) - self.batch_tab_btn.setCursor(Qt.PointingHandCursor) - self.batch_tab_btn.clicked.connect(lambda: self._switch_progress_tab(0)) - self.files_tab_btn = QPushButton("Files") - self.files_tab_btn.setFixedSize(118, 34) - self.files_tab_btn.setCursor(Qt.PointingHandCursor) - self.files_tab_btn.clicked.connect(lambda: self._switch_progress_tab(1)) - progress_tab_row_layout.addWidget(self.batch_tab_btn) - progress_tab_row_layout.addWidget(self.files_tab_btn) - progress_tab_row_layout.addStretch() - self.progress_tab_row.setVisible(False) - - self.progress_content_stack = QStackedWidget() - self.progress_content_stack.addWidget(self.progress_overview_page) - self.progress_content_stack.addWidget(self.progress_files_page) - self.progress_content_stack.setCurrentIndex(1) - - progress_view_layout.addWidget(self.progress_tab_row) - progress_view_layout.addWidget(self.progress_content_stack, 1) - - # Summary button (shown after completion) - icon-only - # Use a simple left-arrow for the back action and place it on the left - self.reset_view_button = QPushButton() - qt_icons.apply_button_icon(self.reset_view_button, "←", color="#dddddd") - self.reset_view_button.setToolTip("Back to File Selection") - self.reset_view_button.clicked.connect(self.reset_to_file_view) - self.reset_view_button.setVisible(False) - # Button to open the translations (translated) folder - icon-only - self.open_translations_button = QPushButton() - qt_icons.apply_button_icon(self.open_translations_button, "📂", color="#dddddd") - self.open_translations_button.setToolTip("Open the translated files folder") - self.open_translations_button.clicked.connect(self.open_output_folder) - self.open_translations_button.setVisible(False) - # Sync translated/ → files/ (RPG Maker only) - self.sync_translated_button = QPushButton() - qt_icons.apply_button_icon(self.sync_translated_button, "🔄", color="#dddddd") - self.sync_translated_button.setToolTip("Sync translated/ → files/\nCopy translated files back into files/ so the next phase starts from the latest state") - self.sync_translated_button.clicked.connect(self._sync_translated_to_files) - self.sync_translated_button.setVisible(False) - # Export active files → game folder (RPG Maker only) - self.export_active_button = QPushButton() - qt_icons.apply_button_icon(self.export_active_button, "📤", color="#dddddd") - self.export_active_button.setToolTip("Export translated files → Game Folder\nCopy the files from this translation run into your game's data directory") - self.export_active_button.clicked.connect(self._export_last_run_files) - self.export_active_button.setVisible(False) - - # Make both buttons the same fixed size and style (icon-only) - icon_btn_style = """ - QPushButton { - background-color: #2d2d30; - color: white; - font-weight: bold; - font-size: 16px; - border: 1px solid #555555; - border-radius: 4px; - min-width: 40px; - max-width: 40px; - min-height: 36px; - max-height: 36px; - } - QPushButton:hover { - background-color: #3e3e42; - border-left-color: #007acc; - } - QPushButton:pressed { - background-color: #007acc; - } - """ - - self.reset_view_button.setStyleSheet(icon_btn_style) - self.open_translations_button.setStyleSheet(icon_btn_style) - self.sync_translated_button.setStyleSheet(icon_btn_style) - self.export_active_button.setStyleSheet(icon_btn_style) - # Size the icons to fill the compact buttons. - for _b in (self.reset_view_button, self.open_translations_button, - self.sync_translated_button, self.export_active_button): - _b.setIconSize(QSize(20, 20)) - - # Create the stop button here so it sits in the same row as the - # back/open buttons. Use a compact icon style to match them but - # make it visually distinct (red) to indicate a destructive action. - stop_button_style = """ - QPushButton { - background-color: #c0392b; /* red */ - color: white; - font-weight: bold; - font-size: 16px; - border: 1px solid #7f2e28; - border-radius: 4px; - min-width: 40px; - max-width: 40px; - min-height: 36px; - max-height: 36px; - } - QPushButton:hover { - background-color: #e04b43; - border-left-color: #ff6b60; - } - QPushButton:pressed { - background-color: #a82a20; - } - """ - - # Use a clear stop-sign emoji so the glyph is rendered as a stop icon - # and not as a colored square on some platforms. - self.stop_button = QPushButton() - qt_icons.apply_button_icon(self.stop_button, "🛑", color="#ffffff") - self.stop_button.setToolTip("Stop Translation") - self.stop_button.clicked.connect(self.stop_translation) - self.stop_button.setStyleSheet(stop_button_style) - self.stop_button.setIconSize(QSize(20, 20)) - self.stop_button.setVisible(False) - - # Place both buttons on the left and totals on the right - buttons_container = QWidget() - # Prevent the buttons row from changing the file list box size when - # buttons/totals are shown — use minimum height that can grow if needed. - buttons_container.setMinimumHeight(64) - buttons_container.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed) - buttons_hbox = QHBoxLayout() - buttons_hbox.setContentsMargins(0, 0, 0, 0) - buttons_hbox.setSpacing(8) - # Back/Open/Stop buttons on the left (stop shown while running) - buttons_hbox.addWidget(self.stop_button) - buttons_hbox.addWidget(self.reset_view_button) - buttons_hbox.addWidget(self.open_translations_button) - buttons_hbox.addWidget(self.sync_translated_button) - buttons_hbox.addWidget(self.export_active_button) - # Spacer between buttons and totals - buttons_hbox.addStretch() - # Totals widget on the right (hidden until start) - self.totals_widget = QWidget() - # Let the totals widget size naturally; the mismatch label starts hidden - # so it won't take extra space until a mismatch occurs. - self.totals_widget.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Preferred) - totals_layout = QVBoxLayout() - totals_layout.setContentsMargins(6, 2, 6, 2) - totals_layout.setSpacing(2) - self.totals_tokens_label = QLabel("Tokens: 0 in / 0 out") - self.totals_tokens_label.setStyleSheet("color: #f1c40f; font-weight: bold;") - totals_layout.addWidget(self.totals_tokens_label) - self.totals_cost_label = QLabel("Cost: $0.0000") - self.totals_cost_label.setStyleSheet("color: #4ec9b0; font-weight: bold;") - totals_layout.addWidget(self.totals_cost_label) - self.totals_time_label = QLabel("Time: 0.0s") - self.totals_time_label.setStyleSheet("color: #4da6ff; font-weight: bold;") - totals_layout.addWidget(self.totals_time_label) - self.totals_mismatch_label = QLabel("") - self.totals_mismatch_label.setStyleSheet("color: #ff4444; font-weight: bold;") - self.totals_mismatch_label.setVisible(False) - totals_layout.addWidget(self.totals_mismatch_label) - self.totals_widget.setLayout(totals_layout) - self.totals_widget.setVisible(False) - buttons_hbox.addWidget(self.totals_widget) - buttons_container.setLayout(buttons_hbox) - progress_view_layout.addWidget(buttons_container) - - progress_view_page.setLayout(progress_view_layout) - self.file_stack.addWidget(progress_view_page) # Index 1 - - # Add stacked widget to main layout and allow it to stretch - # so the input files area can take up available vertical space. - layout.addWidget(self.file_stack, 1) - - # Progress Section (removed from UI) - # The visible progress UI was removed per user request. We keep the - # underlying widgets as attributes so existing logic can update them - # without raising AttributeError, but we do not add them to the - # layout so they are not shown. - - progress_layout = QVBoxLayout() - progress_layout.setSpacing(8) - progress_layout.setContentsMargins(0, 0, 0, 12) - - # Files Translated counter - files_layout = QHBoxLayout() - files_layout.addWidget(QLabel("Files Translated:")) - self.files_translated_label = QLabel("0/0") - self.files_translated_label.setStyleSheet("font-weight: bold; color: #007acc;") - files_layout.addWidget(self.files_translated_label) - files_layout.addStretch() - progress_layout.addLayout(files_layout) - - # Currently translating - translating_layout = QHBoxLayout() - translating_layout.addWidget(QLabel("Translating:")) - self.translating_label = QLabel("—") - self.translating_label.setStyleSheet("font-weight: bold; color: #cccccc;") - translating_layout.addWidget(self.translating_label) - translating_layout.addStretch() - progress_layout.addLayout(translating_layout) - - # Progress bar with label - item_progress_layout = QHBoxLayout() - item_progress_layout.addWidget(QLabel("Progress:")) - self.item_progress_label = QLabel("0/0") - self.item_progress_label.setStyleSheet("font-weight: bold; color: #cccccc;") - item_progress_layout.addWidget(self.item_progress_label) - item_progress_layout.addStretch() - progress_layout.addLayout(item_progress_layout) - - self.item_progress_bar = QProgressBar() - self.item_progress_bar.setStyleSheet(""" - QProgressBar { - border: 1px solid #555555; - border-radius: 3px; - text-align: center; - background-color: #2b2b2b; - color: white; - height: 20px; - } - QProgressBar::chunk { - background-color: #007acc; - border-radius: 2px; - } - """) - progress_layout.addWidget(self.item_progress_bar) - - # NOTE: Do not add progress_layout to the UI. Kept in memory only. - - # Ensure any remaining space is consumed above the settings so - # the Translation Settings block stays anchored to the bottom. - layout.addStretch() - - # Translation Settings Section (moved to bottom) - layout.addWidget(create_horizontal_line()) - layout.addWidget(create_section_header("🌐 Translation Settings")) - - trans_form = QFormLayout() - trans_form.setSpacing(6) - trans_form.setContentsMargins(0, 0, 0, 12) - trans_form.setFieldGrowthPolicy(QFormLayout.FieldsStayAtSizeHint) - trans_form.setLabelAlignment(Qt.AlignRight | Qt.AlignVCenter) - - engine_label = QLabel("Game Engine:") - engine_label.setFixedWidth(100) - engine_label.setAlignment(Qt.AlignLeft | Qt.AlignVCenter) - self.module_combo = QComboBox() - self.module_combo.currentTextChanged.connect(self._on_module_changed) - self.module_combo.setFixedWidth(300) - trans_form.addRow(engine_label, self.module_combo) - - mode_label = QLabel("Mode:") - mode_label.setFixedWidth(100) - mode_label.setAlignment(Qt.AlignLeft | Qt.AlignVCenter) - self.mode_combo = QComboBox() - self.mode_combo.addItem("Translate") - self.mode_combo.addItem("Estimate") - self.mode_combo.addItem(BATCH_MODE_LABEL) - self.mode_combo.setFixedWidth(300) - self.mode_combo.currentTextChanged.connect(self._on_mode_changed) - trans_form.addRow(mode_label, self.mode_combo) - - self.batch_mode_note = QLabel( - BATCH_MODE_BENEFIT_NOTE + "\n" + BATCH_COLLECT_LIVE_CHARGE_NOTE - ) - self.batch_mode_note.setWordWrap(True) - self.batch_mode_note.setStyleSheet("color:#8fbc8f;font-size:12px;padding-left:4px;") - self.batch_mode_note.setVisible(False) - trans_form.addRow("", self.batch_mode_note) - - layout.addLayout(trans_form) - layout.addWidget(create_horizontal_line()) - - # Buttons (right below progress section) - button_layout = QHBoxLayout() - button_layout.addStretch() - self.translate_button = QPushButton("Start Translation") - self.translate_button.clicked.connect(self.start_translation) - self.translate_button.setStyleSheet(""" - QPushButton { - background-color: #007acc; - color: white; - font-weight: bold; - padding: 10px 20px; - font-size: 14px; - border-radius: 5px; - } - QPushButton:hover { - background-color: #106ebe; - } - """) - button_layout.addWidget(self.translate_button) - button_layout.addStretch() - layout.addLayout(button_layout) - - self.mode_combo.setCurrentIndex(self.mode_combo.findText(BATCH_MODE_LABEL)) - left_widget.setLayout(layout) - - # Right side - translation history log viewer - self.translation_log_viewer = LogViewer() - # Mismatch counting is driven by MISMATCH_EVENT stdout markers - # detected in append_log. The log_viewer signal is kept as a - # fallback for in-process mode (e.g. speaker-parse). - self.translation_log_viewer.mismatch_detected.connect(self.on_mismatch_detected) - - # Allow both left and right widgets to expand vertically so the - # log viewer fills the full height to the bottom of the tab. - left_widget.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) - self.translation_log_viewer.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) - - # Add both widgets to the fixed HBox with stretch factors (left 40%, right 60%). - # Keep the left column top-aligned so its header stays at the top, - # but allow the right-hand log viewer to expand vertically to the - # bottom of the tab so it fills available space. - # Let the left widget expand vertically (do not force AlignTop) - # so its internal stretch can push the settings block to the bottom. - main_hbox.addWidget(left_widget, 2) - # Do NOT force AlignTop on the log viewer; with an Expanding - # vertical size policy it will grow to fill the available height. - main_hbox.addWidget(self.translation_log_viewer, 3) - - main_container.setLayout(main_hbox) - - # Ensure main container will expand to fill the tab vertically - main_container.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) - - # Set main layout for this tab - tab_layout = QVBoxLayout() - # Add with stretch so the container expands to fill available space - tab_layout.addWidget(main_container, 1) - tab_layout.setContentsMargins(0, 0, 0, 0) - self.setLayout(tab_layout) - - def setup_module_list(self): - """Set up the module selection list.""" - # Import modules to get the list - try: - sys.path.append(str(self.project_root)) - from modules.rpgmakermvmz import handleMVMZ - from modules.csv import handleCSV - from modules.tyrano import handleTyrano - from modules.kirikiri import handleKirikiri - from modules.json import handleJSON - from modules.lune import handleLune - from modules.yuris import handleYuris - from modules.nscript import handleOnscripter - from modules.wolf import handleWOLF - from modules.wolf2 import handleWOLF2 - from modules.wolfdawn import handleWolfDawn - from modules.regex import handleRegex - from modules.text import handleText - from modules.renpy import handleRenpy - from modules.unity import handleUnity - from modules.images import handleImages - from modules.rpgmakerplugin import handlePlugin - from modules.aquedi4 import handleAquedi4 - from modules.srpg import handleSRPG - - self.modules = [ - ["RPG Maker MV/MZ", [".json"], handleMVMZ], - ["CSV", [".csv"], handleCSV], - ["Tyrano", [".ks"], handleTyrano], - ["Kirikiri", [".ks"], handleKirikiri], - ["JSON", [".json"], handleJSON], - ["Lune", [".l"], handleLune], - ["Yuris", [".json"], handleYuris], - ["NScript", [".nscript"], handleOnscripter], - ["Wolf RPG (WolfDawn)", [".json"], handleWolfDawn], - ["Wolf RPG", [".json"], handleWOLF], - ["Wolf RPG 2", [".txt"], handleWOLF2], - ["Regex", [".txt", ".json", ".script", ".csv"], handleRegex], - ["Text", [".txt", ".srt"], handleText], - ["RenPy", [".rpy"], handleRenpy], - ["Unity", [".unity"], handleUnity], - ["Images", [".png", ".jpg", ".jpeg"], handleImages], - ["RPG Maker Plugin", [".js"], handlePlugin], - ["Aquedi4 Prepared JSON", [".json"], handleAquedi4], - ["SRPG Studio", [".json"], handleSRPG], - ] - - for module in self.modules: - extensions = ", ".join(module[1]) - self.module_combo.addItem(f"{module[0]} ({extensions})") - if self.module_combo.count(): - self._on_module_changed(self.module_combo.currentText()) - - except Exception as e: - # Store error for later logging since log_display might not exist yet - self.module_load_error = f"Warning: Could not load modules: {str(e)}" - # Add a default option - self.module_combo.addItem("RPG Maker MV/MZ (.json)") - self.modules = [["RPG Maker MV/MZ", [".json"], None]] - self._on_module_changed(self.module_combo.currentText()) - - def _on_module_changed(self, text: str): - lowered = text.lower() - if "wolf" in lowered and "wolf rpg 2" not in lowered: - self.engine_changed.emit("wolf") - elif "mv/mz" in lowered: - self.engine_changed.emit("mvmz") - elif "srpg" in lowered: - self.engine_changed.emit("srpg") - - # Update mode dropdown based on engine - current_mode = self.mode_combo.currentText() - self.mode_combo.clear() - self.mode_combo.addItem("Translate") - self.mode_combo.addItem("Estimate") - self.mode_combo.addItem(BATCH_MODE_LABEL) - - # Add Parse Speakers for RPG Maker MV/MZ - if "mv/mz" in lowered: - self.mode_combo.addItem("Parse Speakers") - - # Restore previous selection if it still exists - index = self.mode_combo.findText(current_mode) - if index >= 0: - self.mode_combo.setCurrentIndex(index) - else: - batch_idx = self.mode_combo.findText(BATCH_MODE_LABEL) - self.mode_combo.setCurrentIndex(batch_idx if batch_idx >= 0 else 0) - - # Refresh file list to show only files matching the selected module's extensions - self.refresh_file_lists() - - def _toggle_file_checkbox(self, item): - """Toggle checkbox when clicking anywhere on the item.""" - # Toggle the built-in QListWidgetItem checkbox state - try: - if item.checkState() == Qt.Checked: - item.setCheckState(Qt.Unchecked) - else: - item.setCheckState(Qt.Checked) - except Exception: - pass - - def _remove_file_list_event_filter(self): - if not getattr(self, "_file_list_filter_installed", False): - return - try: - self.file_list.viewport().removeEventFilter(self) - except RuntimeError: - pass - self._file_list_filter_installed = False - - def eventFilter(self, obj, event): - """Intercept mouse presses on the file list. - - If the click is inside the checkbox indicator area, allow the - default Qt handling to toggle the checkbox. If the click is - on the rest of the row, manually toggle the item's check state - and consume the event to prevent further handling (avoids - double toggles). - """ - try: - file_list = self.file_list - viewport = file_list.viewport() - except RuntimeError: - return False - - # We install the filter on the QListWidget viewport, so the - # obj will be the viewport widget when mouse events arrive. - if (obj is viewport or obj is file_list) and event.type() == QEvent.MouseButtonPress: - pos = event.pos() - index = file_list.indexAt(pos) - if not index.isValid(): - return False - - rect = file_list.visualRect(index) - # Approximate checkbox indicator rectangle (style may vary). - # Use a small left inset and a ~20x20 indicator area vertically centered. - indicator_w = 20 - indicator_h = 20 - indicator_x = rect.left() + 4 - indicator_y = rect.top() + (rect.height() - indicator_h) // 2 - indicator_rect = QRect(indicator_x, indicator_y, indicator_w, indicator_h) - - # If the click is inside the indicator area, let Qt handle it - # (it will toggle the check state). Otherwise toggle manually - # and consume the event. - if indicator_rect.contains(pos): - return False - - # Toggle the item and consume the event - item = file_list.item(index.row()) - try: - if item.checkState() == Qt.Checked: - item.setCheckState(Qt.Unchecked) - else: - item.setCheckState(Qt.Checked) - except Exception: - pass - return True - - try: - return super().eventFilter(obj, event) - except RuntimeError: - return False - - def _on_mode_changed(self, mode_text): - """Update the translate button text based on selected mode.""" - if hasattr(self, "batch_mode_note"): - self.batch_mode_note.setVisible(mode_text == BATCH_MODE_LABEL) - if not hasattr(self, "translate_button"): - return - if mode_text == "Translate": - self.translate_button.setText("Start Translation") - elif mode_text == "Estimate": - self.translate_button.setText("Start Estimation") - elif mode_text == BATCH_MODE_LABEL: - self.translate_button.setText("Start Batch Translation") - elif mode_text == "Parse Speakers": - self.translate_button.setText("Parse Speakers") - - def _switch_progress_tab(self, index): - """Switch Batch/Files views; index 0 = batch overview, 1 = per-file list.""" - self._batch_tab_index = index if self.progress_tab_row.isVisible() else -1 - self.progress_content_stack.setCurrentIndex(index) - self.batch_tab_btn.setStyleSheet( - self._progress_tab_btn_active_style if index == 0 else self._progress_tab_btn_style - ) - self.files_tab_btn.setStyleSheet( - self._progress_tab_btn_active_style if index == 1 else self._progress_tab_btn_style - ) - - def _set_progress_view_mode(self, batch_mode, file_count=0): - """Batch runs use a Batch tab for the pipeline; Files tab holds the per-file list.""" - if batch_mode: - self.progress_tab_row.setVisible(True) - self.batch_pipeline_widget.setVisible(True) - self.files_tab_btn.setText(f"Files ({file_count})" if file_count else "Files") - self._switch_progress_tab(0) - else: - self.progress_tab_row.setVisible(False) - self.batch_pipeline_widget.setVisible(False) - self.files_tab_btn.setText("Files") - self._batch_tab_index = -1 - self.progress_content_stack.setCurrentIndex(1) - # Per-file Time is useful for live translate; for batch, Anthropic owns - # the wait and local write time is negligible. - table = getattr(self, "progress_table", None) - if table is not None and table.columnCount() > 5: - table.setColumnHidden(5, bool(batch_mode)) - try: - if hasattr(self, "totals_time_label"): - self.totals_time_label.setVisible(not batch_mode) - except Exception: - pass - - def _on_batch_submit_yes(self): - if hasattr(self, "translation_worker") and self.translation_worker: - self.translation_worker.set_batch_submit_response(True) - - def _on_batch_submit_no(self): - if hasattr(self, "translation_worker") and self.translation_worker: - self.translation_worker.set_batch_submit_response(False) - - def _update_batch_stop_button(self): - """Show stop only while batch is collecting; hide after that.""" - if not getattr(self, "_batch_active", False): - return - try: - self.stop_button.setVisible(self._batch_ui_phase == "collect") - except Exception: - pass - - def _batch_step_style(self, state: str) -> str: - """CSS for a pipeline step chip: idle | active | done.""" - if state == "active": - return ( - "color:#ffffff;background:#007acc;border:1px solid #007acc;" - "border-radius:4px;padding:4px 6px;font-size:11px;font-weight:bold;" - ) - if state == "done": - return ( - "color:#4ec9b0;background:#1e1e1e;border:1px solid #4ec9b0;" - "border-radius:4px;padding:4px 6px;font-size:11px;" - ) - return ( - "color:#888888;background:#1e1e1e;border:1px solid #3e3e42;" - "border-radius:4px;padding:4px 6px;font-size:11px;" - ) - - def _set_batch_steps(self, active_index: int): - """Highlight pipeline steps. active_index 0..3, or 4 when fully done.""" - labels = getattr(self, "_batch_step_labels", None) or [] - for i, lab in enumerate(labels): - if i < active_index: - lab.setStyleSheet(self._batch_step_style("done")) - elif i == active_index and active_index < 4: - lab.setStyleSheet(self._batch_step_style("active")) - elif active_index >= 4: - lab.setStyleSheet(self._batch_step_style("done")) - else: - lab.setStyleSheet(self._batch_step_style("idle")) - - def _update_batch_poll_dashboard(self, statuses): - """Render structured Anthropic batch statuses into the poll panel.""" - if not isinstance(statuses, list) or not statuses: - return - totals = {"processing": 0, "succeeded": 0, "errored": 0, "canceled": 0, "expired": 0} - ids = [] - api_states = [] - expected = 0 - for st in statuses: - ids.append(st.get("id") or "") - api_states.append(st.get("api_status") or "") - expected += int(st.get("request_count") or 0) - for k, v in (st.get("counts") or {}).items(): - if k in totals: - totals[k] += int(v or 0) - done = totals["succeeded"] + totals["errored"] + totals["canceled"] + totals["expired"] - total = max(expected, done + totals["processing"], 1) - if hasattr(self, "batch_poll_id"): - self.batch_poll_id.setText("Batch: " + (", ".join(i for i in ids if i) or "—")) - for key, lab in (getattr(self, "_batch_count_labels", {}) or {}).items(): - lab.setText(f"{key}: {totals.get(key, 0)}") - if hasattr(self, "batch_poll_bar"): - self.batch_poll_bar.setRange(0, total) - self.batch_poll_bar.setValue(min(done, total)) - self.batch_poll_bar.setFormat(f"Requests finished: {done} / {total}") - # Overall bar: process phase spans ~55-80% - frac = done / total if total else 0 - self.batch_overall_bar.setValue(55 + int(20 * frac)) - state_txt = ", ".join(sorted(set(api_states))) or "unknown" - self.batch_poll_status.setText( - f"Anthropic status: {state_txt}\n" - f"{totals['succeeded']} succeeded, {totals['processing']} still processing" - + (f", {totals['errored']} errored" if totals["errored"] else "") - + (f", {totals['canceled']} canceled" if totals["canceled"] else "") - + (f", {totals['expired']} expired" if totals["expired"] else "") - + "." - ) - self.batch_live_status.setText( - f"Polling… {done}/{total} requests finished. Details stay in the Translation Log." - ) - - def _on_batch_phase(self, phase, payload): - """Update the inline batch pipeline panel for the current phase.""" - self._batch_ui_phase = phase - self.batch_pipeline_widget.setVisible(True) - - if phase == "collect": - self._set_batch_steps(0) - self.batch_phase_title.setText("Batch Translate - Pass 1/2: Collect") - self.batch_overall_bar.setRange(0, 100) - self.batch_overall_bar.setValue(15) - self.batch_pipeline_stack.setCurrentIndex(0) - self.batch_submit_yes_btn.setText("Submit Batch") - self.batch_collect_status.setText( - "Scanning files and queueing dialogue for the batch…" - ) - elif phase == "collect_done": - info = payload or {} - n_files = info.get("files", "?") - n_req = info.get("requests", "?") - self._set_batch_steps(1) - self.batch_phase_title.setText("Batch Translate - Collect complete") - self.batch_overall_bar.setValue(30) - self.batch_pipeline_stack.setCurrentIndex(0) - self.batch_collect_status.setText( - f"Finished scanning {n_files} file(s) - {n_req} API request(s) queued.\n" - "Review the cost estimate, then submit once for the whole batch." - ) - self.batch_live_status.setText( - f"All {n_files} files collected. Open the Files tab to inspect per-file rows." - ) - elif phase == "submit": - est = payload or {} - n_files = est.get("files", "?") - n_req = est.get("requests", "?") - self._set_batch_steps(1) - self.batch_phase_title.setText("Batch Translate - Review & Submit") - self.batch_overall_bar.setValue(35) - self.batch_pipeline_stack.setCurrentIndex(1) - self.batch_submit_summary.setText( - f"{n_files} file(s) scanned - {n_req} request(s) queued.\n" - f"Model: {est.get('model') or os.getenv('model', '') or '—'}\n\n" - "One Anthropic submission covers every queued request (50% batch discount). " - "You can stop safely after submit and resume later." - ) - if hasattr(self, "batch_cost_cached"): - self.batch_cost_cached.setText( - f"Batch + cache\n${float(est.get('batch_cached_cost') or 0):.2f}" - ) - self.batch_cost_nocache.setText( - f"Batch worst-case\n${float(est.get('batch_nocache_cost') or 0):.2f}" - ) - self.batch_cost_live.setText( - f"Live API\n${float(est.get('live_cost') or 0):.2f}" - ) - self.batch_submit_yes_btn.setText(f"Submit All ({n_req} requests)") - elif phase == "polling": - self._set_batch_steps(2) - self.batch_phase_title.setText("Batch Translate - Processing") - self.batch_overall_bar.setRange(0, 100) - self.batch_overall_bar.setValue(55) - self.batch_pipeline_stack.setCurrentIndex(2) - self.batch_poll_status.setText("Submitted - waiting for Anthropic to process the batch…") - if hasattr(self, "batch_poll_bar"): - self.batch_poll_bar.setRange(0, 100) - self.batch_poll_bar.setValue(0) - self.batch_poll_bar.setFormat("Waiting for first status…") - elif phase == "poll_status": - self._set_batch_steps(2) - self.batch_pipeline_stack.setCurrentIndex(2) - self._update_batch_poll_dashboard(payload) - elif phase == "consume": - self._batch_consume_started = True - self._set_batch_steps(3) - self.batch_phase_title.setText("Batch Translate - Pass 2/2: Write") - self.batch_overall_bar.setValue(80) - self.batch_pipeline_stack.setCurrentIndex(3) - self.batch_consume_status.setText("Pass 2/2: writing translated files from batch results…") - self._reset_files_for_consume() - if self.progress_tab_row.isVisible(): - self._switch_progress_tab(0) - elif phase == "done": - self._set_batch_steps(4) - self.batch_overall_bar.setValue(100) - self.batch_phase_title.setText("Batch Translate - Complete") - self.batch_pipeline_stack.setCurrentIndex(3) - self.batch_consume_status.setText( - "Pass 2/2 finished. Translations written - use the back arrow to return to the file list." - ) - self.batch_live_status.setText("Batch complete") - - self._update_batch_stop_button() - - def _reset_batch_pipeline_ui(self): - self._batch_active = False - self._batch_ui_phase = None - self._batch_consume_started = False - if hasattr(self, "batch_pipeline_widget"): - self.batch_pipeline_widget.setVisible(False) - if hasattr(self, "batch_live_status"): - self.batch_live_status.setText("") - if hasattr(self, "progress_content_stack"): - self._set_progress_view_mode(False) - elif hasattr(self, "batch_overall_bar"): - self.batch_overall_bar.setRange(0, 100) - self.batch_overall_bar.setValue(0) - - def _reset_files_for_consume(self): - """Clear collect-pass row state before the consume pass writes translations.""" - self.files_completed = 0 - self.files_translated_label.setText(f"0/{self.files_total}") - for filename, item in self.file_progress_items.items(): - try: - item["checkbox"].setChecked(False) - item["label"].setText("Waiting...") - item["progress_bar"].setValue(0) - item["progress_bar"].setMaximum(100) - item["tokens_label"].setText("") - item["cost_label"].setText("") - item["time_label"].setText("") - item["status_label"].setText("") - item.pop("_skip_reason", None) - except Exception: - pass - self._set_progress_row( - filename, - status="Waiting", - status_color="#888888", - progress="-", - tokens="-", - cost="-", - time_s="-", - ) - if hasattr(self, "_applied_file_totals"): - self._applied_file_totals.clear() - self.totals_input_tokens = 0 - self.totals_output_tokens = 0 - self.totals_cost = 0.0 - self.totals_time = 0.0 - try: - if hasattr(self, "totals_tokens_label"): - self.totals_tokens_label.setText("Tokens: 0 in / 0 out") - if hasattr(self, "totals_cost_label"): - self.totals_cost_label.setText("Cost: $0.0000") - if hasattr(self, "totals_time_label"): - self.totals_time_label.setText("Time: 0.0s") - except Exception: - pass - - def mark_file_queued(self, filename): - """Collect pass finished for a file - queued for batch, not translated yet.""" - if filename not in self.file_progress_items: - return - item = self.file_progress_items[filename] - try: - item["label"].setText("Collected") - item["progress_bar"].setMaximum(100) - item["progress_bar"].setValue(100) - item["checkbox"].setChecked(False) - except Exception: - pass - self._set_progress_row( - filename, - status="Collected", - status_color="#d4a017", - progress="queued", - ) - - def _check_model_pricing(self): - """Fetch live pricing for the current model and print it to the log.""" - from dotenv import dotenv_values - from pathlib import Path as _Path - - # Read model from .env file directly so we always get the saved value - env = dotenv_values(_Path(".env")) if _Path(".env").exists() else {} - model = env.get("model") or os.getenv("model", "").strip() - - log = self.translation_log_viewer - - if not model: - log.append_log_message("💰 [PRICING] No model configured — set a model in Settings first.") - return - - log.append_log_message(f"💰 [PRICING] Checking pricing for: {model}") - - try: - from util.translation import _lookup_model_price, _load_litellm_pricing - except Exception as e: - log.append_log_message(f"💰 [PRICING] Could not import pricing module: {e}") - return - - # Force a fresh fetch attempt (bypasses the in-memory TTL check by - # temporarily clearing the in-memory cache timestamp) - try: - import util.translation as _tmod - _tmod._pricing_db_fetched_at = 0.0 - _tmod._pricing_fetch_warned = False - except Exception: - pass - - db = _load_litellm_pricing() - if db is None: - log.append_log_message( - "💰 [PRICING] Could not reach LiteLLM pricing DB — no internet or cache available. " - "Falling back to built-in prices." - ) - else: - log.append_log_message(f"💰 [PRICING] LiteLLM DB loaded ({len(db):,} entries).") - - result = _lookup_model_price(model) - if result: - inp, out = result - source = "LiteLLM" if db else "built-in fallback" - log.append_log_message( - f"💰 [PRICING] {model} — " - f"Input: ${inp:.4f} / 1M tokens | " - f"Output: ${out:.4f} / 1M tokens " - f"(source: {source})" - ) - else: - # Fall back to getPricingConfig for the hardcoded table result - try: - from util.translation import getPricingConfig - cfg = getPricingConfig(model) - log.append_log_message( - f"💰 [PRICING] {model} not found in LiteLLM DB — using built-in fallback: " - f"Input: ${cfg['inputAPICost']:.4f} / 1M tokens | " - f"Output: ${cfg['outputAPICost']:.4f} / 1M tokens" - ) - except Exception as e: - log.append_log_message(f"💰 [PRICING] Could not determine pricing for '{model}': {e}") - - def refresh_file_lists(self): - """Refresh the file list with checkboxes, preserving checked states.""" - # Save current check states from existing QListWidgetItems - checked_files = set() - for i in range(self.file_list.count()): - item = self.file_list.item(i) - try: - if item.checkState() == Qt.Checked: - checked_files.add(item.text()) - except Exception: - pass - - # Get accepted extensions for the currently selected module - accepted_extensions = [] - try: - selected_index = self.module_combo.currentIndex() - if 0 <= selected_index < len(self.modules): - accepted_extensions = self.modules[selected_index][1] # List of extensions like [".json"] - except Exception: - pass - - # Rebuild the list using simple QListWidgetItems with checkboxes - self.file_list.clear() - if self.files_dir.exists(): - for file_path in sorted(self.files_dir.iterdir()): - if file_path.is_file() and file_path.name != '.gitkeep': - # Filter by accepted extensions if any are defined - if accepted_extensions: - file_ext = file_path.suffix.lower() - # Skip files that don't match any accepted extension - if not any(file_ext == ext.lower() for ext in accepted_extensions): - continue - - item = QListWidgetItem(file_path.name) - # Ensure the item is enabled, selectable, and user-checkable - item.setFlags(item.flags() | Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsSelectable) - # Restore checked state if it was previously checked - if file_path.name in checked_files: - item.setCheckState(Qt.Checked) - else: - item.setCheckState(Qt.Unchecked) - self.file_list.addItem(item) - - def select_all_files(self): - """Select all files in the list.""" - for i in range(self.file_list.count()): - item = self.file_list.item(i) - try: - item.setCheckState(Qt.Checked) - except Exception: - pass - - def deselect_all_files(self): - """Deselect all files in the list.""" - for i in range(self.file_list.count()): - item = self.file_list.item(i) - try: - item.setCheckState(Qt.Unchecked) - except Exception: - pass - - def get_selected_files(self): - """Get list of checked files.""" - selected = [] - for i in range(self.file_list.count()): - item = self.file_list.item(i) - try: - if item.checkState() == Qt.Checked: - selected.append(item.text()) - except Exception: - pass - return selected - - def select_files_by_name(self, file_names): - """Check the given relative file names (and uncheck others).""" - wanted = set(file_names or []) - if not wanted: - return - self.refresh_file_lists() - for i in range(self.file_list.count()): - item = self.file_list.item(i) - try: - item.setCheckState(Qt.Checked if item.text() in wanted else Qt.Unchecked) - except Exception: - pass - - def add_input_files(self): - """Add files to the input directory, remembering last used directory.""" - # Restore last used directory from settings if available - start_dir = "" - try: - if self.settings: - start_dir = self.settings.value("last_open_dir", "") or "" - except Exception: - start_dir = "" - - file_paths, _ = QFileDialog.getOpenFileNames( - self, - "Select files to add", - start_dir, - "All Files (*)" - ) - - if file_paths: - # Save the directory used so next time we open the same place - try: - if self.settings and len(file_paths) > 0: - import os - dir_used = os.path.dirname(file_paths[0]) - self.settings.setValue("last_open_dir", dir_used) - except Exception: - pass - try: - import shutil - copied_count = 0 - for file_path in file_paths: - source = Path(file_path) - dest = self.files_dir / source.name - - if dest.exists(): - reply = QMessageBox.question( - self, - "File Exists", - f"File '{source.name}' already exists. Overwrite?", - QMessageBox.Yes | QMessageBox.No - ) - if reply == QMessageBox.No: - continue - - shutil.copy2(source, dest) - copied_count += 1 - - self.refresh_file_lists() - if copied_count > 0: - QMessageBox.information(self, "Files Added", f"Successfully added {copied_count} files.") - - except Exception as e: - QMessageBox.critical(self, "Error", f"Failed to add files:\n{str(e)}") - - def remove_selected_files(self): - """Remove selected (checked) files from the input directory.""" - selected_files = self.get_selected_files() - - if not selected_files: - QMessageBox.information(self, "No Selection", "Please check files to remove.") - return - - reply = QMessageBox.question( - self, - "Confirm Deletion", - f"Are you sure you want to delete {len(selected_files)} file(s)?", - QMessageBox.Yes | QMessageBox.No - ) - - if reply == QMessageBox.Yes: - try: - for filename in selected_files: - file_path = self.files_dir / filename - file_path.unlink() - self.refresh_file_lists() - QMessageBox.information(self, "Files Removed", f"Successfully removed {len(selected_files)} file(s).") - except Exception as e: - QMessageBox.critical(self, "Error", f"Failed to delete files:\n{str(e)}") - - def open_input_folder(self): - """Open the input directory.""" - self.open_folder(self.files_dir) - - def open_output_folder(self): - """Open the output directory.""" - self.open_folder(self.translated_dir) - - def _sync_translated_to_files(self): - """Copy translated/ files back into files/ (only matching names).""" - import shutil - files_dir = Path("files") - transl_dir = Path("translated") - - if not transl_dir.exists(): - QMessageBox.warning(self, "Sync", "translated/ folder not found — nothing to sync.") - return - - active = {fp.name for fp in files_dir.glob("*.json")} if files_dir.exists() else set() - to_copy = [fp for fp in transl_dir.glob("*.json") if not active or fp.name in active] - - if not to_copy: - QMessageBox.warning(self, "Sync", "No matching files found in translated/ to sync.") - return - - files_dir.mkdir(exist_ok=True) - copied = 0 - for src in to_copy: - shutil.copy2(src, files_dir / src.name) - copied += 1 - - QMessageBox.information(self, "Sync Complete", f"Synced {copied} file(s) from translated/ → files/") - self.refresh_file_lists() - - def _export_selected_files(self): - """Export only the currently checked files in the file list.""" - selected = self.get_selected_files() - if not selected: - QMessageBox.warning(self, "Export", "No files are checked — select files to export first.") - return - self._export_active_files(filenames=selected) - - def _export_last_run_files(self): - """Export only the files that were part of the most recent translation run.""" - if not self._last_run_files: - QMessageBox.warning(self, "Export", "No translation run recorded — translate some files first.") - return - self._export_active_files(filenames=self._last_run_files) - - def _export_active_files(self, filenames: list | None = None): - """Export translated files into the game data folder. - - filenames: if provided, only those files are exported; otherwise all - files currently in files/ are used (the original behavior, kept for - back-compat and used by the Workflow tab via inheritance). - """ - files_dir = Path("files") - if filenames is not None: - active = [n for n in filenames if n != ".gitkeep"] - else: - active = sorted( - fp.name for fp in files_dir.glob("*.json") if fp.name != ".gitkeep" - ) if files_dir.exists() else [] - - if not active: - QMessageBox.warning(self, "Export", "No files found in files/ — import game files first.") - return - - # For RPG Maker MV/MZ, try to reuse the project path already set in the Workflow tab - game_data = None - module_text = "" - try: - module_text = self.module_combo.currentText().lower() - except Exception: - pass - is_mvmz = "mv/mz" in module_text - - if is_mvmz: - try: - wt = self.window().workflow_tab - if wt and getattr(wt, "_data_path", None): - game_data = wt._data_path - except Exception: - pass - - if not game_data: - game_data = QFileDialog.getExistingDirectory(self, "Select Game Data Folder to Export Into") - if not game_data: - return - - transl_dir = Path("translated") - exportable_count = sum(1 for name in active if (transl_dir / name).exists()) - reply = QMessageBox.question( - self, - "Export Active Files to Game", - f"Export {exportable_count} file(s) into:\n{game_data}\n\nMake a backup first if needed. Continue?", - QMessageBox.Yes | QMessageBox.No, - QMessageBox.No, - ) - if reply != QMessageBox.Yes: - return - - import shutil - exported = 0 - skipped = 0 - for name in active: - src = transl_dir / name - if src.exists(): - shutil.copy2(src, Path(game_data) / name) - exported += 1 - else: - skipped += 1 - - msg = f"Exported {exported} file(s) to:\n{game_data}" - if skipped: - msg += f"\n({skipped} file(s) not found in translated/ — skipped)" - QMessageBox.information(self, "Export Complete", msg) - - def open_folder(self, folder_path): - """Open a folder in the file explorer.""" - try: - import platform - if platform.system() == "Windows": - subprocess.run(["explorer", str(folder_path)]) - elif platform.system() == "Darwin": # macOS - subprocess.run(["open", str(folder_path)]) - else: # Linux - subprocess.run(["xdg-open", str(folder_path)]) - except Exception as e: - QMessageBox.warning(self, "Error", f"Failed to open folder:\n{str(e)}") - - def _refresh_files_summary(self): - """Update the Files-tab summary line from current row states.""" - if not hasattr(self, "progress_files_summary"): - return - items = getattr(self, "file_progress_items", {}) or {} - total = len(items) - if total == 0: - self.progress_files_summary.setText("No files in this run.") - return - counts = {} - for meta in items.values(): - st = (meta.get("status_text") or "Waiting").split(" ")[0] - counts[st] = counts.get(st, 0) + 1 - parts = [f"{total} file(s)"] - for key in ( - "Done", "Collected", "Writing", "Scanning", "Translating", - "Failed", "Skipped", "Unsupported", "Waiting", - ): - if counts.get(key): - parts.append(f"{counts[key]} {key.lower()}") - self.progress_files_summary.setText(" · ".join(parts)) - - def _set_progress_row(self, filename, *, status=None, status_color=None, - progress=None, tokens=None, cost=None, time_s=None): - """Update one Files-tab table row.""" - meta = (getattr(self, "file_progress_items", {}) or {}).get(filename) - if not meta: - return - row = meta.get("row") - table = getattr(self, "progress_table", None) - if table is None or row is None or row >= table.rowCount(): - return - - def _cell(col, text, color=None): - item = table.item(row, col) - if item is None: - item = QTableWidgetItem("") - table.setItem(row, col, item) - item.setText(str(text)) - if color: - item.setForeground(QBrush(QColor(color))) - - if status is not None: - meta["status_text"] = status - _cell(1, status, status_color or "#cccccc") - if progress is not None: - _cell(2, progress, "#9cdcfe") - if tokens is not None: - _cell(3, tokens, "#f1c40f") - if cost is not None: - _cell(4, cost, "#4ec9b0") - if time_s is not None: - _cell(5, time_s, "#4da6ff") - self._refresh_files_summary() - - def create_progress_item(self, filename): - """Add a Files-tab table row for a file and return a placeholder shim.""" - table = getattr(self, "progress_table", None) - if table is None: - return _ShimWidget() - - row = table.rowCount() - table.insertRow(row) - values = [filename, "Waiting", "-", "-", "-", "-"] - colors = ["#ffffff", "#888888", "#666666", "#666666", "#666666", "#666666"] - for col, (val, color) in enumerate(zip(values, colors)): - item = QTableWidgetItem(val) - item.setForeground(QBrush(QColor(color))) - if col == 0: - item.setData(Qt.UserRole, filename) - table.setItem(row, col, item) - - # Non-Qt shims keep older helpers working without spawning orphan windows. - checkbox = _ShimCheckBox() - checkbox.setEnabled(False) - progress_label = _ShimLabel("Waiting...") - progress_bar = _ShimProgressBar() - tokens_label = _ShimLabel("") - cost_label = _ShimLabel("") - time_label = _ShimLabel("") - status_label = _ShimLabel("") - shim = _ShimWidget() - - self.file_progress_items[filename] = { - "row": row, - "status_text": "Waiting", - "widget": shim, - "checkbox": checkbox, - "label": progress_label, - "progress_bar": progress_bar, - "tokens_label": tokens_label, - "cost_label": cost_label, - "time_label": time_label, - "status_label": status_label, - } - self._refresh_files_summary() - return shim - - def update_file_item_progress(self, filename, current, total): - """Update progress for a specific file.""" - if filename not in self.file_progress_items: - return - item = self.file_progress_items[filename] - try: - item["progress_bar"].setMaximum(total if total > 0 else 100) - item["progress_bar"].setValue(current) - item["label"].setText(f"{current}/{total}") - except Exception: - pass - phase = getattr(self, "_batch_ui_phase", None) - if getattr(self, "_batch_active", False) and phase == "collect": - status, color = "Scanning", "#007acc" - elif getattr(self, "_batch_active", False) and phase == "consume": - status, color = "Writing", "#007acc" - else: - status, color = "Translating", "#007acc" - self._set_progress_row( - filename, - status=status, - status_color=color, - progress=f"{current}/{total}" if total else "-", - ) - - def _apply_success_status_icon(self, item, completion_kind="normal"): - """Mark a successful row; tooltips explain skip / idle when relevant.""" - try: - item["status_label"].setText("✓") - item["status_label"].setStyleSheet( - "color: #4ec9b0; font-weight: bold; font-size: 11px;" - ) - if completion_kind == "skip": - reason = (item.get("_skip_reason") or "").strip() - tip = f"Skipped: {reason}" if reason else "Whole file skipped (paths/fonts only)." - status, color = "Skipped", "#dcdcaa" - elif completion_kind == "idle": - tip = "No translatable lines (non-dialogue content only)." - status, color = "Done", "#4ec9b0" - else: - tip = "" - status, color = "Done", "#4ec9b0" - item["status_label"].setToolTip(tip) - item["status_label"].setVisible(True) - # Mirror into the table when we know the filename - for fname, meta in (getattr(self, "file_progress_items", {}) or {}).items(): - if meta is item: - self._set_progress_row(fname, status=status, status_color=color, progress="✓") - break - except Exception: - pass - - def mark_file_complete(self, filename, success=True, error_message=None, completion_kind="normal"): - """Mark a file as complete or failed.""" - if filename in self.file_progress_items: - item = self.file_progress_items[filename] - # If we have detailed result labels already set (via append_log), - # show them and hide the progress bar to make room. - if success: - item['checkbox'].setChecked(True) - if item.get('tokens_label') and item['tokens_label'].text(): - item['tokens_label'].setVisible(True) - item['cost_label'].setVisible(True) - item['time_label'].setVisible(True) - self._apply_success_status_icon(item, completion_kind) - try: - item['progress_bar'].setVisible(False) - except Exception: - pass - # Clear the transient progress text so completed rows don't show "x/y" - try: - item['label'].setText("") - item['label'].setStyleSheet("color: #888888; font-size: 11px;") - except Exception: - pass - else: - # No parsed results available - show status in status_label - self._apply_success_status_icon(item, completion_kind) - try: - item['progress_bar'].setVisible(False) - except Exception: - pass - try: - item['label'].setText("") - item['label'].setStyleSheet("color: #888888; font-size: 11px;") - except Exception: - pass - else: - # Check if this is an unsupported file type error - is_unsupported = False - if error_message: - error_lower = error_message.lower() - is_unsupported = ( - " not supported" in error_lower or - "not supported" == error_lower or - "unsupported" in error_lower or - "invalid file" in error_lower or - "wrong file type" in error_lower - ) - - # Mark as failed - checkbox unchecked, red X or warning icon - item['checkbox'].setChecked(False) - try: - if is_unsupported: - item['status_label'].setText("⚠") - item['status_label'].setStyleSheet("color: #f1c40f; font-weight: bold; font-size: 13px;") - item['label'].setText("Not Supported") - self._set_progress_row( - filename, - status="Unsupported", - status_color="#f1c40f", - progress="-", - ) - else: - item['status_label'].setText("✗") - item['status_label'].setStyleSheet("color: #f48771; font-weight: bold; font-size: 11px;") - item['label'].setText("Failed") - self._set_progress_row( - filename, - status="Failed", - status_color="#f48771", - progress="-", - ) - item['status_label'].setVisible(True) - if error_message: - item['status_label'].setToolTip(f"Error: {error_message}") - item['widget'].setToolTip(f"Error: {error_message}") - except Exception: - pass - try: - item['progress_bar'].setVisible(False) - except Exception: - pass - # Always refresh table status for success paths that used the icon helper - if success: - tokens = item.get("tokens_label").text() if item.get("tokens_label") else "" - cost = item.get("cost_label").text() if item.get("cost_label") else "" - time_s = item.get("time_label").text() if item.get("time_label") else "" - self._set_progress_row( - filename, - tokens=tokens or None, - cost=cost or None, - time_s=None if getattr(self, "_batch_active", False) else (time_s or None), - ) - - def reset_to_file_view(self): - """Reset back to file selection view.""" - self._reset_batch_pipeline_ui() - self.file_stack.setCurrentIndex(0) - self.reset_view_button.setVisible(False) - # Also hide the open translations button when returning to file view - try: - self.open_translations_button.setVisible(False) - except Exception: - pass - try: - self.sync_translated_button.setVisible(False) - except Exception: - pass - try: - self.export_active_button.setVisible(False) - except Exception: - pass - # Hide totals when returning to file view - try: - if hasattr(self, 'totals_widget') and self.totals_widget: - self.totals_widget.setVisible(False) - except Exception: - pass - self.translate_button.setVisible(True) - self.stop_button.setVisible(False) - self.refresh_file_lists() - - def start_translation(self, skip_confirm: bool = False, forced_resume_state: str | None = None): - """Start the translation process. - - skip_confirm: when True the confirmation dialog is bypassed (used when - called programmatically from the Workflow tab so the user doesn't need - an extra click to confirm what they just explicitly requested). - forced_resume_state: when set (from the Batch history tab), force Batch - Translate mode and resume with this state without the Resume? dialog. - """ - # Get checked files - selected_files = self.get_selected_files() - - if not selected_files: - QMessageBox.warning(self, "No Files Selected", "Please check at least one file to translate.") - return - - # Get selected module - selected_index = self.module_combo.currentIndex() - if selected_index < 0 or selected_index >= len(self.modules): - QMessageBox.warning(self, "No Module", "Please select a translation module.") - return - - selected_module = self.modules[selected_index] - - # Get mode from dropdown - mode = self.mode_combo.currentText() - if forced_resume_state: - # Batch history Resume always runs Batch Translate. - idx = self.mode_combo.findText(BATCH_MODE_LABEL) - if idx >= 0: - self.mode_combo.setCurrentIndex(idx) - mode = BATCH_MODE_LABEL - estimate_only = (mode == "Estimate") - parse_speakers = (mode == "Parse Speakers") - batch_mode = (mode == BATCH_MODE_LABEL) - batch_resume_state = None - - if batch_mode: - load_dotenv() - sys.path.insert(0, str(self.project_root)) - try: - from util.translation import isClaudeNative, batchRunState - except Exception as e: - QMessageBox.warning(self, "Batch Translate", f"Could not load batch support: {e}") - return - model = os.getenv("model", "") - if not isClaudeNative(model): - QMessageBox.warning( - self, - "Batch Translate", - "Batch Translate requires a Claude model with the API URL unset or " - "pointing at anthropic.com.\n\nChange your model/API settings and try again.", - ) - return - if forced_resume_state: - batch_resume_state = forced_resume_state - elif skip_confirm: - # Workflow auto-start: each phase is an independent batch run. - # Never resume stale queue/results left over from a prior phase. - from util.translation import clearBatchFiles, batchRunState as _brs - prior = _brs() - if prior: - # Log discard so operators notice unpaid queue / in-flight work. - print( - f"[BATCH] Workflow start discarding prior batch state ({prior}).", - flush=True, - ) - clearBatchFiles() - batch_resume_state = None - else: - batch_resume_state = batchRunState() - if batch_resume_state: - if batch_resume_state == "queued": - reply = QMessageBox.question( - self, - "Resume Queued Batch?", - "A previous collect finished but the batch was not submitted.\n" - "The queue is still in log/batch_requests.json.\n\n" - "Resume and submit that queue?\n\n" - "Choosing No discards the queue and re-collects " - "(speaker/name strings bill at live rates again).", - QMessageBox.Yes | QMessageBox.No, - ) - else: - reply = QMessageBox.question( - self, - "Resume Batch?", - f"A previous batch run was interrupted ({batch_resume_state}).\n\n" - "Resume it instead of re-collecting?\n\n" - "Re-collecting discards the current run and can bill again " - "(live collect charges + a new batch submission).", - QMessageBox.Yes | QMessageBox.No, - ) - if reply != QMessageBox.Yes: - batch_resume_state = None - - # Confirm start (skipped when called programmatically from the Workflow tab - # or when Batch history already confirmed Resume). - if not skip_confirm and not forced_resume_state: - if batch_mode and not batch_resume_state: - reply = QMessageBox.question( - self, - "Start Batch Translate", - f"Start batch translation for {len(selected_files)} file(s) using {selected_module[0]}?\n\n" - "Pass 1 collects dialogue for the batch; you confirm the estimate, then Anthropic " - "processes it (50% off). Pass 2 writes translated files.\n\n" - f"⚠ {BATCH_COLLECT_LIVE_CHARGE_NOTE}", - QMessageBox.Yes | QMessageBox.No, - ) - else: - action = mode.lower() - reply = QMessageBox.question( - self, - f"Start {mode}", - f"Start {action} for {len(selected_files)} file(s) using {selected_module[0]}?", - QMessageBox.Yes | QMessageBox.No, - ) - if reply != QMessageBox.Yes: - return - - if True: - # Switch to progress view - self.file_stack.setCurrentIndex(1) - - # Initialize Files-tab table with all selected files - self.progress_list.clear() - self.file_progress_items.clear() - if hasattr(self, "progress_table"): - self.progress_table.setRowCount(0) - if hasattr(self, "progress_files_summary"): - self.progress_files_summary.setText("No files in this run.") - - for filename in selected_files: - self.create_progress_item(filename) - - # Toggle button visibility - self.translate_button.setVisible(False) - if batch_mode: - # Shown during collect; hidden once collection finishes (see _on_batch_phase) - self.stop_button.setVisible(batch_resume_state is None) - else: - self.stop_button.setVisible(True) - # Show totals footer and reset totals when starting translation - try: - self.totals_input_tokens = 0 - self.totals_output_tokens = 0 - self.totals_cost = 0.0 - self.totals_time = 0.0 - # Reset seen filenames for this run so totals can be applied anew - try: - self._applied_file_totals.clear() - except Exception: - self._applied_file_totals = set() - if hasattr(self, 'totals_tokens_label'): - self.totals_tokens_label.setText("Tokens: 0 in / 0 out") - if hasattr(self, 'totals_cost_label'): - self.totals_cost_label.setText("Cost: $0.0000") - if hasattr(self, 'totals_time_label'): - self.totals_time_label.setText("Time: 0.0s") - if hasattr(self, 'totals_mismatch_label'): - self.totals_mismatch_label.setText("") - self.totals_mismatch_label.setVisible(False) - self.totals_mismatch_count = 0 - if hasattr(self, 'totals_widget') and self.totals_widget: - self.totals_widget.setVisible(True) - except Exception: - pass - # Hide all post-run buttons while a new run is in progress - for _btn_attr in ( - "open_translations_button", - "reset_view_button", - "sync_translated_button", - "export_active_button", - ): - try: - getattr(self, _btn_attr).setVisible(False) - except Exception: - pass - - # Initialize progress tracking - self.files_completed = 0 - self.files_total = len(selected_files) - self._batch_active = batch_mode - self._batch_ui_phase = None - self._batch_consume_started = False - self._finish_pending = None - if batch_mode: - self._set_progress_view_mode(True, len(selected_files)) - self.batch_overall_bar.setValue(0) - self.batch_pipeline_stack.setCurrentIndex(0) - self.batch_live_status.setText("") - # Resume-from-fetched jumps straight to write; arm the UI before the - # worker starts so early cost lines are not dropped as "pre-consume". - if batch_resume_state == "fetched": - self._on_batch_phase("consume", None) - else: - self._set_progress_view_mode(False, len(selected_files)) - self._batch_active = False - self._batch_ui_phase = None - self._batch_consume_started = False - self.files_translated_label.setText(f"0/{self.files_total}") - self.translating_label.setText("Starting...") - self.item_progress_label.setText("0/0") - self.item_progress_bar.setValue(0) - self.item_progress_bar.setMaximum(100) - - # Remember which files this run covers so the post-run export button - # can export exactly those files rather than all active files. - self._last_run_files = list(selected_files) - - # Create and start translation worker - self.translation_worker = TranslationWorker( - self.project_root, - selected_module, - estimate_only, - selected_files, - parse_speakers=parse_speakers, - batch_mode=batch_mode, - batch_resume_state=batch_resume_state, - ) - - # Connect signals - self.translation_worker.log_signal.connect(self.append_log) - self.translation_worker.progress_signal.connect(self.update_file_progress) - self.translation_worker.item_progress_signal.connect(self.update_item_progress) - self.translation_worker.file_error_signal.connect(self.on_file_error) - self.translation_worker.finished_signal.connect(self.on_translation_finished) - self.translation_worker.batch_phase_signal.connect(self._on_batch_phase) - # Prepare a per-run log file in log/history and start tailing it so - # the right-hand log panel shows only this run's new lines. - try: - history_dir = self.project_root / 'log' / 'history' - history_dir.mkdir(parents=True, exist_ok=True) - - # Clean up old log files, keeping only the 10 most recent - try: - log_files = sorted(history_dir.glob("translationHistory_*.txt"), key=lambda p: p.stat().st_mtime, reverse=True) - # Keep only the 10 most recent, delete the rest - for old_log in log_files[10:]: - try: - old_log.unlink() - except Exception: - pass - except Exception: - pass - - # Use timestamp (safe filename) for sorting - fname = datetime.datetime.now().strftime('translationHistory_%Y%m%d_%H%M%S.txt') - run_log_path = history_dir / fname - # Don't create the file yet - it will be created when first log is written - - # Create the per-run log immediately with a header so the Translation - # Log panel is never blank while a batch resume/consume runs. - try: - mode = BATCH_MODE_LABEL if batch_mode else mode - with open(run_log_path, "a", encoding="utf-8") as f: - f.write( - f"[RUN] {mode}" - + (f" resume={batch_resume_state}" if batch_resume_state else "") - + f" files={len(selected_files)}\n" - ) - for name in selected_files: - f.write(f"[RUN] - {name}\n") - f.flush() - except Exception: - pass - - # Export env var so subprocess workers inherit the path - try: - os.environ['TRANSLATION_RUN_LOG'] = str(run_log_path) - except Exception: - pass - - # Per-game overlays (game.md / quirks / custom): Workflow folder → DAZED_GAME_ROOT - try: - game_root = "" - if self.settings: - game_root = str( - self.settings.value("last_game_folder", "") or "" - ).strip() - if game_root and Path(game_root).is_dir(): - os.environ["DAZED_GAME_ROOT"] = game_root - else: - os.environ.pop("DAZED_GAME_ROOT", None) - except Exception: - pass - - # Try to create a hard link at legacy location so modules that - # still write to log/translationHistory.txt end up in this file. - # This will be created when the run_log_path file is first written to - legacy = self.project_root / 'log' / 'translationHistory.txt' - try: - # Remove any existing legacy file - if legacy.exists(): - try: - legacy.unlink() - except Exception: - pass - except Exception: - pass - - # Clear UI log and start tailing the per-run file (tailer will handle non-existent files) - self.translation_log_viewer.clear_log() - self.translation_log_viewer.start_tail(run_log_path) - except Exception: - # Fallback to legacy file if anything goes wrong - try: - self.translation_log_viewer.clear_log() - self.translation_log_viewer.start_tail(self.project_root / 'log' / 'translationHistory.txt') - except Exception: - pass - - # Start the worker - self.translation_worker.start() - - def append_log(self, message): - # Detect mismatch markers emitted to stdout by translation.py. - # This is the primary, non-racy detection path for subprocess mode. - if isinstance(message, str) and message.startswith("MISMATCH_EVENT:"): - self.on_mismatch_detected() - return # marker is internal, not displayed - # Forward error messages to the log viewer directly. These worker-level - # errors are not written to the log file so the tail won't capture them. - if isinstance(message, str) and '\u274c' in message: - try: - if hasattr(self, 'translation_log_viewer') and self.translation_log_viewer: - self.translation_log_viewer.append_log_message(message) - except Exception: - pass - # During batch collect/poll/submit, per-file cost lines are not final translations. - # Accept costs once consume has started (and after done) - resume→consume is fast - # enough that log lines from the pool thread can arrive before/after the phase - # signal is processed on the UI thread. - if getattr(self, "_batch_active", False): - phase = getattr(self, "_batch_ui_phase", None) - consume_started = getattr(self, "_batch_consume_started", False) - if not consume_started and phase not in ("consume", "done"): - return - try: - stripped = _strip_ansi(message) - pattern = ( - r'^\s*(?P[^:]+):.*?\[Input:\s*(?P\d+)\].*?\[Output:\s*(?P\d+)\]' - r'.*?\[Cost:\s*\$(?P[\d,\.]+)\].*?\[(?P