Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
56f81ef892 | ||
|
|
968c673e58 | ||
|
|
174ff4232f | ||
|
|
c43659005e | ||
|
|
e3cdd1a9ca | ||
|
|
ed31aaf632 | ||
|
|
0820c40b14 | ||
|
|
0ac2810515 | ||
|
|
1c3fd9e4cf | ||
|
|
e96c19ace6 | ||
|
|
4e678c7e13 | ||
|
|
7ee40d6aca | ||
|
|
55a4f958ee | ||
|
|
6c633e9167 | ||
|
|
566158c132 | ||
|
|
bf43d1799d | ||
|
|
7293b4b826 | ||
|
|
34f2d712b3 | ||
|
|
5ef2612df6 | ||
|
|
290c265d5e | ||
|
|
ed418ac11d | ||
|
|
e6abf2e541 | ||
|
|
f5d95547c5 | ||
|
|
795bd8f1aa | ||
|
|
c4e2a5ef0d | ||
|
|
f57d1994cf | ||
|
|
6482a56479 | ||
|
|
8ddfd01dff | ||
|
|
ac551abd55 | ||
|
|
711858ce2c | ||
|
|
5cca6235e1 | ||
|
|
115ab2a0f2 | ||
|
|
bf5dfddee6 | ||
|
|
23a89a618b | ||
|
|
4ad384dc9f | ||
|
|
5ec06e7d49 | ||
|
|
7f75aa8209 | ||
|
|
cd2e5c9150 | ||
|
|
f477703b1a | ||
|
|
6b3125b624 | ||
|
|
bb5c4204de | ||
|
|
43f5589a9d | ||
|
|
10d3f470e7 | ||
|
|
c72c8e777c | ||
|
|
b51e094af1 | ||
|
|
04844997d5 | ||
|
|
7835211a25 | ||
|
|
6bb6fcb406 | ||
|
|
ad8a6e520e | ||
|
|
fe22fe83b0 | ||
|
|
afbd5203fe | ||
|
|
4b175c7cf2 | ||
|
|
334c0654ca | ||
|
|
42b7979585 | ||
|
|
c994c2eeb0 | ||
|
|
e4fe8122ff | ||
|
|
4cef65e70e | ||
|
|
24d5588f09 | ||
|
|
7883b8865b | ||
|
|
47fe6341b4 | ||
|
|
9d2e8bed00 | ||
|
|
eedf347866 | ||
|
|
0d0d3b7871 | ||
|
|
f80c86b00d | ||
|
|
f5d1def330 | ||
|
|
4a28bfb859 | ||
|
|
4adda262c0 | ||
|
|
1d02802172 | ||
|
|
28c0b2c401 | ||
|
|
1e89b566d1 | ||
|
|
ccb350fa08 | ||
|
|
42336145dd | ||
|
|
5d2da7e91c | ||
|
|
e69195e6ef | ||
|
|
1520b971ac | ||
|
|
d426bac87e | ||
|
|
866746c5a1 | ||
|
|
58889d47b0 | ||
|
|
ae148eb591 | ||
|
|
14ed98c316 | ||
|
|
eacbe05934 | ||
|
|
e7fb89f137 | ||
|
|
033c809b95 | ||
|
|
f492cc2506 | ||
|
|
495a50f666 | ||
|
|
8c6e292f4f | ||
|
|
1d8cb8eabd | ||
|
|
a3ee95e83f | ||
|
|
a98f75c1a9 | ||
|
|
f4f3a4c207 | ||
|
|
bcd76de0c2 | ||
|
|
93c556a063 | ||
|
|
343ac29759 | ||
|
|
ce61b2e312 | ||
|
|
3d90bf94ac | ||
|
|
de8cdfa278 | ||
|
|
9e9561988d | ||
|
|
a074b8b0ad | ||
|
|
a2dc28f4d4 | ||
|
|
3b5c2ceb8f | ||
|
|
52a562039e | ||
|
|
6291f89ae9 | ||
|
|
be36796d5d | ||
|
|
219e2fc6ae | ||
|
|
196cd93008 | ||
|
|
4230db4c40 | ||
|
|
6834d223c9 | ||
|
|
67dbb0a8fb | ||
|
|
023884d6d8 | ||
|
|
aa98547456 | ||
|
|
528343f375 | ||
|
|
7fa8b23c4e | ||
|
|
3171066f0c | ||
|
|
c1c93ea00a | ||
|
|
795dee84ad | ||
|
|
eb4575df23 | ||
|
|
f58b254aed | ||
|
|
e073c32e64 | ||
|
|
509477bafa | ||
|
|
867d373eb9 | ||
|
|
f727c5ca70 | ||
|
|
e48df7c106 | ||
|
|
395cd75038 | ||
|
|
e60a1f7e42 | ||
|
|
1fe5a73c98 | ||
|
|
37db948d2f | ||
|
|
63679405a9 | ||
|
|
156fa75c76 | ||
|
|
c7631cc852 | ||
|
|
6f09efd018 | ||
|
|
f1903eb6ac | ||
|
|
977b2a736c | ||
|
|
1f51e3edd3 | ||
|
|
5f5dc85ae0 | ||
|
|
0b15633995 | ||
|
|
8bc51dd270 | ||
|
|
d16c7603ce | ||
|
|
970fc6c9c0 | ||
|
|
d2f1c12de9 | ||
|
|
5d9409d391 | ||
|
|
0195788a54 | ||
|
|
1fd8fe21b7 | ||
|
|
89a7116c87 | ||
|
|
6b85a08ccb | ||
|
|
b9c3fd7c6c | ||
|
|
a35bea9053 | ||
|
|
e2802a2737 | ||
|
|
e86ad435e5 | ||
|
|
f2b42ef00e | ||
|
|
7a8df9b34e | ||
|
|
c9450d2ca5 | ||
|
|
efe0c48877 | ||
|
|
33361275fe | ||
|
|
05fcf00a9c | ||
|
|
ef1063018f | ||
|
|
9b88a1ba55 | ||
|
|
6ede337658 | ||
|
|
0c7a52fd86 | ||
|
|
5babc936b4 | ||
|
|
742a77234f | ||
|
|
f2c558c8f4 | ||
|
|
c540361160 | ||
|
|
26dcf9368b | ||
|
|
95f647fb54 | ||
|
|
d0865015f9 | ||
|
|
3a87fb032f | ||
|
|
6b997d2909 | ||
|
|
c1fb4541dc | ||
|
|
0f025da06b | ||
|
|
60b22c7731 | ||
|
|
f6c813602c | ||
|
|
bc474fecff | ||
|
|
ad91e0ec34 | ||
|
|
17936a868e | ||
|
|
d08ab54d2b | ||
|
|
bc4ae106c1 | ||
|
|
ec7656efb3 | ||
|
|
8fac957f1b | ||
|
|
c848ed473b | ||
|
|
9ddd30fe1f | ||
|
|
627240ad00 | ||
|
|
2c7fc8a6ee | ||
|
|
b70c4ba57e | ||
|
|
89d5fb8116 | ||
|
|
d736242016 | ||
|
|
dfb9256abe | ||
|
|
2f5780f40e | ||
|
|
010274e703 | ||
|
|
ee2999f2cd | ||
|
|
26d4481ee1 | ||
|
|
c5174d2c0a | ||
|
|
bbabc9feed | ||
|
|
507aedeae1 | ||
|
|
c5faab4bd7 | ||
|
|
3278447d6a | ||
|
|
a0e59564af | ||
|
|
6e96fb6405 | ||
|
|
f1cb92900c | ||
|
|
065eaa4745 | ||
|
|
082bb08e48 | ||
|
|
f286c4c6dc | ||
|
|
669ef26667 | ||
|
|
fd10aab6f3 | ||
|
|
0d7e746516 | ||
|
|
2cb2f3dfb1 | ||
|
|
578da66f34 | ||
|
|
814c171a90 | ||
|
|
b3b420beec | ||
|
|
93be50eda1 | ||
|
|
9c44001601 | ||
|
|
e03336836b | ||
|
|
b75277a3cb | ||
|
|
f611512747 | ||
|
|
accb3efc54 | ||
|
|
f415c0d4d3 | ||
|
|
7fff2f154c | ||
|
|
85481ad119 | ||
|
|
143c9d9b30 | ||
|
|
5cfb8b4be7 | ||
|
|
1b5b8e0e31 | ||
|
|
11caeedef5 | ||
|
|
9c997c9616 | ||
|
|
ac8907041a | ||
|
|
22086793ae | ||
|
|
101012d9aa | ||
|
|
cab0a7631f | ||
|
|
375a185aef | ||
|
|
a691b9af58 | ||
|
|
311cdaeae1 | ||
|
|
4d0aabc13e | ||
|
|
b644da8556 | ||
|
|
fa6e3fce1a |
@@ -0,0 +1,63 @@
|
||||
name: Build multi-arch Docker Image
|
||||
|
||||
on:
|
||||
# Build and push
|
||||
#release:
|
||||
# types: [published]
|
||||
# Build only
|
||||
#push: it
|
||||
# branches: [main]
|
||||
# TODO - enable build on pull requests if build times can be reduced
|
||||
# pull_request:
|
||||
workflow_dispatch:
|
||||
|
||||
env:
|
||||
IMAGE_REPOSITORY: ghcr.io/denizsafak/abogen
|
||||
|
||||
jobs:
|
||||
build:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Login to Github Container Registry
|
||||
# Only if we need to push an image
|
||||
# if: ${{ github.event_name == 'release' && github.event.action == 'published' }}
|
||||
uses: docker/login-action@v3
|
||||
with:
|
||||
registry: ghcr.io
|
||||
username: ${{ github.actor }}
|
||||
password: ${{ secrets.GITHUB_TOKEN }}
|
||||
|
||||
# Setup for buildx
|
||||
- name: Set up QEMU
|
||||
uses: docker/setup-qemu-action@v3
|
||||
|
||||
- name: Set up Docker Buildx
|
||||
id: buildx
|
||||
uses: docker/setup-buildx-action@v3
|
||||
|
||||
# Debugging information
|
||||
- name: Docker info
|
||||
run: docker info
|
||||
|
||||
- name: Buildx inspect
|
||||
run: docker buildx inspect
|
||||
|
||||
# Build and (optionally) push the image
|
||||
- name: Build image
|
||||
uses: docker/build-push-action@v6
|
||||
with:
|
||||
context: ./abogen
|
||||
file: ./abogen/Dockerfile
|
||||
# platforms: linux/amd64,linux/arm/v7,linux/arm64,linux/ppc64le,linux/s390x
|
||||
# platforms: linux/amd64,linux/arm64
|
||||
platforms: linux/amd64 # using the solution mentioned in https://github.com/denizsafak/abogen/issues/46
|
||||
# Only push if we are publishing a release
|
||||
# push: ${{ github.event_name == 'release' && github.event.action == 'published' }}
|
||||
push: true
|
||||
# Use a 'temp' tag, that won't be pushed, for non-release builds
|
||||
tags: ${{ env.IMAGE_REPOSITORY }}:${{ github.event.release.tag_name || 'latest' }}
|
||||
# Use a cache to reduce build times
|
||||
cache-to: type=gha,mode=max
|
||||
cache-from: type=gha
|
||||
@@ -0,0 +1 @@
|
||||
3.12
|
||||
@@ -1,5 +1,207 @@
|
||||
# 1.2.5
|
||||
- Added new option: `Override item settings with current selection` in the queue manager. When enabled, all items in the queue will be processed using the current global settings selected in the main GUI, overriding their individual settings. When disabled, each item will retain its own specific settings.
|
||||
- Fixed `Error "Could not load the Qt platform plugin "xcb"` error that occurred in some Linux distributions due to missing `libxcb-cursor0` library by conditionally loading the bundled library when the system version is unavailable, issue mentioned by @bmcgonag in #101.
|
||||
- Fixed the `No module named pip` error that occurred for users who installed Abogen via the [**uv**](https://github.com/astral-sh/uv) installer.
|
||||
- Fixed defaults for `replace_single_newlines` not being applied correctly in some cases.
|
||||
- Fixed `Save chapters separately for queued epubs is ignored`, issue mentioned by @dymas-cz in #109.
|
||||
- Fixed incorrect sentence segmentation when using spaCy, where text would erroneously split after opening parentheses.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.2.4
|
||||
- **Subtitle generation is now available for all languages!** Abogen now supports subtitle generation for non-English languages using audio duration-based timing. Available modes include `Line`, `Sentence`, and `Sentence + Comma`. (Note: Word-level subtitle modes remain English-only due to Kokoro's timestamp token limitations.)
|
||||
- New option: **"Use spaCy for sentence segmentation"** You can now use [spaCy](https://spacy.io/) to automatically detect sentence boundaries and produce cleaner, more readable subtitles. Quick summary:
|
||||
- **What it does:** Splits text into natural sentences so subtitle entries read better and align more naturally with speech.
|
||||
- **Why this helps:** The previous punctuation-based splitting could break sentences incorrectly at common abbreviations (e.g. "Mr.", "Dr.", "Prof.") or initials, producing wrong subtitle breaks. spaCy avoids those false splits by using linguistic rules to detect real sentence boundaries.
|
||||
- **For Non-English:** spaCy runs **before** audio generation to create better sentence chunks for TTS.
|
||||
- **For English:** spaCy runs **during** subtitle generation to find accurate sentence breaks after TTS.
|
||||
- **Note:** spaCy segmentation is only applied when subtitle mode is `Sentence` or `Sentence + Comma`. When turned off, it falls back to simple punctuation-based splitting.
|
||||
- New option: **Pre-download models and voices for offline use** You can now pre-download all required Kokoro models, voices, and spaCy language models using this option in the settings menu. Allowing you to use Abogen completely offline without any internet connection.
|
||||
- Added support for `.` separator in timestamps (e.g. `HH:MM:SS.ms`) for timestamp-based text files.
|
||||
- Optimized regex compilation and eliminated busy-wait loops.
|
||||
- Possibly fixed `Silent truncation of long paragraphs` issue mentioned in [#91](https://github.com/denizsafak/abogen/issues/91) by [@xklzlxr](https://github.com/xklzlxr)
|
||||
- Fixed unused regex patterns and variable naming conventions.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.2.3
|
||||
- Same as 1.2.2, re-released to fix an issue with subtitle timing when using timestamp-based text files.
|
||||
|
||||
# 1.2.2
|
||||
- **You can now voice your subtitle files!** Simply add `.srt`, `.ass` or `.vtt` files to generate timed audio. Alternatively, add a text file with timestamps in `HH:MM:SS` or `HH:MM:SS,ms` format to generate audio that matches the timestamps. See [here](https://github.com/denizsafak/abogen?tab=readme-ov-file#about-timestamp-based-text-files) for detailed instructions.
|
||||
- New option: **"Use silent gaps between subtitles"**: Prevents unnecessary audio speed-up by letting speech continue into the silent gaps between subtitles.
|
||||
- New option: **"Subtitle speed adjustment method"**: Choose how to speed up audio when needed:
|
||||
- **TTS Regeneration (better quality):** Re-generates the audio at a faster speed for more natural sound.
|
||||
- **FFmpeg Time-stretch (better speed):** Quickly speeds up the generated audio.
|
||||
- Added support for embedding cover images in M4B files. Abogen now automatically extracts cover images from EPUB and PDF files. You can also manually specify a cover image using the `<<METADATA_COVER_PATH:path>>` tag in your text file. (To prevent MPV from showing the cover image, you can add `audio-display=no` to your MPV config file.)
|
||||
- Fixed `[WinError 1114] A dynamic link library (DLL) initialization routine failed` error on Windows, pre-loading PyTorch DLLs before initializing PyQt6 to avoid DLL initialization errors, mentioned in #98 by @ephr0n.
|
||||
- Potential fix for `CUDA GPU is not available` issue, by ensuring PyTorch is installed correctly with CUDA support on Windows using the installer script.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.2.1
|
||||
- Upgraded Abogen's interface from PyQt5 to PyQt6 for better compatibility and long-term support.
|
||||
- Added tooltip indicators in queue manager to display book handler options (`Save chapters separately` and `Merge chapters at the end`) for queued items.
|
||||
- Added `Open processed file` and `Open input file` options for items in the queue manager, instead of just `Open file` option.
|
||||
- Added loading gif animation to book handler window.
|
||||
- Fixed light theme slider colors in voice mixer for better visibility (for non-Windows users).
|
||||
- Fixed subtitle word-count splitting logic for more accurate segmentation.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.2.0
|
||||
- Added `Line` option to subtitle generation modes, allowing subtitles to be generated based on line breaks in the text, by @mleg in #94.
|
||||
- Added a loading indicator to the book handler window for better user experience during book preprocessing.
|
||||
- Fixed `cannot access local variable 'is_narrow'` error when subtitle format `SRT` was selected, mentioned by @Kinasa0096 in #88.
|
||||
- Fixed folder and filename sanitization to properly handle OS-specific illegal characters (Windows, Linux, macOS), ensuring compatibility across all platforms when creating chapter folders and files.
|
||||
- Fixed `/` and `\` path display by normalizing paths.
|
||||
- Fixed book reprocessing issue where books were being processed every time the chapters window was opened, improving performance when reopening the same book.
|
||||
- Fixed taskbar icon not appearing correctly in Windows.
|
||||
- Fixed “Go to folder” button not opening the chapter output directory when only separate chapters were generated.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.1.9
|
||||
- Fixed the issue where spaces were deleted before punctuation marks while generating subtitles.
|
||||
- Fixed markdown TOC generation breaks when "Replace single newlines" is enabled.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.1.8
|
||||
- Added `.md` (Markdown) file extension support by @brianxiadong in PR #75
|
||||
- Added new option `Configure silence between chapters` that lets you configure the silence between chapters, mentioned by @lfperez1982 in #79
|
||||
- Better indicators and options while displaying and managing the input and processing files.
|
||||
- Improved the markdown logic to better handle various markdown structures and cases.
|
||||
- Fixed subtitle splitting before commas by combining punctuation with preceding words.
|
||||
- Fixed save options not working correctly in queue mode, mentioned by @jborza in #78
|
||||
- Fixed `No Qt platform plugin could be initialized` error, mentioned by @sunrainxyz in #59
|
||||
- Fixed ordered list numbers not being included in EPUB content conversion. The numbers are now properly included in the converted content, mentioned by @jefro108 in #47
|
||||
- Potentially fixed subtitle generation stucks at 9:59:59, mentioned by @bolaykim in #73
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.1.7
|
||||
- Added MPS GPU acceleration support for Silicon Mac, mentioned in https://github.com/denizsafak/abogen/issues/32#issuecomment-3155902040 by @jefro108. **Please read the [Mac](https://github.com/denizsafak/abogen?tab=readme-ov-file#mac) section in the documentation again, as it requires additional configuration.**
|
||||
- Added word-by-word karaoke highlighting feature by @robmckinnon in PR #65
|
||||
- Fixed sleep inhibition error occurring on some Linux systems that do not use systemd, mentioned in #67 by @hendrack
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.1.6
|
||||
- Improved EPUB chapter detection: Now reliably detects chapters from NAV HTML (TOC) files, even in non-standard EPUBs, fixes the issue mentioned by @jefro108 in #33
|
||||
- Fixed SRT subtitle numbering issue, mentioned by @page-muncher in #41
|
||||
- Fixed missing chapter contents issue in some EPUB files.
|
||||
- Windows installer script now prompts the user to install the CUDA version of PyTorch even if no NVIDIA GPU is detected.
|
||||
- Abogen now includes Mandarin Chinese (misaki[zh]) by default; manual installation is no longer required.
|
||||
|
||||
# 1.1.5
|
||||
- Changed the temporary directory path to user's cache directory, which is more appropriate for storing cache files and avoids issues with unintended cleanup.
|
||||
- Fixed the isssue where extra metadata information was not being saved to M4B files when they have no chapters, ensuring that all metadata is correctly written to the output file.
|
||||
- Fixed sleep prevention process not ending if program exited using Ctrl+C or kill.
|
||||
- Improved automatic filename suffixing to better prevent overwriting files with the same name, even if they have different extensions.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# 1.1.4
|
||||
- Fixed extra metadata information not being saved to M4B files, ensuring that all metadata is correctly written to the output file.
|
||||
- Reformatted the code using Black for better readability and consistency.
|
||||
|
||||
# 1.1.3
|
||||
- `M4B (with chapters)` generation is faster now, as it directly generates `m4b` files instead of converting from `wav`, which significantly reduces processing time, fixes the issue mentioned by @Milor123 in #39
|
||||
- Better sleep state handling for Linux.
|
||||
- The app window now tries to fit the screen if its height would exceed the available display area.
|
||||
- Fixed issue where the app would not restart properly on Windows.
|
||||
- Fixed last sentence/subtitle entry timing in generated subtitles, the end time of the final subtitle entry now correctly matches the end of the audio chunk, preventing zero or invalid timings at the end.
|
||||
|
||||
# v1.1.2
|
||||
- Now you can play the audio files while they are processing.
|
||||
- Audio and subtitle files are now written directly to disk during generation, which significantly reduces memory usage.
|
||||
- Added a better logic for detecting chapters from the epub, mentioned by @jefro108 in #33
|
||||
- Added a new option: `Reset to default settings`, allowing users to reset all settings to their default values.
|
||||
- Added a new option: `Disable Kokoro's internet access`. This lets you prevent Kokoro from downloading models or voices from HuggingFace Hub, which can help avoid long waiting times if your computer is offline.
|
||||
- HuggingFace Hub telemetry is now disabled by default for improved privacy. (HuggingFace Hub is used by Kokoro to download its models)
|
||||
- cPotential fix for #37 and #38, where the program was becoming slow while processing large files.
|
||||
- Fixed `Open folder` and `Open file` buttons in the queue manager GUI.
|
||||
- Improvements in code structure.
|
||||
|
||||
# v1.1.1
|
||||
- Fixed adding wrong file in queue for EPUB and PDF files, ensuring the correct file is added to the queue.
|
||||
- Reformatted the code using Black.
|
||||
|
||||
# v1.1.0
|
||||
- Added queue system for processing multiple items, allowing users to add multiple files and process them in a queue, mentioned by @jborza in #30 (Special thanks to @jborza for implementing this feature in PR #35)
|
||||
- Added a feature that allows selecting multiple items in book handler (in right click menu) by @jborza in #31, that fixes #28
|
||||
- Added dark theme support, allowing users to switch between light and dark themes in the settings.
|
||||
- Added auto-accept system to the chapter options dialog in conversion process, allowing the dialog to auto-accept after a certain time if no action is taken.
|
||||
- Added new option: `Configure max lines in log window` that allows configuring the maximum number of lines to display in the log window.
|
||||
- Improvements in documentation and code.
|
||||
|
||||
# v1.0.9
|
||||
- Added chunking/segmenting system that fixes memory outage issues when processing large audio files.
|
||||
- Added new option: `Subtitle format`, allowing users to choose between `srt` , `ass (wide)`, `ass (narrow)`, and `ass (centered wide)` and `ass (centered narrow)`
|
||||
- Improved chapter filename generation with smart word-boundary truncation at 80 characters, preventing mid-word cuts in filenames.
|
||||
- `Composer` and `Genre` metadata fields for M4B files are now editable from the text editor.
|
||||
- Improvements in documentation and code.
|
||||
|
||||
# v1.0.8
|
||||
- Added support for AMD GPUs in Linux (Special thanks to @hg000125 for his contribution in #23)
|
||||
- Added voice preview caching system that stores generated previews in the cache folder, mentioned by @jborza in #22
|
||||
- Added extra metadata support for chaptered M4B files, ensuring better compatibility with audiobook players.
|
||||
- Added new option: `Separate chapters audio format`, allowing to choose between `wav`, `mp4`, `flac` and `opus` formats for chaptered audio files.
|
||||
- Added a download tracker that displays informative messages while downloading Kokoro models or voices from HuggingFace.
|
||||
- Skipping PyTorch CUDA installation if GPU is not NVIDIA in WINDOWS_INSTALL.bat script, preventing unnecessary installation of PyTorch.
|
||||
- Removed `abogen_` prefix that was adding to converted books in temp directory.
|
||||
- Fixed voice preview player keeps playing silently at the background after preview ends.
|
||||
- Fixed not writing separate chapters audio when output is OPUS.
|
||||
- Improved input box background color handling, fixed display issues in Linux.
|
||||
- Updated profile and voice mixer icons, better visibility and aesthetics in voice mixer.
|
||||
- Better sleep state handling for Linux.
|
||||
- Improvements in documentation and code.
|
||||
|
||||
# v1.0.7
|
||||
- Improve chaptered audio generation by outputting directly as `m4b` instead of converting from `wav`.
|
||||
- Ignore chapter markers and single newlines when calculating text length, improving the accuracy of the text length calculation.
|
||||
- Prevent cancellation if process is at 99%, ensuring the process is not interrupted at the last moment.
|
||||
- Improved process handling for subpprocess calls, ensuring better management of subprocesses.
|
||||
- Improved PDF handling, ignoring empty pages/chapters and better chapter handling.
|
||||
- Added `Save in a project folder with metadata` option in the book handler, allowing users to save the converted items in a project folder with available metadata files. Useful if you want to work with the converted files in the future, issue mentioned by @Darthagnon in #15
|
||||
- Added `Go to folder` button in input box, allowing users to open the folder containing the converted file.
|
||||
- Added `.opus` as output format for generated audio files, which is a more efficient format for audio files.
|
||||
- Added `Create desktop shortcut and install` option to Linux version, allowing users to create a shortcut and install
|
||||
- Added "Playing..." indicator for "Preview" button in the voice mixer.
|
||||
|
||||
# v1.0.6
|
||||
- Added `Insert chapter marker` button in text editor to insert chapter markers at the current cursor position.
|
||||
- Added `Preview` button in voice mixer to preview the voice mix with the selected settings.
|
||||
- Fixed `f-string: unmatched '['` error in Voice preview, mentioned in #14
|
||||
- Fixed the issue with the content before first chapter not being included in the output.
|
||||
- Fixed m4b chapter generation opens CMD window in Windows.
|
||||
|
||||
# v1.0.5
|
||||
- Added new output format: `m4b`, enabling chapter metadata in audiobooks. Special thanks to @jborza for implementing this feature in PR #10.
|
||||
- Better approach for determining the correct configuration folder for Linux and MacOS, using platformdirs. (Fixes Docker issue #12)
|
||||
- Improvements in documentation and code.
|
||||
|
||||
# v1.0.4
|
||||
- Merge pull request [#7](https://github.com/denizsafak/abogen/pull/7) by @jborza that improves voice preview and documentation.
|
||||
- Fixed the issue when a voice is selected, the voice mixer tries to pre-select that voice and ignores existing profiles.
|
||||
- Fixed the error while renaming the default "New profile" in the voice mixer.
|
||||
- Fixed subtitle_combo enabling/disabling when a voice in the voice mixer is selected.
|
||||
- Prevented using special characters in the profile name to avoid conflicts.
|
||||
- Improved invalid profile handling in the voice mixer.
|
||||
|
||||
# v1.0.3
|
||||
- Added voice mixing, allowing multiple voices to be combined into a single “Mixed Voice”, a feature mentioned by @PulsarFTW in #1. Special thanks to @jborza for making this possible through his contributions in #5.
|
||||
- Added profile system to voice mixer, allowing users to create and manage multiple voice profiles.
|
||||
- Improvements in the voice mixer, mostly for organizing controls and enhancing user experience.
|
||||
- Added icons for flags and genders in the GUI, making it easier to identify different options.
|
||||
- Improved the content and chapter extraction process for EPUB files, ensuring better handling of various structures.
|
||||
- Switched to platformdirs for determining the correct desktop path, instead of using old methods.
|
||||
- Fixed preview voices was not using GPU acceleration, which was causing performance issues.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# v1.0.2
|
||||
- Enhanced EPUB handling by treating all items in chapter list (including anchors) as chapters, improving navigation and organization for poorly structured books, mentioned by @Darthagnon in #4
|
||||
- Fixed the issue with some chapters in EPUB files had missing content.
|
||||
- Fixed the issue with some EPUB files only having one chapter caused the program to ignore the entire book.
|
||||
- Fixed "utf-8' codec can't decode byte" error, mentioned by @nigelp in #3
|
||||
- Added "Replace single newlines with spaces" option in the menu. This can be useful for texts that have imaginary line breaks.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# v1.0.1
|
||||
- Added abogen-cli command for better troubleshooting and error handling.
|
||||
- Switched from setuptools to hatchling for packaging.
|
||||
- Added classifiers to the package metadata.
|
||||
- Fixed "No module named 'docopt'" and "setuptools.build_meta" import errors while using .bat installer in Windows, mentioned by @nigelp in #2
|
||||
- Improvements in code and documentation.
|
||||
|
||||
@@ -2,70 +2,178 @@
|
||||
|
||||
[](https://github.com/denizsafak/abogen/actions)
|
||||
[](https://github.com/denizsafak/abogen/releases/latest)
|
||||

|
||||
[](https://pypi.org/project/abogen/)
|
||||
[](https://github.com/denizsafak/abogen/releases/latest)
|
||||
[](https://github.com/psf/black)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
|
||||
Abogen is a powerful text-to-speech conversion tool that makes it easy to turn ePub, PDF, or text files into high-quality audio with matching subtitles in seconds. Use it for audiobooks, voiceovers for Instagram, YouTube, TikTok, or any project that needs natural-sounding text-to-speech, using [Kokoro-82M](https://huggingface.co/hexgrad/Kokoro-82M).
|
||||
<a href="https://trendshift.io/repositories/14433" target="_blank"><img src="https://trendshift.io/api/badge/repositories/14433" alt="denizsafak%2Fabogen | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
|
||||
|
||||
Abogen is a powerful text-to-speech conversion tool that makes it easy to turn ePub, PDF, text, markdown, or subtitle files into high-quality audio with matching subtitles in seconds. Use it for audiobooks, voiceovers for Instagram, YouTube, TikTok, or any project that needs natural-sounding text-to-speech, using [Kokoro-82M](https://huggingface.co/hexgrad/Kokoro-82M).
|
||||
|
||||
<img title="Abogen Main" src='https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/abogen.png' width="380"> <img title="Abogen Processing" src='https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/abogen2.png' width="380">
|
||||
|
||||
## Demo
|
||||
https://github.com/user-attachments/assets/cb66512d-0a52-48c3-bda4-f1e6a03fb8d6
|
||||
|
||||
https://github.com/user-attachments/assets/094ba3df-7d66-494a-bc31-0e4b41d0b865
|
||||
|
||||
> This demo was generated in just 5 seconds, producing ∼1 minute of audio with perfectly synced subtitles. To create a similar video, see [the demo guide](https://github.com/denizsafak/abogen/tree/main/demo).
|
||||
|
||||
## `How to install?`
|
||||
### Windows
|
||||
Go to [espeak-ng latest release](https://github.com/espeak-ng/espeak-ng/releases/latest) download and run the *.msi file.
|
||||
```bash
|
||||
# For NVIDIA GPUs:
|
||||
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
||||
## `How to install?` <a href="https://pypi.org/project/abogen/" target="_blank"><img src="https://img.shields.io/pypi/pyversions/abogen" alt="Abogen Compatible PyPi Python Versions" align="right" style="margin-top:6px;"></a>
|
||||
|
||||
# Install abogen
|
||||
pip install abogen
|
||||
```
|
||||
Alternatively, for an easier setup on Windows:
|
||||
### `Windows`
|
||||
Go to [espeak-ng latest release](https://github.com/espeak-ng/espeak-ng/releases/latest) download and run the *.msi file.
|
||||
|
||||
#### <b>OPTION 1: Install using script</b>
|
||||
1. [Download](https://github.com/denizsafak/abogen/archive/refs/heads/main.zip) the repository
|
||||
2. Extract the ZIP file
|
||||
3. Run `WINDOWS_INSTALL.bat` by double-clicking it
|
||||
|
||||
This method handles everything automatically - installing all dependencies including CUDA in a self-contained environment without requiring a separate Python installation. (You still need to install [espeak-ng](https://github.com/espeak-ng/espeak-ng/releases/latest).)
|
||||
|
||||
### Mac
|
||||
```bash
|
||||
# Install espeak-ng
|
||||
brew install espeak-ng
|
||||
> [!NOTE]
|
||||
> You don't need to install Python separately. The script will install Python automatically.
|
||||
|
||||
# Install abogen
|
||||
pip install abogen # (I have not tested it)
|
||||
#### <b>OPTION 2: Install using uv</b>
|
||||
First, [install uv](https://docs.astral.sh/uv/getting-started/installation/) if you haven't already.
|
||||
|
||||
```bash
|
||||
# For NVIDIA GPUs (CUDA 12.8) - Recommended
|
||||
uv tool install --python 3.12 abogen[cuda]
|
||||
|
||||
# For NVIDIA GPUs (CUDA 12.6) - Older drivers
|
||||
uv tool install --python 3.12 abogen[cuda126]
|
||||
|
||||
# For NVIDIA GPUs (CUDA 13.0) - Newer drivers
|
||||
uv tool install --python 3.12 abogen[cuda130]
|
||||
|
||||
# For AMD GPUs or without GPU (CPU) - ROCm is not available on Windows. Use Linux if you have AMD GPU
|
||||
uv tool install --python 3.12 abogen
|
||||
```
|
||||
### Linux
|
||||
```bash
|
||||
# Install espeak-ng
|
||||
|
||||
# Ubuntu/Debian
|
||||
sudo apt install espeak-ng
|
||||
# Arch Linux
|
||||
sudo pacman -S espeak-ng
|
||||
# Fedora
|
||||
sudo dnf install espeak-ng
|
||||
<details>
|
||||
<summary><b>Alternative: Install using pip (click to expand)</b></summary>
|
||||
|
||||
```bash
|
||||
# Create a virtual environment (optional)
|
||||
mkdir abogen && cd abogen
|
||||
python -m venv venv
|
||||
venv\Scripts\activate
|
||||
|
||||
# For NVIDIA GPUs:
|
||||
# We need to use an older version of PyTorch (2.8.0) until this issue is fixed: https://github.com/pytorch/pytorch/issues/166628
|
||||
pip install torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
|
||||
|
||||
# For AMD GPUs:
|
||||
# Not supported yet, because ROCm is not available on Windows. Use Linux if you have AMD GPU.
|
||||
|
||||
# Install abogen
|
||||
pip install abogen
|
||||
```
|
||||
> If you get "No matching distribution found" error, try installing it on supported Python (3.10 to 3.12). You can use [pyenv](https://github.com/pyenv/pyenv) to manage multiple Python versions easily in Linux. Watch this [video](https://www.youtube.com/watch?v=MVyb-nI4KyI) by NetworkChuck for a quick guide.
|
||||
|
||||
Then simply run by typing:
|
||||
</details>
|
||||
|
||||
### `Mac`
|
||||
|
||||
First, [install uv](https://docs.astral.sh/uv/getting-started/installation/) if you haven't already.
|
||||
|
||||
```bash
|
||||
# Install espeak-ng
|
||||
brew install espeak-ng
|
||||
|
||||
# Install abogen (Automatically handles Silicon Mac/MPS support)
|
||||
uv tool install --python 3.12 abogen
|
||||
```
|
||||
|
||||
<details>
|
||||
<summary><b>Alternative: Install using pip (click to expand)</b></summary>
|
||||
|
||||
```bash
|
||||
# Install espeak-ng
|
||||
brew install espeak-ng
|
||||
|
||||
# Create a virtual environment (recommended)
|
||||
mkdir abogen && cd abogen
|
||||
python3 -m venv venv
|
||||
source venv/bin/activate
|
||||
|
||||
# Install abogen
|
||||
pip3 install abogen
|
||||
|
||||
# For Silicon Mac (M1, M2 etc.)
|
||||
# After installing abogen, we need to install Kokoro's development version which includes MPS support.
|
||||
pip3 install git+https://github.com/hexgrad/kokoro.git
|
||||
```
|
||||
|
||||
</details>
|
||||
|
||||
### `Linux`
|
||||
|
||||
First, [install uv](https://docs.astral.sh/uv/getting-started/installation/) if you haven't already.
|
||||
|
||||
```bash
|
||||
# Install espeak-ng
|
||||
sudo apt install espeak-ng # Ubuntu/Debian
|
||||
sudo pacman -S espeak-ng # Arch Linux
|
||||
sudo dnf install espeak-ng # Fedora
|
||||
|
||||
# For NVIDIA GPUs or without GPU (CPU) - No need to include [CUDA] in here.
|
||||
uv tool install --python 3.12 abogen
|
||||
|
||||
# For AMD GPUs (ROCm 6.4)
|
||||
uv tool install --python 3.12 abogen[rocm]
|
||||
```
|
||||
|
||||
<details>
|
||||
<summary><b>Alternative: Install using pip (click to expand)</b></summary>
|
||||
|
||||
```bash
|
||||
# Install espeak-ng
|
||||
sudo apt install espeak-ng # Ubuntu/Debian
|
||||
sudo pacman -S espeak-ng # Arch Linux
|
||||
sudo dnf install espeak-ng # Fedora
|
||||
|
||||
# Create a virtual environment (recommended)
|
||||
mkdir abogen && cd abogen
|
||||
python3 -m venv venv
|
||||
source venv/bin/activate
|
||||
|
||||
# Install abogen
|
||||
pip3 install abogen
|
||||
|
||||
# For NVIDIA GPUs:
|
||||
# Already supported, no need to install CUDA separately.
|
||||
|
||||
# For AMD GPUs:
|
||||
# After installing abogen, we need to uninstall the existing torch package
|
||||
pip3 uninstall torch
|
||||
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4
|
||||
```
|
||||
</details>
|
||||
|
||||
|
||||
> See [How to fix "CUDA GPU is not available. Using CPU" warning?](#cuda-warning)
|
||||
|
||||
> See [How to fix "WARNING: The script abogen-cli is installed in '/home/username/.local/bin' which is not on PATH" error in Linux?](#path-warning)
|
||||
|
||||
> See [How to fix "No matching distribution found" error?](#no-matching-distribution-found)
|
||||
|
||||
> See [How to fix "[WinError 1114] A dynamic link library (DLL) initialization routine failed" error?](#WinError-1114)
|
||||
|
||||
> Special thanks to [@hg000125](https://github.com/hg000125) for his contribution in [#23](https://github.com/denizsafak/abogen/issues/23). AMD GPU support is possible thanks to his work.
|
||||
|
||||
## `How to run?`
|
||||
|
||||
You can simply run this command to start Abogen:
|
||||
|
||||
```bash
|
||||
abogen
|
||||
```
|
||||
> [!TIP]
|
||||
> If you installed Abogen using the Windows installer `(WINDOWS_INSTALL.bat)`, It should have created a shortcut in the same folder, or your desktop. You can run it from there. If you lost the shortcut, Abogen is located in `python_embedded/Scripts/abogen.exe`. You can run it from there directly.
|
||||
|
||||
## `How to use?`
|
||||
1) Drag and drop any ePub, PDF, or text file (or use the built-in text editor)
|
||||
1) Drag and drop any ePub, PDF, text, markdown, or subtitle file (or use the built-in text editor)
|
||||
2) Configure the settings:
|
||||
- Set speech speed
|
||||
- Select a voice (or create a custom voice using voice mixer)
|
||||
@@ -79,37 +187,84 @@ abogen
|
||||
|
||||
Here’s Abogen in action: in this demo, it processes ∼3,000 characters of text in just 11 seconds and turns it into 3 minutes and 28 seconds of audio, and I have a low-end **RTX 2060 Mobile laptop GPU**. Your results may vary depending on your hardware.
|
||||
|
||||
## `Key Features`
|
||||
- **Supported formats**: `ePub`, `PDF`, or `.TXT` files (or use built-in text editor)
|
||||
- **Speed**: Adjust speech rate from `0.1x` to `2.0x`
|
||||
- **Voices**: First letter of the language code (e.g., `a` for American English, `b` for British English, etc.), second letter is for `m` for male and `f` for female.
|
||||
- **Voice mixer**: Create custom voices by mixing different voice models with a profile system.
|
||||
- **Generate subtitles**: `Disabled`, `Sentence`, `Sentence + Comma`, `1 word`, `2 words`, `3 words`, etc. (Represents the number of words in each subtitle entry)
|
||||
- **Output formats**: `.WAV`, `.FLAC`, `.MP3`, and `M4B (with chapters)` (Special thanks to [@jborza](https://github.com/jborza) for chapter support in PR [#10](https://github.com/denizsafak/abogen/pull/10))
|
||||
- **Save location**: `Save next to input file`, `Save to desktop`, or `Choose output folder`
|
||||
- **Chapter Control**: Select specific `chapters` from ePUBs or `chapters + pages` from PDFs.
|
||||
- **Options**:
|
||||
- **Replace single newlines with spaces**: Replaces single newlines with spaces in the text. This is useful for texts that have imaginary line breaks.
|
||||
- **Configure max words per subtitle**: Configures the maximum number of words per subtitle entry.
|
||||
- **Create desktop shortcut**: Creates a shortcut on your desktop for easy access.
|
||||
- **Open config.json directory**: Opens the directory where the configuration file is stored.
|
||||
- **Open temp directory**: Opens the temporary directory where converted text files are stored.
|
||||
- **Clear all temporary files**: Deletes all temporary files created during the conversion process.
|
||||
- **Check for updates at startup**: Automatically checks for updates when the program starts.
|
||||
- **After conversion**: `Open file`, `Go to folder`, `New conversion`, or `Go back`.
|
||||
## `Configuration`
|
||||
|
||||
| Options | Description |
|
||||
|---------|-------------|
|
||||
| **Input Box** | Drag and drop `ePub`, `PDF`, `.TXT`, `.MD`, `.SRT`, `.ASS` or `.VTT` files (or use built-in text editor) |
|
||||
| **Queue options** | Add multiple files to a queue and process them in batch, with individual settings for each file. See [Queue mode](#queue-mode) for more details. |
|
||||
| **Speed** | Adjust speech rate from `0.1x` to `2.0x` |
|
||||
| **Select Voice** | First letter of the language code (e.g., `a` for American English, `b` for British English, etc.), second letter is for `m` for male and `f` for female. |
|
||||
| **Voice mixer** | Create custom voices by mixing different voice models with a profile system. See [Voice Mixer](#voice-mixer) for more details. |
|
||||
| **Voice preview** | Listen to the selected voice before processing. |
|
||||
| **Generate subtitles** | `Disabled`, `Line`, `Sentence`, `Sentence + Comma`, `Sentence + Highlighting`, `1 word`, `2 words`, `3 words`, etc. (Represents the number of words in each subtitle entry) |
|
||||
| **Output voice format** | `.WAV`, `.FLAC`, `.MP3`, `.OPUS (best compression)` and `M4B (with chapters)` |
|
||||
| **Output subtitle format** | Configures the subtitle format as `SRT (standard)`, `ASS (wide)`, `ASS (narrow)`, `ASS (centered wide)`, or `ASS (centered narrow)`. |
|
||||
| **Replace single newlines with spaces** | Replaces single newlines with spaces in the text. This is useful for texts that have imaginary line breaks. |
|
||||
| **Save location** | `Save next to input file`, `Save to desktop`, or `Choose output folder` |
|
||||
|
||||
> Special thanks to [@brianxiadong](https://github.com/brianxiadong) for adding markdown support in PR [#75](https://github.com/denizsafak/abogen/pull/75)
|
||||
|
||||
> Special thanks to [@jborza](https://github.com/jborza) for chapter support in PR [#10](https://github.com/denizsafak/abogen/pull/10)
|
||||
|
||||
> Special thanks to [@mleg](https://github.com/mleg) for adding `Line` option in subtitle generation in PR [#94](https://github.com/denizsafak/abogen/pull/94)
|
||||
|
||||
| Book handler options | Description |
|
||||
|---------|-------------|
|
||||
| **Chapter Control** | Select specific `chapters` from ePUBs or markdown files or `chapters + pages` from PDFs. |
|
||||
| **Save each chapter separately** | Save each chapter in e-books as a separate audio file. |
|
||||
| **Create a merged version** | Create a single audio file that combines all chapters. (If `Save each chapter separately` is disabled, this option will be the default behavior.) |
|
||||
| **Save in a project folder with metadata** | Save the converted items in a project folder with available metadata files. |
|
||||
|
||||
| Menu options | Description |
|
||||
|---------|-------------|
|
||||
| **Theme** | Change the application's theme using `System`, `Light`, or `Dark` options. |
|
||||
| **Configure max words per subtitle** | Configures the maximum number of words per subtitle entry. |
|
||||
| **Configure silence between chapters** | Configures the duration of silence between chapters (in seconds). |
|
||||
| **Configure max lines in log window** | Configures the maximum number of lines to display in the log window. |
|
||||
| **Separate chapters audio format** | Configures the audio format for separate chapters as `wav`, `flac`, `mp3`, or `opus`. |
|
||||
| **Create desktop shortcut** | Creates a shortcut on your desktop for easy access. |
|
||||
| **Open config directory** | Opens the directory where the configuration file is stored. |
|
||||
| **Open cache directory** | Opens the cache directory where converted text files are stored. |
|
||||
| **Clear cache files** | Deletes cache files created during the conversion or preview. |
|
||||
| **Use silent gaps between subtitles** | Prevents unnecessary audio speed-up by letting speech continue into the silent gaps between subtitle etries. In short, it ignores the end times in subtitle entries and uses the silent space until the beginning of the next subtitle entry. When disabled, it speeds up the audio to fit the exact time interval specified in the subtitle. (for subtitle files). |
|
||||
| **Subtitle speed adjustment method** | Choose how to speed up audio when needed: `TTS Regeneration (better quality)` re-generates the audio at a faster speed, while `FFmpeg Time-stretch (better speed)` quickly speeds up the generated audio. (for subtitle files). |
|
||||
| **Use spaCy for sentence segmentation** | When this option is enabled, Abogen uses [spaCy](https://spacy.io/) to detect sentence boundaries more accurately, instead of using punctuation marks (like periods, question marks, etc.) to split sentences, which could incorrectly cut off phrases like "Mr." or "Dr.". With spaCy, sentences are divided more accurately. For non-English text, spaCy runs **before** audio generation to create sentence chunks. For English text, spaCy runs **during** subtitle generation to improve timing and readability. spaCy is only used when subtitle mode is `Sentence` or `Sentence + Comma`. If you prefer the old punctuation splitting method, you can turn this option off. |
|
||||
| **Pre-download models and voices for offline use** | Opens a window that displays the available models and voices. Click `Download all` button to download all required models and voices, allowing you to use Abogen completely offline without any internet connection. |
|
||||
| **Disable Kokoro's internet access** | Prevents Kokoro from downloading models or voices from HuggingFace Hub, useful for offline use. |
|
||||
| **Check for updates at startup** | Automatically checks for updates when the program starts. |
|
||||
| **Reset to default settings** | Resets all settings to their default values. |
|
||||
|
||||
> Special thanks to [@robmckinnon](https://github.com/robmckinnon) for adding Sentence + Highlighting feature in PR [#65](https://github.com/denizsafak/abogen/pull/65)
|
||||
|
||||
## `Voice Mixer`
|
||||
<img title="Abogen Voice Mixer" src='https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/voice_mixer.png'>
|
||||
|
||||
With voice mixer, you can create custom voices by mixing different voice models. You can adjust the weight of each voice and save your custom voice as a profile for future use. The voice mixer allows you to create unique and personalized voices. (Huge thanks to [@jborza](https://github.com/jborza) for making this possible through his contributions in [#5](https://github.com/denizsafak/abogen/pull/5))
|
||||
With voice mixer, you can create custom voices by mixing different voice models. You can adjust the weight of each voice and save your custom voice as a profile for future use. The voice mixer allows you to create unique and personalized voices.
|
||||
|
||||
> Special thanks to [@jborza](https://github.com/jborza) for making this possible through his contributions in [#5](https://github.com/denizsafak/abogen/pull/5)
|
||||
|
||||
## `Queue Mode`
|
||||
<img title="Abogen queue mode" src='https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/queue.png'>
|
||||
|
||||
Abogen supports **queue mode**, allowing you to add multiple files to a processing queue. This is useful if you want to convert several files in one batch.
|
||||
|
||||
- You can add text files (`.txt`) and subtitle files (`.srt`, `.ass`, `.vtt`) directly using the **Add files** button in the Queue Manager or by dragging and dropping them into the queue list. To add PDF, EPUB, or markdown files, use the input box in the main window and click the **Add to Queue** button.
|
||||
- Each file in the queue keeps the configuration settings that were active when it was added. Changing the main window configuration afterward does **not** affect files already in the queue.
|
||||
- You can enable the **Override item settings with current selection** option to force all items in the queue to use the configuration currently selected in the main window, overriding their saved settings.
|
||||
- You can view each file's configuration by hovering over them.
|
||||
|
||||
Abogen will process each item in the queue automatically, saving outputs as configured.
|
||||
|
||||
> Special thanks to [@jborza](https://github.com/jborza) for adding queue mode in PR [#35](https://github.com/denizsafak/abogen/pull/35)
|
||||
|
||||
## `About Chapter Markers`
|
||||
When you process ePUB or PDF files, abogen converts them into text files stored in your temporary directory. When you click "Edit," you're actually modifying these converted text files. In these text files, you'll notice tags that look like this:
|
||||
When you process ePUB, PDF or markdown files, Abogen converts them into text files stored in your cache directory. When you click "Edit," you're actually modifying these converted text files. In these text files, you'll notice tags that look like this:
|
||||
|
||||
```
|
||||
<<CHAPTER_MARKER:Chapter Title>>
|
||||
```
|
||||
These are chapter markers. They are automatically added when you process ePUB or PDF files, based on the chapters you select. They serve an important purpose:
|
||||
These are chapter markers. They are automatically added when you process ePUB, PDF or markdown files, based on the chapters you select. They serve an important purpose:
|
||||
- Allow you to split the text into separate audio files for each chapter
|
||||
- Save time by letting you reprocess only specific chapters if errors occur, rather than the entire file
|
||||
|
||||
@@ -122,10 +277,44 @@ This is the beginning of my text...
|
||||
<<CHAPTER_MARKER:Main Content>>
|
||||
Here's another part...
|
||||
```
|
||||
When you process the text file, abogen will detect these markers automatically and ask if you want to save each chapter separately and create a merged version.
|
||||
When you process the text file, Abogen will detect these markers automatically and ask if you want to save each chapter separately and create a merged version.
|
||||
|
||||

|
||||
|
||||
## `About Metadata Tags`
|
||||
Similar to chapter markers, it is possible to add metadata tags for `M4B` files. This is useful for audiobook players that support metadata, allowing you to add information like title, author, year, etc. Abogen automatically adds these tags when you process ePUB, PDF or markdown files, but you can also add them manually to your text files. Add metadata tags **at the beginning of your text file** like this:
|
||||
```
|
||||
<<METADATA_TITLE:Title>>
|
||||
<<METADATA_ARTIST:Author>>
|
||||
<<METADATA_ALBUM:Album Title>>
|
||||
<<METADATA_YEAR:Year>>
|
||||
<<METADATA_ALBUM_ARTIST:Album Artist>>
|
||||
<<METADATA_COMPOSER:Narrator>>
|
||||
<<METADATA_GENRE:Audiobook>>
|
||||
<<METADATA_COVER_PATH:path/to/cover.jpg>>
|
||||
```
|
||||
> Note: `METADATA_COVER_PATH` is used to embed a cover image into the generated M4B file. Abogen automatically extracts the cover from EPUB and PDF files and adds this tag for you.
|
||||
|
||||
## `About Timestamp-based Text Files`
|
||||
Similar to converting subtitle files to audio, Abogen can automatically detect text files that contain timestamps in `HH:MM:SS`, `HH:MM:SS,ms` or `HH:MM:SS.ms` format. When timestamps are found inside your text file, Abogen will ask if you want to use them for audio timing. This is useful for creating timed narrations, scripts, or transcripts where you need exact control over when each segment is spoken.
|
||||
|
||||
Format your text file like this:
|
||||
```
|
||||
00:00:00
|
||||
This is the first segment of text.
|
||||
|
||||
00:00:15
|
||||
This is the second segment, starting at 15 seconds.
|
||||
|
||||
00:00:45
|
||||
And this is the third segment, starting at 45 seconds.
|
||||
```
|
||||
|
||||
**Important notes:**
|
||||
- Timestamps must be in `HH:MM:SS`, `HH:MM:SS,ms` or `HH:MM:SS.ms` format (e.g., `00:05:30` for 5 minutes 30 seconds, or `00:05:30.500` for 5 minutes 30.5 seconds)
|
||||
- Milliseconds are optional and provide precision up to 1/1000th of a second
|
||||
- Text before the first timestamp (if any) will automatically start at `00:00:00`
|
||||
- When using timestamps, the subtitle generation mode setting is ignored
|
||||
|
||||
## `Supported Languages`
|
||||
```
|
||||
@@ -140,14 +329,19 @@ When you process the text file, abogen will detect these markers automatically a
|
||||
```
|
||||
For a complete list of supported languages and voices, refer to Kokoro's [VOICES.md](https://huggingface.co/hexgrad/Kokoro-82M/blob/main/VOICES.md). To listen to sample audio outputs, see [SAMPLES.md](https://huggingface.co/hexgrad/Kokoro-82M/blob/main/SAMPLES.md).
|
||||
|
||||
> See [How to fix Japanese audio not working?](#japanese-audio-not-working)
|
||||
|
||||
## `MPV Config`
|
||||
I highly recommend using [MPV](https://mpv.io/installation/) to play your audio files, as it supports displaying subtitles even without a video track. Here's my `mpv.conf`:
|
||||
```
|
||||
# --- MPV Settings ---
|
||||
save-position-on-quit
|
||||
keep-open=yes
|
||||
--audio-device=openal
|
||||
--sub-margin-x=235
|
||||
--sub-pos=60
|
||||
audio-display=no
|
||||
# --- Subtitle ---
|
||||
sub-ass-override=no
|
||||
sub-margin-y=50
|
||||
sub-margin-x=50
|
||||
# --- Audio Quality ---
|
||||
audio-spdif=ac3,dts,eac3,truehd,dts-hd
|
||||
audio-channels=auto
|
||||
@@ -184,12 +378,24 @@ Abogen launches automatically inside the container.
|
||||
- You can access it via a web browser at [http://localhost:5800](http://localhost:5800) or connect to it using a VNC client at `localhost:5900`.
|
||||
- You can use `/shared` directory to share files between your host and the container.
|
||||
- For later use, start it with `docker start abogen` and stop it with `docker stop abogen`.
|
||||
- Pass in `-e WEB_AUDIO="1"` for `docker run` to enable audio.
|
||||
|
||||
Known issues:
|
||||
- Audio preview is not working inside container (ALSA error).
|
||||
- `Open temp directory` and `Open configuration directory` options in settings not working. (Tried pcmanfm, did not work with abogen).
|
||||
- Audio preview is not working inside container (ALSA error) if using a VNC client.
|
||||
- `Open cache directory` and `Open configuration directory` options in settings not working. (Tried pcmanfm, did not work with Abogen).
|
||||
|
||||
(Special thanks to [@geo38](https://www.reddit.com/user/geo38/) from Reddit, who provided the Dockerfile and instructions in [this comment](https://www.reddit.com/r/selfhosted/comments/1k8x1yo/comment/mpe0bz8/).)
|
||||
> Special thanks to [@geo38](https://www.reddit.com/user/geo38/) from Reddit, who provided the Dockerfile and instructions in [this comment](https://www.reddit.com/r/selfhosted/comments/1k8x1yo/comment/mpe0bz8/).
|
||||
|
||||
## `🌐 Web Application`
|
||||
|
||||
A web-based version of Abogen has been developed by [@jeremiahsb](https://github.com/jeremiahsb).
|
||||
|
||||
**Access the repository here:** [jeremiahsb/abogen](https://github.com/jeremiahsb/abogen)
|
||||
|
||||
> [!NOTE]
|
||||
> I intend to merge this implementation into the main repository in the future once existing conflicts are resolved. Until then, please be aware that the web version is maintained independently and may not always be in sync with the latest updates in this repository.
|
||||
|
||||
> Special thanks to [@jeremiahsb](https://github.com/jeremiahsb) for implementing the web app!
|
||||
|
||||
## `Similar Projects`
|
||||
Abogen is a standalone project, but it is inspired by and shares some similarities with other projects. Here are a few:
|
||||
@@ -197,24 +403,141 @@ Abogen is a standalone project, but it is inspired by and shares some similariti
|
||||
- [autiobooks](https://github.com/plusuncold/autiobooks): Automatically convert epubs to audiobooks
|
||||
- [pdf-narrator](https://github.com/mateogon/pdf-narrator): Convert your PDFs and EPUBs into audiobooks effortlessly.
|
||||
- [epub_to_audiobook](https://github.com/p0n1/epub_to_audiobook): EPUB to audiobook converter, optimized for Audiobookshelf
|
||||
- [ebook2audiobook](https://github.com/DrewThomasson/ebook2audiobook): Convert ebooks to audiobooks with chapters and metadata using dynamic AI models and voice cloning
|
||||
|
||||
## `Roadmap`
|
||||
- [ ] Improve PDF support for better text extraction.
|
||||
- [ ] Add OCR scan feature for PDF files using docling/teserract.
|
||||
- [x] Add chapter metadata for .m4a files. (Issue [#9](https://github.com/denizsafak/abogen/issues/9), PR [#10](https://github.com/denizsafak/abogen/pull/10))
|
||||
- [ ] Add support for different languages in GUI.
|
||||
- [x] Add voice formula feature that enables mixing different voice models. (Issue [#1](https://github.com/denizsafak/abogen/issues/1), PR [#5](https://github.com/denizsafak/abogen/pull/5))
|
||||
- [ ] Add support for kokoro-onnx.
|
||||
- [ ] Add dark mode.
|
||||
- [ ] Add support for kokoro-onnx (If it's necessary).
|
||||
- [x] Add dark mode.
|
||||
|
||||
## `Troubleshooting`
|
||||
If you encounter any issues while running Abogen, try launching it from the command line with:
|
||||
```
|
||||
abogen-cli
|
||||
```
|
||||
|
||||
If you installed using the Windows installer `(WINDOWS_INSTALL.bat)`, go to `python_embedded/Scripts` and run:
|
||||
```
|
||||
abogen-cli.exe
|
||||
```
|
||||
|
||||
This will start Abogen in command-line mode and display detailed error messages. Please open a new issue on the [Issues](https://github.com/denizsafak/abogen/issues) page with the error message and a description of your problem.
|
||||
|
||||
## `Common Issues & Solutions`
|
||||
|
||||
<details><summary><b>
|
||||
<a name="about-abogen">About the name "abogen"</a>
|
||||
</b></summary>
|
||||
|
||||
> The name **"abogen"** comes from a shortened form of **"audiobook generator"**, which is the purpose of this project.
|
||||
>
|
||||
> After releasing the project, I learned from [community feedback](https://news.ycombinator.com/item?id=44853064#44857237) that the prefix *"abo"* can unfortunately be understood as an ethnic slur in certain regions (particularly Australia and New Zealand). This was something I was not aware of when naming the project, as English is not my first language.
|
||||
>
|
||||
> I want to make it clear that the name was chosen only for its technical meaning, with **no offensive intent**. I’m grateful to those who kindly pointed this out, as it helps ensure the project remains respectful and welcoming to everyone.
|
||||
|
||||
</details>
|
||||
|
||||
<details><summary><b>
|
||||
<a name="cuda-warning">How to fix "CUDA GPU is not available. Using CPU" warning?</a>
|
||||
</b></summary>
|
||||
|
||||
> This message means PyTorch could not use your GPU and has fallen back to the CPU. On Windows, Abogen only supports NVIDIA GPUs with CUDA. AMD GPUs are not supported on Windows (they are only supported on Linux with ROCm). Abogen will still work on the CPU, but processing will be slower compared to a supported GPU.
|
||||
>
|
||||
> If you have a compatible NVIDIA GPU on Windows and still see this warning:
|
||||
> Open your terminal in the Abogen folder (the folder that contains `python_embedded`) and type:
|
||||
> ```bash
|
||||
> python_embedded\python.exe -m pip install --force-reinstall torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
|
||||
> ```
|
||||
>
|
||||
> If this does not resolve the issue and you are using an older NVIDIA GPU that does not support CUDA 12.8, you can try installing an older version of PyTorch that supports your GPU. For example, for CUDA 12.6, run:
|
||||
> ```bash
|
||||
> python_embedded\python.exe -m pip install --force-reinstall torch==2.8.0+cu126 torchvision==0.23.0+cu126 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu126
|
||||
> ```
|
||||
>
|
||||
> If you have an AMD GPU, you need to use Linux and follow the Linux/ROCm [instructions](#linux). If you want to keep running on CPU, no action is required, but performance will just be reduced. See [#32](https://github.com/denizsafak/abogen/issues/32) for more details.
|
||||
>
|
||||
> If you used `uv` to install Abogen, you can uninstall and try reinstalling with another CUDA version:
|
||||
> ```bash
|
||||
> uv tool uninstall abogen
|
||||
> # First, try CUDA 13.0 for newer drivers
|
||||
> uv tool install --python 3.12 abogen[cuda130]
|
||||
> # If that doesn't work, try CUDA 12.6 for older drivers
|
||||
> uv tool install --python 3.12 abogen[cuda126]
|
||||
> ```
|
||||
|
||||
</details>
|
||||
|
||||
<details><summary><b>
|
||||
<a name="path-warning">How to fix "WARNING: The script abogen-cli is installed in '/home/username/.local/bin' which is not on PATH" error in Linux?</a>
|
||||
</b></summary>
|
||||
|
||||
> Run the following command to add Abogen to your PATH:
|
||||
> ```bash
|
||||
> echo "export PATH=\"/home/$USER/.local/bin:\$PATH\"" >> ~/.bashrc && source ~/.bashrc
|
||||
> ```
|
||||
|
||||
</details>
|
||||
|
||||
<details><summary><b>
|
||||
<a name="no-matching-distribution-found">How to fix "No matching distribution found" error?<a>
|
||||
</b></summary>
|
||||
|
||||
> Try installing Abogen on supported Python (3.10 to 3.12) versions. I recommend installing with [uv](https://docs.astral.sh/uv/getting-started/installation/). You can also use [pyenv](https://github.com/pyenv/pyenv) to manage multiple Python versions easily on Linux. Watch this [video](https://www.youtube.com/watch?v=MVyb-nI4KyI) by NetworkChuck for a quick guide.
|
||||
|
||||
</details>
|
||||
|
||||
<details><summary><b>
|
||||
<a name="WinError-1114">How to fix "[WinError 1114] A dynamic link library (DLL) initialization routine failed" error?</a>
|
||||
</b></summary>
|
||||
|
||||
> I faced this error when trying to run Abogen in a virtual Windows machine without GPU support. Here's how I fixed it:
|
||||
> If you installed Abogen using the Windows installer `(WINDOWS_INSTALL.bat)`, go to Abogen's folder (that contains `python_embedded`), open your terminal there and run:
|
||||
> ```bash
|
||||
> python_embedded\python.exe -m pip install --force-reinstall torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
|
||||
> ```
|
||||
> If you installed Abogen using pip, open your terminal in the virtual environment and run:
|
||||
> ```bash
|
||||
> pip install torch==2.8.0 torchaudio==2.8.0 torchvision==0.23.0
|
||||
> ```
|
||||
|
||||
</details>
|
||||
|
||||
<details><summary><b>
|
||||
<a name="japanese-audio-not-working">How to fix Japanese audio not working?</a>
|
||||
</b></summary>
|
||||
|
||||
> Japanese audio may require additional configuration.
|
||||
> I'm not sure about the exact solution, but it seems to be related to installing additional dependencies for Japanese support in Kokoro. Please check [#56](https://github.com/denizsafak/abogen/issues/56) for more information.
|
||||
|
||||
</details>
|
||||
|
||||
<details><summary><b>
|
||||
<a name="use-uv-instead-of-pip">How to uninstall Abogen?</a>
|
||||
</b></summary>
|
||||
|
||||
> - From the settings menu, go to `Open configuration directory` and delete the directory.
|
||||
> - From the settings menu, go to `Open cache directory` and delete the directory.
|
||||
> - If you installed Abogen using pip, type:
|
||||
>```bash
|
||||
>pip uninstall abogen # uninstalls abogen
|
||||
>pip cache purge # removes pip cache
|
||||
>```
|
||||
>- If you installed Abogen using uv, type:
|
||||
>```bash
|
||||
>uv tool uninstall abogen # uninstalls abogen
|
||||
>uv cache clear # removes uv cache
|
||||
>```
|
||||
> - If you installed Abogen using the Windows installer (WINDOWS_INSTALL.bat), just remove the folder that contains Abogen. It installs everything inside `python_embedded` folder, no other directories are created.
|
||||
> - If you installed espeak-ng, you need to remove it separately.
|
||||
|
||||
</details>
|
||||
|
||||
## `Contributing`
|
||||
I welcome contributions! If you have ideas for new features, improvements, or bug fixes, please fork the repository and submit a pull request.
|
||||
|
||||
### For developers and contributors
|
||||
If you'd like to modify the code and contribute to development, you can [download the repository](https://github.com/denizsafak/abogen/archive/refs/heads/main.zip), extract it and run the following commands to build **or** install the package:
|
||||
```bash
|
||||
@@ -228,16 +551,20 @@ Feel free to explore the code and make any changes you like.
|
||||
|
||||
## `Credits`
|
||||
- Abogen uses [Kokoro](https://github.com/hexgrad/kokoro) for its high-quality, natural-sounding text-to-speech synthesis. Huge thanks to the Kokoro team for making this possible.
|
||||
- Thanks to the [spaCy](https://spacy.io/) project for its sentence-segmentation tools, which help Abogen produce cleaner, more natural sentence segmentation.
|
||||
- Thanks to [@wojiushixiaobai](https://github.com/wojiushixiaobai) for [Embedded Python](https://github.com/wojiushixiaobai/Python-Embed-Win64) packages. These modified packages include pip pre-installed, enabling Abogen to function as a standalone application without requiring users to separately install Python in Windows.
|
||||
- Thanks to creators of [EbookLib](https://github.com/aerkalov/ebooklib), a Python library for reading and writing ePub files, which is used for extracting text from ePub files.
|
||||
- Special thanks to the [PyQt](https://www.riverbankcomputing.com/software/pyqt/) team for providing the cross-platform GUI toolkit that powers Abogen's interface.
|
||||
- Icons: [US](https://icons8.com/icon/aRiu1GGi6Aoe/usa), [Great Britain](https://icons8.com/icon/t3NE3BsOAQwq/great-britain), [Spain](https://icons8.com/icon/ly7tzANRt33n/spain), [France](https://icons8.com/icon/3muzEmi4dpD5/france), [India](https://icons8.com/icon/esGVrxg9VCJ1/india), [Italy](https://icons8.com/icon/PW8KZnP7qXzO/italy), [Japan](https://icons8.com/icon/McQbrq9qaQye/japan), [Brazil](https://icons8.com/icon/zHmH8HpOmM90/brazil), [China](https://icons8.com/icon/Ej50Oe3crXwF/china), [Female](https://icons8.com/icon/uI49hxbpxTkp/female), [Male](https://icons8.com/icon/12351/male) and [Voice Id](https://icons8.com/icon/GskSeVoroQ7u/voice-id) icons by [Icons8](https://icons8.com/).
|
||||
- Icons: [US](https://icons8.com/icon/aRiu1GGi6Aoe/usa), [Great Britain](https://icons8.com/icon/t3NE3BsOAQwq/great-britain), [Spain](https://icons8.com/icon/ly7tzANRt33n/spain), [France](https://icons8.com/icon/3muzEmi4dpD5/france), [India](https://icons8.com/icon/esGVrxg9VCJ1/india), [Italy](https://icons8.com/icon/PW8KZnP7qXzO/italy), [Japan](https://icons8.com/icon/McQbrq9qaQye/japan), [Brazil](https://icons8.com/icon/zHmH8HpOmM90/brazil), [China](https://icons8.com/icon/Ej50Oe3crXwF/china), [Female](https://icons8.com/icon/uI49hxbpxTkp/female), [Male](https://icons8.com/icon/12351/male), [Adjust](https://icons8.com/icon/21698/adjust) and [Voice Id](https://icons8.com/icon/GskSeVoroQ7u/voice-id) icons by [Icons8](https://icons8.com/).
|
||||
|
||||
## `License`
|
||||
This project is available under the MIT License - see the [LICENSE](https://github.com/denizsafak/abogen/blob/main/LICENSE) file for details.
|
||||
[Kokoro](https://github.com/hexgrad/kokoro) is licensed under [Apache-2.0](https://github.com/hexgrad/kokoro/blob/main/LICENSE) which allows commercial use, modification, distribution, and private use.
|
||||
|
||||
> [!IMPORTANT]
|
||||
> Subtitle generation currently works only for English. This is because Kokoro provides timestamp tokens only for English text. If you want subtitles in other languages, please request this feature in the [Kokoro project](https://github.com/hexgrad/kokoro). For more technical details, see [this line](https://github.com/hexgrad/kokoro/blob/6d87f4ae7abc2d14dbc4b3ef2e5f19852e861ac2/kokoro/pipeline.py#L383) in the Kokoro's code.
|
||||
## `Star History`
|
||||
[](https://www.star-history.com/#denizsafak/abogen&Date)
|
||||
|
||||
> Tags: audiobook, kokoro, text-to-speech, TTS, audiobook generator, audiobooks, text to speech, audiobook maker, audiobook creator, audiobook generator, voice-synthesis, text to audio, text to audio converter, text to speech converter, text to speech generator, text to speech software, text to speech app, epub to audio, pdf to audio, content-creation, media-generation
|
||||
> [!NOTE]
|
||||
> Abogen supports subtitle generation for all languages. However, word-level subtitle modes (e.g., "1 word", "2 words", "3 words", etc.) are only available for English because [Kokoro provides timestamp tokens only for English text](https://github.com/hexgrad/kokoro/blob/6d87f4ae7abc2d14dbc4b3ef2e5f19852e861ac2/kokoro/pipeline.py#L383). For non-English languages, Abogen uses a duration-based fallback that supports sentence-level and comma-based subtitle modes ("Line", "Sentence", "Sentence + Comma"). If you need word-level subtitles for other languages, please request that feature in the [Kokoro project](https://github.com/hexgrad/kokoro).
|
||||
|
||||
> Tags: audiobook, kokoro, text-to-speech, TTS, audiobook generator, audiobooks, text to speech, audiobook maker, audiobook creator, audiobook generator, voice-synthesis, text to audio, text to audio converter, text to speech converter, text to speech generator, text to speech software, text to speech app, epub to audio, pdf to audio, markdown to audio, subtitle to audio, srt to audio, ass to audio, vtt to audio, webvtt to audio, content-creation, media-generation
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
@echo off
|
||||
setlocal
|
||||
setlocal EnableDelayedExpansion
|
||||
cd /d "%~dp0"
|
||||
|
||||
:: Set misaki language
|
||||
@@ -8,9 +8,6 @@ cd /d "%~dp0"
|
||||
:: Japanese: "ja"
|
||||
set MISAKI_LANG=en
|
||||
|
||||
:: Set PyTorch CUDA version
|
||||
set CUDA_VERSION=128
|
||||
|
||||
:::
|
||||
::: _ ____ ___ ____ _____ _ _
|
||||
::: / \ | __ ) / _ \ / ___|| ____| \ | |
|
||||
@@ -23,7 +20,6 @@ set CUDA_VERSION=128
|
||||
for /f "delims=: tokens=*" %%A in ('findstr /b ::: "%~f0"') do @echo(%%A
|
||||
|
||||
set CURRENT_DIR="%CD%"
|
||||
setlocal enabledelayedexpansion
|
||||
set NAME=abogen
|
||||
set PROJECTFOLDER=abogen
|
||||
set RUN=python_embedded\Scripts\abogen.exe
|
||||
@@ -140,10 +136,10 @@ if exist "%VERSION_FILE%" (
|
||||
REM Python embedded download configuration for different architectures
|
||||
if "%PROCESSOR_ARCHITECTURE%"=="x86" (
|
||||
set PYTHON_EMBEDDED_FILE=%PROJECTFOLDER%\python_embedded_win32.zip
|
||||
set PYTHON_EMBEDDED_URL=https://github.com/wojiushixiaobai/Python-Embed-Win64/releases/download/3.12.8/python-3.12.8-embed-win32.zip
|
||||
set PYTHON_EMBEDDED_URL=https://github.com/wojiushixiaobai/Python-Embed-Win64/releases/download/3.12.12/python-3.12.12-embed-win32.zip
|
||||
) else (
|
||||
set PYTHON_EMBEDDED_FILE=%PROJECTFOLDER%\python_embedded_amd64.zip
|
||||
set PYTHON_EMBEDDED_URL=https://github.com/wojiushixiaobai/Python-Embed-Win64/releases/download/3.12.8/python-3.12.8-embed-amd64.zip
|
||||
set PYTHON_EMBEDDED_URL=https://github.com/wojiushixiaobai/Python-Embed-Win64/releases/download/3.12.12/python-3.12.12-embed-amd64.zip
|
||||
)
|
||||
|
||||
:: Check if Python exists
|
||||
@@ -219,20 +215,38 @@ if errorlevel 1 (
|
||||
exit /b
|
||||
)
|
||||
|
||||
:: Install progress's fixed version
|
||||
echo Installing fixed version of progress...
|
||||
%PYTHON_CONSOLE_PATH% -m pip install --force-reinstall https://github.com/denizsafak/abogen/raw/refs/heads/main/abogen/resources/progress-1.6-py3-none-any.whl --no-warn-script-location
|
||||
if errorlevel 1 (
|
||||
echo Failed to install fixed version of progress.
|
||||
pause
|
||||
exit /b
|
||||
)
|
||||
|
||||
:: Install setup requirements
|
||||
echo Installing setup requirements...
|
||||
%PYTHON_CONSOLE_PATH% -m pip install --upgrade setuptools wheel sphinx hatchling progress --no-warn-script-location
|
||||
%PYTHON_CONSOLE_PATH% -m pip install --upgrade setuptools setuptools-scm wheel sphinx hatchling editables --no-warn-script-location
|
||||
if errorlevel 1 (
|
||||
echo Failed to install setup requirements.
|
||||
pause
|
||||
exit /b
|
||||
)
|
||||
|
||||
:: Install gpustat
|
||||
echo Installing gpustat...
|
||||
%PYTHON_CONSOLE_PATH% -m pip install gpustat --no-warn-script-location
|
||||
if errorlevel 1 (
|
||||
echo Failed to install gpustat.
|
||||
pause
|
||||
exit /b
|
||||
)
|
||||
|
||||
:: Install project and dependencies from pyproject.toml
|
||||
echo Checking and installing project dependencies...
|
||||
if exist %PYPROJECT_FILE% (
|
||||
echo Installing project from pyproject.toml...
|
||||
%PYTHON_CONSOLE_PATH% -m pip install . --no-warn-script-location
|
||||
%PYTHON_CONSOLE_PATH% -m pip install -e . --no-warn-script-location
|
||||
if errorlevel 1 (
|
||||
echo Failed to install from pyproject.toml.
|
||||
pause
|
||||
@@ -246,7 +260,7 @@ if exist %PYPROJECT_FILE% (
|
||||
:: Install misaki again if MISAKI_LANG is not set to "en"
|
||||
if "%MISAKI_LANG%" NEQ "en" (
|
||||
echo Configuring language pack: %MISAKI_LANG%
|
||||
%PYTHON_CONSOLE_PATH% -m pip install misaki[lang] --upgrade --no-warn-script-location
|
||||
%PYTHON_CONSOLE_PATH% -m pip install misaki[%MISAKI_LANG%] --upgrade --no-warn-script-location
|
||||
if errorlevel 1 (
|
||||
echo Failed to install misaki language pack.
|
||||
pause
|
||||
@@ -254,21 +268,52 @@ if "%MISAKI_LANG%" NEQ "en" (
|
||||
)
|
||||
)
|
||||
|
||||
:: Check if torch is installed with CUDA support
|
||||
echo Checking CUDA availability...
|
||||
for /f %%i in ('%PYTHON_CONSOLE_PATH% -c "from torch.cuda import is_available; print(is_available())"') do set cuda_available=%%i
|
||||
:: Check for NVIDIA GPU via is_nvidia()
|
||||
for /f %%i in ('%PYTHON_CONSOLE_PATH% -c "from abogen.is_nvidia import check; print(check())"') do set IS_NVIDIA=%%i
|
||||
|
||||
if "%cuda_available%"=="False" (
|
||||
echo Installing PyTorch with CUDA %CUDA_VERSION% support...
|
||||
%PYTHON_CONSOLE_PATH% -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu%CUDA_VERSION% --no-warn-script-location
|
||||
:: Check if torch is installed with CUDA support
|
||||
echo.
|
||||
echo Checking CUDA availability...
|
||||
if /I "%IS_NVIDIA%"=="true" (
|
||||
for /f %%i in ('%PYTHON_CONSOLE_PATH% -c "from torch.cuda import is_available; print(is_available())"') do set cuda_available=%%i
|
||||
|
||||
if "%cuda_available%"=="False" (
|
||||
echo "Installing PyTorch with CUDA (12.8) support..."
|
||||
:: We need to use an older version of PyTorch (2.8.0) until this issue is fixed: https://github.com/pytorch/pytorch/issues/166628
|
||||
:: Solution mentioned by @mazenemam19 in #99:
|
||||
%PYTHON_CONSOLE_PATH% -m pip install torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128 --no-warn-script-location
|
||||
echo.
|
||||
if errorlevel 1 (
|
||||
echo Failed to install PyTorch.
|
||||
pause
|
||||
exit /b
|
||||
)
|
||||
) else (
|
||||
echo CUDA is available on NVIDIA GPU.
|
||||
)
|
||||
) else (
|
||||
echo CUDA is available.
|
||||
echo.
|
||||
echo Unable to detect an NVIDIA GPU in your system.
|
||||
echo.
|
||||
echo Do you want to install PyTorch anyway?
|
||||
echo.
|
||||
echo If you DO have an NVIDIA GPU, please press Y.
|
||||
echo If you DO NOT have an NVIDIA GPU, please press N.
|
||||
echo.
|
||||
choice /C YN /M "Y=Yes, N=No"
|
||||
if errorlevel 2 (
|
||||
echo Skipping PyTorch installation.
|
||||
) else (
|
||||
echo "Installing PyTorch with CUDA (12.8) support..."
|
||||
:: We need to use an older version of PyTorch (2.8.0) until this issue is fixed: https://github.com/pytorch/pytorch/issues/166628
|
||||
:: Solution mentioned by @mazenemam19 in #99:
|
||||
%PYTHON_CONSOLE_PATH% -m pip install torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128 --no-warn-script-location
|
||||
if errorlevel 1 (
|
||||
echo Failed to install PyTorch.
|
||||
pause
|
||||
exit /b
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
:: Ask user if they want to create a desktop shortcut
|
||||
|
||||
@@ -12,8 +12,10 @@ RUN apt-get update \
|
||||
python3 \
|
||||
python3-venv \
|
||||
python3-pip \
|
||||
python3-pyqt5 \
|
||||
python3-pyqt6 \
|
||||
espeak-ng \
|
||||
libxcb-cursor0 \
|
||||
libgl1 \
|
||||
&& apt-get clean \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
|
||||
@@ -1 +1 @@
|
||||
1.0.6
|
||||
1.2.5
|
||||
|
After Width: | Height: | Size: 18 KiB |
|
Before Width: | Height: | Size: 241 B After Width: | Height: | Size: 343 B |
|
Before Width: | Height: | Size: 585 B |
@@ -0,0 +1,31 @@
|
||||
<?xml version="1.0" encoding="utf-8"?>
|
||||
|
||||
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
|
||||
<!-- Uploaded to: SVG Repo, www.svgrepo.com, Generator: SVG Repo Mixer Tools -->
|
||||
<svg height="800px" width="800px" version="1.1" id="_x32_" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink"
|
||||
viewBox="0 0 512 512" xml:space="preserve">
|
||||
<style type="text/css">
|
||||
.st0{fill:#808080;}
|
||||
</style>
|
||||
<g>
|
||||
<path class="st0" d="M502.325,307.303l-39.006-30.805c-6.215-4.908-9.665-12.429-9.668-20.348c0-0.084,0-0.168,0-0.252
|
||||
c-0.014-7.936,3.44-15.478,9.667-20.396l39.007-30.806c8.933-7.055,12.093-19.185,7.737-29.701l-17.134-41.366
|
||||
c-4.356-10.516-15.167-16.86-26.472-15.532l-49.366,5.8c-7.881,0.926-15.656-1.966-21.258-7.586
|
||||
c-0.059-0.06-0.118-0.119-0.177-0.178c-5.597-5.602-8.476-13.36-7.552-21.225l5.799-49.363
|
||||
c1.328-11.305-5.015-22.116-15.531-26.472L337.004,1.939c-10.516-4.356-22.646-1.196-29.701,7.736l-30.805,39.005
|
||||
c-4.908,6.215-12.43,9.665-20.349,9.668c-0.084,0-0.168,0-0.252,0c-7.935,0.014-15.477-3.44-20.395-9.667L204.697,9.675
|
||||
c-7.055-8.933-19.185-12.092-29.702-7.736L133.63,19.072c-10.516,4.356-16.86,15.167-15.532,26.473l5.799,49.366
|
||||
c0.926,7.881-1.964,15.656-7.585,21.257c-0.059,0.059-0.118,0.118-0.178,0.178c-5.602,5.598-13.36,8.477-21.226,7.552
|
||||
l-49.363-5.799c-11.305-1.328-22.116,5.015-26.472,15.531L1.939,174.996c-4.356,10.516-1.196,22.646,7.736,29.701l39.006,30.805
|
||||
c6.215,4.908,9.665,12.429,9.668,20.348c0,0.084,0,0.167,0,0.251c0.014,7.935-3.44,15.477-9.667,20.395L9.675,307.303
|
||||
c-8.933,7.055-12.092,19.185-7.736,29.701l17.134,41.365c4.356,10.516,15.168,16.86,26.472,15.532l49.366-5.799
|
||||
c7.882-0.926,15.656,1.965,21.258,7.586c0.059,0.059,0.118,0.119,0.178,0.178c5.597,5.603,8.476,13.36,7.552,21.226l-5.799,49.364
|
||||
c-1.328,11.305,5.015,22.116,15.532,26.472l41.366,17.134c10.516,4.356,22.646,1.196,29.701-7.736l30.804-39.005
|
||||
c4.908-6.215,12.43-9.665,20.348-9.669c0.084,0,0.168,0,0.251,0c7.936-0.014,15.478,3.44,20.396,9.667l30.806,39.007
|
||||
c7.055,8.933,19.185,12.093,29.701,7.736l41.366-17.134c10.516-4.356,16.86-15.168,15.532-26.472l-5.8-49.366
|
||||
c-0.926-7.881,1.965-15.656,7.586-21.257c0.059-0.059,0.119-0.119,0.178-0.178c5.602-5.597,13.36-8.476,21.225-7.552l49.364,5.799
|
||||
c11.305,1.328,22.117-5.015,26.472-15.531l17.134-41.365C514.418,326.488,511.258,314.358,502.325,307.303z M281.292,329.698
|
||||
c-39.68,16.436-85.172-2.407-101.607-42.087c-16.436-39.68,2.407-85.171,42.087-101.608c39.68-16.436,85.172,2.407,101.608,42.088
|
||||
C339.815,267.771,320.972,313.262,281.292,329.698z"/>
|
||||
</g>
|
||||
</svg>
|
||||
|
After Width: | Height: | Size: 2.6 KiB |
|
Before Width: | Height: | Size: 331 B After Width: | Height: | Size: 376 B |
@@ -1,13 +1,21 @@
|
||||
from utils import get_version
|
||||
from abogen.utils import get_version
|
||||
|
||||
# Program Information
|
||||
PROGRAM_NAME = "abogen"
|
||||
PROGRAM_DESCRIPTION = (
|
||||
"Generate audiobooks from EPUBs, PDFs and text with synchronized captions."
|
||||
)
|
||||
PROGRAM_DESCRIPTION = "Generate audiobooks from EPUBs, PDFs, text and subtitles with synchronized captions."
|
||||
GITHUB_URL = "https://github.com/denizsafak/abogen"
|
||||
VERSION = get_version()
|
||||
|
||||
# Settings
|
||||
CHAPTER_OPTIONS_COUNTDOWN = 30 # Countdown seconds for chapter options
|
||||
SUBTITLE_FORMATS = [
|
||||
("srt", "SRT (standard)"),
|
||||
("ass_wide", "ASS (wide)"),
|
||||
("ass_narrow", "ASS (narrow)"),
|
||||
("ass_centered_wide", "ASS (centered wide)"),
|
||||
("ass_centered_narrow", "ASS (centered narrow)"),
|
||||
]
|
||||
|
||||
# Language description mapping
|
||||
LANGUAGE_DESCRIPTIONS = {
|
||||
"a": "American English",
|
||||
@@ -21,16 +29,39 @@ LANGUAGE_DESCRIPTIONS = {
|
||||
"z": "Mandarin Chinese",
|
||||
}
|
||||
|
||||
# Supported sound formats
|
||||
SUPPORTED_SOUND_FORMATS = [
|
||||
"wav",
|
||||
"mp3",
|
||||
"opus",
|
||||
"m4b",
|
||||
"flac",
|
||||
]
|
||||
|
||||
# Supported subtitle formats
|
||||
SUPPORTED_SUBTITLE_FORMATS = [
|
||||
"srt",
|
||||
"ass",
|
||||
"vtt",
|
||||
]
|
||||
|
||||
# Supported input formats
|
||||
SUPPORTED_INPUT_FORMATS = [
|
||||
"epub",
|
||||
"pdf",
|
||||
"txt",
|
||||
"srt",
|
||||
"ass",
|
||||
"vtt",
|
||||
]
|
||||
|
||||
# Supported languages for subtitle generation
|
||||
# Currently, only 'a (American English)' and 'b (British English)' are supported for subtitle generation.
|
||||
# This is because tokens that contain timestamps are not generated for other languages in the Kokoro pipeline.
|
||||
# Please refer to: https://github.com/hexgrad/kokoro/blob/6d87f4ae7abc2d14dbc4b3ef2e5f19852e861ac2/kokoro/pipeline.py
|
||||
# 383 English processing (unchanged)
|
||||
# 384 if self.lang_code in 'ab':
|
||||
SUPPORTED_LANGUAGES_FOR_SUBTITLE_GENERATION = [
|
||||
"a",
|
||||
"b",
|
||||
]
|
||||
SUPPORTED_LANGUAGES_FOR_SUBTITLE_GENERATION = list(LANGUAGE_DESCRIPTIONS.keys())
|
||||
|
||||
# Voice and sample text constants
|
||||
VOICES_INTERNAL = [
|
||||
@@ -102,3 +133,30 @@ SAMPLE_VOICE_TEXTS = {
|
||||
"p": "Este é um exemplo da voz selecionada.",
|
||||
"z": "这是所选语音的示例。",
|
||||
}
|
||||
|
||||
COLORS = {
|
||||
"BLUE": "#007dff",
|
||||
"RED": "#c0392b",
|
||||
"ORANGE": "#FFA500",
|
||||
"GREEN": "#42ad4a",
|
||||
"GREEN_BG": "rgba(66, 173, 73, 0.1)",
|
||||
"GREEN_BG_HOVER": "rgba(66, 173, 73, 0.15)",
|
||||
"GREEN_BORDER": "#42ad4a",
|
||||
"BLUE_BG": "rgba(0, 102, 255, 0.05)",
|
||||
"BLUE_BG_HOVER": "rgba(0, 102, 255, 0.1)",
|
||||
"BLUE_BORDER_HOVER": "#6ab0de",
|
||||
"YELLOW_BACKGROUND": "rgba(255, 221, 51, 0.40)",
|
||||
"GREY_BACKGROUND": "rgba(128, 128, 128, 0.15)",
|
||||
"GREY_BORDER": "#808080",
|
||||
"RED_BACKGROUND": "rgba(232, 78, 60, 0.15)",
|
||||
"RED_BG": "rgba(232, 78, 60, 0.10)",
|
||||
"RED_BG_HOVER": "rgba(232, 78, 60, 0.15)",
|
||||
# Theme palette colors
|
||||
"DARK_BG": "#202326",
|
||||
"DARK_BASE": "#141618",
|
||||
"DARK_ALT": "#2c2f31",
|
||||
"DARK_BUTTON": "#292c30",
|
||||
"DARK_DISABLED": "#535353",
|
||||
"LIGHT_BG": "#eff0f1",
|
||||
"LIGHT_DISABLED": "#9a9999",
|
||||
}
|
||||
|
||||
@@ -0,0 +1,45 @@
|
||||
log_callback = None
|
||||
show_warning_signal_emitter = None # Renamed for clarity
|
||||
|
||||
|
||||
def set_log_callback(cb):
|
||||
global log_callback
|
||||
log_callback = cb
|
||||
|
||||
|
||||
def set_show_warning_signal_emitter(emitter): # Renamed for clarity
|
||||
global show_warning_signal_emitter
|
||||
show_warning_signal_emitter = emitter
|
||||
|
||||
|
||||
from huggingface_hub import hf_hub_download
|
||||
|
||||
|
||||
def tracked_hf_hub_download(*args, **kwargs):
|
||||
try:
|
||||
local_kwargs = dict(kwargs)
|
||||
local_kwargs["local_files_only"] = True
|
||||
hf_hub_download(*args, **local_kwargs)
|
||||
except Exception:
|
||||
repo_id = kwargs.get("repo_id", "<unknown repo>")
|
||||
filename = kwargs.get("filename", "<unknown file>")
|
||||
if filename.endswith(".pth"):
|
||||
msg = f"\nDownloading model '{filename}' from Hugging Face ({repo_id}). This may take a while. Please wait..."
|
||||
if show_warning_signal_emitter: # Check if the emitter is set
|
||||
show_warning_signal_emitter.emit(
|
||||
"Downloading Model",
|
||||
f"Downloading model '{filename}' from Hugging Face repository '{repo_id}'. This may take a while, please wait.",
|
||||
)
|
||||
else:
|
||||
msg = f"\nDownloading '{filename}' from Hugging Face ({repo_id}). Please wait..."
|
||||
if log_callback:
|
||||
print(msg, flush=True)
|
||||
log_callback(msg)
|
||||
else:
|
||||
print(msg, flush=True)
|
||||
return hf_hub_download(*args, **kwargs)
|
||||
|
||||
|
||||
import huggingface_hub
|
||||
|
||||
huggingface_hub.hf_hub_download = tracked_hf_hub_download
|
||||
@@ -0,0 +1,22 @@
|
||||
import gpustat
|
||||
|
||||
|
||||
def check():
|
||||
try:
|
||||
stats = gpustat.new_query()
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
nvidia_keywords = ["nvidia", "rtx", "gtx", "quadro", "tesla", "titan", "mx"]
|
||||
for gpu in stats.gpus:
|
||||
name = gpu.name.lower()
|
||||
if any(keyword in name for keyword in nvidia_keywords):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
stats = gpustat.new_query()
|
||||
for gpu in stats.gpus:
|
||||
print(gpu.name)
|
||||
print(check())
|
||||
@@ -1,14 +1,122 @@
|
||||
import os
|
||||
import sys
|
||||
import platform
|
||||
from PyQt5.QtWidgets import QApplication
|
||||
from PyQt5.QtGui import QIcon
|
||||
import atexit
|
||||
import signal
|
||||
from abogen.utils import get_resource_path, load_config, prevent_sleep_end
|
||||
|
||||
|
||||
# Fix PyTorch DLL loading issue ([WinError 1114]) on Windows before importing PyQt6
|
||||
if platform.system() == "Windows":
|
||||
import ctypes
|
||||
from importlib.util import find_spec
|
||||
|
||||
try:
|
||||
if (
|
||||
(spec := find_spec("torch"))
|
||||
and spec.origin
|
||||
and os.path.exists(
|
||||
dll_path := os.path.join(os.path.dirname(spec.origin), "lib", "c10.dll")
|
||||
)
|
||||
):
|
||||
ctypes.CDLL(os.path.normpath(dll_path))
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
|
||||
# Qt platform plugin detection (fixes #59)
|
||||
try:
|
||||
from PyQt6.QtCore import QLibraryInfo
|
||||
|
||||
# Get the path to the plugins directory
|
||||
plugins = QLibraryInfo.path(QLibraryInfo.LibraryPath.PluginsPath)
|
||||
|
||||
# Normalize path to use the OS-native separators and absolute path
|
||||
platform_dir = os.path.normpath(os.path.join(plugins, "platforms"))
|
||||
|
||||
# Ensure we work with an absolute path for clarity
|
||||
platform_dir = os.path.abspath(platform_dir)
|
||||
|
||||
if os.path.isdir(platform_dir):
|
||||
os.environ["QT_QPA_PLATFORM_PLUGIN_PATH"] = platform_dir
|
||||
print("QT_QPA_PLATFORM_PLUGIN_PATH set to:", platform_dir)
|
||||
else:
|
||||
print("PyQt6 platform plugins not found at", platform_dir)
|
||||
except ImportError:
|
||||
print("PyQt6 not installed.")
|
||||
|
||||
|
||||
# Pre-load "libxcb-cursor" on Linux (fixes #101)
|
||||
if platform.system() == "Linux":
|
||||
arch = platform.machine().lower()
|
||||
lib_filename = {"x86_64": "libxcb-cursor-amd64.so.0", "amd64": "libxcb-cursor-amd64.so.0", "aarch64": "libxcb-cursor-arm64.so.0", "arm64": "libxcb-cursor-arm64.so.0"}.get(arch)
|
||||
if lib_filename:
|
||||
import ctypes
|
||||
try:
|
||||
# Try to load the system libxcb-cursor.so.0 first
|
||||
ctypes.CDLL('libxcb-cursor.so.0', mode=ctypes.RTLD_GLOBAL)
|
||||
except OSError:
|
||||
# System lib not available, load the bundled version
|
||||
lib_path = get_resource_path('abogen.libs', lib_filename)
|
||||
if lib_path:
|
||||
try:
|
||||
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
|
||||
except OSError:
|
||||
# If it fails (e.g. wrong glibc version on very old systems),
|
||||
# we simply ignore it and hope the system has the library.
|
||||
pass
|
||||
|
||||
|
||||
# Set application ID for Windows taskbar icon
|
||||
if platform.system() == "Windows":
|
||||
try:
|
||||
from abogen.constants import PROGRAM_NAME, VERSION
|
||||
import ctypes
|
||||
|
||||
app_id = f"{PROGRAM_NAME}.{VERSION}"
|
||||
ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(app_id)
|
||||
except Exception as e:
|
||||
print("Warning: failed to set AppUserModelID:", e)
|
||||
|
||||
from PyQt6.QtWidgets import QApplication
|
||||
from PyQt6.QtGui import QIcon
|
||||
from PyQt6.QtCore import (
|
||||
QLibraryInfo,
|
||||
qInstallMessageHandler,
|
||||
QtMsgType,
|
||||
)
|
||||
|
||||
# Add the directory to Python path
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__)))
|
||||
from gui import abogen
|
||||
from utils import get_resource_path
|
||||
from constants import PROGRAM_NAME, VERSION
|
||||
|
||||
# Set Hugging Face Hub environment variables
|
||||
os.environ["HF_HUB_DISABLE_TELEMETRY"] = "1" # Disable Hugging Face telemetry
|
||||
os.environ["HF_HUB_ETAG_TIMEOUT"] = "10" # Metadata request timeout (seconds)
|
||||
os.environ["HF_HUB_DOWNLOAD_TIMEOUT"] = "10" # File download timeout (seconds)
|
||||
os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1" # Disable symlinks warning
|
||||
if load_config().get("disable_kokoro_internet", False):
|
||||
print("INFO: Kokoro's internet access is disabled.")
|
||||
os.environ["HF_HUB_OFFLINE"] = "1" # Disable Hugging Face Hub internet access
|
||||
|
||||
from abogen.gui import abogen
|
||||
from abogen.constants import PROGRAM_NAME, VERSION
|
||||
|
||||
# Set environment variables for AMD ROCm
|
||||
os.environ["MIOPEN_FIND_MODE"] = "FAST"
|
||||
os.environ["MIOPEN_CONV_PRECISE_ROCM_TUNING"] = "0"
|
||||
|
||||
# Reset sleep states
|
||||
atexit.register(prevent_sleep_end)
|
||||
|
||||
|
||||
# Also handle signals (Ctrl+C, kill, etc.)
|
||||
def _cleanup_sleep(signum, frame):
|
||||
prevent_sleep_end()
|
||||
sys.exit(0)
|
||||
|
||||
|
||||
signal.signal(signal.SIGINT, _cleanup_sleep)
|
||||
signal.signal(signal.SIGTERM, _cleanup_sleep)
|
||||
|
||||
# Ensure sys.stdout and sys.stderr are valid in GUI mode
|
||||
if sys.stdout is None:
|
||||
@@ -20,12 +128,27 @@ if sys.stderr is None:
|
||||
if platform.system() == "Darwin" and platform.processor() == "arm":
|
||||
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
|
||||
|
||||
# Set application ID for Windows taskbar icon
|
||||
if platform.system() == "Windows":
|
||||
import ctypes
|
||||
|
||||
app_id = f"{PROGRAM_NAME}.{VERSION}"
|
||||
ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(app_id)
|
||||
# Custom message handler to filter out specific Qt warnings
|
||||
def qt_message_handler(mode, context, message):
|
||||
# In PyQt6, the mode is an enum, so we compare with the enum members
|
||||
if "Wayland does not support QWindow::requestActivate()" in message:
|
||||
return # Suppress this specific message
|
||||
if "setGrabPopup called with a parent, QtWaylandClient" in message:
|
||||
return
|
||||
|
||||
if mode == QtMsgType.QtWarningMsg:
|
||||
print(f"Qt Warning: {message}")
|
||||
elif mode == QtMsgType.QtCriticalMsg:
|
||||
print(f"Qt Critical: {message}")
|
||||
elif mode == QtMsgType.QtFatalMsg:
|
||||
print(f"Qt Fatal: {message}")
|
||||
elif mode == QtMsgType.QtInfoMsg:
|
||||
print(f"Qt Info: {message}")
|
||||
|
||||
|
||||
# Install the custom message handler
|
||||
qInstallMessageHandler(qt_message_handler)
|
||||
|
||||
# Handle Wayland on Linux GNOME
|
||||
if platform.system() == "Linux":
|
||||
@@ -48,9 +171,16 @@ def main():
|
||||
if icon_path:
|
||||
app.setWindowIcon(QIcon(icon_path))
|
||||
|
||||
# Set the .desktop name on Linux
|
||||
if platform.system() == "Linux":
|
||||
try:
|
||||
app.setDesktopFileName("abogen")
|
||||
except AttributeError:
|
||||
pass
|
||||
|
||||
ex = abogen()
|
||||
ex.show()
|
||||
sys.exit(app.exec_())
|
||||
sys.exit(app.exec())
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
@@ -0,0 +1,590 @@
|
||||
"""
|
||||
Pre-download dialog and worker for Abogen
|
||||
|
||||
This module consolidates pre-download logic for Kokoro voices and model
|
||||
and spaCy language models. The code favors clarity, avoids duplication,
|
||||
and handles optional dependencies gracefully.
|
||||
"""
|
||||
|
||||
from typing import List, Optional, Tuple
|
||||
import importlib
|
||||
import importlib.util
|
||||
|
||||
from PyQt6.QtWidgets import (
|
||||
QDialog,
|
||||
QVBoxLayout,
|
||||
QHBoxLayout,
|
||||
QLabel,
|
||||
QPushButton,
|
||||
QSpacerItem,
|
||||
QSizePolicy,
|
||||
)
|
||||
from PyQt6.QtCore import QThread, pyqtSignal
|
||||
|
||||
from abogen.constants import COLORS, VOICES_INTERNAL
|
||||
from abogen.spacy_utils import SPACY_MODELS
|
||||
import abogen.hf_tracker
|
||||
|
||||
|
||||
# Helpers
|
||||
def _unique_sorted_models() -> List[str]:
|
||||
"""Return a sorted list of unique spaCy model package names."""
|
||||
return sorted(set(SPACY_MODELS.values()))
|
||||
|
||||
|
||||
def _is_package_installed(pkg_name: str) -> bool:
|
||||
"""Return True if a package with the given name can be imported (site-packages)."""
|
||||
try:
|
||||
return importlib.util.find_spec(pkg_name) is not None
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
|
||||
# NOTE: explicit HF cache helper removed; we use try_to_load_from_cache in-scope where needed
|
||||
|
||||
|
||||
class PreDownloadWorker(QThread):
|
||||
"""Worker thread to download required models/voices.
|
||||
|
||||
Emits human-readable messages via `progress`. Uses `category_done` to indicate
|
||||
a category (voices/model/spacy) finished successfully. Emits `error` on exception
|
||||
and `finished` after all work completes.
|
||||
"""
|
||||
|
||||
# Emit (category, status, message)
|
||||
progress = pyqtSignal(str, str, str)
|
||||
category_done = pyqtSignal(str)
|
||||
finished = pyqtSignal()
|
||||
error = pyqtSignal(str)
|
||||
|
||||
def __init__(self, parent=None):
|
||||
super().__init__(parent)
|
||||
self._cancelled = False
|
||||
# repo and filenames used for Kokoro model
|
||||
self._repo_id = "hexgrad/Kokoro-82M"
|
||||
self._model_files = ["kokoro-v1_0.pth", "config.json"]
|
||||
# Track download success per category
|
||||
self._voices_success = False
|
||||
self._model_success = False
|
||||
self._spacy_success = False
|
||||
# Suppress HF tracker warnings during downloads
|
||||
self._original_emitter = abogen.hf_tracker.show_warning_signal_emitter
|
||||
|
||||
def cancel(self) -> None:
|
||||
self._cancelled = True
|
||||
|
||||
def run(self) -> None:
|
||||
# Suppress HF tracker warnings during downloads
|
||||
abogen.hf_tracker.show_warning_signal_emitter = None
|
||||
try:
|
||||
self._download_kokoro_voices()
|
||||
if self._cancelled:
|
||||
return
|
||||
if self._voices_success:
|
||||
self.category_done.emit("voices")
|
||||
|
||||
self._download_kokoro_model()
|
||||
if self._cancelled:
|
||||
return
|
||||
if self._model_success:
|
||||
self.category_done.emit("model")
|
||||
|
||||
self._download_spacy_models()
|
||||
if self._cancelled:
|
||||
return
|
||||
if self._spacy_success:
|
||||
self.category_done.emit("spacy")
|
||||
|
||||
self.finished.emit()
|
||||
except Exception as exc: # pragma: no cover - best-effort reporting
|
||||
self.error.emit(str(exc))
|
||||
finally:
|
||||
# Restore original emitter
|
||||
abogen.hf_tracker.show_warning_signal_emitter = self._original_emitter
|
||||
|
||||
# Kokoro voices
|
||||
def _download_kokoro_voices(self) -> None:
|
||||
self._voices_success = True
|
||||
try:
|
||||
from huggingface_hub import hf_hub_download, try_to_load_from_cache
|
||||
except Exception:
|
||||
self.progress.emit(
|
||||
"voice", "warning", "huggingface_hub not installed, skipping voices..."
|
||||
)
|
||||
self._voices_success = False
|
||||
return
|
||||
|
||||
voice_list = VOICES_INTERNAL
|
||||
for idx, voice in enumerate(voice_list, start=1):
|
||||
if self._cancelled:
|
||||
self._voices_success = False
|
||||
return
|
||||
filename = f"voices/{voice}.pt"
|
||||
if try_to_load_from_cache(repo_id=self._repo_id, filename=filename):
|
||||
self.progress.emit(
|
||||
"voice",
|
||||
"installed",
|
||||
f"{idx}/{len(voice_list)}: {voice} already present",
|
||||
)
|
||||
continue
|
||||
self.progress.emit(
|
||||
"voice", "downloading", f"{idx}/{len(voice_list)}: {voice}..."
|
||||
)
|
||||
try:
|
||||
hf_hub_download(repo_id=self._repo_id, filename=filename)
|
||||
self.progress.emit("voice", "downloaded", f"{voice} downloaded")
|
||||
except Exception as exc:
|
||||
self.progress.emit(
|
||||
"voice", "warning", f"could not download {voice}: {exc}"
|
||||
)
|
||||
self._voices_success = False
|
||||
|
||||
# Kokoro model
|
||||
def _download_kokoro_model(self) -> None:
|
||||
self._model_success = True
|
||||
try:
|
||||
from huggingface_hub import hf_hub_download, try_to_load_from_cache
|
||||
except Exception:
|
||||
self.progress.emit(
|
||||
"model", "warning", "huggingface_hub not installed, skipping model..."
|
||||
)
|
||||
self._model_success = False
|
||||
return
|
||||
for fname in self._model_files:
|
||||
if self._cancelled:
|
||||
self._model_success = False
|
||||
return
|
||||
category = "config" if fname == "config.json" else "model"
|
||||
if try_to_load_from_cache(repo_id=self._repo_id, filename=fname):
|
||||
self.progress.emit(
|
||||
category, "installed", f"file {fname} already present"
|
||||
)
|
||||
continue
|
||||
self.progress.emit(category, "downloading", f"file {fname}...")
|
||||
try:
|
||||
hf_hub_download(repo_id=self._repo_id, filename=fname)
|
||||
self.progress.emit(category, "downloaded", f"file {fname} downloaded")
|
||||
except Exception as exc:
|
||||
self.progress.emit(
|
||||
category, "warning", f"could not download file {fname}: {exc}"
|
||||
)
|
||||
self._model_success = False
|
||||
|
||||
# spaCy models
|
||||
def _download_spacy_models(self) -> None:
|
||||
"""Download spaCy models. Prefer missing models provided by parent.
|
||||
|
||||
Parent dialog will populate _spacy_models_missing during checking.
|
||||
"""
|
||||
self._spacy_success = True
|
||||
# Determine which models to process: prefer parent-provided missing list to avoid
|
||||
# re-checking everything; otherwise use the full unique list.
|
||||
parent = self.parent()
|
||||
models_to_process: List[str] = _unique_sorted_models()
|
||||
try:
|
||||
if (
|
||||
parent is not None
|
||||
and hasattr(parent, "_spacy_models_missing")
|
||||
and parent._spacy_models_missing
|
||||
):
|
||||
models_to_process = list(dict.fromkeys(parent._spacy_models_missing))
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# If spaCy is not available to run the CLI, skip gracefully
|
||||
try:
|
||||
import spacy.cli as _spacy_cli
|
||||
except Exception:
|
||||
self.progress.emit(
|
||||
"spacy", "warning", "spaCy not available, skipping spaCy models..."
|
||||
)
|
||||
self._spacy_success = False
|
||||
return
|
||||
|
||||
for idx, model_name in enumerate(models_to_process, start=1):
|
||||
if self._cancelled:
|
||||
self._spacy_success = False
|
||||
return
|
||||
if _is_package_installed(model_name):
|
||||
self.progress.emit(
|
||||
"spacy",
|
||||
"installed",
|
||||
f"{idx}/{len(models_to_process)}: {model_name} already installed",
|
||||
)
|
||||
continue
|
||||
self.progress.emit(
|
||||
"spacy",
|
||||
"downloading",
|
||||
f"{idx}/{len(models_to_process)}: {model_name}...",
|
||||
)
|
||||
try:
|
||||
_spacy_cli.download(model_name)
|
||||
self.progress.emit("spacy", "downloaded", f"{model_name} downloaded")
|
||||
except Exception as exc:
|
||||
self.progress.emit(
|
||||
"spacy", "warning", f"could not download {model_name}: {exc}"
|
||||
)
|
||||
self._spacy_success = False
|
||||
|
||||
|
||||
class PreDownloadDialog(QDialog):
|
||||
"""Dialog to show and control pre-download process."""
|
||||
|
||||
VOICE_PREFIX = "Kokoro voices: "
|
||||
MODEL_PREFIX = "Kokoro model: "
|
||||
CONFIG_PREFIX = "Kokoro config: "
|
||||
SPACY_PREFIX = "spaCy models: "
|
||||
|
||||
def __init__(self, parent=None):
|
||||
super().__init__(parent)
|
||||
self.setWindowTitle("Pre-download Models and Voices")
|
||||
self.setMinimumWidth(500)
|
||||
self.worker: Optional[PreDownloadWorker] = None
|
||||
self.has_missing = False
|
||||
self._spacy_models_checked: List[tuple] = []
|
||||
self._spacy_models_missing: List[str] = []
|
||||
self._status_worker = None
|
||||
|
||||
# Map keywords to (label, prefix) - labels filled after UI creation
|
||||
self.status_map = {
|
||||
"voice": (None, self.VOICE_PREFIX),
|
||||
"spacy": (None, self.SPACY_PREFIX),
|
||||
"model": (None, self.MODEL_PREFIX),
|
||||
"config": (None, self.CONFIG_PREFIX),
|
||||
}
|
||||
|
||||
self.category_map = {
|
||||
"voices": ["voice"],
|
||||
"model": ["model", "config"],
|
||||
"spacy": ["spacy"],
|
||||
}
|
||||
|
||||
self._setup_ui()
|
||||
self._start_status_check()
|
||||
|
||||
def _setup_ui(self) -> None:
|
||||
layout = QVBoxLayout(self)
|
||||
layout.setSpacing(0)
|
||||
layout.setContentsMargins(15, 0, 15, 15)
|
||||
|
||||
desc = QLabel(
|
||||
"You can pre-download all required models and voices for offline use.\n"
|
||||
"This includes Kokoro voices, Kokoro model (and config), and spaCy models."
|
||||
)
|
||||
desc.setWordWrap(True)
|
||||
layout.addWidget(desc)
|
||||
|
||||
# Status rows
|
||||
status_layout = QVBoxLayout()
|
||||
status_title = QLabel("<b>Current Status:</b>")
|
||||
status_layout.addWidget(status_title)
|
||||
|
||||
self.voices_status = QLabel(self.VOICE_PREFIX + "⏳ Checking...")
|
||||
row = QHBoxLayout()
|
||||
row.addWidget(self.voices_status)
|
||||
row.addStretch()
|
||||
status_layout.addLayout(row)
|
||||
|
||||
self.model_status = QLabel(self.MODEL_PREFIX + "⏳ Checking...")
|
||||
row = QHBoxLayout()
|
||||
row.addWidget(self.model_status)
|
||||
row.addStretch()
|
||||
status_layout.addLayout(row)
|
||||
|
||||
self.config_status = QLabel(self.CONFIG_PREFIX + "⏳ Checking...")
|
||||
row = QHBoxLayout()
|
||||
row.addWidget(self.config_status)
|
||||
row.addStretch()
|
||||
status_layout.addLayout(row)
|
||||
|
||||
self.spacy_status = QLabel(self.SPACY_PREFIX + "⏳ Checking...")
|
||||
row = QHBoxLayout()
|
||||
row.addWidget(self.spacy_status)
|
||||
row.addStretch()
|
||||
status_layout.addLayout(row)
|
||||
|
||||
# register labels
|
||||
self.status_map["voice"] = (self.voices_status, self.VOICE_PREFIX)
|
||||
self.status_map["model"] = (self.model_status, self.MODEL_PREFIX)
|
||||
self.status_map["config"] = (self.config_status, self.CONFIG_PREFIX)
|
||||
self.status_map["spacy"] = (self.spacy_status, self.SPACY_PREFIX)
|
||||
|
||||
layout.addLayout(status_layout)
|
||||
|
||||
layout.addItem(
|
||||
QSpacerItem(0, 20, QSizePolicy.Policy.Minimum, QSizePolicy.Policy.Fixed)
|
||||
)
|
||||
|
||||
# Buttons
|
||||
button_row = QHBoxLayout()
|
||||
button_row.setSpacing(10)
|
||||
self.download_btn = QPushButton("Download all")
|
||||
self.download_btn.setMinimumWidth(100)
|
||||
self.download_btn.setMinimumHeight(35)
|
||||
self.download_btn.setEnabled(False)
|
||||
self.download_btn.clicked.connect(self._start_download)
|
||||
button_row.addWidget(self.download_btn)
|
||||
|
||||
self.close_btn = QPushButton("Close")
|
||||
self.close_btn.setMinimumWidth(100)
|
||||
self.close_btn.setMinimumHeight(35)
|
||||
self.close_btn.clicked.connect(self._handle_close)
|
||||
button_row.addWidget(self.close_btn)
|
||||
|
||||
layout.addLayout(button_row)
|
||||
self.adjustSize()
|
||||
|
||||
# Status checking worker
|
||||
class StatusCheckWorker(QThread):
|
||||
voices_checked = pyqtSignal(bool, list)
|
||||
model_checked = pyqtSignal(bool)
|
||||
config_checked = pyqtSignal(bool)
|
||||
spacy_model_checking = pyqtSignal(str)
|
||||
spacy_model_result = pyqtSignal(str, bool)
|
||||
spacy_checked = pyqtSignal(bool, list)
|
||||
|
||||
def run(self):
|
||||
parent = self.parent()
|
||||
if parent is None:
|
||||
return
|
||||
|
||||
voices_ok, missing_voices = parent._check_kokoro_voices()
|
||||
self.voices_checked.emit(voices_ok, missing_voices)
|
||||
|
||||
model_ok = parent._check_kokoro_model()
|
||||
self.model_checked.emit(model_ok)
|
||||
|
||||
config_ok = parent._check_kokoro_config()
|
||||
self.config_checked.emit(config_ok)
|
||||
|
||||
# Check spaCy models by package name to detect site-package installs
|
||||
unique = _unique_sorted_models()
|
||||
missing: List[str] = []
|
||||
for name in unique:
|
||||
self.spacy_model_checking.emit(name)
|
||||
ok = _is_package_installed(name)
|
||||
self.spacy_model_result.emit(name, ok)
|
||||
if not ok:
|
||||
missing.append(name)
|
||||
parent._spacy_models_missing = missing
|
||||
self.spacy_checked.emit(len(missing) == 0, missing)
|
||||
|
||||
def _start_status_check(self) -> None:
|
||||
self._status_worker = self.StatusCheckWorker(self)
|
||||
self._status_worker.voices_checked.connect(self._update_voices_status)
|
||||
self._status_worker.model_checked.connect(self._update_model_status)
|
||||
self._status_worker.config_checked.connect(self._update_config_status)
|
||||
self._status_worker.spacy_model_checking.connect(self._spacy_model_checking)
|
||||
self._status_worker.spacy_model_result.connect(self._spacy_model_result)
|
||||
self._status_worker.spacy_checked.connect(self._update_spacy_status)
|
||||
|
||||
# These are initialized in __init__ to keep consistent object state
|
||||
|
||||
# Set checking visual state
|
||||
for lbl in (
|
||||
self.voices_status,
|
||||
self.model_status,
|
||||
self.config_status,
|
||||
self.spacy_status,
|
||||
):
|
||||
lbl.setStyleSheet(f"color: {COLORS['ORANGE']};")
|
||||
|
||||
self.spacy_status.setText(self.SPACY_PREFIX + "⏳ Checking...")
|
||||
self._status_worker.start()
|
||||
|
||||
# UI update callbacks
|
||||
def _spacy_model_checking(self, name: str) -> None:
|
||||
self.spacy_status.setText(f"{self.SPACY_PREFIX}Checking {name}...")
|
||||
|
||||
def _spacy_model_result(self, name: str, ok: bool) -> None:
|
||||
self._spacy_models_checked.append((name, ok))
|
||||
if not ok and name not in self._spacy_models_missing:
|
||||
self._spacy_models_missing.append(name)
|
||||
checked = len(self._spacy_models_checked)
|
||||
missing_count = len(self._spacy_models_missing)
|
||||
if missing_count:
|
||||
self.spacy_status.setText(
|
||||
f"{self.SPACY_PREFIX}{checked} checked, {missing_count} missing..."
|
||||
)
|
||||
else:
|
||||
self.spacy_status.setText(f"{self.SPACY_PREFIX}{checked} checked...")
|
||||
|
||||
def _update_voices_status(self, ok: bool, missing: List[str]) -> None:
|
||||
if ok:
|
||||
self._set_status("voice", "✓ Downloaded", COLORS["GREEN"])
|
||||
else:
|
||||
self.has_missing = True
|
||||
if missing:
|
||||
self._set_status(
|
||||
"voice", f"✗ Missing {len(missing)} voices", COLORS["RED"]
|
||||
)
|
||||
else:
|
||||
self._set_status("voice", "✗ Not downloaded", COLORS["RED"])
|
||||
|
||||
def _update_model_status(self, ok: bool) -> None:
|
||||
if ok:
|
||||
self._set_status("model", "✓ Downloaded", COLORS["GREEN"])
|
||||
else:
|
||||
self.has_missing = True
|
||||
self._set_status("model", "✗ Not downloaded", COLORS["RED"])
|
||||
|
||||
def _update_config_status(self, ok: bool) -> None:
|
||||
if ok:
|
||||
self._set_status("config", "✓ Downloaded", COLORS["GREEN"])
|
||||
else:
|
||||
self.has_missing = True
|
||||
self._set_status("config", "✗ Not downloaded", COLORS["RED"])
|
||||
|
||||
def _update_spacy_status(self, ok: bool, missing: List[str]) -> None:
|
||||
if ok:
|
||||
self._set_status("spacy", "✓ Downloaded", COLORS["GREEN"])
|
||||
else:
|
||||
self.has_missing = True
|
||||
if missing:
|
||||
self._set_status(
|
||||
"spacy", f"✗ Missing {len(missing)} model(s)", COLORS["RED"]
|
||||
)
|
||||
else:
|
||||
self._set_status("spacy", "✗ Not downloaded", COLORS["RED"])
|
||||
self.download_btn.setEnabled(self.has_missing)
|
||||
|
||||
def _set_status(self, key: str, text: str, color: str) -> None:
|
||||
lbl, prefix = self.status_map.get(key, (None, ""))
|
||||
if not lbl:
|
||||
return
|
||||
lbl.setText(prefix + text)
|
||||
lbl.setStyleSheet(f"color: {color};")
|
||||
|
||||
# Helper checks
|
||||
def _check_kokoro_voices(self) -> Tuple[bool, List[str]]:
|
||||
"""Return (ok, missing_list) for Kokoro voices check."""
|
||||
missing = []
|
||||
try:
|
||||
from huggingface_hub import try_to_load_from_cache
|
||||
|
||||
for voice in VOICES_INTERNAL:
|
||||
if not try_to_load_from_cache(
|
||||
repo_id="hexgrad/Kokoro-82M", filename=f"voices/{voice}.pt"
|
||||
):
|
||||
missing.append(voice)
|
||||
except Exception:
|
||||
# If HF missing, report all as missing
|
||||
return False, list(VOICES_INTERNAL)
|
||||
return (len(missing) == 0), missing
|
||||
|
||||
def _check_kokoro_model(self) -> bool:
|
||||
try:
|
||||
from huggingface_hub import try_to_load_from_cache
|
||||
|
||||
return (
|
||||
try_to_load_from_cache(
|
||||
repo_id="hexgrad/Kokoro-82M", filename="kokoro-v1_0.pth"
|
||||
)
|
||||
is not None
|
||||
)
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
def _check_kokoro_config(self) -> bool:
|
||||
try:
|
||||
from huggingface_hub import try_to_load_from_cache
|
||||
|
||||
return (
|
||||
try_to_load_from_cache(
|
||||
repo_id="hexgrad/Kokoro-82M", filename="config.json"
|
||||
)
|
||||
is not None
|
||||
)
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
def _check_spacy_models(self) -> bool:
|
||||
unique = _unique_sorted_models()
|
||||
missing = [m for m in unique if not _is_package_installed(m)]
|
||||
self._spacy_models_missing = missing
|
||||
return len(missing) == 0
|
||||
|
||||
# Download control
|
||||
def _start_download(self) -> None:
|
||||
self.download_btn.setEnabled(False)
|
||||
self.download_btn.setText("Downloading...")
|
||||
# mark the start of downloads; this triggers the labels
|
||||
self._on_progress("system", "starting", "Processing, please wait...")
|
||||
self.worker = PreDownloadWorker(self)
|
||||
self.worker.progress.connect(self._on_progress)
|
||||
self.worker.category_done.connect(self._on_category_done)
|
||||
self.worker.finished.connect(self._on_download_finished)
|
||||
self.worker.error.connect(self._on_download_error)
|
||||
self.worker.start()
|
||||
|
||||
def _on_progress(self, category: str, status: str, message: str) -> None:
|
||||
"""Map worker (category, status, message) to UI label updates.
|
||||
|
||||
Status is one of: 'downloading', 'installed', 'downloaded', 'warning', 'starting'.
|
||||
Category is one of: 'voice', 'model', 'spacy', 'config', or 'system'.
|
||||
"""
|
||||
try:
|
||||
# If the category targets a specific label, update directly
|
||||
if category in self.status_map:
|
||||
lbl, prefix = self.status_map[category]
|
||||
if not lbl:
|
||||
return
|
||||
# Compose message and set color based on status token
|
||||
full_text = prefix + message
|
||||
if len(full_text) > 60:
|
||||
display_text = full_text[:57] + "..."
|
||||
lbl.setText(display_text)
|
||||
lbl.setToolTip(full_text)
|
||||
else:
|
||||
lbl.setText(full_text)
|
||||
lbl.setToolTip("") # Clear tooltip if not needed
|
||||
if status == "downloading":
|
||||
lbl.setStyleSheet(f"color: {COLORS['ORANGE']};")
|
||||
elif status in ("installed", "downloaded"):
|
||||
lbl.setStyleSheet(f"color: {COLORS['GREEN']};")
|
||||
elif status == "warning":
|
||||
lbl.setStyleSheet(f"color: {COLORS['RED']};")
|
||||
elif status == "error":
|
||||
lbl.setStyleSheet(f"color: {COLORS['RED']};")
|
||||
return
|
||||
|
||||
# System-level messages
|
||||
if category == "system":
|
||||
if status == "starting":
|
||||
for k in self.status_map:
|
||||
lbl, prefix = self.status_map[k]
|
||||
if lbl:
|
||||
lbl.setText(prefix + "Processing, please wait...")
|
||||
lbl.setStyleSheet(f"color: {COLORS['ORANGE']};")
|
||||
# other system statuses don't require action
|
||||
return
|
||||
except Exception:
|
||||
# Do not let UI thread crash on unexpected worker message
|
||||
pass
|
||||
|
||||
def _on_category_done(self, category: str) -> None:
|
||||
for key in self.category_map.get(category, []):
|
||||
self._set_status(key, "✓ Downloaded", COLORS["GREEN"])
|
||||
|
||||
def _on_download_finished(self) -> None:
|
||||
self.has_missing = False
|
||||
self.download_btn.setText("Download all")
|
||||
self.download_btn.setEnabled(False)
|
||||
|
||||
def _on_download_error(self, error_msg: str) -> None:
|
||||
self.download_btn.setText("Download all")
|
||||
self.download_btn.setEnabled(True)
|
||||
for key in self.status_map:
|
||||
self._set_status(key, f"✗ Error - {error_msg}", COLORS["RED"])
|
||||
|
||||
def _handle_close(self) -> None:
|
||||
if self.worker and self.worker.isRunning():
|
||||
self.worker.cancel()
|
||||
self.worker.wait(2000)
|
||||
self.accept()
|
||||
|
||||
def closeEvent(self, event) -> None:
|
||||
if self.worker and self.worker.isRunning():
|
||||
self.worker.cancel()
|
||||
self.worker.wait(2000)
|
||||
super().closeEvent(event)
|
||||
@@ -0,0 +1,860 @@
|
||||
# a simple window with a list of items in the queue, no checkboxes
|
||||
# button to remove an item from the queue
|
||||
# button to clear the queue
|
||||
|
||||
from PyQt6.QtWidgets import (
|
||||
QDialog,
|
||||
QVBoxLayout,
|
||||
QHBoxLayout,
|
||||
QDialogButtonBox,
|
||||
QPushButton,
|
||||
QListWidget,
|
||||
QListWidgetItem,
|
||||
QFileIconProvider,
|
||||
QLabel,
|
||||
QWidget,
|
||||
QSizePolicy,
|
||||
QAbstractItemView,
|
||||
QCheckBox,
|
||||
)
|
||||
from PyQt6.QtCore import QFileInfo, Qt
|
||||
from abogen.constants import COLORS
|
||||
from copy import deepcopy
|
||||
from PyQt6.QtGui import QFontMetrics
|
||||
from abogen.utils import load_config, save_config
|
||||
|
||||
# Define attributes that are safe to override with global settings
|
||||
OVERRIDE_FIELDS = [
|
||||
"lang_code",
|
||||
"speed",
|
||||
"voice",
|
||||
"save_option",
|
||||
"output_folder",
|
||||
"subtitle_mode",
|
||||
"output_format",
|
||||
"replace_single_newlines",
|
||||
"use_silent_gaps",
|
||||
"subtitle_speed_method",
|
||||
]
|
||||
|
||||
|
||||
class ElidedLabel(QLabel):
|
||||
def __init__(self, text):
|
||||
super().__init__(text)
|
||||
self._full_text = text
|
||||
self.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Preferred)
|
||||
self.setTextFormat(Qt.TextFormat.PlainText)
|
||||
|
||||
def setText(self, text):
|
||||
self._full_text = text
|
||||
super().setText(text)
|
||||
self.update()
|
||||
|
||||
def resizeEvent(self, event):
|
||||
metrics = QFontMetrics(self.font())
|
||||
elided = metrics.elidedText(
|
||||
self._full_text, Qt.TextElideMode.ElideRight, self.width()
|
||||
)
|
||||
super().setText(elided)
|
||||
super().resizeEvent(event)
|
||||
|
||||
def fullText(self):
|
||||
return self._full_text
|
||||
|
||||
|
||||
class QueueListItemWidget(QWidget):
|
||||
def __init__(self, file_name, char_count):
|
||||
super().__init__()
|
||||
layout = QHBoxLayout()
|
||||
layout.setContentsMargins(12, 0, 6, 0)
|
||||
layout.setSpacing(0)
|
||||
import os
|
||||
|
||||
name_label = ElidedLabel(os.path.basename(file_name))
|
||||
char_label = QLabel(f"Chars: {char_count}")
|
||||
char_label.setStyleSheet(f"color: {COLORS['LIGHT_DISABLED']};")
|
||||
char_label.setAlignment(
|
||||
Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter
|
||||
)
|
||||
char_label.setSizePolicy(
|
||||
QSizePolicy.Policy.Minimum, QSizePolicy.Policy.Preferred
|
||||
)
|
||||
layout.addWidget(name_label, 1)
|
||||
layout.addWidget(char_label, 0)
|
||||
self.setLayout(layout)
|
||||
|
||||
|
||||
class DroppableQueueListWidget(QListWidget):
|
||||
def __init__(self, parent_dialog):
|
||||
super().__init__()
|
||||
self.parent_dialog = parent_dialog
|
||||
self.setAcceptDrops(True)
|
||||
# Overlay for drag hover
|
||||
self.drag_overlay = QLabel("", self)
|
||||
self.drag_overlay.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
self.drag_overlay.setStyleSheet(
|
||||
f"border:2px dashed {COLORS['BLUE_BORDER_HOVER']}; border-radius:5px; padding:20px; background:{COLORS['BLUE_BG_HOVER']};"
|
||||
)
|
||||
self.drag_overlay.setVisible(False)
|
||||
self.drag_overlay.setAttribute(
|
||||
Qt.WidgetAttribute.WA_TransparentForMouseEvents, True
|
||||
)
|
||||
|
||||
def dragEnterEvent(self, event):
|
||||
if event.mimeData().hasUrls():
|
||||
for url in event.mimeData().urls():
|
||||
file_path = url.toLocalFile().lower()
|
||||
if url.isLocalFile() and (
|
||||
file_path.endswith(".txt")
|
||||
or file_path.endswith((".srt", ".ass", ".vtt"))
|
||||
):
|
||||
self.drag_overlay.resize(self.size())
|
||||
self.drag_overlay.setVisible(True)
|
||||
event.acceptProposedAction()
|
||||
return
|
||||
self.drag_overlay.setVisible(False)
|
||||
event.ignore()
|
||||
|
||||
def dragMoveEvent(self, event):
|
||||
if event.mimeData().hasUrls():
|
||||
for url in event.mimeData().urls():
|
||||
file_path = url.toLocalFile().lower()
|
||||
if url.isLocalFile() and (
|
||||
file_path.endswith(".txt")
|
||||
or file_path.endswith((".srt", ".ass", ".vtt"))
|
||||
):
|
||||
event.acceptProposedAction()
|
||||
return
|
||||
event.ignore()
|
||||
|
||||
def dragLeaveEvent(self, event):
|
||||
self.drag_overlay.setVisible(False)
|
||||
event.accept()
|
||||
|
||||
def dropEvent(self, event):
|
||||
self.drag_overlay.setVisible(False)
|
||||
if event.mimeData().hasUrls():
|
||||
file_paths = [
|
||||
url.toLocalFile()
|
||||
for url in event.mimeData().urls()
|
||||
if url.isLocalFile()
|
||||
and (
|
||||
url.toLocalFile().lower().endswith(".txt")
|
||||
or url.toLocalFile().lower().endswith((".srt", ".ass", ".vtt"))
|
||||
)
|
||||
]
|
||||
if file_paths:
|
||||
self.parent_dialog.add_files_from_paths(file_paths)
|
||||
event.acceptProposedAction()
|
||||
else:
|
||||
event.ignore()
|
||||
else:
|
||||
event.ignore()
|
||||
|
||||
def resizeEvent(self, event):
|
||||
super().resizeEvent(event)
|
||||
if hasattr(self, "drag_overlay"):
|
||||
self.drag_overlay.resize(self.size())
|
||||
|
||||
|
||||
class QueueManager(QDialog):
|
||||
def __init__(self, parent, queue: list, title="Queue Manager", size=(600, 700)):
|
||||
super().__init__()
|
||||
self.queue = queue
|
||||
self._original_queue = deepcopy(
|
||||
queue
|
||||
) # Store a deep copy of the original queue
|
||||
self.parent = parent
|
||||
self.config = load_config() # Load config for persistence
|
||||
|
||||
layout = QVBoxLayout()
|
||||
layout.setContentsMargins(15, 15, 15, 15) # set main layout margins
|
||||
layout.setSpacing(12) # set spacing between widgets in main layout
|
||||
# list of queued items
|
||||
self.listwidget = DroppableQueueListWidget(self)
|
||||
self.listwidget.setSelectionMode(
|
||||
QAbstractItemView.SelectionMode.ExtendedSelection
|
||||
)
|
||||
self.listwidget.setAlternatingRowColors(True)
|
||||
self.listwidget.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu)
|
||||
self.listwidget.customContextMenuRequested.connect(self.show_context_menu)
|
||||
# Add informative instructions at the top
|
||||
instructions = QLabel(
|
||||
"<h2>How Queue Works?</h2>"
|
||||
"You can add text and subtitle files (.txt, .srt, .ass, .vtt) directly using the '<b>Add files</b>' button below. "
|
||||
"To add PDF, EPUB or markdown files, use the input box in the main window and click the <b>'Add to Queue'</b> button. "
|
||||
"By default, each file in the queue keeps the configuration settings active when they were added. "
|
||||
"Enabling the <b>'Override item settings with current selection'</b> option below will force all items to use the configuration currently selected in the main window. "
|
||||
"You can view each file's configuration by hovering over them."
|
||||
)
|
||||
instructions.setAlignment(Qt.AlignmentFlag.AlignLeft)
|
||||
instructions.setWordWrap(True)
|
||||
layout.addWidget(instructions)
|
||||
|
||||
# Override Checkbox
|
||||
self.override_chk = QCheckBox("Override item settings with current selection")
|
||||
self.override_chk.setToolTip(
|
||||
"If checked, all items in the queue will be processed using the \n"
|
||||
"settings currently selected in the main window, ignoring their saved state."
|
||||
)
|
||||
# Load saved state (default to False)
|
||||
self.override_chk.setChecked(self.config.get("queue_override_settings", False))
|
||||
# Trigger process_queue to update tooltips immediately when toggled
|
||||
self.override_chk.stateChanged.connect(self.process_queue)
|
||||
self.override_chk.setStyleSheet("margin-bottom: 8px;")
|
||||
layout.addWidget(self.override_chk)
|
||||
|
||||
# Overlay label for empty queue
|
||||
self.empty_overlay = QLabel(
|
||||
"Drag and drop your text or subtitle files here or use the 'Add files' button.",
|
||||
self.listwidget,
|
||||
)
|
||||
self.empty_overlay.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
self.empty_overlay.setStyleSheet(
|
||||
f"color: {COLORS['LIGHT_DISABLED']}; background: transparent; padding: 20px;"
|
||||
)
|
||||
self.empty_overlay.setWordWrap(True)
|
||||
self.empty_overlay.setAttribute(
|
||||
Qt.WidgetAttribute.WA_TransparentForMouseEvents, True
|
||||
)
|
||||
self.empty_overlay.hide()
|
||||
# add queue items to the list
|
||||
self.process_queue()
|
||||
|
||||
button_row = QHBoxLayout()
|
||||
button_row.setContentsMargins(0, 0, 0, 0) # optional: no margins for button row
|
||||
button_row.setSpacing(7) # set spacing between buttons
|
||||
# Add files button
|
||||
add_files_button = QPushButton("Add files")
|
||||
add_files_button.setFixedHeight(40)
|
||||
add_files_button.clicked.connect(self.add_more_files)
|
||||
button_row.addWidget(add_files_button)
|
||||
|
||||
# Remove button
|
||||
self.remove_button = QPushButton("Remove selected")
|
||||
self.remove_button.setFixedHeight(40)
|
||||
self.remove_button.clicked.connect(self.remove_item)
|
||||
button_row.addWidget(self.remove_button)
|
||||
|
||||
# Clear button
|
||||
self.clear_button = QPushButton("Clear Queue")
|
||||
self.clear_button.setFixedHeight(40)
|
||||
self.clear_button.clicked.connect(self.clear_queue)
|
||||
button_row.addWidget(self.clear_button)
|
||||
|
||||
layout.addLayout(button_row)
|
||||
layout.addWidget(self.listwidget)
|
||||
|
||||
# Connect selection change to update button state
|
||||
self.listwidget.currentItemChanged.connect(self.update_button_states)
|
||||
self.listwidget.itemSelectionChanged.connect(self.update_button_states)
|
||||
|
||||
buttons = QDialogButtonBox(
|
||||
QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel,
|
||||
self,
|
||||
)
|
||||
buttons.accepted.connect(self.accept)
|
||||
buttons.rejected.connect(self.reject)
|
||||
|
||||
layout.addWidget(buttons)
|
||||
|
||||
self.setLayout(layout)
|
||||
|
||||
self.setWindowTitle(title)
|
||||
self.resize(*size)
|
||||
|
||||
self.update_button_states()
|
||||
|
||||
def process_queue(self):
|
||||
"""Process the queue items."""
|
||||
import os
|
||||
|
||||
self.listwidget.clear()
|
||||
if not self.queue:
|
||||
self.empty_overlay.show()
|
||||
self.update_button_states()
|
||||
return
|
||||
else:
|
||||
self.empty_overlay.hide()
|
||||
|
||||
# Get current global settings and checkbox state for overrides
|
||||
current_global_settings = self.get_current_attributes()
|
||||
is_override_active = self.override_chk.isChecked()
|
||||
|
||||
icon_provider = QFileIconProvider()
|
||||
for item in self.queue:
|
||||
# Dynamic Attribute Retrieval Helper
|
||||
def get_val(attr, default=""):
|
||||
# If override is ON and attr is overrideable, use global setting
|
||||
if is_override_active and attr in OVERRIDE_FIELDS:
|
||||
return current_global_settings.get(attr, default)
|
||||
# Otherwise return the item's saved attribute
|
||||
return getattr(item, attr, default)
|
||||
|
||||
# Determine display file path (prefer save_base_path for original file)
|
||||
display_file_path = getattr(item, "save_base_path", None) or item.file_name
|
||||
processing_file_path = item.file_name
|
||||
|
||||
# Normalize paths for consistent display (fixes Windows path separator issues)
|
||||
display_file_path = (
|
||||
os.path.normpath(display_file_path)
|
||||
if display_file_path
|
||||
else display_file_path
|
||||
)
|
||||
processing_file_path = (
|
||||
os.path.normpath(processing_file_path)
|
||||
if processing_file_path
|
||||
else processing_file_path
|
||||
)
|
||||
|
||||
# Only show the file name, not the full path
|
||||
display_name = display_file_path
|
||||
|
||||
if os.path.sep in display_file_path:
|
||||
display_name = os.path.basename(display_file_path)
|
||||
# Get icon for the display file
|
||||
icon = icon_provider.icon(QFileInfo(display_file_path))
|
||||
list_item = QListWidgetItem()
|
||||
|
||||
# Tooltip Generation
|
||||
tooltip = ""
|
||||
# If override is active, add the warning header on its own line
|
||||
if is_override_active:
|
||||
tooltip += "<b style='color: #ff9900;'>(Global Override Active)</b><br>"
|
||||
|
||||
output_folder = get_val("output_folder")
|
||||
# For plain .txt inputs we don't need to show a separate processing file
|
||||
show_processing = True
|
||||
try:
|
||||
if isinstance(
|
||||
display_file_path, str
|
||||
) and display_file_path.lower().endswith(".txt"):
|
||||
show_processing = False
|
||||
except Exception:
|
||||
show_processing = True
|
||||
|
||||
tooltip += f"<b>Input File:</b> {display_file_path}<br>"
|
||||
if (
|
||||
show_processing
|
||||
and processing_file_path
|
||||
and processing_file_path != display_file_path
|
||||
):
|
||||
tooltip += f"<b>Processing File:</b> {processing_file_path}<br>"
|
||||
|
||||
tooltip += (
|
||||
f"<b>Language:</b> {get_val('lang_code')}<br>"
|
||||
f"<b>Speed:</b> {get_val('speed')}<br>"
|
||||
f"<b>Voice:</b> {get_val('voice')}<br>"
|
||||
f"<b>Save Option:</b> {get_val('save_option')}<br>"
|
||||
)
|
||||
if output_folder not in (None, "", "None"):
|
||||
tooltip += f"<b>Output Folder:</b> {output_folder}<br>"
|
||||
tooltip += (
|
||||
f"<b>Subtitle Mode:</b> {get_val('subtitle_mode')}<br>"
|
||||
f"<b>Output Format:</b> {get_val('output_format')}<br>"
|
||||
f"<b>Characters:</b> {getattr(item, 'total_char_count', '')}<br>"
|
||||
f"<b>Replace Single Newlines:</b> {get_val('replace_single_newlines', True)}<br>"
|
||||
f"<b>Use Silent Gaps:</b> {get_val('use_silent_gaps', False)}<br>"
|
||||
f"<b>Speed Method:</b> {get_val('subtitle_speed_method', 'tts')}"
|
||||
)
|
||||
# Add book handler options if present (Preserve logic: specific to file structure)
|
||||
save_chapters_separately = getattr(item, "save_chapters_separately", None)
|
||||
merge_chapters_at_end = getattr(item, "merge_chapters_at_end", None)
|
||||
if save_chapters_separately is not None:
|
||||
tooltip += f"<br><b>Save chapters separately:</b> {'Yes' if save_chapters_separately else 'No'}"
|
||||
# Only show merge option if saving chapters separately
|
||||
if save_chapters_separately and merge_chapters_at_end is not None:
|
||||
tooltip += f"<br><b>Merge chapters at the end:</b> {'Yes' if merge_chapters_at_end else 'No'}"
|
||||
list_item.setToolTip(tooltip)
|
||||
list_item.setIcon(icon)
|
||||
# Store both paths for context menu
|
||||
list_item.setData(
|
||||
Qt.ItemDataRole.UserRole,
|
||||
{
|
||||
"display_path": display_file_path,
|
||||
"processing_path": processing_file_path,
|
||||
},
|
||||
)
|
||||
# Use custom widget for display
|
||||
char_count = getattr(item, "total_char_count", 0)
|
||||
widget = QueueListItemWidget(display_file_path, char_count)
|
||||
self.listwidget.addItem(list_item)
|
||||
self.listwidget.setItemWidget(list_item, widget)
|
||||
self.update_button_states()
|
||||
|
||||
def remove_item(self):
|
||||
items = self.listwidget.selectedItems()
|
||||
if not items:
|
||||
return
|
||||
from PyQt6.QtWidgets import QMessageBox
|
||||
|
||||
# Remove by index to ensure correct mapping
|
||||
rows = sorted([self.listwidget.row(item) for item in items], reverse=True)
|
||||
# Warn user if removing multiple files
|
||||
if len(rows) > 1:
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
"Confirm Remove",
|
||||
f"Are you sure you want to remove {len(rows)} selected items from the queue?",
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
for row in rows:
|
||||
if 0 <= row < len(self.queue):
|
||||
del self.queue[row]
|
||||
self.process_queue()
|
||||
self.update_button_states()
|
||||
|
||||
def clear_queue(self):
|
||||
from PyQt6.QtWidgets import QMessageBox
|
||||
|
||||
if len(self.queue) > 1:
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
"Confirm Clear Queue",
|
||||
f"Are you sure you want to clear {len(self.queue)} items from the queue?",
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
self.queue.clear()
|
||||
self.listwidget.clear()
|
||||
self.empty_overlay.resize(
|
||||
self.listwidget.size()
|
||||
) # Ensure overlay is sized correctly
|
||||
self.empty_overlay.show() # Show the overlay when queue is empty
|
||||
self.update_button_states()
|
||||
|
||||
def get_queue(self):
|
||||
return self.queue
|
||||
|
||||
def get_current_attributes(self):
|
||||
# Fetch current attribute values from the parent abogen GUI
|
||||
attrs = {}
|
||||
parent = self.parent
|
||||
if parent is not None:
|
||||
# lang_code: use parent's get_voice_formula and get_selected_lang
|
||||
if hasattr(parent, "get_voice_formula") and hasattr(
|
||||
parent, "get_selected_lang"
|
||||
):
|
||||
voice_formula = parent.get_voice_formula()
|
||||
attrs["lang_code"] = parent.get_selected_lang(voice_formula)
|
||||
attrs["voice"] = voice_formula
|
||||
else:
|
||||
attrs["lang_code"] = getattr(parent, "selected_lang", "")
|
||||
attrs["voice"] = getattr(parent, "selected_voice", "")
|
||||
# speed
|
||||
if hasattr(parent, "speed_slider"):
|
||||
attrs["speed"] = parent.speed_slider.value() / 100.0
|
||||
else:
|
||||
attrs["speed"] = getattr(parent, "speed", 1.0)
|
||||
# save_option
|
||||
attrs["save_option"] = getattr(parent, "save_option", "")
|
||||
# output_folder
|
||||
attrs["output_folder"] = getattr(parent, "selected_output_folder", "")
|
||||
# subtitle_mode
|
||||
if hasattr(parent, "get_actual_subtitle_mode"):
|
||||
attrs["subtitle_mode"] = parent.get_actual_subtitle_mode()
|
||||
else:
|
||||
attrs["subtitle_mode"] = getattr(parent, "subtitle_mode", "")
|
||||
# output_format
|
||||
attrs["output_format"] = getattr(parent, "selected_format", "")
|
||||
# total_char_count
|
||||
attrs["total_char_count"] = getattr(parent, "char_count", "")
|
||||
# replace_single_newlines
|
||||
attrs["replace_single_newlines"] = getattr(
|
||||
parent, "replace_single_newlines", True
|
||||
)
|
||||
# use_silent_gaps
|
||||
attrs["use_silent_gaps"] = getattr(parent, "use_silent_gaps", False)
|
||||
# subtitle_speed_method
|
||||
attrs["subtitle_speed_method"] = getattr(
|
||||
parent, "subtitle_speed_method", "tts"
|
||||
)
|
||||
# book handler options
|
||||
attrs["save_chapters_separately"] = getattr(
|
||||
parent, "save_chapters_separately", None
|
||||
)
|
||||
attrs["merge_chapters_at_end"] = getattr(
|
||||
parent, "merge_chapters_at_end", None
|
||||
)
|
||||
else:
|
||||
# fallback: empty values
|
||||
attrs = {
|
||||
k: ""
|
||||
for k in [
|
||||
"lang_code",
|
||||
"speed",
|
||||
"voice",
|
||||
"save_option",
|
||||
"output_folder",
|
||||
"subtitle_mode",
|
||||
"output_format",
|
||||
"total_char_count",
|
||||
"replace_single_newlines",
|
||||
]
|
||||
}
|
||||
attrs["save_chapters_separately"] = None
|
||||
attrs["merge_chapters_at_end"] = None
|
||||
return attrs
|
||||
|
||||
def add_files_from_paths(self, file_paths):
|
||||
from abogen.utils import calculate_text_length
|
||||
from PyQt6.QtWidgets import QMessageBox
|
||||
import os
|
||||
|
||||
current_attrs = self.get_current_attributes()
|
||||
duplicates = []
|
||||
for file_path in file_paths:
|
||||
|
||||
class QueueItem:
|
||||
pass
|
||||
|
||||
item = QueueItem()
|
||||
item.file_name = file_path
|
||||
item.save_base_path = (
|
||||
file_path # For .txt files, processing and save paths are the same
|
||||
)
|
||||
for attr, value in current_attrs.items():
|
||||
setattr(item, attr, value)
|
||||
# Override subtitle_mode to "Disabled" for subtitle files
|
||||
if file_path.lower().endswith((".srt", ".ass", ".vtt")):
|
||||
item.subtitle_mode = "Disabled"
|
||||
# Read file content and calculate total_char_count using calculate_text_length
|
||||
try:
|
||||
with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
|
||||
file_content = f.read()
|
||||
item.total_char_count = calculate_text_length(file_content)
|
||||
except Exception:
|
||||
item.total_char_count = 0
|
||||
# Prevent adding duplicate items to the queue (check all attributes)
|
||||
is_duplicate = False
|
||||
for queued_item in self.queue:
|
||||
if (
|
||||
getattr(queued_item, "file_name", None)
|
||||
== getattr(item, "file_name", None)
|
||||
and getattr(queued_item, "lang_code", None)
|
||||
== getattr(item, "lang_code", None)
|
||||
and getattr(queued_item, "speed", None)
|
||||
== getattr(item, "speed", None)
|
||||
and getattr(queued_item, "voice", None)
|
||||
== getattr(item, "voice", None)
|
||||
and getattr(queued_item, "save_option", None)
|
||||
== getattr(item, "save_option", None)
|
||||
and getattr(queued_item, "output_folder", None)
|
||||
== getattr(item, "output_folder", None)
|
||||
and getattr(queued_item, "subtitle_mode", None)
|
||||
== getattr(item, "subtitle_mode", None)
|
||||
and getattr(queued_item, "output_format", None)
|
||||
== getattr(item, "output_format", None)
|
||||
and getattr(queued_item, "total_char_count", None)
|
||||
== getattr(item, "total_char_count", None)
|
||||
and getattr(queued_item, "replace_single_newlines", True)
|
||||
== getattr(item, "replace_single_newlines", True)
|
||||
and getattr(queued_item, "use_silent_gaps", False)
|
||||
== getattr(item, "use_silent_gaps", False)
|
||||
and getattr(queued_item, "subtitle_speed_method", "tts")
|
||||
== getattr(item, "subtitle_speed_method", "tts")
|
||||
and getattr(queued_item, "save_base_path", None)
|
||||
== getattr(item, "save_base_path", None)
|
||||
and getattr(queued_item, "save_chapters_separately", None)
|
||||
== getattr(item, "save_chapters_separately", None)
|
||||
and getattr(queued_item, "merge_chapters_at_end", None)
|
||||
== getattr(item, "merge_chapters_at_end", None)
|
||||
):
|
||||
is_duplicate = True
|
||||
break
|
||||
if is_duplicate:
|
||||
duplicates.append(os.path.basename(file_path))
|
||||
continue
|
||||
self.queue.append(item)
|
||||
if duplicates:
|
||||
QMessageBox.warning(
|
||||
self,
|
||||
"Duplicate Item(s)",
|
||||
f"Skipping {len(duplicates)} file(s) with the same attributes, already in the queue.",
|
||||
)
|
||||
self.process_queue()
|
||||
self.update_button_states()
|
||||
|
||||
def add_more_files(self):
|
||||
from PyQt6.QtWidgets import QFileDialog
|
||||
|
||||
# Allow .txt, .srt, .ass, and .vtt files
|
||||
files, _ = QFileDialog.getOpenFileNames(
|
||||
self,
|
||||
"Select text or subtitle files",
|
||||
"",
|
||||
"Supported Files (*.txt *.srt *.ass *.vtt)",
|
||||
)
|
||||
if not files:
|
||||
return
|
||||
self.add_files_from_paths(files)
|
||||
|
||||
def resizeEvent(self, event):
|
||||
super().resizeEvent(event)
|
||||
if hasattr(self, "empty_overlay"):
|
||||
self.empty_overlay.resize(self.listwidget.size())
|
||||
|
||||
def update_button_states(self):
|
||||
# Enable Remove if at least one item is selected, else disable
|
||||
if hasattr(self, "remove_button"):
|
||||
selected_count = len(self.listwidget.selectedItems())
|
||||
self.remove_button.setEnabled(selected_count > 0)
|
||||
if selected_count > 1:
|
||||
self.remove_button.setText(f"Remove selected ({selected_count})")
|
||||
else:
|
||||
self.remove_button.setText("Remove selected")
|
||||
# Disable Clear if queue is empty
|
||||
if hasattr(self, "clear_button"):
|
||||
self.clear_button.setEnabled(bool(self.queue))
|
||||
|
||||
def show_context_menu(self, pos):
|
||||
from PyQt6.QtWidgets import QMenu
|
||||
from PyQt6.QtGui import QAction, QDesktopServices
|
||||
from PyQt6.QtCore import QUrl
|
||||
import os
|
||||
|
||||
global_pos = self.listwidget.viewport().mapToGlobal(pos)
|
||||
selected_items = self.listwidget.selectedItems()
|
||||
menu = QMenu(self)
|
||||
if len(selected_items) == 1:
|
||||
# Add Remove action
|
||||
remove_action = QAction("Remove this item", self)
|
||||
remove_action.triggered.connect(self.remove_item)
|
||||
menu.addAction(remove_action)
|
||||
|
||||
# Get paths for determining if it's a document input
|
||||
item = selected_items[0]
|
||||
paths = item.data(Qt.ItemDataRole.UserRole)
|
||||
if isinstance(paths, dict):
|
||||
display_path = paths.get("display_path", "")
|
||||
processing_path = paths.get("processing_path", "")
|
||||
else:
|
||||
display_path = paths
|
||||
processing_path = paths
|
||||
|
||||
doc_exts = (".md", ".markdown", ".pdf", ".epub")
|
||||
is_document_input = (
|
||||
isinstance(display_path, str)
|
||||
and display_path.lower().endswith(doc_exts)
|
||||
) or (
|
||||
isinstance(processing_path, str)
|
||||
and processing_path.lower().endswith(doc_exts)
|
||||
)
|
||||
|
||||
# Add Open file action(s)
|
||||
def open_file_by_path(path_label: str):
|
||||
from PyQt6.QtWidgets import QMessageBox
|
||||
|
||||
p = display_path if path_label == "display" else processing_path
|
||||
if not p:
|
||||
QMessageBox.warning(
|
||||
self, "File Not Found", "Path is not available."
|
||||
)
|
||||
return
|
||||
|
||||
# Find the queue item and resolve the target path
|
||||
target_path = None
|
||||
for q in self.queue:
|
||||
if (
|
||||
getattr(q, "save_base_path", None) == display_path
|
||||
or q.file_name == display_path
|
||||
):
|
||||
if path_label == "display":
|
||||
target_path = (
|
||||
getattr(q, "save_base_path", None) or q.file_name
|
||||
)
|
||||
else:
|
||||
target_path = q.file_name
|
||||
break
|
||||
if (
|
||||
getattr(q, "save_base_path", None) == processing_path
|
||||
or q.file_name == processing_path
|
||||
):
|
||||
if path_label == "display":
|
||||
target_path = (
|
||||
getattr(q, "save_base_path", None) or q.file_name
|
||||
)
|
||||
else:
|
||||
target_path = q.file_name
|
||||
break
|
||||
|
||||
# Fallback to the raw path if resolution failed
|
||||
if not target_path:
|
||||
target_path = p
|
||||
|
||||
if not os.path.exists(target_path):
|
||||
QMessageBox.warning(
|
||||
self, "File Not Found", f"The file does not exist."
|
||||
)
|
||||
return
|
||||
QDesktopServices.openUrl(QUrl.fromLocalFile(target_path))
|
||||
|
||||
if is_document_input:
|
||||
# For documents, show two open options
|
||||
open_processed_action = QAction("Open processed file", self)
|
||||
open_processed_action.triggered.connect(
|
||||
lambda: open_file_by_path("processing")
|
||||
)
|
||||
menu.addAction(open_processed_action)
|
||||
|
||||
open_input_action = QAction("Open input file", self)
|
||||
open_input_action.triggered.connect(
|
||||
lambda: open_file_by_path("display")
|
||||
)
|
||||
menu.addAction(open_input_action)
|
||||
else:
|
||||
# For plain text files, show single open option
|
||||
open_file_action = QAction("Open file", self)
|
||||
open_file_action.triggered.connect(lambda: open_file_by_path("display"))
|
||||
menu.addAction(open_file_action)
|
||||
|
||||
# Add Go to folder action
|
||||
# If the queued item represents a converted document (markdown, pdf, epub)
|
||||
# show two actions: Go to processed file (the cached .txt) and Go to input file (original source)
|
||||
|
||||
from PyQt6.QtWidgets import QMessageBox
|
||||
|
||||
def open_folder_for(path_label: str):
|
||||
# path_label should be either 'display' or 'processing'
|
||||
p = display_path if path_label == "display" else processing_path
|
||||
if not p:
|
||||
QMessageBox.warning(
|
||||
self, "File Not Found", "Path is not available."
|
||||
)
|
||||
return
|
||||
# If the stored path is the display path (original) but the actual file may be
|
||||
# stored on the queue object differently, try to resolve via the queue entry.
|
||||
target_path = None
|
||||
for q in self.queue:
|
||||
if (
|
||||
getattr(q, "save_base_path", None) == display_path
|
||||
or q.file_name == display_path
|
||||
):
|
||||
if path_label == "display":
|
||||
target_path = (
|
||||
getattr(q, "save_base_path", None) or q.file_name
|
||||
)
|
||||
else:
|
||||
target_path = q.file_name
|
||||
break
|
||||
if (
|
||||
getattr(q, "save_base_path", None) == processing_path
|
||||
or q.file_name == processing_path
|
||||
):
|
||||
if path_label == "display":
|
||||
target_path = (
|
||||
getattr(q, "save_base_path", None) or q.file_name
|
||||
)
|
||||
else:
|
||||
target_path = q.file_name
|
||||
break
|
||||
# Fallback to the raw path if resolution failed
|
||||
if not target_path:
|
||||
target_path = p
|
||||
|
||||
if not os.path.exists(target_path):
|
||||
QMessageBox.warning(
|
||||
self,
|
||||
"File Not Found",
|
||||
f"The file does not exist: {target_path}",
|
||||
)
|
||||
return
|
||||
folder = os.path.dirname(target_path)
|
||||
if os.path.exists(folder):
|
||||
QDesktopServices.openUrl(QUrl.fromLocalFile(folder))
|
||||
|
||||
if is_document_input:
|
||||
processed_action = QAction("Go to processed file", self)
|
||||
processed_action.triggered.connect(
|
||||
lambda: open_folder_for("processing")
|
||||
)
|
||||
menu.addAction(processed_action)
|
||||
|
||||
input_action = QAction("Go to input file", self)
|
||||
input_action.triggered.connect(lambda: open_folder_for("display"))
|
||||
menu.addAction(input_action)
|
||||
else:
|
||||
# Default behavior for non-document inputs: single "Go to folder" action
|
||||
go_to_folder_action = QAction("Go to folder", self)
|
||||
|
||||
def go_to_folder():
|
||||
item = selected_items[0]
|
||||
paths = item.data(Qt.ItemDataRole.UserRole)
|
||||
if isinstance(paths, dict):
|
||||
file_path = paths.get(
|
||||
"display_path", paths.get("processing_path", "")
|
||||
)
|
||||
else:
|
||||
file_path = paths # Fallback for old format
|
||||
# Find the queue item
|
||||
for q in self.queue:
|
||||
if (
|
||||
getattr(q, "save_base_path", None) == file_path
|
||||
or q.file_name == file_path
|
||||
):
|
||||
target_path = (
|
||||
getattr(q, "save_base_path", None) or q.file_name
|
||||
)
|
||||
if not os.path.exists(target_path):
|
||||
QMessageBox.warning(
|
||||
self, "File Not Found", f"The file does not exist."
|
||||
)
|
||||
return
|
||||
folder = os.path.dirname(target_path)
|
||||
if os.path.exists(folder):
|
||||
QDesktopServices.openUrl(QUrl.fromLocalFile(folder))
|
||||
break
|
||||
|
||||
go_to_folder_action.triggered.connect(go_to_folder)
|
||||
menu.addAction(go_to_folder_action)
|
||||
|
||||
elif len(selected_items) > 1:
|
||||
remove_action = QAction(f"Remove selected ({len(selected_items)})", self)
|
||||
remove_action.triggered.connect(self.remove_item)
|
||||
menu.addAction(remove_action)
|
||||
# Always add Clear Queue
|
||||
clear_action = QAction("Clear Queue", self)
|
||||
clear_action.triggered.connect(self.clear_queue)
|
||||
menu.addAction(clear_action)
|
||||
menu.exec(global_pos)
|
||||
|
||||
def accept(self):
|
||||
# Save the override state to config so it persists globally
|
||||
self.config["queue_override_settings"] = self.override_chk.isChecked()
|
||||
save_config(self.config)
|
||||
|
||||
super().accept()
|
||||
|
||||
def reject(self):
|
||||
# Cancel: restore original queue
|
||||
from PyQt6.QtWidgets import QMessageBox
|
||||
|
||||
# Warn if user changed a lot (e.g., more than 1 items difference)
|
||||
original_count = len(self._original_queue)
|
||||
current_count = len(self.queue)
|
||||
if abs(original_count - current_count) > 1:
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
"Confirm Cancel",
|
||||
f"Are you sure you want to cancel and discard all changes?",
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
self.queue.clear()
|
||||
self.queue.extend(deepcopy(self._original_queue))
|
||||
super().reject()
|
||||
|
||||
def keyPressEvent(self, event):
|
||||
from PyQt6.QtCore import Qt
|
||||
|
||||
if event.key() == Qt.Key.Key_Delete:
|
||||
self.remove_item()
|
||||
else:
|
||||
super().keyPressEvent(event)
|
||||
@@ -0,0 +1,21 @@
|
||||
# represents a queued item - book, chapters, voice, etc.
|
||||
from dataclasses import dataclass
|
||||
|
||||
|
||||
@dataclass
|
||||
class QueuedItem:
|
||||
file_name: str
|
||||
lang_code: str
|
||||
speed: float
|
||||
voice: str
|
||||
save_option: str
|
||||
output_folder: str
|
||||
subtitle_mode: str
|
||||
output_format: str
|
||||
total_char_count: int
|
||||
replace_single_newlines: bool = True
|
||||
use_silent_gaps: bool = False
|
||||
subtitle_speed_method: str = "tts"
|
||||
save_base_path: str = None
|
||||
save_chapters_separately: bool = None
|
||||
merge_chapters_at_end: bool = None
|
||||
@@ -0,0 +1,161 @@
|
||||
"""
|
||||
Lazy-loaded spaCy utilities for sentence segmentation.
|
||||
"""
|
||||
|
||||
# Cached spaCy module and models (lazy loaded)
|
||||
_spacy = None
|
||||
_nlp_cache = {}
|
||||
|
||||
# Language code to spaCy model mapping
|
||||
SPACY_MODELS = {
|
||||
"a": "en_core_web_sm", # American English
|
||||
"b": "en_core_web_sm", # British English
|
||||
"e": "es_core_news_sm", # Spanish
|
||||
"f": "fr_core_news_sm", # French
|
||||
"i": "it_core_news_sm", # Italian
|
||||
"p": "pt_core_news_sm", # Brazilian Portuguese
|
||||
"z": "zh_core_web_sm", # Mandarin Chinese
|
||||
"j": "ja_core_news_sm", # Japanese
|
||||
"h": "xx_sent_ud_sm", # Hindi (multi-language model)
|
||||
}
|
||||
|
||||
|
||||
def _load_spacy():
|
||||
"""Lazy load spaCy module."""
|
||||
global _spacy
|
||||
if _spacy is None:
|
||||
try:
|
||||
import spacy
|
||||
|
||||
_spacy = spacy
|
||||
except ImportError:
|
||||
return None
|
||||
return _spacy
|
||||
|
||||
|
||||
def get_spacy_model(lang_code, log_callback=None):
|
||||
"""
|
||||
Get or load a spaCy model for the given language code.
|
||||
Downloads the model automatically if not available.
|
||||
|
||||
Args:
|
||||
lang_code: Language code (a, b, e, f, etc.)
|
||||
log_callback: Optional function to log messages
|
||||
|
||||
Returns:
|
||||
Loaded spaCy model or None if unavailable
|
||||
"""
|
||||
|
||||
def log(msg, is_error=False):
|
||||
# Prefer GUI log callback when provided to avoid spamming stdout.
|
||||
if log_callback:
|
||||
color = "red" if is_error else "grey"
|
||||
try:
|
||||
log_callback((msg, color))
|
||||
except Exception:
|
||||
# Fallback to printing if callback misbehaves
|
||||
print(msg)
|
||||
else:
|
||||
print(msg)
|
||||
|
||||
# Check if model is cached
|
||||
if lang_code in _nlp_cache:
|
||||
return _nlp_cache[lang_code]
|
||||
|
||||
# Check if language is supported
|
||||
model_name = SPACY_MODELS.get(lang_code)
|
||||
if not model_name:
|
||||
log(f"\nspaCy: No model mapping for language '{lang_code}'...")
|
||||
return None
|
||||
|
||||
# Lazy load spaCy
|
||||
spacy = _load_spacy()
|
||||
if spacy is None:
|
||||
log("\nspaCy: Module not installed, falling back to default segmentation...")
|
||||
return None
|
||||
|
||||
# Try to load the model
|
||||
try:
|
||||
log(f"\nLoading spaCy model '{model_name}'...")
|
||||
# sentence segmentation involving parentheses, quotes, and complex structure.
|
||||
# We only disable heavier components we don't need like NER.
|
||||
nlp = spacy.load(
|
||||
model_name,
|
||||
disable=["ner", "tagger", "lemmatizer", "attribute_ruler"],
|
||||
)
|
||||
|
||||
# Ensure a sentence segmentation strategy is in place
|
||||
# The parser provides sents, but if it's missing (unlikely for core models), fallback to sentencizer
|
||||
if "parser" not in nlp.pipe_names and "sentencizer" not in nlp.pipe_names:
|
||||
nlp.add_pipe("sentencizer")
|
||||
|
||||
_nlp_cache[lang_code] = nlp
|
||||
return nlp
|
||||
except OSError:
|
||||
# Model not found, attempt download
|
||||
log(f"\nspaCy: Downloading model '{model_name}'...")
|
||||
try:
|
||||
from spacy.cli import download
|
||||
|
||||
download(model_name)
|
||||
# Retry loading with the same fix
|
||||
nlp = spacy.load(
|
||||
model_name,
|
||||
disable=["ner", "tagger", "lemmatizer", "attribute_ruler"],
|
||||
)
|
||||
if "parser" not in nlp.pipe_names and "sentencizer" not in nlp.pipe_names:
|
||||
nlp.add_pipe("sentencizer")
|
||||
|
||||
_nlp_cache[lang_code] = nlp
|
||||
log(f"spaCy model '{model_name}' downloaded and loaded")
|
||||
return nlp
|
||||
except Exception as e:
|
||||
log(
|
||||
f"\nspaCy: Failed to download model '{model_name}': {e}...",
|
||||
is_error=True,
|
||||
)
|
||||
return None
|
||||
except Exception as e:
|
||||
log(f"\nspaCy: Error loading model '{model_name}': {e}...", is_error=True)
|
||||
return None
|
||||
|
||||
|
||||
def segment_sentences(text, lang_code, log_callback=None):
|
||||
"""
|
||||
Segment text into sentences using spaCy.
|
||||
|
||||
Args:
|
||||
text: Text to segment
|
||||
lang_code: Language code
|
||||
log_callback: Optional function to log messages
|
||||
|
||||
Returns:
|
||||
List of sentence strings, or None if spaCy unavailable
|
||||
"""
|
||||
nlp = get_spacy_model(lang_code, log_callback)
|
||||
if nlp is None:
|
||||
return None
|
||||
|
||||
# Ensure spaCy can handle large texts by adjusting max_length if necessary
|
||||
try:
|
||||
text_len = len(text or "")
|
||||
if text_len and hasattr(nlp, "max_length") and text_len > nlp.max_length:
|
||||
# increase a bit beyond the text length to be safe
|
||||
nlp.max_length = text_len + 1000
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Process text and extract sentences
|
||||
doc = nlp(text)
|
||||
return [sent.text.strip() for sent in doc.sents if sent.text.strip()]
|
||||
|
||||
|
||||
def is_spacy_available():
|
||||
"""Check if spaCy can be imported."""
|
||||
return _load_spacy() is not None
|
||||
|
||||
|
||||
def clear_cache():
|
||||
"""Clear the model cache to free memory."""
|
||||
global _nlp_cache
|
||||
_nlp_cache.clear()
|
||||
@@ -1,15 +1,41 @@
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
import warnings
|
||||
import platform
|
||||
import shutil
|
||||
import subprocess
|
||||
import re
|
||||
from threading import Thread
|
||||
|
||||
# suppress warnings and disable HF hub symlink warnings
|
||||
os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
|
||||
warnings.filterwarnings("ignore")
|
||||
|
||||
# Pre-compile frequently used regex patterns for better performance
|
||||
_WHITESPACE_PATTERN = re.compile(r"[^\S\n]+")
|
||||
_MULTIPLE_NEWLINES_PATTERN = re.compile(r"\n{3,}")
|
||||
_SINGLE_NEWLINE_PATTERN = re.compile(r"(?<!\n)\n(?!\n)")
|
||||
_CHAPTER_MARKER_PATTERN = re.compile(r"<<CHAPTER_MARKER:.*?>>")
|
||||
_METADATA_PATTERN = re.compile(r"<<METADATA_[^:]+:[^>]*>>")
|
||||
|
||||
|
||||
def detect_encoding(file_path):
|
||||
import chardet
|
||||
import charset_normalizer
|
||||
|
||||
with open(file_path, "rb") as f:
|
||||
raw_data = f.read()
|
||||
detected_encoding = None
|
||||
for detectors in (charset_normalizer, chardet):
|
||||
try:
|
||||
result = detectors.detect(raw_data)["encoding"]
|
||||
except Exception:
|
||||
continue
|
||||
if result is not None:
|
||||
detected_encoding = result
|
||||
break
|
||||
encoding = detected_encoding if detected_encoding else "utf-8"
|
||||
return encoding.lower()
|
||||
|
||||
|
||||
def get_resource_path(package, resource):
|
||||
"""
|
||||
@@ -75,33 +101,141 @@ def get_user_config_path():
|
||||
if os.path.exists(custom_dir):
|
||||
config_dir = custom_dir
|
||||
else:
|
||||
config_dir = user_config_dir("abogen", appauthor=False, roaming=True)
|
||||
config_dir = user_config_dir(
|
||||
"abogen", appauthor=False, roaming=True, ensure_exists=True
|
||||
)
|
||||
else:
|
||||
# Windows and fallback case
|
||||
config_dir = user_config_dir("abogen", appauthor=False, roaming=True)
|
||||
config_dir = user_config_dir(
|
||||
"abogen", appauthor=False, roaming=True, ensure_exists=True
|
||||
)
|
||||
|
||||
os.makedirs(config_dir, exist_ok=True)
|
||||
return os.path.join(config_dir, "config.json")
|
||||
|
||||
|
||||
# Define cache path
|
||||
def get_user_cache_path(folder=None):
|
||||
from platformdirs import user_cache_dir
|
||||
|
||||
cache_dir = user_cache_dir(
|
||||
"abogen", appauthor=False, opinion=True, ensure_exists=True
|
||||
)
|
||||
if folder:
|
||||
cache_dir = os.path.join(cache_dir, folder)
|
||||
# Ensure the directory exists
|
||||
os.makedirs(cache_dir, exist_ok=True)
|
||||
return cache_dir
|
||||
|
||||
|
||||
_sleep_procs = {"Darwin": None, "Linux": None} # Store sleep prevention processes
|
||||
|
||||
|
||||
def clean_text(text, *args, **kwargs):
|
||||
# Load replace_single_newlines from config
|
||||
cfg = load_config()
|
||||
replace_single_newlines = cfg.get("replace_single_newlines", False)
|
||||
replace_single_newlines = cfg.get("replace_single_newlines", True)
|
||||
# Collapse all whitespace (excluding newlines) into single spaces per line and trim edges
|
||||
lines = [re.sub(r"[^\S\n]+", " ", line).strip() for line in text.splitlines()]
|
||||
# Use pre-compiled pattern for better performance
|
||||
lines = [_WHITESPACE_PATTERN.sub(" ", line).strip() for line in text.splitlines()]
|
||||
text = "\n".join(lines)
|
||||
# Standardize paragraph breaks (multiple newlines become exactly two) and trim overall whitespace
|
||||
text = re.sub(r"\n{3,}", "\n\n", text).strip()
|
||||
# Use pre-compiled pattern for better performance
|
||||
text = _MULTIPLE_NEWLINES_PATTERN.sub("\n\n", text).strip()
|
||||
# Optionally replace single newlines with spaces, but preserve double newlines
|
||||
if replace_single_newlines:
|
||||
text = re.sub(r"(?<!\n)\n(?!\n)", " ", text)
|
||||
# Use pre-compiled pattern for better performance
|
||||
text = _SINGLE_NEWLINE_PATTERN.sub(" ", text)
|
||||
return text
|
||||
|
||||
|
||||
default_encoding = sys.getfilesystemencoding()
|
||||
|
||||
|
||||
def create_process(cmd, stdin=None, text=True, capture_output=False):
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Configure root logger to output to console if not already configured
|
||||
root = logging.getLogger()
|
||||
if not root.handlers:
|
||||
handler = logging.StreamHandler(sys.stdout)
|
||||
formatter = logging.Formatter("%(message)s")
|
||||
handler.setFormatter(formatter)
|
||||
root.addHandler(handler)
|
||||
root.setLevel(logging.INFO)
|
||||
|
||||
# Determine shell usage: use shell only for string commands
|
||||
use_shell = isinstance(cmd, str)
|
||||
kwargs = {
|
||||
"shell": use_shell,
|
||||
"stdout": subprocess.PIPE,
|
||||
"stderr": subprocess.STDOUT,
|
||||
"bufsize": 1, # Line buffered
|
||||
}
|
||||
|
||||
if text:
|
||||
# Configure for text I/O
|
||||
kwargs["text"] = True
|
||||
kwargs["encoding"] = default_encoding
|
||||
kwargs["errors"] = "replace"
|
||||
else:
|
||||
# Configure for binary I/O
|
||||
kwargs["text"] = False
|
||||
# For binary mode, 'encoding' and 'errors' arguments must not be passed to Popen
|
||||
kwargs["bufsize"] = 0 # Use unbuffered mode for binary data
|
||||
|
||||
if stdin is not None:
|
||||
kwargs["stdin"] = stdin
|
||||
|
||||
if platform.system() == "Windows":
|
||||
startupinfo = subprocess.STARTUPINFO()
|
||||
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
|
||||
startupinfo.wShowWindow = subprocess.SW_HIDE
|
||||
kwargs.update(
|
||||
{"startupinfo": startupinfo, "creationflags": subprocess.CREATE_NO_WINDOW}
|
||||
)
|
||||
|
||||
# Print the command being executed
|
||||
print(f"Executing: {cmd if isinstance(cmd, str) else ' '.join(cmd)}")
|
||||
|
||||
proc = subprocess.Popen(cmd, **kwargs)
|
||||
|
||||
# Stream output to console in real-time if not capturing
|
||||
if proc.stdout and not capture_output:
|
||||
|
||||
def _stream_output(stream):
|
||||
if text:
|
||||
# For text mode, read character by character for real-time output
|
||||
while True:
|
||||
char = stream.read(1)
|
||||
if not char:
|
||||
break
|
||||
# Direct write to stdout for immediate feedback
|
||||
sys.stdout.write(char)
|
||||
sys.stdout.flush()
|
||||
else:
|
||||
# For binary mode, read small chunks
|
||||
while True:
|
||||
chunk = stream.read(1) # Read byte by byte for real-time output
|
||||
if not chunk:
|
||||
break
|
||||
try:
|
||||
# Try to decode binary data for display
|
||||
sys.stdout.write(
|
||||
chunk.decode(default_encoding, errors="replace")
|
||||
)
|
||||
sys.stdout.flush()
|
||||
except Exception:
|
||||
pass
|
||||
stream.close()
|
||||
|
||||
# Start a daemon thread to handle output streaming
|
||||
Thread(target=_stream_output, args=(proc.stdout,), daemon=True).start()
|
||||
|
||||
return proc
|
||||
|
||||
|
||||
def load_config():
|
||||
try:
|
||||
with open(get_user_config_path(), "r", encoding="utf-8") as f:
|
||||
@@ -119,50 +253,91 @@ def save_config(config):
|
||||
|
||||
|
||||
def calculate_text_length(text):
|
||||
# Remove double newlines (replace them with single newlines)
|
||||
cleaned_text = text.replace("\n\n", "")
|
||||
# Use pre-compiled patterns for better performance
|
||||
# Ignore chapter markers and metadata patterns in a single pass
|
||||
text = _CHAPTER_MARKER_PATTERN.sub("", text)
|
||||
text = _METADATA_PATTERN.sub("", text)
|
||||
# Ignore newlines and leading/trailing spaces
|
||||
text = text.replace("\n", "").strip()
|
||||
# Calculate character count
|
||||
char_count = len(cleaned_text)
|
||||
char_count = len(text)
|
||||
return char_count
|
||||
|
||||
|
||||
def get_gpu_acceleration(enabled):
|
||||
from torch.cuda import is_available
|
||||
"""
|
||||
Check GPU acceleration availability.
|
||||
|
||||
Note: On Windows, torch DLLs must be pre-loaded in main.py before PyQt6
|
||||
to avoid DLL initialization errors.
|
||||
"""
|
||||
try:
|
||||
import torch
|
||||
from torch.cuda import is_available as cuda_available
|
||||
|
||||
if not enabled:
|
||||
return "CUDA GPU available but using CPU.", False
|
||||
return "GPU available but using CPU.", False
|
||||
|
||||
if is_available():
|
||||
# Check for Apple Silicon MPS
|
||||
if platform.system() == "Darwin" and platform.processor() == "arm":
|
||||
if torch.backends.mps.is_available():
|
||||
return "MPS GPU available and enabled.", True
|
||||
else:
|
||||
return "MPS GPU not available on Apple Silicon. Using CPU.", False
|
||||
|
||||
# Check for CUDA
|
||||
if cuda_available():
|
||||
return "CUDA GPU available and enabled.", True
|
||||
return "CUDA GPU is not available. Using CPU.", False
|
||||
|
||||
# Gather CUDA diagnostic info if not available
|
||||
try:
|
||||
cuda_devices = torch.cuda.device_count()
|
||||
cuda_error = (
|
||||
torch.cuda.get_device_name(0)
|
||||
if cuda_devices > 0
|
||||
else "No devices found"
|
||||
)
|
||||
except Exception as e:
|
||||
cuda_error = str(e)
|
||||
return f"CUDA GPU is not available. Using CPU. ({cuda_error})", False
|
||||
except Exception as e:
|
||||
return f"Error checking GPU: {e}", False
|
||||
|
||||
|
||||
def prevent_sleep_start():
|
||||
from abogen.constants import PROGRAM_NAME
|
||||
|
||||
system = platform.system()
|
||||
if system == "Windows":
|
||||
import ctypes
|
||||
|
||||
ctypes.windll.kernel32.SetThreadExecutionState(
|
||||
0x80000000 | 0x00000001 | 0x00000040
|
||||
) # ES_CONTINUOUS | ES_SYSTEM_REQUIRED | ES_AWAYMODE_REQUIRED
|
||||
)
|
||||
elif system == "Darwin":
|
||||
_sleep_procs["Darwin"] = subprocess.Popen(["caffeinate"])
|
||||
_sleep_procs["Darwin"] = create_process(["caffeinate"])
|
||||
elif system == "Linux":
|
||||
try:
|
||||
_sleep_procs["Linux"] = subprocess.Popen(
|
||||
# Add program name and reason for inhibition
|
||||
program_name = PROGRAM_NAME
|
||||
reason = "Prevent sleep during abogen process"
|
||||
# Only attempt to use systemd-inhibit if it's available on the system.
|
||||
if shutil.which("systemd-inhibit"):
|
||||
_sleep_procs["Linux"] = create_process(
|
||||
[
|
||||
"systemd-inhibit",
|
||||
f"--who={program_name}",
|
||||
f"--why={reason}",
|
||||
"--what=sleep",
|
||||
"--why=TextToAudiobook conversion",
|
||||
"--mode=block",
|
||||
"sleep",
|
||||
"999999",
|
||||
"infinity",
|
||||
]
|
||||
)
|
||||
except Exception:
|
||||
try:
|
||||
subprocess.Popen(["xdg-screensaver", "reset"])
|
||||
except Exception:
|
||||
pass
|
||||
else:
|
||||
# Non-systemd distro or systemd tools not installed: skip inhibition rather than crash
|
||||
print(
|
||||
"systemd-inhibit not found: skipping sleep inhibition on this Linux system."
|
||||
)
|
||||
|
||||
|
||||
def prevent_sleep_end():
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
import json
|
||||
import os
|
||||
from PyQt5.QtWidgets import (
|
||||
from PyQt6.QtWidgets import (
|
||||
QDialog,
|
||||
QVBoxLayout,
|
||||
QCheckBox,
|
||||
@@ -22,18 +22,21 @@ from PyQt5.QtWidgets import (
|
||||
QFileDialog,
|
||||
QSplitter,
|
||||
QMenu,
|
||||
QAction,
|
||||
QApplication,
|
||||
QComboBox,
|
||||
)
|
||||
from PyQt5.QtCore import Qt, QTimer, QPoint, QRect, QSize
|
||||
from PyQt5.QtGui import QPixmap, QIcon, QColor
|
||||
from constants import (
|
||||
from PyQt6.QtCore import Qt, QTimer, QPoint, QRect, QSize
|
||||
from PyQt6.QtGui import QPixmap, QIcon, QAction
|
||||
from abogen.constants import (
|
||||
VOICES_INTERNAL,
|
||||
SUPPORTED_LANGUAGES_FOR_SUBTITLE_GENERATION,
|
||||
LANGUAGE_DESCRIPTIONS,
|
||||
COLORS,
|
||||
)
|
||||
from utils import get_resource_path
|
||||
from voice_profiles import (
|
||||
import re
|
||||
import platform
|
||||
from abogen.utils import get_resource_path
|
||||
from abogen.voice_profiles import (
|
||||
load_profiles,
|
||||
save_profiles,
|
||||
delete_profile,
|
||||
@@ -87,7 +90,7 @@ class FlowLayout(QLayout):
|
||||
return len(self._item_list)
|
||||
|
||||
def expandingDirections(self):
|
||||
return Qt.Orientations(Qt.Orientation(0))
|
||||
return Qt.Orientation(0)
|
||||
|
||||
def hasHeightForWidth(self):
|
||||
return True
|
||||
@@ -128,10 +131,14 @@ class FlowLayout(QLayout):
|
||||
for item in self._item_list:
|
||||
style = self.parentWidget().style() if self.parentWidget() else QStyle()
|
||||
layout_spacing_x = style.layoutSpacing(
|
||||
QSizePolicy.PushButton, QSizePolicy.PushButton, Qt.Horizontal
|
||||
QSizePolicy.ControlType.PushButton,
|
||||
QSizePolicy.ControlType.PushButton,
|
||||
Qt.Orientation.Horizontal,
|
||||
)
|
||||
layout_spacing_y = style.layoutSpacing(
|
||||
QSizePolicy.PushButton, QSizePolicy.PushButton, Qt.Vertical
|
||||
QSizePolicy.ControlType.PushButton,
|
||||
QSizePolicy.ControlType.PushButton,
|
||||
Qt.Orientation.Vertical,
|
||||
)
|
||||
space_x = spacing if spacing >= 0 else layout_spacing_x
|
||||
space_y = spacing if spacing >= 0 else layout_spacing_y
|
||||
@@ -159,7 +166,7 @@ class VoiceMixer(QWidget):
|
||||
super().__init__()
|
||||
self.voice_name = voice_name
|
||||
self.setFixedWidth(VOICE_MIXER_WIDTH)
|
||||
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
|
||||
self.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Expanding)
|
||||
|
||||
# TODO Set CSS for rounded corners
|
||||
# self.setObjectName("VoiceMixer")
|
||||
@@ -169,7 +176,7 @@ class VoiceMixer(QWidget):
|
||||
|
||||
# Name label at the top
|
||||
name = voice_name
|
||||
layout.addWidget(QLabel(name), alignment=Qt.AlignCenter)
|
||||
layout.addWidget(QLabel(name), alignment=Qt.AlignmentFlag.AlignCenter)
|
||||
|
||||
# Voice name label with gender icon
|
||||
is_female = self.voice_name in VOICES_INTERNAL and self.voice_name[1] == "f"
|
||||
@@ -177,7 +184,9 @@ class VoiceMixer(QWidget):
|
||||
# Icons layout (flag and gender)
|
||||
icons_layout = QHBoxLayout()
|
||||
icons_layout.setSpacing(3)
|
||||
icons_layout.setAlignment(Qt.AlignCenter) # Center the icons horizontally
|
||||
icons_layout.setAlignment(
|
||||
Qt.AlignmentFlag.AlignCenter
|
||||
) # Center the icons horizontally
|
||||
|
||||
# Flag icon
|
||||
flag_icon_path = get_resource_path(
|
||||
@@ -190,11 +199,21 @@ class VoiceMixer(QWidget):
|
||||
gender_label = QLabel()
|
||||
flag_pixmap = QPixmap(flag_icon_path)
|
||||
flag_label.setPixmap(
|
||||
flag_pixmap.scaled(16, 16, Qt.KeepAspectRatio, Qt.SmoothTransformation)
|
||||
flag_pixmap.scaled(
|
||||
16,
|
||||
16,
|
||||
Qt.AspectRatioMode.KeepAspectRatio,
|
||||
Qt.TransformationMode.SmoothTransformation,
|
||||
)
|
||||
)
|
||||
gender_pixmap = QPixmap(gender_icon_path)
|
||||
gender_label.setPixmap(
|
||||
gender_pixmap.scaled(16, 16, Qt.KeepAspectRatio, Qt.SmoothTransformation)
|
||||
gender_pixmap.scaled(
|
||||
16,
|
||||
16,
|
||||
Qt.AspectRatioMode.KeepAspectRatio,
|
||||
Qt.TransformationMode.SmoothTransformation,
|
||||
)
|
||||
)
|
||||
icons_layout.addWidget(flag_label)
|
||||
icons_layout.addWidget(gender_label)
|
||||
@@ -206,7 +225,7 @@ class VoiceMixer(QWidget):
|
||||
self.checkbox = QCheckBox()
|
||||
self.checkbox.setChecked(initial_status)
|
||||
self.checkbox.stateChanged.connect(self.toggle_inputs)
|
||||
layout.addWidget(self.checkbox, alignment=Qt.AlignCenter)
|
||||
layout.addWidget(self.checkbox, alignment=Qt.AlignmentFlag.AlignCenter)
|
||||
|
||||
# Spinbox and slider
|
||||
self.spin_box = QDoubleSpinBox()
|
||||
@@ -215,12 +234,17 @@ class VoiceMixer(QWidget):
|
||||
self.spin_box.setDecimals(2)
|
||||
self.spin_box.setValue(initial_weight)
|
||||
|
||||
self.slider = QSlider(Qt.Vertical)
|
||||
self.slider = QSlider(Qt.Orientation.Vertical)
|
||||
self.slider.setRange(0, 100)
|
||||
self.slider.setValue(int(initial_weight * 100))
|
||||
self.slider.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)
|
||||
self.slider.setSizePolicy(
|
||||
QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Expanding
|
||||
)
|
||||
self.slider.setFixedWidth(SLIDER_WIDTH)
|
||||
|
||||
# Apply slider styling after widget is added to window (see showEvent)
|
||||
self._slider_style_applied = False
|
||||
|
||||
# Connect controls
|
||||
self.slider.valueChanged.connect(lambda val: self.spin_box.setValue(val / 100))
|
||||
self.spin_box.valueChanged.connect(
|
||||
@@ -230,23 +254,67 @@ class VoiceMixer(QWidget):
|
||||
# Layout for slider and labels
|
||||
slider_layout = QVBoxLayout()
|
||||
slider_layout.addWidget(self.spin_box)
|
||||
slider_layout.addWidget(QLabel("1", alignment=Qt.AlignCenter))
|
||||
slider_layout.addWidget(QLabel("1", alignment=Qt.AlignmentFlag.AlignCenter))
|
||||
|
||||
slider_center_layout = QHBoxLayout()
|
||||
slider_center_layout.addWidget(self.slider, alignment=Qt.AlignHCenter)
|
||||
slider_center_layout.addWidget(
|
||||
self.slider, alignment=Qt.AlignmentFlag.AlignHCenter
|
||||
)
|
||||
slider_center_layout.setContentsMargins(0, 0, 0, 0)
|
||||
|
||||
slider_center_widget = QWidget()
|
||||
slider_center_widget.setLayout(slider_center_layout)
|
||||
|
||||
slider_layout.addWidget(slider_center_widget, stretch=1)
|
||||
slider_layout.addWidget(QLabel("0", alignment=Qt.AlignCenter))
|
||||
slider_layout.addWidget(QLabel("0", alignment=Qt.AlignmentFlag.AlignCenter))
|
||||
slider_layout.setStretch(2, 1)
|
||||
|
||||
layout.addLayout(slider_layout, stretch=1)
|
||||
self.setLayout(layout)
|
||||
self.toggle_inputs()
|
||||
|
||||
def showEvent(self, event):
|
||||
super().showEvent(event)
|
||||
# Apply slider styling once when widget is shown and has access to parent
|
||||
if not self._slider_style_applied:
|
||||
self._slider_style_applied = True
|
||||
|
||||
# Fix slider in Windows
|
||||
if platform.system() == "Windows":
|
||||
appstyle = QApplication.instance().style().objectName().lower()
|
||||
if appstyle != "windowsvista":
|
||||
# Set custom groove color for disabled state using COLORS["GREY_BACKGROUND"]
|
||||
self.slider.setStyleSheet(
|
||||
f"""
|
||||
QSlider::groove:vertical:disabled {{
|
||||
background: {COLORS.get("GREY_BACKGROUND")};
|
||||
width: 4px;
|
||||
border-radius: 4px;
|
||||
}}
|
||||
"""
|
||||
)
|
||||
else:
|
||||
# Apply same fix for Light theme on non-Windows systems
|
||||
# Get theme from parent window's config
|
||||
parent_window = self.window()
|
||||
theme = "system"
|
||||
while parent_window:
|
||||
if hasattr(parent_window, "config"):
|
||||
theme = parent_window.config.get("theme", "system")
|
||||
break
|
||||
parent_window = parent_window.parent()
|
||||
|
||||
if theme == "light":
|
||||
self.slider.setStyleSheet(
|
||||
f"""
|
||||
QSlider::groove:vertical:disabled {{
|
||||
background: {COLORS.get("GREY_BACKGROUND")};
|
||||
width: 4px;
|
||||
border-radius: 4px;
|
||||
}}
|
||||
"""
|
||||
)
|
||||
|
||||
def toggle_inputs(self):
|
||||
is_enabled = self.checkbox.isChecked()
|
||||
self.spin_box.setEnabled(is_enabled)
|
||||
@@ -264,18 +332,16 @@ class HoverLabel(QLabel):
|
||||
self.voice_name = voice_name
|
||||
self.setMouseTracking(True)
|
||||
self.setStyleSheet(
|
||||
"background-color: #e0e0e0; border-radius: 4px; padding: 3px 6px 3px 6px; margin: 2px;"
|
||||
"background-color: rgba(140, 140, 140, 0.15); border-radius: 4px; padding: 3px 6px 3px 6px; margin: 2px;"
|
||||
)
|
||||
|
||||
# Create delete button
|
||||
self.delete_button = QPushButton("×", self)
|
||||
self.delete_button.setFixedSize(16, 16)
|
||||
self.delete_button = QPushButton("×", self)
|
||||
self.delete_button.setFixedSize(16, 16)
|
||||
self.delete_button.setStyleSheet(
|
||||
"""
|
||||
QPushButton {
|
||||
background-color: #ff5555;
|
||||
f"""
|
||||
QPushButton {{
|
||||
background-color: {COLORS.get("RED")};
|
||||
color: white;
|
||||
border-radius: 7px;
|
||||
font-weight: bold;
|
||||
@@ -283,16 +349,18 @@ class HoverLabel(QLabel):
|
||||
border: none;
|
||||
padding: 0px;
|
||||
margin: 0px;
|
||||
}
|
||||
QPushButton:hover {
|
||||
}}
|
||||
QPushButton:hover {{
|
||||
background-color: red;
|
||||
}
|
||||
}}
|
||||
"""
|
||||
)
|
||||
# Make sure the entire button is clickable, not just the text
|
||||
self.delete_button.setFocusPolicy(Qt.NoFocus)
|
||||
self.delete_button.setAttribute(Qt.WA_TransparentForMouseEvents, False)
|
||||
self.delete_button.setCursor(Qt.PointingHandCursor)
|
||||
self.delete_button.setFocusPolicy(Qt.FocusPolicy.NoFocus)
|
||||
self.delete_button.setAttribute(
|
||||
Qt.WidgetAttribute.WA_TransparentForMouseEvents, False
|
||||
)
|
||||
self.delete_button.setCursor(Qt.CursorShape.PointingHandCursor)
|
||||
self.delete_button.hide()
|
||||
|
||||
def resizeEvent(self, event):
|
||||
@@ -310,6 +378,14 @@ class HoverLabel(QLabel):
|
||||
class VoiceFormulaDialog(QDialog):
|
||||
def __init__(self, parent=None, initial_state=None, selected_profile=None):
|
||||
super().__init__(parent)
|
||||
# Store original profile/mix state for restoration on cancel
|
||||
self._original_profile_name = None
|
||||
self._original_mixed_voice_state = None
|
||||
if parent is not None:
|
||||
self._original_profile_name = getattr(parent, "selected_profile_name", None)
|
||||
self._original_mixed_voice_state = getattr(
|
||||
parent, "mixed_voice_state", None
|
||||
)
|
||||
profiles = load_profiles()
|
||||
self._virtual_new_profile = False
|
||||
if not profiles:
|
||||
@@ -332,7 +408,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
if parent is not None and hasattr(parent, "subtitle_combo"):
|
||||
self.subtitle_combo = parent.subtitle_combo
|
||||
# Create main container layout with profile section and mixer section
|
||||
splitter = QSplitter(Qt.Horizontal)
|
||||
splitter = QSplitter(Qt.Orientation.Horizontal)
|
||||
# Profile section
|
||||
profile_widget = QWidget()
|
||||
profile_layout = QVBoxLayout(profile_widget)
|
||||
@@ -346,6 +422,11 @@ class VoiceFormulaDialog(QDialog):
|
||||
profile_layout.addLayout(header_layout)
|
||||
# Profile list
|
||||
self.profile_list = QListWidget()
|
||||
self.profile_list.setSelectionMode(QListWidget.SelectionMode.SingleSelection)
|
||||
self.profile_list.setSelectionBehavior(QListWidget.SelectionBehavior.SelectRows)
|
||||
self.profile_list.setStyleSheet(
|
||||
"QListWidget::item:selected { background: palette(highlight); color: palette(highlighted-text); }"
|
||||
)
|
||||
icon = QIcon(get_resource_path("abogen.assets", "profile.png"))
|
||||
if self._virtual_new_profile:
|
||||
item = QListWidgetItem(icon, "New profile")
|
||||
@@ -358,7 +439,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
idx = list(profiles.keys()).index(self.current_profile)
|
||||
self.profile_list.setCurrentRow(idx)
|
||||
profile_layout.addWidget(self.profile_list)
|
||||
self.profile_list.setContextMenuPolicy(Qt.CustomContextMenu)
|
||||
self.profile_list.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu)
|
||||
self.profile_list.customContextMenuRequested.connect(
|
||||
self.show_profile_context_menu
|
||||
)
|
||||
@@ -379,7 +460,9 @@ class VoiceFormulaDialog(QDialog):
|
||||
|
||||
self.setWindowTitle("Voice Mixer")
|
||||
self.setWindowFlags(
|
||||
Qt.Window | Qt.WindowCloseButtonHint | Qt.WindowMaximizeButtonHint
|
||||
Qt.WindowType.Window
|
||||
| Qt.WindowType.WindowCloseButtonHint
|
||||
| Qt.WindowType.WindowMaximizeButtonHint
|
||||
)
|
||||
self.setMinimumSize(MIN_WINDOW_WIDTH, MIN_WINDOW_HEIGHT)
|
||||
self.resize(INITIAL_WINDOW_WIDTH, INITIAL_WINDOW_HEIGHT)
|
||||
@@ -438,21 +521,25 @@ class VoiceFormulaDialog(QDialog):
|
||||
|
||||
# Separator
|
||||
separator = QFrame()
|
||||
separator.setFrameShadow(QFrame.Sunken)
|
||||
separator.setFrameShadow(QFrame.Shadow.Sunken)
|
||||
mixer_layout.addWidget(separator)
|
||||
|
||||
# Voice list scroll area
|
||||
self.scroll_area = QScrollArea()
|
||||
self.scroll_area.setWidgetResizable(True)
|
||||
self.scroll_area.setHorizontalScrollBarPolicy(Qt.ScrollBarAsNeeded)
|
||||
self.scroll_area.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded)
|
||||
self.scroll_area.setHorizontalScrollBarPolicy(
|
||||
Qt.ScrollBarPolicy.ScrollBarAsNeeded
|
||||
)
|
||||
self.scroll_area.setVerticalScrollBarPolicy(
|
||||
Qt.ScrollBarPolicy.ScrollBarAsNeeded
|
||||
)
|
||||
self.scroll_area.viewport().installEventFilter(self)
|
||||
|
||||
self.voice_list_widget = QWidget()
|
||||
self.voice_list_layout = QHBoxLayout()
|
||||
self.voice_list_widget.setLayout(self.voice_list_layout)
|
||||
self.voice_list_widget.setSizePolicy(
|
||||
QSizePolicy.Expanding, QSizePolicy.Expanding
|
||||
QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Expanding
|
||||
)
|
||||
self.scroll_area.setWidget(self.voice_list_widget)
|
||||
mixer_layout.addWidget(self.scroll_area, stretch=1)
|
||||
@@ -467,6 +554,9 @@ class VoiceFormulaDialog(QDialog):
|
||||
ok_button.setDefault(True)
|
||||
ok_button.setFocus()
|
||||
|
||||
# Connect buttons
|
||||
clear_all_button.clicked.connect(self.clear_all_voices)
|
||||
ok_button.clicked.connect(self.accept)
|
||||
# Connect buttons
|
||||
clear_all_button.clicked.connect(self.clear_all_voices)
|
||||
ok_button.clicked.connect(self.accept)
|
||||
@@ -503,7 +593,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
|
||||
def keyPressEvent(self, event):
|
||||
# Bind Delete key to delete_profile when a profile is selected
|
||||
if event.key() == Qt.Key_Delete and self.profile_list.hasFocus():
|
||||
if event.key() == Qt.Key.Key_Delete and self.profile_list.hasFocus():
|
||||
item = self.profile_list.currentItem()
|
||||
if item:
|
||||
self.delete_profile(item)
|
||||
@@ -532,10 +622,12 @@ class VoiceFormulaDialog(QDialog):
|
||||
self,
|
||||
"Unsaved Changes",
|
||||
msg,
|
||||
QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel,
|
||||
QMessageBox.Save,
|
||||
QMessageBox.StandardButton.Save
|
||||
| QMessageBox.StandardButton.Discard
|
||||
| QMessageBox.StandardButton.Cancel,
|
||||
QMessageBox.StandardButton.Save,
|
||||
)
|
||||
if ret == QMessageBox.Save:
|
||||
if ret == QMessageBox.StandardButton.Save:
|
||||
# Save all using stored states
|
||||
profiles = load_profiles()
|
||||
for i in dirty_indices:
|
||||
@@ -560,7 +652,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
self.update_profile_save_buttons()
|
||||
self.update_profile_list_colors()
|
||||
return True
|
||||
elif ret == QMessageBox.Discard:
|
||||
elif ret == QMessageBox.StandardButton.Discard:
|
||||
# Discard all modifications
|
||||
self._profile_states.clear()
|
||||
for i in dirty_indices:
|
||||
@@ -582,17 +674,19 @@ class VoiceFormulaDialog(QDialog):
|
||||
else:
|
||||
# Fallback to original logic for 0 or 1 dirty profile
|
||||
box = QMessageBox(self)
|
||||
box.setIcon(QMessageBox.Warning)
|
||||
box.setIcon(QMessageBox.Icon.Warning)
|
||||
box.setWindowTitle("Unsaved Changes")
|
||||
box.setText(
|
||||
"You have unsaved changes in your profile. Do you want to save the changes?"
|
||||
)
|
||||
box.setStandardButtons(
|
||||
QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel
|
||||
QMessageBox.StandardButton.Save
|
||||
| QMessageBox.StandardButton.Discard
|
||||
| QMessageBox.StandardButton.Cancel
|
||||
)
|
||||
box.setDefaultButton(QMessageBox.Save)
|
||||
ret = box.exec_()
|
||||
if ret == QMessageBox.Save:
|
||||
box.setDefaultButton(QMessageBox.StandardButton.Save)
|
||||
ret = box.exec()
|
||||
if ret == QMessageBox.StandardButton.Save:
|
||||
for i in range(self.profile_list.count()):
|
||||
item = self.profile_list.item(i)
|
||||
name = item.text().lstrip("*")
|
||||
@@ -606,7 +700,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
if hasattr(parent, "populate_profiles_in_voice_combo"):
|
||||
parent.populate_profiles_in_voice_combo()
|
||||
return True
|
||||
elif ret == QMessageBox.Discard:
|
||||
elif ret == QMessageBox.StandardButton.Discard:
|
||||
profiles = load_profiles()
|
||||
for i in range(self.profile_list.count()):
|
||||
item = self.profile_list.item(i)
|
||||
@@ -687,7 +781,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
return voice_mixer
|
||||
|
||||
def handle_voice_checkbox(self, voice_mixer, state):
|
||||
if state == Qt.Checked:
|
||||
if state == Qt.CheckState.Checked.value:
|
||||
self.last_enabled_voice = voice_mixer.voice_name
|
||||
self.update_weighted_sums()
|
||||
|
||||
@@ -734,10 +828,12 @@ class VoiceFormulaDialog(QDialog):
|
||||
percentage = weight / total * 100
|
||||
# Make the voice name bold and include percentage
|
||||
voice_label = HoverLabel(
|
||||
f'<b><span style="color:#1976d2">{name}: {percentage:.1f}%</span></b>',
|
||||
f'<b><span style="color:{COLORS.get("BLUE")}">{name}: {percentage:.1f}%</span></b>',
|
||||
name,
|
||||
)
|
||||
voice_label.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
|
||||
voice_label.setSizePolicy(
|
||||
QSizePolicy.Policy.Preferred, QSizePolicy.Policy.Preferred
|
||||
)
|
||||
voice_label.delete_button.clicked.connect(
|
||||
lambda _, vn=name: self.disable_voice_by_name(vn)
|
||||
)
|
||||
@@ -757,7 +853,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
mixer.checkbox.setChecked(False)
|
||||
|
||||
def eventFilter(self, source, event):
|
||||
if source is self.scroll_area.viewport() and event.type() == event.Wheel:
|
||||
if source is self.scroll_area.viewport() and event.type() == event.Type.Wheel:
|
||||
# Skip if over an enabled slider
|
||||
if any(
|
||||
mixer.slider.underMouse() and mixer.slider.isEnabled()
|
||||
@@ -860,10 +956,10 @@ class VoiceFormulaDialog(QDialog):
|
||||
self,
|
||||
"Invalid Profiles",
|
||||
msg,
|
||||
QMessageBox.Yes | QMessageBox.Cancel,
|
||||
QMessageBox.Yes,
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.Cancel,
|
||||
QMessageBox.StandardButton.Yes,
|
||||
)
|
||||
if reply == QMessageBox.Yes:
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
for i, name in reversed(zero):
|
||||
self.profile_list.takeItem(i)
|
||||
delete_profile(name)
|
||||
@@ -916,6 +1012,13 @@ class VoiceFormulaDialog(QDialog):
|
||||
super().accept()
|
||||
|
||||
def reject(self):
|
||||
# Restore parent's profile/mix state on cancel
|
||||
parent = self.parent()
|
||||
if parent is not None:
|
||||
if hasattr(self, "_original_profile_name"):
|
||||
parent.selected_profile_name = self._original_profile_name
|
||||
if hasattr(self, "_original_mixed_voice_state"):
|
||||
parent.mixed_voice_state = self._original_mixed_voice_state
|
||||
# Prompt to save if unsaved changes, then check for zero-weight error after save
|
||||
if self._has_unsaved_changes():
|
||||
if not self._prompt_save_changes():
|
||||
@@ -925,6 +1028,13 @@ class VoiceFormulaDialog(QDialog):
|
||||
super().reject()
|
||||
|
||||
def closeEvent(self, event):
|
||||
# Restore parent's profile/mix state on close
|
||||
parent = self.parent()
|
||||
if parent is not None:
|
||||
if hasattr(self, "_original_profile_name"):
|
||||
parent.selected_profile_name = self._original_profile_name
|
||||
if hasattr(self, "_original_mixed_voice_state"):
|
||||
parent.mixed_voice_state = self._original_mixed_voice_state
|
||||
# Prompt to save if unsaved changes, then check for zero-weight error after save
|
||||
if self._has_unsaved_changes():
|
||||
if not self._prompt_save_changes():
|
||||
@@ -935,6 +1045,19 @@ class VoiceFormulaDialog(QDialog):
|
||||
return
|
||||
super().closeEvent(event)
|
||||
|
||||
def _parse_rgba_to_qcolor(self, rgba_str):
|
||||
from PyQt6.QtCore import Qt
|
||||
from PyQt6.QtGui import QColor
|
||||
|
||||
"""Helper to convert 'rgba(R,G,B,A_float)' string to QColor."""
|
||||
match = re.match(r"rgba\((\d+),\s*(\d+),\s*(\d+),\s*([\d.]+)\)", rgba_str)
|
||||
if match:
|
||||
r, g, b = int(match.group(1)), int(match.group(2)), int(match.group(3))
|
||||
a_float = float(match.group(4))
|
||||
a_int = int(a_float * 255)
|
||||
return QColor(r, g, b, a_int)
|
||||
return Qt.GlobalColor.transparent
|
||||
|
||||
def mark_profile_modified(self):
|
||||
item = self.profile_list.currentItem()
|
||||
if item and not item.text().startswith("*"):
|
||||
@@ -1034,7 +1157,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
self, "Import Profiles", "", "JSON Files (*.json)"
|
||||
)
|
||||
if path:
|
||||
from voice_profiles import load_profiles, save_profiles
|
||||
from abogen.voice_profiles import load_profiles, save_profiles
|
||||
|
||||
# Try to read the file and count profiles
|
||||
try:
|
||||
@@ -1079,9 +1202,12 @@ class VoiceFormulaDialog(QDialog):
|
||||
msg += f"\nThis will overwrite an existing profile."
|
||||
msg += "\nContinue?"
|
||||
reply = QMessageBox.question(
|
||||
self, "Import Profile", msg, QMessageBox.Yes | QMessageBox.No
|
||||
self,
|
||||
"Import Profile",
|
||||
msg,
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.Yes:
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
profiles.update(imported_profiles)
|
||||
save_profiles(profiles)
|
||||
@@ -1096,9 +1222,12 @@ class VoiceFormulaDialog(QDialog):
|
||||
msg += f"\n{len(collisions)} profile(s) will be overwritten."
|
||||
msg += "\nContinue?"
|
||||
reply = QMessageBox.question(
|
||||
self, "Import Profiles", msg, QMessageBox.Yes | QMessageBox.No
|
||||
self,
|
||||
"Import Profiles",
|
||||
msg,
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.Yes:
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
profiles.update(imported_profiles)
|
||||
save_profiles(profiles)
|
||||
@@ -1139,7 +1268,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
menu.addAction(dup_act)
|
||||
menu.addAction(export_act)
|
||||
menu.addAction(delete_act)
|
||||
act = menu.exec_(self.profile_list.viewport().mapToGlobal(pos))
|
||||
act = menu.exec(self.profile_list.viewport().mapToGlobal(pos))
|
||||
if act == rename_act:
|
||||
self.rename_profile(item)
|
||||
elif act == delete_act:
|
||||
@@ -1266,9 +1395,9 @@ class VoiceFormulaDialog(QDialog):
|
||||
self,
|
||||
"Delete Profile",
|
||||
f"Delete profile '{name}'?",
|
||||
QMessageBox.Yes | QMessageBox.No,
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply == QMessageBox.Yes:
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
delete_profile(name)
|
||||
row = self.profile_list.row(item)
|
||||
self.profile_list.takeItem(row)
|
||||
@@ -1321,17 +1450,24 @@ class VoiceFormulaDialog(QDialog):
|
||||
self.profile_list.setItemWidget(item, widget)
|
||||
|
||||
def update_profile_list_colors(self):
|
||||
from PyQt6.QtCore import Qt
|
||||
|
||||
profiles = load_profiles()
|
||||
for i in range(self.profile_list.count()):
|
||||
item = self.profile_list.item(i)
|
||||
name = item.text().lstrip("*")
|
||||
if self._virtual_new_profile and name == "New profile":
|
||||
item.setBackground(QColor("#fff59d")) # yellow
|
||||
color = self._parse_rgba_to_qcolor(COLORS.get("YELLOW_BACKGROUND"))
|
||||
item.setData(Qt.ItemDataRole.BackgroundRole, color)
|
||||
elif item.text().startswith("*"):
|
||||
item.setBackground(QColor("#fff59d")) # yellow
|
||||
color = self._parse_rgba_to_qcolor(COLORS.get("YELLOW_BACKGROUND"))
|
||||
item.setData(Qt.ItemDataRole.BackgroundRole, color)
|
||||
else:
|
||||
item.setData(
|
||||
Qt.ItemDataRole.BackgroundRole,
|
||||
self.profile_list.palette().base().color(),
|
||||
)
|
||||
weights = profiles.get(name, {}).get("voices", [])
|
||||
# Defensive: only sum if weights is a list of (voice, weight) pairs
|
||||
total = 0
|
||||
if isinstance(weights, list):
|
||||
for entry in weights:
|
||||
@@ -1342,9 +1478,8 @@ class VoiceFormulaDialog(QDialog):
|
||||
):
|
||||
total += entry[1]
|
||||
if total == 0:
|
||||
item.setBackground(QColor("#ffcdd2")) # light red
|
||||
else:
|
||||
item.setBackground(QColor("white"))
|
||||
color = self._parse_rgba_to_qcolor(COLORS.get("RED_BACKGROUND"))
|
||||
item.setData(Qt.ItemDataRole.BackgroundRole, color)
|
||||
self.update_profile_save_buttons()
|
||||
|
||||
def preview_current_mix(self):
|
||||
@@ -1376,6 +1511,8 @@ class VoiceFormulaDialog(QDialog):
|
||||
# Mark when playback starts
|
||||
if parent.preview_playing:
|
||||
self._started = True
|
||||
# Update button text to "Playing..." when playback starts
|
||||
self.btn_preview_mix.setText("Playing...")
|
||||
# Once started and then stopped, re-enable
|
||||
elif getattr(self, "_started", False):
|
||||
self.btn_preview_mix.setEnabled(True)
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
import re
|
||||
from constants import VOICES_INTERNAL
|
||||
from abogen.constants import VOICES_INTERNAL
|
||||
|
||||
|
||||
# Calls parsing and loads the voice to gpu or cpu
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
import os
|
||||
import json
|
||||
from utils import get_user_config_path
|
||||
from abogen.utils import get_user_config_path
|
||||
|
||||
|
||||
def _get_profiles_path():
|
||||
|
||||
@@ -0,0 +1,90 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Build PyPI package (wheel and sdist) to `dist` folder for abogen."""
|
||||
|
||||
import subprocess
|
||||
import os
|
||||
import shutil
|
||||
import tempfile
|
||||
|
||||
|
||||
def main():
|
||||
script_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
output_dir = os.path.join(script_dir, "dist")
|
||||
|
||||
print("🔧 abogen PyPI Package Builder")
|
||||
print("=" * 40)
|
||||
print(f"📁 Script directory: {script_dir}")
|
||||
print(f"📦 Output directory: {output_dir}")
|
||||
|
||||
# Try to print package version if present
|
||||
version = None
|
||||
version_file = os.path.join(script_dir, "abogen", "VERSION")
|
||||
if os.path.isfile(version_file):
|
||||
try:
|
||||
with open(version_file, "r", encoding="utf-8") as vf:
|
||||
version = vf.read().strip()
|
||||
except Exception:
|
||||
version = None
|
||||
if version:
|
||||
print(f"🔖 Package version: {version}")
|
||||
|
||||
# Check if build module is installed, install if not
|
||||
# Temporarily remove script_dir from sys.path to avoid importing local build.py
|
||||
import sys
|
||||
original_path = sys.path[:]
|
||||
try:
|
||||
sys.path = [p for p in sys.path if os.path.abspath(p) != script_dir]
|
||||
import build
|
||||
except ImportError:
|
||||
print("📦 Installing build module...")
|
||||
subprocess.run([sys.executable, "-m", "pip", "install", "build"], check=True)
|
||||
finally:
|
||||
sys.path = original_path
|
||||
|
||||
# Create output directory
|
||||
print(f"📂 Preparing output directory: {output_dir}")
|
||||
if os.path.exists(output_dir):
|
||||
shutil.rmtree(output_dir)
|
||||
os.makedirs(output_dir, exist_ok=True)
|
||||
|
||||
print("🏗️ Building PyPI package...")
|
||||
print(" Using temporary directory to avoid module conflicts...")
|
||||
|
||||
# Run from temp directory to avoid local build.py shadowing the build module
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
print(f" Temp directory: {tmpdir}")
|
||||
print(" Running: python -m build -o <output_dir> <source_dir>")
|
||||
|
||||
result = subprocess.run(
|
||||
[sys.executable, "-m", "build", "-o", output_dir, script_dir],
|
||||
check=False,
|
||||
cwd=tmpdir,
|
||||
)
|
||||
|
||||
print("\n" + "=" * 40)
|
||||
if result.returncode == 0:
|
||||
print("✅ Build successful!")
|
||||
print(f"📦 Files created in {output_dir}:")
|
||||
|
||||
files = os.listdir(output_dir)
|
||||
if files:
|
||||
for f in files:
|
||||
file_path = os.path.join(output_dir, f)
|
||||
size = os.path.getsize(file_path)
|
||||
print(f" 📄 {f} ({size:,} bytes)")
|
||||
else:
|
||||
print(" (No files found)")
|
||||
|
||||
print("\n🚀 Ready for upload with:\n")
|
||||
print(" - To test on Test PyPI:")
|
||||
print(f" python -m twine upload --repository testpypi {output_dir}/*")
|
||||
print("\n - To upload to PyPI (when ready):")
|
||||
print(f" python -m twine upload {output_dir}/*")
|
||||
else:
|
||||
print("❌ Build failed!")
|
||||
print(f" Exit code: {result.returncode}")
|
||||
sys.exit(result.returncode)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -7,9 +7,10 @@ https://github.com/user-attachments/assets/9e4fc237-a3cd-46bd-b82c-c608336d6411
|
||||
### What You Need
|
||||
|
||||
- A background image (bg.jpg)
|
||||
- The subtitle file (.srt) **(created by Abogen)**
|
||||
- The subtitle file (.ass) **(created by Abogen)**
|
||||
- Select `ass(centered narrow)` as subtitle format in settings.
|
||||
- The audio recording (.wav) **(created by Abogen)**
|
||||
- [Python](https://www.python.org/downloads/) and FFmpeg installed on your computer:
|
||||
- FFmpeg installed on your computer:
|
||||
|
||||
```bash
|
||||
# Windows
|
||||
@@ -22,30 +23,22 @@ brew install ffmpeg
|
||||
sudo apt install ffmpeg
|
||||
```
|
||||
|
||||
### Step 1: Process the Subtitle File
|
||||
## Create the Video (.mp4) (Recommended)
|
||||
|
||||
Run this command to process Abogen's subtitle file:
|
||||
Run this FFmpeg command to create the video:
|
||||
|
||||
```
|
||||
python convert.py your_subtitle.srt
|
||||
ffmpeg -loop 1 -framerate 24 -i bg.jpg -i demo.wav -vf "ass=demo.ass" -c:v libx264 -pix_fmt yuv420p -preset slow -crf 18 -c:a aac -b:a 192k -movflags +faststart -shortest demo.mp4
|
||||
```
|
||||
|
||||
This creates a properly formatted subtitle file called "your_subtitle_demo.ass" with centered text and appropriate styling.
|
||||
## For Smaller Filesize (.webm)
|
||||
|
||||
### Step 2: Create the Video (.webm)
|
||||
|
||||
Run this FFmpeg command to create the tiny video:
|
||||
If you need a smaller video file, use this command:
|
||||
|
||||
```
|
||||
ffmpeg -loop 1 -framerate 24 -i bg.jpg -i audio.wav -vf "ass=your_subtitle_demo.ass" -c:v libvpx-vp9 -b:v 0 -crf 30 -c:a libopus -shortest demo.webm
|
||||
ffmpeg -loop 1 -framerate 24 -i bg.jpg -i demo.wav -vf "ass=demo.ass" -c:v libvpx-vp9 -b:v 0 -crf 30 -c:a libopus -shortest demo.webm
|
||||
```
|
||||
|
||||
## For Higher Quality (But Larger) Video (.mp4)
|
||||
|
||||
If you need better quality for distribution, use this command instead:
|
||||
|
||||
```
|
||||
ffmpeg -loop 1 -framerate 24 -i bg.jpg -i audio.wav -vf "ass=your_subtitle_demo.ass" -c:v libx264 -preset slow -crf 18 -movflags +faststart -c:a copy -shortest demo.mp4
|
||||
```
|
||||
|
||||
This creates an MP4 file that's compatible with more devices but larger in size.
|
||||
|
||||
|
Before Width: | Height: | Size: 23 KiB After Width: | Height: | Size: 63 KiB |
|
Before Width: | Height: | Size: 41 KiB After Width: | Height: | Size: 148 KiB |
|
Before Width: | Height: | Size: 32 KiB After Width: | Height: | Size: 12 KiB |
|
Before Width: | Height: | Size: 11 KiB After Width: | Height: | Size: 31 KiB |
@@ -1,220 +0,0 @@
|
||||
# FFmpeg commands for creating videos:
|
||||
#
|
||||
# For WebM (smaller filesize):
|
||||
# ffmpeg -loop 1 -framerate 24 -i bg.jpg -i audio.wav -vf "ass=your_subtitle_demo.ass" -c:v libvpx-vp9 -b:v 0 -crf 30 -c:a libopus -shortest demo.webm
|
||||
#
|
||||
# For MP4 (higher quality):
|
||||
# ffmpeg -loop 1 -framerate 24 -i bg.jpg -i audio.wav -vf "ass=your_subtitle_demo.ass" -c:v libx264 -preset slow -crf 18 -movflags +faststart -c:a copy -shortest demo.mp4
|
||||
#
|
||||
# Note: Replace 'audio.wav' with your audio file and 'your_subtitle_demo.ass' with your processed subtitle file
|
||||
|
||||
|
||||
import re
|
||||
import sys
|
||||
from datetime import timedelta
|
||||
|
||||
|
||||
def parse_time(s):
|
||||
# For ASS format
|
||||
if ":" in s and "." in s:
|
||||
h, m, s = s.split(":")
|
||||
sec, cs = s.split(".")
|
||||
return timedelta(
|
||||
hours=int(h), minutes=int(m), seconds=int(sec), milliseconds=int(cs) * 10
|
||||
)
|
||||
# For SRT format (00:00:00,000)
|
||||
elif ":" in s and "," in s:
|
||||
parts = s.split(":")
|
||||
h = int(parts[0])
|
||||
m = int(parts[1])
|
||||
sec_parts = parts[2].split(",")
|
||||
sec = int(sec_parts[0])
|
||||
ms = int(sec_parts[1])
|
||||
return timedelta(hours=h, minutes=m, seconds=sec, milliseconds=ms)
|
||||
return None
|
||||
|
||||
|
||||
def format_time(t):
|
||||
total_seconds = int(t.total_seconds())
|
||||
cs = int((t.total_seconds() - total_seconds) * 100)
|
||||
h = total_seconds // 3600
|
||||
m = (total_seconds % 3600) // 60
|
||||
s = total_seconds % 60
|
||||
return f"{h}:{m:02}:{s:02}.{cs:02}"
|
||||
|
||||
|
||||
# Desired script info and style
|
||||
DESIRED_SCRIPT_INFO = """[Script Info]
|
||||
Title: Centered Subs
|
||||
ScriptType: v4.00+
|
||||
PlayResX: 1280
|
||||
PlayResY: 720
|
||||
"""
|
||||
|
||||
DESIRED_STYLE = """[V4+ Styles]
|
||||
Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
|
||||
Style: Default,Arial,60,&H00FFFFFF,&H000000FF,&H00000000,&H64000000,-1,0,0,0,100,100,0,0,1,2,0,5,10,10,10,1
|
||||
"""
|
||||
|
||||
|
||||
def process_subtitle_file(input_file):
|
||||
output_file = input_file.replace(".ass", "_demo.ass").replace(".srt", "_demo.ass")
|
||||
if output_file == input_file:
|
||||
output_file = f"{input_file}_demo.ass"
|
||||
|
||||
try:
|
||||
with open(input_file, "r", encoding="utf-8") as f:
|
||||
lines = f.readlines()
|
||||
except FileNotFoundError:
|
||||
print(f"❌ Error: Input file '{input_file}' not found")
|
||||
sys.exit(1)
|
||||
except Exception as e:
|
||||
print(f"❌ Error reading file: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
# Check if it's an SRT file
|
||||
is_srt = input_file.lower().endswith(".srt")
|
||||
|
||||
if is_srt:
|
||||
return convert_srt_to_ass(input_file, output_file, lines)
|
||||
|
||||
# Process ASS file
|
||||
header = []
|
||||
events = []
|
||||
in_events = False
|
||||
script_info_found = False
|
||||
styles_found = False
|
||||
|
||||
for line in lines:
|
||||
if line.strip().startswith("[Script Info]"):
|
||||
script_info_found = True
|
||||
header.append(DESIRED_SCRIPT_INFO)
|
||||
continue
|
||||
elif line.strip().startswith("[V4+ Styles]"):
|
||||
styles_found = True
|
||||
header.append(DESIRED_STYLE)
|
||||
continue
|
||||
elif line.strip().startswith("[Events]"):
|
||||
in_events = True
|
||||
header.append(line)
|
||||
elif not in_events:
|
||||
if not script_info_found and not styles_found:
|
||||
header.append(line)
|
||||
else:
|
||||
if line.startswith("Dialogue:"):
|
||||
events.append(line)
|
||||
else:
|
||||
header.append(line)
|
||||
|
||||
parsed_events = []
|
||||
for line in events:
|
||||
match = re.match(r"Dialogue: \d,([^,]+),([^,]+),", line)
|
||||
if not match:
|
||||
parsed_events.append((None, None, line))
|
||||
continue
|
||||
start = parse_time(match.group(1))
|
||||
end = parse_time(match.group(2))
|
||||
parsed_events.append((start, end, line))
|
||||
|
||||
# Cut overlaps
|
||||
fixed_lines = []
|
||||
for i in range(len(parsed_events)):
|
||||
start, end, line = parsed_events[i]
|
||||
if start is None:
|
||||
fixed_lines.append(line)
|
||||
continue
|
||||
|
||||
# Check for next subtitle
|
||||
if i + 1 < len(parsed_events):
|
||||
next_start, _, _ = parsed_events[i + 1]
|
||||
if end > next_start:
|
||||
end = next_start # cut current subtitle to stop at next one's start
|
||||
|
||||
fixed_line = re.sub(
|
||||
r"Dialogue: \d,[^,]+,[^,]+,",
|
||||
f"Dialogue: 0,{format_time(start)},{format_time(end)},",
|
||||
line,
|
||||
count=1,
|
||||
)
|
||||
fixed_lines.append(fixed_line)
|
||||
|
||||
try:
|
||||
with open(output_file, "w", encoding="utf-8") as f:
|
||||
f.writelines(header)
|
||||
f.writelines(fixed_lines)
|
||||
print(f"✅ Successfully processed. Output file: {output_file}")
|
||||
except Exception as e:
|
||||
print(f"❌ Error writing output file: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
def convert_srt_to_ass(input_file, output_file, lines):
|
||||
"""Convert SRT format to ASS format."""
|
||||
events = []
|
||||
subtitle_blocks = []
|
||||
current_block = []
|
||||
|
||||
# Parse SRT file
|
||||
for line in lines:
|
||||
line = line.strip()
|
||||
if not line:
|
||||
if current_block:
|
||||
subtitle_blocks.append(current_block)
|
||||
current_block = []
|
||||
else:
|
||||
current_block.append(line)
|
||||
|
||||
# Don't forget the last block
|
||||
if current_block:
|
||||
subtitle_blocks.append(current_block)
|
||||
|
||||
# Process subtitle blocks
|
||||
for block in subtitle_blocks:
|
||||
if len(block) < 3:
|
||||
continue # Invalid block
|
||||
|
||||
# Skip the subtitle number
|
||||
timing_line = block[1]
|
||||
|
||||
# Parse timing information
|
||||
timing_match = re.match(
|
||||
r"(\d+:\d+:\d+,\d+)\s+-->\s+(\d+:\d+:\d+,\d+)", timing_line
|
||||
)
|
||||
if not timing_match:
|
||||
continue
|
||||
|
||||
start_time = parse_time(timing_match.group(1))
|
||||
end_time = parse_time(timing_match.group(2))
|
||||
|
||||
# Combine text lines
|
||||
text = "\\N".join(block[2:])
|
||||
|
||||
# Create ASS Dialogue line
|
||||
dialogue_line = f"Dialogue: 0,{format_time(start_time)},{format_time(end_time)},Default,,0,0,0,,{text}\n"
|
||||
events.append(dialogue_line)
|
||||
|
||||
# Create ASS file
|
||||
try:
|
||||
with open(output_file, "w", encoding="utf-8") as f:
|
||||
f.write(DESIRED_SCRIPT_INFO)
|
||||
f.write(DESIRED_STYLE)
|
||||
f.write("[Events]\n")
|
||||
f.write(
|
||||
"Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text\n"
|
||||
)
|
||||
f.writelines(events)
|
||||
print(f"✅ Successfully converted SRT to ASS. Output file: {output_file}")
|
||||
return True
|
||||
except Exception as e:
|
||||
print(f"❌ Error writing output file: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) < 2:
|
||||
print("❌ Error: No input file specified")
|
||||
print(f"Usage: {sys.argv[0]} <input_file.ass|input_file.srt>")
|
||||
sys.exit(1)
|
||||
|
||||
input_file = sys.argv[1]
|
||||
process_subtitle_file(input_file)
|
||||
@@ -0,0 +1,130 @@
|
||||
[Script Info]
|
||||
Title: Generated by Abogen
|
||||
ScriptType: v4.00+
|
||||
|
||||
[Events]
|
||||
Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
|
||||
Dialogue: 0,0:00:00.35,0:00:00.93,Default,,90,90,0,,{\an5}Abogen
|
||||
Dialogue: 0,0:00:00.93,0:00:01.07,Default,,90,90,0,,{\an5}is
|
||||
Dialogue: 0,0:00:01.07,0:00:01.27,Default,,90,90,0,,{\an5}your
|
||||
Dialogue: 0,0:00:01.27,0:00:01.86,Default,,90,90,0,,{\an5}all-in-one
|
||||
Dialogue: 0,0:00:01.86,0:00:02.22,Default,,90,90,0,,{\an5}tool
|
||||
Dialogue: 0,0:00:02.22,0:00:02.35,Default,,90,90,0,,{\an5}for
|
||||
Dialogue: 0,0:00:02.35,0:00:02.73,Default,,90,90,0,,{\an5}turning
|
||||
Dialogue: 0,0:00:02.73,0:00:03.13,Default,,90,90,0,,{\an5}text
|
||||
Dialogue: 0,0:00:03.13,0:00:03.35,Default,,90,90,0,,{\an5}into
|
||||
Dialogue: 0,0:00:03.35,0:00:03.70,Default,,90,90,0,,{\an5}something
|
||||
Dialogue: 0,0:00:03.70,0:00:03.79,Default,,90,90,0,,{\an5}you
|
||||
Dialogue: 0,0:00:03.79,0:00:03.98,Default,,90,90,0,,{\an5}can
|
||||
Dialogue: 0,0:00:03.98,0:00:04.25,Default,,90,90,0,,{\an5}see
|
||||
Dialogue: 0,0:00:04.25,0:00:04.37,Default,,90,90,0,,{\an5}and
|
||||
Dialogue: 0,0:00:04.37,0:00:05.37,Default,,90,90,0,,{\an5}hear.
|
||||
Dialogue: 0,0:00:05.37,0:00:05.79,Default,,90,90,0,,{\an5}Easily
|
||||
Dialogue: 0,0:00:05.79,0:00:06.23,Default,,90,90,0,,{\an5}convert
|
||||
Dialogue: 0,0:00:06.23,0:00:07.11,Default,,90,90,0,,{\an5}ePub,
|
||||
Dialogue: 0,0:00:07.11,0:00:08.17,Default,,90,90,0,,{\an5}PDF,
|
||||
Dialogue: 0,0:00:08.17,0:00:08.33,Default,,90,90,0,,{\an5}or
|
||||
Dialogue: 0,0:00:08.33,0:00:08.67,Default,,90,90,0,,{\an5}text
|
||||
Dialogue: 0,0:00:08.67,0:00:08.98,Default,,90,90,0,,{\an5}file
|
||||
Dialogue: 0,0:00:08.98,0:00:09.23,Default,,90,90,0,,{\an5}into
|
||||
Dialogue: 0,0:00:09.23,0:00:09.91,Default,,90,90,0,,{\an5}audio
|
||||
Dialogue: 0,0:00:09.91,0:00:10.09,Default,,90,90,0,,{\an5}with
|
||||
Dialogue: 0,0:00:10.09,0:00:10.69,Default,,90,90,0,,{\an5}subtitles
|
||||
Dialogue: 0,0:00:10.69,0:00:10.87,Default,,90,90,0,,{\an5}that
|
||||
Dialogue: 0,0:00:10.87,0:00:11.06,Default,,90,90,0,,{\an5}light
|
||||
Dialogue: 0,0:00:11.06,0:00:11.26,Default,,90,90,0,,{\an5}up
|
||||
Dialogue: 0,0:00:11.26,0:00:11.59,Default,,90,90,0,,{\an5}every
|
||||
Dialogue: 0,0:00:11.59,0:00:11.90,Default,,90,90,0,,{\an5}word
|
||||
Dialogue: 0,0:00:11.90,0:00:12.02,Default,,90,90,0,,{\an5}as
|
||||
Dialogue: 0,0:00:12.02,0:00:12.17,Default,,90,90,0,,{\an5}it’s
|
||||
Dialogue: 0,0:00:12.17,0:00:13.46,Default,,90,90,0,,{\an5}spoken.
|
||||
Dialogue: 0,0:00:13.46,0:00:13.59,Default,,90,90,0,,{\an5}The
|
||||
Dialogue: 0,0:00:13.59,0:00:13.92,Default,,90,90,0,,{\an5}voice
|
||||
Dialogue: 0,0:00:13.92,0:00:14.07,Default,,90,90,0,,{\an5}you
|
||||
Dialogue: 0,0:00:14.07,0:00:14.86,Default,,90,90,0,,{\an5}hear,
|
||||
Dialogue: 0,0:00:14.86,0:00:14.98,Default,,90,90,0,,{\an5}and
|
||||
Dialogue: 0,0:00:14.98,0:00:15.08,Default,,90,90,0,,{\an5}the
|
||||
Dialogue: 0,0:00:15.08,0:00:15.67,Default,,90,90,0,,{\an5}subtitles
|
||||
Dialogue: 0,0:00:15.67,0:00:15.81,Default,,90,90,0,,{\an5}you
|
||||
Dialogue: 0,0:00:15.81,0:00:16.13,Default,,90,90,0,,{\an5}see
|
||||
Dialogue: 0,0:00:16.13,0:00:16.27,Default,,90,90,0,,{\an5}are
|
||||
Dialogue: 0,0:00:16.27,0:00:16.57,Default,,90,90,0,,{\an5}both
|
||||
Dialogue: 0,0:00:16.57,0:00:17.03,Default,,90,90,0,,{\an5}created
|
||||
Dialogue: 0,0:00:17.03,0:00:17.96,Default,,90,90,0,,{\an5}automatically
|
||||
Dialogue: 0,0:00:17.96,0:00:18.14,Default,,90,90,0,,{\an5}by
|
||||
Dialogue: 0,0:00:18.14,0:00:18.67,Default,,90,90,0,,{\an5}Abogen
|
||||
Dialogue: 0,0:00:18.67,0:00:19.02,Default,,90,90,0,,{\an5}using
|
||||
Dialogue: 0,0:00:19.02,0:00:20.10,Default,,90,90,0,,{\an5}Kokoro,
|
||||
Dialogue: 0,0:00:20.10,0:00:20.27,Default,,90,90,0,,{\an5}an
|
||||
Dialogue: 0,0:00:20.27,0:00:20.88,Default,,90,90,0,,{\an5}open-weight
|
||||
Dialogue: 0,0:00:20.88,0:00:21.55,Default,,90,90,0,,{\an5}TTS
|
||||
Dialogue: 0,0:00:21.55,0:00:21.94,Default,,90,90,0,,{\an5}model
|
||||
Dialogue: 0,0:00:21.94,0:00:22.14,Default,,90,90,0,,{\an5}with
|
||||
Dialogue: 0,0:00:22.14,0:00:22.78,Default,,90,90,0,,{\an5}82
|
||||
Dialogue: 0,0:00:22.78,0:00:23.14,Default,,90,90,0,,{\an5}million
|
||||
Dialogue: 0,0:00:23.14,0:00:24.55,Default,,90,90,0,,{\an5}parameters.
|
||||
Dialogue: 0,0:00:24.55,0:00:25.05,Default,,90,90,0,,{\an5}Everything
|
||||
Dialogue: 0,0:00:25.05,0:00:25.35,Default,,90,90,0,,{\an5}works
|
||||
Dialogue: 0,0:00:25.35,0:00:25.58,Default,,90,90,0,,{\an5}right
|
||||
Dialogue: 0,0:00:25.58,0:00:25.68,Default,,90,90,0,,{\an5}on
|
||||
Dialogue: 0,0:00:25.68,0:00:25.80,Default,,90,90,0,,{\an5}your
|
||||
Dialogue: 0,0:00:25.80,0:00:26.69,Default,,90,90,0,,{\an5}computer,
|
||||
Dialogue: 0,0:00:26.69,0:00:26.87,Default,,90,90,0,,{\an5}so
|
||||
Dialogue: 0,0:00:26.87,0:00:27.06,Default,,90,90,0,,{\an5}your
|
||||
Dialogue: 0,0:00:27.06,0:00:27.47,Default,,90,90,0,,{\an5}files
|
||||
Dialogue: 0,0:00:27.47,0:00:27.73,Default,,90,90,0,,{\an5}stay
|
||||
Dialogue: 0,0:00:27.73,0:00:28.28,Default,,90,90,0,,{\an5}private
|
||||
Dialogue: 0,0:00:28.28,0:00:28.44,Default,,90,90,0,,{\an5}and
|
||||
Dialogue: 0,0:00:28.44,0:00:29.51,Default,,90,90,0,,{\an5}secure.
|
||||
Dialogue: 0,0:00:29.90,0:00:30.03,Default,,90,90,0,,{\an5}The
|
||||
Dialogue: 0,0:00:30.03,0:00:30.37,Default,,90,90,0,,{\an5}easy
|
||||
Dialogue: 0,0:00:30.37,0:00:30.95,Default,,90,90,0,,{\an5}interface
|
||||
Dialogue: 0,0:00:30.95,0:00:31.23,Default,,90,90,0,,{\an5}lets
|
||||
Dialogue: 0,0:00:31.23,0:00:31.71,Default,,90,90,0,,{\an5}anyone
|
||||
Dialogue: 0,0:00:31.71,0:00:32.13,Default,,90,90,0,,{\an5}create
|
||||
Dialogue: 0,0:00:32.13,0:00:33.33,Default,,90,90,0,,{\an5}audiobooks,
|
||||
Dialogue: 0,0:00:33.33,0:00:34.07,Default,,90,90,0,,{\an5}voiceovers
|
||||
Dialogue: 0,0:00:34.07,0:00:34.23,Default,,90,90,0,,{\an5}for
|
||||
Dialogue: 0,0:00:34.23,0:00:35.30,Default,,90,90,0,,{\an5}Instagram,
|
||||
Dialogue: 0,0:00:35.30,0:00:35.93,Default,,90,90,0,,{\an5}YouTube
|
||||
Dialogue: 0,0:00:35.93,0:00:36.11,Default,,90,90,0,,{\an5}or
|
||||
Dialogue: 0,0:00:36.11,0:00:36.92,Default,,90,90,0,,{\an5}TikTok,
|
||||
Dialogue: 0,0:00:36.92,0:00:37.07,Default,,90,90,0,,{\an5}or
|
||||
Dialogue: 0,0:00:37.07,0:00:37.30,Default,,90,90,0,,{\an5}just
|
||||
Dialogue: 0,0:00:37.30,0:00:38.00,Default,,90,90,0,,{\an5}high-quality
|
||||
Dialogue: 0,0:00:38.00,0:00:38.82,Default,,90,90,0,,{\an5}text-to-speech
|
||||
Dialogue: 0,0:00:38.82,0:00:38.97,Default,,90,90,0,,{\an5}in
|
||||
Dialogue: 0,0:00:38.97,0:00:40.16,Default,,90,90,0,,{\an5}seconds.
|
||||
Dialogue: 0,0:00:40.16,0:00:40.45,Default,,90,90,0,,{\an5}Whether
|
||||
Dialogue: 0,0:00:40.45,0:00:40.57,Default,,90,90,0,,{\an5}you
|
||||
Dialogue: 0,0:00:40.57,0:00:40.77,Default,,90,90,0,,{\an5}want
|
||||
Dialogue: 0,0:00:40.77,0:00:40.87,Default,,90,90,0,,{\an5}to
|
||||
Dialogue: 0,0:00:40.87,0:00:41.58,Default,,90,90,0,,{\an5}listen,
|
||||
Dialogue: 0,0:00:41.58,0:00:41.92,Default,,90,90,0,,{\an5}create
|
||||
Dialogue: 0,0:00:41.92,0:00:42.65,Default,,90,90,0,,{\an5}content,
|
||||
Dialogue: 0,0:00:42.65,0:00:42.79,Default,,90,90,0,,{\an5}or
|
||||
Dialogue: 0,0:00:42.79,0:00:43.01,Default,,90,90,0,,{\an5}bring
|
||||
Dialogue: 0,0:00:43.01,0:00:43.12,Default,,90,90,0,,{\an5}your
|
||||
Dialogue: 0,0:00:43.12,0:00:43.42,Default,,90,90,0,,{\an5}words
|
||||
Dialogue: 0,0:00:43.42,0:00:43.54,Default,,90,90,0,,{\an5}to
|
||||
Dialogue: 0,0:00:43.54,0:00:44.33,Default,,90,90,0,,{\an5}life,
|
||||
Dialogue: 0,0:00:44.33,0:00:44.90,Default,,90,90,0,,{\an5}Abogen
|
||||
Dialogue: 0,0:00:44.90,0:00:45.17,Default,,90,90,0,,{\an5}makes
|
||||
Dialogue: 0,0:00:45.17,0:00:45.35,Default,,90,90,0,,{\an5}it
|
||||
Dialogue: 0,0:00:45.35,0:00:46.12,Default,,90,90,0,,{\an5}fast,
|
||||
Dialogue: 0,0:00:46.12,0:00:46.65,Default,,90,90,0,,{\an5}fun,
|
||||
Dialogue: 0,0:00:46.65,0:00:46.77,Default,,90,90,0,,{\an5}and
|
||||
Dialogue: 0,0:00:46.77,0:00:48.07,Default,,90,90,0,,{\an5}effortless.
|
||||
Dialogue: 0,0:00:48.07,0:00:48.76,Default,,90,90,0,,{\an5}Experience
|
||||
Dialogue: 0,0:00:48.76,0:00:48.87,Default,,90,90,0,,{\an5}your
|
||||
Dialogue: 0,0:00:48.87,0:00:49.61,Default,,90,90,0,,{\an5}stories,
|
||||
Dialogue: 0,0:00:49.61,0:00:50.15,Default,,90,90,0,,{\an5}scripts,
|
||||
Dialogue: 0,0:00:50.15,0:00:50.29,Default,,90,90,0,,{\an5}and
|
||||
Dialogue: 0,0:00:50.29,0:00:50.87,Default,,90,90,0,,{\an5}ideas
|
||||
Dialogue: 0,0:00:50.87,0:00:50.97,Default,,90,90,0,,{\an5}in
|
||||
Dialogue: 0,0:00:50.97,0:00:51.10,Default,,90,90,0,,{\an5}a
|
||||
Dialogue: 0,0:00:51.10,0:00:51.46,Default,,90,90,0,,{\an5}whole
|
||||
Dialogue: 0,0:00:51.46,0:00:51.65,Default,,90,90,0,,{\an5}new
|
||||
Dialogue: 0,0:00:51.65,0:00:51.95,Default,,90,90,0,,{\an5}way
|
||||
Dialogue: 0,0:00:51.95,0:00:52.10,Default,,90,90,0,,{\an5}with
|
||||
Dialogue: 0,0:00:52.10,0:00:53.20,Default,,90,90,0,,{\an5}Abogen.
|
||||
@@ -1,480 +0,0 @@
|
||||
1
|
||||
00:00:00,350 --> 00:00:00,962
|
||||
Abogen
|
||||
|
||||
2
|
||||
00:00:00,962 --> 00:00:01,100
|
||||
is
|
||||
|
||||
3
|
||||
00:00:01,100 --> 00:00:01,300
|
||||
your
|
||||
|
||||
4
|
||||
00:00:01,300 --> 00:00:01,887
|
||||
all-in-one
|
||||
|
||||
5
|
||||
00:00:01,887 --> 00:00:02,250
|
||||
tool
|
||||
|
||||
6
|
||||
00:00:02,250 --> 00:00:02,375
|
||||
for
|
||||
|
||||
7
|
||||
00:00:02,375 --> 00:00:02,762
|
||||
turning
|
||||
|
||||
8
|
||||
00:00:02,762 --> 00:00:03,162
|
||||
text
|
||||
|
||||
9
|
||||
00:00:03,162 --> 00:00:03,375
|
||||
into
|
||||
|
||||
10
|
||||
00:00:03,375 --> 00:00:03,725
|
||||
something
|
||||
|
||||
11
|
||||
00:00:03,725 --> 00:00:03,825
|
||||
you
|
||||
|
||||
12
|
||||
00:00:03,825 --> 00:00:04,025
|
||||
can
|
||||
|
||||
13
|
||||
00:00:04,025 --> 00:00:04,312
|
||||
see
|
||||
|
||||
14
|
||||
00:00:04,312 --> 00:00:04,450
|
||||
and
|
||||
|
||||
15
|
||||
00:00:04,450 --> 00:00:05,450
|
||||
hear.
|
||||
|
||||
16
|
||||
00:00:05,450 --> 00:00:05,900
|
||||
Easily
|
||||
|
||||
17
|
||||
00:00:05,900 --> 00:00:06,337
|
||||
convert
|
||||
|
||||
18
|
||||
00:00:06,337 --> 00:00:07,187
|
||||
ePub,
|
||||
|
||||
19
|
||||
00:00:07,187 --> 00:00:08,250
|
||||
PDF,
|
||||
|
||||
20
|
||||
00:00:08,250 --> 00:00:08,412
|
||||
or
|
||||
|
||||
21
|
||||
00:00:08,412 --> 00:00:08,800
|
||||
text
|
||||
|
||||
22
|
||||
00:00:08,800 --> 00:00:09,162
|
||||
file
|
||||
|
||||
23
|
||||
00:00:09,162 --> 00:00:09,425
|
||||
into
|
||||
|
||||
24
|
||||
00:00:09,425 --> 00:00:10,037
|
||||
audio
|
||||
|
||||
25
|
||||
00:00:10,037 --> 00:00:10,237
|
||||
with
|
||||
|
||||
26
|
||||
00:00:10,237 --> 00:00:10,875
|
||||
subtitles
|
||||
|
||||
27
|
||||
00:00:10,875 --> 00:00:11,087
|
||||
that
|
||||
|
||||
28
|
||||
00:00:11,087 --> 00:00:11,287
|
||||
light
|
||||
|
||||
29
|
||||
00:00:11,287 --> 00:00:11,487
|
||||
up
|
||||
|
||||
30
|
||||
00:00:11,487 --> 00:00:11,824
|
||||
every
|
||||
|
||||
31
|
||||
00:00:11,824 --> 00:00:12,175
|
||||
word
|
||||
|
||||
32
|
||||
00:00:12,175 --> 00:00:12,300
|
||||
as
|
||||
|
||||
33
|
||||
00:00:12,300 --> 00:00:12,449
|
||||
it’s
|
||||
|
||||
34
|
||||
00:00:12,449 --> 00:00:13,787
|
||||
spoken.
|
||||
|
||||
35
|
||||
00:00:13,787 --> 00:00:13,925
|
||||
The
|
||||
|
||||
36
|
||||
00:00:13,925 --> 00:00:14,275
|
||||
voice
|
||||
|
||||
37
|
||||
00:00:14,275 --> 00:00:14,425
|
||||
you
|
||||
|
||||
38
|
||||
00:00:14,425 --> 00:00:15,212
|
||||
hear,
|
||||
|
||||
39
|
||||
00:00:15,212 --> 00:00:15,337
|
||||
and
|
||||
|
||||
40
|
||||
00:00:15,337 --> 00:00:15,437
|
||||
the
|
||||
|
||||
41
|
||||
00:00:15,437 --> 00:00:16,024
|
||||
subtitles
|
||||
|
||||
42
|
||||
00:00:16,024 --> 00:00:16,162
|
||||
you
|
||||
|
||||
43
|
||||
00:00:16,162 --> 00:00:16,487
|
||||
see
|
||||
|
||||
44
|
||||
00:00:16,487 --> 00:00:16,649
|
||||
are
|
||||
|
||||
45
|
||||
00:00:16,649 --> 00:00:16,975
|
||||
both
|
||||
|
||||
46
|
||||
00:00:16,975 --> 00:00:17,449
|
||||
created
|
||||
|
||||
47
|
||||
00:00:17,449 --> 00:00:18,362
|
||||
automatically
|
||||
|
||||
48
|
||||
00:00:18,362 --> 00:00:18,550
|
||||
by
|
||||
|
||||
49
|
||||
00:00:18,550 --> 00:00:19,074
|
||||
Abogen
|
||||
|
||||
50
|
||||
00:00:19,074 --> 00:00:19,425
|
||||
using
|
||||
|
||||
51
|
||||
00:00:19,425 --> 00:00:20,512
|
||||
Kokoro,
|
||||
|
||||
52
|
||||
00:00:20,512 --> 00:00:20,637
|
||||
a
|
||||
|
||||
53
|
||||
00:00:20,637 --> 00:00:21,237
|
||||
powerful
|
||||
|
||||
54
|
||||
00:00:21,237 --> 00:00:22,050
|
||||
text-to-speech
|
||||
|
||||
55
|
||||
00:00:22,050 --> 00:00:23,225
|
||||
engine.
|
||||
|
||||
56
|
||||
00:00:23,225 --> 00:00:23,750
|
||||
Everything
|
||||
|
||||
57
|
||||
00:00:23,750 --> 00:00:24,062
|
||||
works
|
||||
|
||||
58
|
||||
00:00:24,062 --> 00:00:24,312
|
||||
right
|
||||
|
||||
59
|
||||
00:00:24,312 --> 00:00:24,412
|
||||
on
|
||||
|
||||
60
|
||||
00:00:24,412 --> 00:00:24,524
|
||||
your
|
||||
|
||||
61
|
||||
00:00:24,524 --> 00:00:25,437
|
||||
computer,
|
||||
|
||||
62
|
||||
00:00:25,437 --> 00:00:25,625
|
||||
so
|
||||
|
||||
63
|
||||
00:00:25,625 --> 00:00:25,812
|
||||
your
|
||||
|
||||
64
|
||||
00:00:25,812 --> 00:00:26,225
|
||||
files
|
||||
|
||||
65
|
||||
00:00:26,225 --> 00:00:26,524
|
||||
stay
|
||||
|
||||
66
|
||||
00:00:26,524 --> 00:00:27,112
|
||||
private
|
||||
|
||||
67
|
||||
00:00:27,112 --> 00:00:27,274
|
||||
and
|
||||
|
||||
68
|
||||
00:00:27,274 --> 00:00:28,337
|
||||
secure.
|
||||
|
||||
69
|
||||
00:00:28,725 --> 00:00:28,862
|
||||
The
|
||||
|
||||
70
|
||||
00:00:28,862 --> 00:00:29,225
|
||||
easy
|
||||
|
||||
71
|
||||
00:00:29,225 --> 00:00:29,824
|
||||
interface
|
||||
|
||||
72
|
||||
00:00:29,824 --> 00:00:30,112
|
||||
lets
|
||||
|
||||
73
|
||||
00:00:30,112 --> 00:00:30,637
|
||||
anyone
|
||||
|
||||
74
|
||||
00:00:30,637 --> 00:00:31,062
|
||||
create
|
||||
|
||||
75
|
||||
00:00:31,062 --> 00:00:32,299
|
||||
audiobooks,
|
||||
|
||||
76
|
||||
00:00:32,299 --> 00:00:33,075
|
||||
voiceovers
|
||||
|
||||
77
|
||||
00:00:33,075 --> 00:00:33,237
|
||||
for
|
||||
|
||||
78
|
||||
00:00:33,237 --> 00:00:34,325
|
||||
Instagram,
|
||||
|
||||
79
|
||||
00:00:34,325 --> 00:00:34,962
|
||||
YouTube
|
||||
|
||||
80
|
||||
00:00:34,962 --> 00:00:35,137
|
||||
or
|
||||
|
||||
81
|
||||
00:00:35,137 --> 00:00:35,950
|
||||
TikTok,
|
||||
|
||||
82
|
||||
00:00:35,950 --> 00:00:36,100
|
||||
or
|
||||
|
||||
83
|
||||
00:00:36,100 --> 00:00:36,325
|
||||
just
|
||||
|
||||
84
|
||||
00:00:36,325 --> 00:00:37,024
|
||||
high-quality
|
||||
|
||||
85
|
||||
00:00:37,024 --> 00:00:37,850
|
||||
text-to-speech
|
||||
|
||||
86
|
||||
00:00:37,850 --> 00:00:38,000
|
||||
in
|
||||
|
||||
87
|
||||
00:00:38,000 --> 00:00:39,187
|
||||
seconds.
|
||||
|
||||
88
|
||||
00:00:39,187 --> 00:00:39,475
|
||||
Whether
|
||||
|
||||
89
|
||||
00:00:39,475 --> 00:00:39,625
|
||||
you
|
||||
|
||||
90
|
||||
00:00:39,625 --> 00:00:39,825
|
||||
want
|
||||
|
||||
91
|
||||
00:00:39,825 --> 00:00:39,924
|
||||
to
|
||||
|
||||
92
|
||||
00:00:39,924 --> 00:00:40,662
|
||||
listen,
|
||||
|
||||
93
|
||||
00:00:40,662 --> 00:00:41,000
|
||||
create
|
||||
|
||||
94
|
||||
00:00:41,000 --> 00:00:41,774
|
||||
content,
|
||||
|
||||
95
|
||||
00:00:41,774 --> 00:00:41,924
|
||||
or
|
||||
|
||||
96
|
||||
00:00:41,924 --> 00:00:42,137
|
||||
bring
|
||||
|
||||
97
|
||||
00:00:42,137 --> 00:00:42,250
|
||||
your
|
||||
|
||||
98
|
||||
00:00:42,250 --> 00:00:42,575
|
||||
words
|
||||
|
||||
99
|
||||
00:00:42,575 --> 00:00:42,700
|
||||
to
|
||||
|
||||
100
|
||||
00:00:42,700 --> 00:00:43,512
|
||||
life,
|
||||
|
||||
101
|
||||
00:00:43,512 --> 00:00:44,075
|
||||
Abogen
|
||||
|
||||
102
|
||||
00:00:44,075 --> 00:00:44,350
|
||||
makes
|
||||
|
||||
103
|
||||
00:00:44,350 --> 00:00:44,525
|
||||
it
|
||||
|
||||
104
|
||||
00:00:44,525 --> 00:00:45,299
|
||||
fast,
|
||||
|
||||
105
|
||||
00:00:45,299 --> 00:00:45,825
|
||||
fun,
|
||||
|
||||
106
|
||||
00:00:45,825 --> 00:00:45,950
|
||||
and
|
||||
|
||||
107
|
||||
00:00:45,950 --> 00:00:47,275
|
||||
effortless.
|
||||
|
||||
108
|
||||
00:00:47,275 --> 00:00:47,962
|
||||
Experience
|
||||
|
||||
109
|
||||
00:00:47,962 --> 00:00:48,075
|
||||
your
|
||||
|
||||
110
|
||||
00:00:48,075 --> 00:00:48,812
|
||||
stories,
|
||||
|
||||
111
|
||||
00:00:48,812 --> 00:00:49,350
|
||||
scripts,
|
||||
|
||||
112
|
||||
00:00:49,350 --> 00:00:49,500
|
||||
and
|
||||
|
||||
113
|
||||
00:00:49,500 --> 00:00:50,075
|
||||
ideas
|
||||
|
||||
114
|
||||
00:00:50,075 --> 00:00:50,174
|
||||
in
|
||||
|
||||
115
|
||||
00:00:50,174 --> 00:00:50,299
|
||||
a
|
||||
|
||||
116
|
||||
00:00:50,299 --> 00:00:50,662
|
||||
whole
|
||||
|
||||
117
|
||||
00:00:50,662 --> 00:00:50,850
|
||||
new
|
||||
|
||||
118
|
||||
00:00:50,850 --> 00:00:51,150
|
||||
way
|
||||
|
||||
119
|
||||
00:00:51,150 --> 00:00:51,299
|
||||
with
|
||||
|
||||
120
|
||||
00:00:51,299 --> 00:00:52,424
|
||||
Abogen.
|
||||
|
||||
|
After Width: | Height: | Size: 103 KiB |
|
Before Width: | Height: | Size: 28 KiB After Width: | Height: | Size: 69 KiB |
@@ -13,8 +13,10 @@ license = "MIT"
|
||||
requires-python = ">=3.10, <3.13"
|
||||
keywords = ["audiobook", "epub", "pdf", "text-to-speech", "subtitle", "tts", "kokoro", "accessibility", "book-converter", "voice-synthesis", "multilingual", "chapter-management", "subtitles", "content-creation", "media-generation"]
|
||||
dependencies = [
|
||||
"PyQt5>=5.15.11",
|
||||
"pip",
|
||||
"PyQt6>=6.10.0",
|
||||
"kokoro>=0.9.4",
|
||||
"misaki[zh]>=0.9.4",
|
||||
"ebooklib>=0.19",
|
||||
"beautifulsoup4>=4.13.4",
|
||||
"PyMuPDF>=1.25.5",
|
||||
@@ -23,7 +25,9 @@ dependencies = [
|
||||
"pygame>=2.6.1",
|
||||
"charset_normalizer>=3.4.1",
|
||||
"chardet>=5.2.0",
|
||||
"static_ffmpeg>=2.13"
|
||||
"static_ffmpeg>=2.13",
|
||||
"Markdown>=3.9",
|
||||
"spacy>=3.8.7"
|
||||
]
|
||||
|
||||
classifiers = [
|
||||
@@ -67,3 +71,77 @@ packages = ["abogen"]
|
||||
[tool.hatch.version]
|
||||
path = "abogen/VERSION"
|
||||
pattern = "^(?P<version>.+)$"
|
||||
|
||||
# --- OPTIONAL DEPENDENCIES ---
|
||||
|
||||
[project.optional-dependencies]
|
||||
# NVIDIA GPU (Windows) (CUDA 12.6) # uv tool install abogen[cuda126]
|
||||
cuda126 = ["torch"]
|
||||
# NVIDIA GPU (Windows) (CUDA 12.8) # uv tool install abogen[cuda]
|
||||
cuda = ["torch"]
|
||||
# NVIDIA GPU (Windows) (CUDA 13.0) # uv tool install abogen[cuda130]
|
||||
cuda130 = ["torch"]
|
||||
# AMD GPU (Linux) (ROCm 6.4) # uv tool install abogen[rocm]
|
||||
rocm = ["torch", "pytorch-triton-rocm"]
|
||||
|
||||
# --- KOKORO CONFIGURATION (for macOS) ---
|
||||
|
||||
[tool.uv.sources]
|
||||
kokoro = [
|
||||
{ git = "https://github.com/hexgrad/kokoro.git", marker = "sys_platform == 'darwin'" },
|
||||
{ index = "pypi", marker = "sys_platform != 'darwin'" }
|
||||
]
|
||||
|
||||
# --- TORCH CONFIGURATION ---
|
||||
|
||||
torch = [
|
||||
# ROCm 6.4 Nightly (AMD)
|
||||
{ index = "pytorch-rocm-64-nightly", marker = "extra == 'rocm'" },
|
||||
|
||||
# CUDA 13.0 (NVIDIA)
|
||||
{ index = "pytorch-cuda-130", marker = "extra == 'cuda130' and extra != 'rocm'" },
|
||||
|
||||
# CUDA 12.6 (NVIDIA)
|
||||
{ index = "pytorch-cuda-126", marker = "extra == 'cuda126' and extra != 'rocm' and extra != 'cuda130'" },
|
||||
|
||||
# CUDA 12.8 (NVIDIA)
|
||||
{ index = "pytorch-cuda-128", marker = "extra == 'cuda' and extra != 'rocm' and extra != 'cuda130' and extra != 'cuda126'" }
|
||||
]
|
||||
|
||||
# --- TRITON CONFIGURATION ---
|
||||
|
||||
pytorch-triton-rocm = [
|
||||
{ index = "pytorch-rocm-64-nightly", marker = "extra == 'rocm'" }
|
||||
]
|
||||
|
||||
# --- INDEX DEFINITIONS ---
|
||||
|
||||
# PyPI Index
|
||||
[[tool.uv.index]]
|
||||
name = "pypi"
|
||||
url = "https://pypi.org/simple"
|
||||
default = true
|
||||
|
||||
# CUDA 12.6 Index
|
||||
[[tool.uv.index]]
|
||||
name = "pytorch-cuda-126"
|
||||
url = "https://download.pytorch.org/whl/cu126"
|
||||
explicit = true
|
||||
|
||||
# CUDA 12.8 Index
|
||||
[[tool.uv.index]]
|
||||
name = "pytorch-cuda-128"
|
||||
url = "https://download.pytorch.org/whl/cu128"
|
||||
explicit = true
|
||||
|
||||
# CUDA 13.0 Index
|
||||
[[tool.uv.index]]
|
||||
name = "pytorch-cuda-130"
|
||||
url = "https://download.pytorch.org/whl/cu130"
|
||||
explicit = true
|
||||
|
||||
# ROCm 6.4 Nightly Index
|
||||
[[tool.uv.index]]
|
||||
name = "pytorch-rocm-64-nightly"
|
||||
url = "https://download.pytorch.org/whl/nightly/rocm6.4"
|
||||
explicit = true
|
||||
|
||||