Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
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 | ||
|
|
8f8ad3ec26 | ||
|
|
67b35d1dfb | ||
|
|
15f1eb7f1e | ||
|
|
146fcb3c1f | ||
|
|
412fb3bc7e | ||
|
|
d9b6081a06 | ||
|
|
674fc14813 | ||
|
|
fa144561fb | ||
|
|
1d27f6fd2e | ||
|
|
a03120d66f | ||
|
|
68b44a3516 | ||
|
|
3bbb88d446 | ||
|
|
a1ccd51ea0 | ||
|
|
13cf6f20a4 | ||
|
|
4311ed51d4 | ||
|
|
9170d9ae3e | ||
|
|
593d21675e | ||
|
|
9aebdecc60 | ||
|
|
290d246e80 | ||
|
|
5ac6e3bb2a | ||
|
|
6bf189312a | ||
|
|
22fdccd78d | ||
|
|
62aeaecf89 | ||
|
|
a7a41fa4f6 | ||
|
|
1ca4d0d5f9 | ||
|
|
e09535fa83 | ||
|
|
0863b25d82 | ||
|
|
b02416e068 | ||
|
|
8daa3cb3b5 | ||
|
|
20f31f82fd | ||
|
|
d5388c62be | ||
|
|
07a2fe2d5b | ||
|
|
58d9073ace | ||
|
|
2cbb2f8aa7 | ||
|
|
f7d8d5ccd8 | ||
|
|
e7d479d846 | ||
|
|
036eed3142 | ||
|
|
0d86d107e9 | ||
|
|
56b336a529 | ||
|
|
b1b9b8d42f | ||
|
|
18f95c74e0 | ||
|
|
3a3f6e34e5 | ||
|
|
02b9990004 | ||
|
|
4acf70954e | ||
|
|
04e16722e2 | ||
|
|
68dbdf123d | ||
|
|
854ad1adea | ||
|
|
7cf3884a4b | ||
|
|
9e75870c6d | ||
|
|
28813b8975 | ||
|
|
7c55a6f6fa | ||
|
|
0a7d352209 | ||
|
|
d500a42d8e | ||
|
|
be4a258589 | ||
|
|
f1677dcaf4 | ||
|
|
82fe975d3c | ||
|
|
3497c01053 | ||
|
|
06ad35fd74 | ||
|
|
b0f3589618 | ||
|
|
8cff2ad980 | ||
|
|
ed9036b7b7 | ||
|
|
d18933d362 | ||
|
|
27cdd264b1 | ||
|
|
fe09b2140b | ||
|
|
4462ee7ed4 | ||
|
|
31c8d00ae7 | ||
|
|
8583f229f1 | ||
|
|
44e555c83e | ||
|
|
4a7f1767b9 | ||
|
|
572a3c7285 | ||
|
|
a6bdc9c086 | ||
|
|
7135ea6222 | ||
|
|
bb3cb27b25 |
@@ -1,5 +1,71 @@
|
|||||||
- Added `abogen-cli` command for better troubleshooting and error handling.
|
# 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.
|
- Switched from setuptools to hatchling for packaging.
|
||||||
- Added classifiers to the package metadata.
|
- 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 https://github.com/denizsafak/abogen/issues/2
|
- 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.
|
- Improvements in code and documentation.
|
||||||
|
|||||||
@@ -1,7 +1,8 @@
|
|||||||
# abogen <img width="40px" title="abogen icon" src="https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/abogen/assets/icon.ico" align="right" style="padding-left: 10px; padding-top:5px;">
|
# abogen <img width="40px" title="abogen icon" src="https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/abogen/assets/icon.ico" align="right" style="padding-left: 10px; padding-top:5px;">
|
||||||
|
|
||||||
[](https://github.com/denizsafak/abogen/actions)
|
[](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/denizsafak/abogen/releases/latest)
|
||||||
[](https://github.com/psf/black)
|
[](https://github.com/psf/black)
|
||||||
[](https://opensource.org/licenses/MIT)
|
[](https://opensource.org/licenses/MIT)
|
||||||
@@ -13,16 +14,18 @@ Abogen is a powerful text-to-speech conversion tool that makes it easy to turn e
|
|||||||
## Demo
|
## Demo
|
||||||
https://github.com/user-attachments/assets/cb66512d-0a52-48c3-bda4-f1e6a03fb8d6
|
https://github.com/user-attachments/assets/cb66512d-0a52-48c3-bda4-f1e6a03fb8d6
|
||||||
|
|
||||||
|
|
||||||
> 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).
|
> 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?`
|
## `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>
|
||||||
### Windows
|
### Windows
|
||||||
Go to [espeak-ng latest release](https://github.com/espeak-ng/espeak-ng/releases/latest) download and run the *.msi file.
|
Go to [espeak-ng latest release](https://github.com/espeak-ng/espeak-ng/releases/latest) download and run the *.msi file.
|
||||||
```bash
|
```bash
|
||||||
# For NVIDIA GPUs:
|
# For NVIDIA GPUs:
|
||||||
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
pip install torch torchvision torchaudio --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
|
# Install abogen
|
||||||
pip install abogen
|
pip install abogen
|
||||||
```
|
```
|
||||||
@@ -54,20 +57,36 @@ sudo dnf install espeak-ng
|
|||||||
|
|
||||||
# Install abogen
|
# Install abogen
|
||||||
pip 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.
|
|
||||||
|
|
||||||
Then simply run by typing:
|
# For NVIDIA GPUs:
|
||||||
|
# Already supported, no need to install CUDA separately.
|
||||||
|
|
||||||
|
# For AMD GPUs:
|
||||||
|
pip uninstall torch # After installing abogen, we need to uninstall the existing torch package
|
||||||
|
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4
|
||||||
|
```
|
||||||
|
> If you get `WARNING: The script abogen-cli is installed in '/home/username/.local/bin' which is not on PATH.` error, run the following command to add it to your PATH:
|
||||||
|
>```bash
|
||||||
|
>echo "export PATH=\"/home/$USER/.local/bin:\$PATH\"" >> ~/.bashrc && source ~/.bashrc
|
||||||
|
>```
|
||||||
|
|
||||||
|
> 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.
|
||||||
|
|
||||||
|
> 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?`
|
||||||
|
If you installed using pip, you can simply run the following command to start Abogen:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
abogen
|
abogen
|
||||||
```
|
```
|
||||||
|
> If you installed 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?`
|
## `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, or text file (or use the built-in text editor)
|
||||||
2) Configure the settings:
|
2) Configure the settings:
|
||||||
- Set speech speed
|
- Set speech speed
|
||||||
- Select a voice
|
- Select a voice (or create a custom voice using voice mixer)
|
||||||
- Select subtitle generation style (by sentence, word, etc.)
|
- Select subtitle generation style (by sentence, word, etc.)
|
||||||
- Select output format
|
- Select output format
|
||||||
- Select where to save the output
|
- Select where to save the output
|
||||||
@@ -82,19 +101,53 @@ Here’s Abogen in action: in this demo, it processes ∼3,000 characters of tex
|
|||||||
- **Supported formats**: `ePub`, `PDF`, or `.TXT` files (or use built-in text editor)
|
- **Supported formats**: `ePub`, `PDF`, or `.TXT` files (or use built-in text editor)
|
||||||
- **Speed**: Adjust speech rate from `0.1x` to `2.0x`
|
- **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.
|
- **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)
|
- **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`, or `.MP3`
|
- **Output formats**: `.WAV`, `.FLAC`, `.MP3`, `.OPUS` 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`
|
- **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.
|
- **Chapter Control**: Select specific `chapters` from ePUBs or `chapters + pages` from PDFs.
|
||||||
|
- **Save each chapter separately**: Save each chapter in e-books as a separate audio file.
|
||||||
|
- **Project folder**: Save the converted items in a project folder with available metadata files.
|
||||||
- **Options**:
|
- **Options**:
|
||||||
- **Configure max words per subtitle**: Automatically configures the maximum number of words per subtitle entry.
|
- **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.
|
||||||
|
- **Separate chapters audio format**: Configures the audio format for separate chapters (e.g., `WAV`, `FLAC`, `MP3`, `OPUS`).
|
||||||
- **Create desktop shortcut**: Creates a shortcut on your desktop for easy access.
|
- **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 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.
|
- **Open temp directory**: Opens the temporary directory where converted text files are stored.
|
||||||
- **Clear all teporary files**: Deletes all temporary files created during the conversion process.
|
- **Clear temporary files**: Deletes temporary files created during the conversion or preview.
|
||||||
- **Check for updates at startup**: Automatically checks for updates when the program starts.
|
- **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`.
|
- **After conversion**: `Open file`, `Go to folder`, `New conversion`, or `Go back`.
|
||||||
|
|
||||||
|
## `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))
|
||||||
|
|
||||||
|
## `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:
|
||||||
|
|
||||||
|
```
|
||||||
|
<<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:
|
||||||
|
- 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
|
||||||
|
|
||||||
|
You can manually add these markers to plain text files for the same benefits. Simply include them in your text like this:
|
||||||
|
|
||||||
|
```
|
||||||
|
<<CHAPTER_MARKER:Introduction>>
|
||||||
|
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.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
|
||||||
## `Supported Languages`
|
## `Supported Languages`
|
||||||
```
|
```
|
||||||
# 🇺🇸 'a' => American English, 🇬🇧 'b' => British English
|
# 🇺🇸 'a' => American English, 🇬🇧 'b' => British English
|
||||||
@@ -116,21 +169,63 @@ keep-open=yes
|
|||||||
--audio-device=openal
|
--audio-device=openal
|
||||||
--sub-margin-x=235
|
--sub-margin-x=235
|
||||||
--sub-pos=60
|
--sub-pos=60
|
||||||
|
# --- Audio Quality ---
|
||||||
|
audio-spdif=ac3,dts,eac3,truehd,dts-hd
|
||||||
|
audio-channels=auto
|
||||||
|
audio-samplerate=48000
|
||||||
|
volume-max=200
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## `Docker Guide`
|
||||||
|
If you want to run Abogen in a Docker container:
|
||||||
|
1) [Download the repository](https://github.com/denizsafak/abogen/archive/refs/heads/main.zip) and extract, or clone it using git.
|
||||||
|
2) Go to `abogen` folder. You should see `Dockerfile` there.
|
||||||
|
3) Open your termminal in that directory and run the following commands:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Build the Docker image:
|
||||||
|
docker build --progress plain -t abogen .
|
||||||
|
|
||||||
|
# Note that building the image may take a while.
|
||||||
|
# After building is complete, run the Docker container:
|
||||||
|
|
||||||
|
# Windows
|
||||||
|
docker run --name abogen -v %cd%:/shared -p 5800:5800 -p 5900:5900 --gpus all abogen
|
||||||
|
|
||||||
|
# Linux
|
||||||
|
docker run --name abogen -v $(pwd):/shared -p 5800:5800 -p 5900:5900 --gpus all abogen
|
||||||
|
|
||||||
|
# MacOS
|
||||||
|
docker run --name abogen -v $(pwd):/shared -p 5800:5800 -p 5900:5900 abogen
|
||||||
|
|
||||||
|
# We expose port 5800 for use by a web browser, 5900 if you want to connect with a VNC client.
|
||||||
|
```
|
||||||
|
|
||||||
|
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`.
|
||||||
|
|
||||||
|
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).
|
||||||
|
|
||||||
|
(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/).)
|
||||||
|
|
||||||
## `Similar Projects`
|
## `Similar Projects`
|
||||||
Abogen is a standalone project, but it is inspired by and shares some similarities with other projects. Here are a few:
|
Abogen is a standalone project, but it is inspired by and shares some similarities with other projects. Here are a few:
|
||||||
- [audiblez](https://github.com/santinic/audiblez): Generate audiobooks from e-books. **(Has CLI and GUI support)**
|
- [audiblez](https://github.com/santinic/audiblez): Generate audiobooks from e-books. **(Has CLI and GUI support)**
|
||||||
- [autiobooks](https://github.com/plusuncold/autiobooks): Automatically convert epubs to audiobooks
|
- [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.
|
- [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
|
- [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`
|
## `Roadmap`
|
||||||
- [ ] Improve PDF support for better text extraction.
|
- [ ] Add OCR scan feature for PDF files using docling.
|
||||||
- [ ] Add chapter metadata for .m4a files using ffmpeg-bin.
|
- [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.
|
- [ ] Add support for different languages in GUI.
|
||||||
- [ ] Add voice formula feature that enables mixing different voice models. https://github.com/denizsafak/abogen/issues/1
|
- [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 support for kokoro-onnx (If it's necessary).
|
||||||
- [ ] Add dark mode.
|
- [ ] Add dark mode.
|
||||||
|
|
||||||
## `Troubleshooting`
|
## `Troubleshooting`
|
||||||
@@ -147,7 +242,8 @@ If you'd like to modify the code and contribute to development, you can [downloa
|
|||||||
```bash
|
```bash
|
||||||
# Go to the directory where you extracted the repository and run:
|
# Go to the directory where you extracted the repository and run:
|
||||||
pip install -e . # Installs the package in editable mode
|
pip install -e . # Installs the package in editable mode
|
||||||
python -m build # Builds the package in dist folder
|
pip install build # Install the build package
|
||||||
|
python -m build # Builds the package in dist folder (optional)
|
||||||
abogen # Opens the GUI
|
abogen # Opens the GUI
|
||||||
```
|
```
|
||||||
Feel free to explore the code and make any changes you like.
|
Feel free to explore the code and make any changes you like.
|
||||||
@@ -155,7 +251,9 @@ Feel free to explore the code and make any changes you like.
|
|||||||
## `Credits`
|
## `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.
|
- 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 [@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 [@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.
|
- 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), [Adjust](https://icons8.com/icon/21698/adjust) and [Voice Id](https://icons8.com/icon/GskSeVoroQ7u/voice-id) icons by [Icons8](https://icons8.com/).
|
||||||
|
|
||||||
## `License`
|
## `License`
|
||||||
This project is available under the MIT License - see the [LICENSE](https://github.com/denizsafak/abogen/blob/main/LICENSE) file for details.
|
This project is available under the MIT License - see the [LICENSE](https://github.com/denizsafak/abogen/blob/main/LICENSE) file for details.
|
||||||
|
|||||||
@@ -46,8 +46,7 @@ if exist "%VERSION_FILE%" (
|
|||||||
set "LOCAL_VERSION_CLEAN=!LOCAL_VERSION:.=!"
|
set "LOCAL_VERSION_CLEAN=!LOCAL_VERSION:.=!"
|
||||||
|
|
||||||
:: First verify GitHub is accessible by checking HTTP status code
|
:: First verify GitHub is accessible by checking HTTP status code
|
||||||
curl -s -o nul -w "%%{http_code}" "%VERSION_URL%" > "%TEMP%\abogen_http_status.txt"
|
for /f %%i in ('curl -s -o nul -w "%%{http_code}" "%VERSION_URL%"') do set abogen_http_status=%%i
|
||||||
set /p abogen_http_status=<"%TEMP%\abogen_http_status.txt"
|
|
||||||
|
|
||||||
if not "!abogen_http_status!"=="200" (
|
if not "!abogen_http_status!"=="200" (
|
||||||
echo Failed to access GitHub repository ^(HTTP status: !abogen_http_status!^). Continuing with current version.
|
echo Failed to access GitHub repository ^(HTTP status: !abogen_http_status!^). Continuing with current version.
|
||||||
@@ -55,14 +54,8 @@ if exist "%VERSION_FILE%" (
|
|||||||
)
|
)
|
||||||
|
|
||||||
:: Get the remote version (only if GitHub is accessible)
|
:: Get the remote version (only if GitHub is accessible)
|
||||||
curl -s -o "%TEMP%\abogen_remote_version.txt" "%VERSION_URL%"
|
for /f "delims=" %%i in ('curl -s "%VERSION_URL%"') do set abogen_remote_version=%%i
|
||||||
findstr "^" "%TEMP%\abogen_remote_version.txt" >nul 2>&1
|
|
||||||
if errorlevel 1 (
|
|
||||||
echo Failed to retrieve version information. Continuing with current version.
|
|
||||||
goto continue_with_current_version
|
|
||||||
)
|
|
||||||
|
|
||||||
set /p abogen_remote_version=<"%TEMP%\abogen_remote_version.txt"
|
|
||||||
if "!abogen_remote_version!"=="" (
|
if "!abogen_remote_version!"=="" (
|
||||||
echo Empty version information received. Continuing with current version.
|
echo Empty version information received. Continuing with current version.
|
||||||
goto continue_with_current_version
|
goto continue_with_current_version
|
||||||
@@ -97,8 +90,7 @@ if exist "%VERSION_FILE%" (
|
|||||||
echo Downloading the latest update...
|
echo Downloading the latest update...
|
||||||
|
|
||||||
:: Test the zip URL before downloading
|
:: Test the zip URL before downloading
|
||||||
curl -s -o nul -w "%%{http_code}" "%UPDATE_ZIP_URL%" > "%TEMP%\abogen_zip_status.txt"
|
for /f %%i in ('curl -s -o nul -w "%%{http_code}" "%UPDATE_ZIP_URL%"') do set abogen_zip_status=%%i
|
||||||
set /p abogen_zip_status=<"%TEMP%\abogen_zip_status.txt"
|
|
||||||
|
|
||||||
if not "!abogen_zip_status!"=="302" (
|
if not "!abogen_zip_status!"=="302" (
|
||||||
echo Failed to access update zip file ^(HTTP status: !abogen_zip_status!^). Continuing with current version.
|
echo Failed to access update zip file ^(HTTP status: !abogen_zip_status!^). Continuing with current version.
|
||||||
@@ -137,7 +129,7 @@ if exist "%VERSION_FILE%" (
|
|||||||
)
|
)
|
||||||
) else (
|
) else (
|
||||||
echo Current version: !LOCAL_VERSION!, Remote version: !abogen_remote_version!
|
echo Current version: !LOCAL_VERSION!, Remote version: !abogen_remote_version!
|
||||||
echo You are usering the latest version.
|
echo You are using the latest version.
|
||||||
)
|
)
|
||||||
) else (
|
) else (
|
||||||
echo VERSION file not found. Cannot check for updates.
|
echo VERSION file not found. Cannot check for updates.
|
||||||
@@ -227,15 +219,33 @@ if errorlevel 1 (
|
|||||||
exit /b
|
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
|
:: Install setup requirements
|
||||||
echo Installing setup requirements...
|
echo Installing setup requirements...
|
||||||
%PYTHON_CONSOLE_PATH% -m pip install --upgrade setuptools wheel sphinx hatchling --no-warn-script-location
|
%PYTHON_CONSOLE_PATH% -m pip install --upgrade setuptools setuptools-scm wheel sphinx hatchling --no-warn-script-location
|
||||||
if errorlevel 1 (
|
if errorlevel 1 (
|
||||||
echo Failed to install setup requirements.
|
echo Failed to install setup requirements.
|
||||||
pause
|
pause
|
||||||
exit /b
|
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
|
:: Install project and dependencies from pyproject.toml
|
||||||
echo Checking and installing project dependencies...
|
echo Checking and installing project dependencies...
|
||||||
if exist %PYPROJECT_FILE% (
|
if exist %PYPROJECT_FILE% (
|
||||||
@@ -262,21 +272,28 @@ if "%MISAKI_LANG%" NEQ "en" (
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
:: 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
|
||||||
|
|
||||||
:: Check if torch is installed with CUDA support
|
:: Check if torch is installed with CUDA support
|
||||||
echo Checking CUDA availability...
|
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
|
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" (
|
if "%cuda_available%"=="False" (
|
||||||
echo Installing PyTorch with CUDA %CUDA_VERSION% support...
|
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
|
%PYTHON_CONSOLE_PATH% -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu%CUDA_VERSION% --no-warn-script-location
|
||||||
echo.
|
echo.
|
||||||
if errorlevel 1 (
|
if errorlevel 1 (
|
||||||
echo Failed to install PyTorch.
|
echo Failed to install PyTorch.
|
||||||
pause
|
pause
|
||||||
exit /b
|
exit /b
|
||||||
|
)
|
||||||
|
) else (
|
||||||
|
echo CUDA is available on NVIDIA GPU.
|
||||||
)
|
)
|
||||||
) else (
|
) else (
|
||||||
echo CUDA is available.
|
echo GPU is not NVIDIA. Skipping PyTorch CUDA installation.
|
||||||
)
|
)
|
||||||
|
|
||||||
:: Ask user if they want to create a desktop shortcut
|
:: Ask user if they want to create a desktop shortcut
|
||||||
|
|||||||
@@ -0,0 +1,42 @@
|
|||||||
|
# Special thanks to @geo38 from Reddit, who provided this Dockerfile:
|
||||||
|
# https://www.reddit.com/r/selfhosted/comments/1k8x1yo/comment/mpe0bz8/
|
||||||
|
|
||||||
|
# Use a docker base image that runs a window manager that can be viewed
|
||||||
|
# outside the image with a web browser or VNC client.
|
||||||
|
# https://github.com/jlesage/docker-baseimage-gui
|
||||||
|
FROM jlesage/baseimage-gui:debian-12-v4
|
||||||
|
|
||||||
|
# Load stuff needed by abogen
|
||||||
|
RUN apt-get update \
|
||||||
|
&& apt-get install -y \
|
||||||
|
python3 \
|
||||||
|
python3-venv \
|
||||||
|
python3-pip \
|
||||||
|
python3-pyqt5 \
|
||||||
|
espeak-ng \
|
||||||
|
&& apt-get clean \
|
||||||
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
# The base image will run /startapp.sh on launch.
|
||||||
|
#
|
||||||
|
# The base image runs that script as user 'app' uid=1000. That user
|
||||||
|
# does not exist in the base image but is created at run time.
|
||||||
|
#
|
||||||
|
# We need to install abogen in python venv (requirement of newer python3).
|
||||||
|
#
|
||||||
|
# The python venv has to be writable by the 'app' user as abogen dynamically
|
||||||
|
# installs python packages, so create the venv as that user
|
||||||
|
#
|
||||||
|
# We intend to share the /shared directory with the host using a bind volume
|
||||||
|
# in order to access any source files and the created files.
|
||||||
|
|
||||||
|
RUN echo '#!/bin/bash\nsource /app/venv/bin/activate\nexec abogen' > /startapp.sh \
|
||||||
|
&& chmod 555 /startapp.sh \
|
||||||
|
&& mkdir /app /shared \
|
||||||
|
&& chown 1000:1000 /app /shared \
|
||||||
|
&& chmod 755 /app /shared
|
||||||
|
USER 1000:1000
|
||||||
|
RUN python3 -m venv /app/venv
|
||||||
|
RUN /bin/bash -c "source /app/venv/bin/activate && pip install abogen"
|
||||||
|
# Change back to user ROOT as the startup scripts inside base image needs it
|
||||||
|
USER root
|
||||||
@@ -1 +1 @@
|
|||||||
1.0.0
|
1.0.8
|
||||||
|
After Width: | Height: | Size: 442 B |
|
After Width: | Height: | Size: 571 B |
|
After Width: | Height: | Size: 786 B |
|
After Width: | Height: | Size: 521 B |
|
After Width: | Height: | Size: 372 B |
|
After Width: | Height: | Size: 465 B |
|
After Width: | Height: | Size: 381 B |
|
After Width: | Height: | Size: 441 B |
|
After Width: | Height: | Size: 617 B |
|
After Width: | Height: | Size: 431 B |
|
After Width: | Height: | Size: 18 KiB |
|
After Width: | Height: | Size: 426 B |
|
After Width: | Height: | Size: 343 B |
|
After Width: | Height: | Size: 376 B |
@@ -1,4 +1,4 @@
|
|||||||
from utils import get_version, get_user_config_path
|
from utils import get_version
|
||||||
|
|
||||||
# Program Information
|
# Program Information
|
||||||
PROGRAM_NAME = "abogen"
|
PROGRAM_NAME = "abogen"
|
||||||
@@ -102,16 +102,3 @@ SAMPLE_VOICE_TEXTS = {
|
|||||||
"p": "Este é um exemplo da voz selecionada.",
|
"p": "Este é um exemplo da voz selecionada.",
|
||||||
"z": "这是所选语音的示例。",
|
"z": "这是所选语音的示例。",
|
||||||
}
|
}
|
||||||
|
|
||||||
# flags mapping for voice display
|
|
||||||
FLAGS = {
|
|
||||||
"a": "🇺🇸",
|
|
||||||
"b": "🇬🇧",
|
|
||||||
"e": "🇪🇸",
|
|
||||||
"f": "🇫🇷",
|
|
||||||
"h": "🇮🇳",
|
|
||||||
"i": "🇮🇹",
|
|
||||||
"j": "🇯🇵",
|
|
||||||
"p": "🇧🇷",
|
|
||||||
"z": "🇨🇳",
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -2,17 +2,42 @@ import os
|
|||||||
import re
|
import re
|
||||||
import tempfile
|
import tempfile
|
||||||
import time
|
import time
|
||||||
|
import chardet
|
||||||
|
import charset_normalizer
|
||||||
|
import hashlib # For generating unique cache filenames
|
||||||
|
from platformdirs import user_desktop_dir
|
||||||
from PyQt5.QtCore import QThread, pyqtSignal, Qt
|
from PyQt5.QtCore import QThread, pyqtSignal, Qt
|
||||||
from PyQt5.QtWidgets import QCheckBox, QVBoxLayout, QDialog, QLabel, QDialogButtonBox
|
from PyQt5.QtWidgets import QCheckBox, QVBoxLayout, QDialog, QLabel, QDialogButtonBox
|
||||||
import soundfile as sf
|
import soundfile as sf
|
||||||
from utils import clean_text
|
from utils import clean_text, create_process
|
||||||
from constants import PROGRAM_NAME, LANGUAGE_DESCRIPTIONS, SAMPLE_VOICE_TEXTS
|
from constants import PROGRAM_NAME, LANGUAGE_DESCRIPTIONS, SAMPLE_VOICE_TEXTS
|
||||||
|
from voice_formulas import get_new_voice
|
||||||
|
import hf_tracker
|
||||||
|
import static_ffmpeg
|
||||||
|
import threading # for efficient waiting
|
||||||
|
import subprocess
|
||||||
|
|
||||||
|
|
||||||
def get_sample_voice_text(lang_code):
|
def get_sample_voice_text(lang_code):
|
||||||
return SAMPLE_VOICE_TEXTS.get(lang_code, SAMPLE_VOICE_TEXTS["a"])
|
return SAMPLE_VOICE_TEXTS.get(lang_code, SAMPLE_VOICE_TEXTS["a"])
|
||||||
|
|
||||||
|
|
||||||
|
def detect_encoding(file_path):
|
||||||
|
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()
|
||||||
|
|
||||||
|
|
||||||
class ChapterOptionsDialog(QDialog):
|
class ChapterOptionsDialog(QDialog):
|
||||||
def __init__(self, chapter_count, parent=None):
|
def __init__(self, chapter_count, parent=None):
|
||||||
super().__init__(parent)
|
super().__init__(parent)
|
||||||
@@ -108,6 +133,7 @@ class ConversionThread(QThread):
|
|||||||
use_gpu=True,
|
use_gpu=True,
|
||||||
): # Add use_gpu parameter
|
): # Add use_gpu parameter
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
self._chapter_options_event = threading.Event()
|
||||||
self.np = np_module
|
self.np = np_module
|
||||||
self.KPipeline = kpipeline_class
|
self.KPipeline = kpipeline_class
|
||||||
self.file_name = file_name
|
self.file_name = file_name
|
||||||
@@ -132,7 +158,11 @@ class ConversionThread(QThread):
|
|||||||
self.max_subtitle_words = 50 # Default value, will be overridden from GUI
|
self.max_subtitle_words = 50 # Default value, will be overridden from GUI
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
|
print(
|
||||||
|
f"\nVoice: {self.voice}\nLanguage: {self.lang_code}\nSpeed: {self.speed}\nGPU: {self.use_gpu}\nFile: {self.file_name}\nSubtitle mode: {self.subtitle_mode}\nOutput format: {self.output_format}\nSave option: {self.save_option}\n"
|
||||||
|
)
|
||||||
try:
|
try:
|
||||||
|
hf_tracker.set_log_callback(lambda msg: self.log_updated.emit(msg))
|
||||||
# Show configuration
|
# Show configuration
|
||||||
self.log_updated.emit("Configuration:")
|
self.log_updated.emit("Configuration:")
|
||||||
# Use display_path for logs if available, otherwise use the actual file name
|
# Use display_path for logs if available, otherwise use the actual file name
|
||||||
@@ -153,6 +183,8 @@ class ConversionThread(QThread):
|
|||||||
self.log_updated.emit(f"- Subtitle mode: {self.subtitle_mode}")
|
self.log_updated.emit(f"- Subtitle mode: {self.subtitle_mode}")
|
||||||
self.log_updated.emit(f"- Output format: {self.output_format}")
|
self.log_updated.emit(f"- Output format: {self.output_format}")
|
||||||
self.log_updated.emit(f"- Save option: {self.save_option}")
|
self.log_updated.emit(f"- Save option: {self.save_option}")
|
||||||
|
if self.replace_single_newlines:
|
||||||
|
self.log_updated.emit(f"- Replace single newlines: Yes")
|
||||||
|
|
||||||
# Display save_chapters_separately flag if it's set
|
# Display save_chapters_separately flag if it's set
|
||||||
if hasattr(self, "save_chapters_separately"):
|
if hasattr(self, "save_chapters_separately"):
|
||||||
@@ -167,6 +199,9 @@ class ConversionThread(QThread):
|
|||||||
self.log_updated.emit(
|
self.log_updated.emit(
|
||||||
f"- Merge chapters at the end: {'Yes' if merge_at_end else 'No'}"
|
f"- Merge chapters at the end: {'Yes' if merge_at_end else 'No'}"
|
||||||
)
|
)
|
||||||
|
# Display the separate chapters format if it's set
|
||||||
|
separate_format = getattr(self, 'separate_chapters_format', 'wav')
|
||||||
|
self.log_updated.emit(f"- Separate chapters format: {separate_format}")
|
||||||
|
|
||||||
if self.save_option == "Choose output folder":
|
if self.save_option == "Choose output folder":
|
||||||
self.log_updated.emit(
|
self.log_updated.emit(
|
||||||
@@ -183,17 +218,30 @@ class ConversionThread(QThread):
|
|||||||
if self.is_direct_text:
|
if self.is_direct_text:
|
||||||
text = self.file_name # Treat file_name as direct text input
|
text = self.file_name # Treat file_name as direct text input
|
||||||
else:
|
else:
|
||||||
with open(self.file_name, "r", encoding="utf-8") as file:
|
encoding = detect_encoding(self.file_name)
|
||||||
|
with open(
|
||||||
|
self.file_name, "r", encoding=encoding, errors="replace"
|
||||||
|
) as file:
|
||||||
text = file.read()
|
text = file.read()
|
||||||
|
|
||||||
# Clean up text using utility function
|
# Clean up text using utility function
|
||||||
text = clean_text(text)
|
text = clean_text(text)
|
||||||
|
|
||||||
|
# Remove metadata markers from the text to be processed
|
||||||
|
metadata_pattern = r"<<METADATA_[^:]+:[^>]*>>"
|
||||||
|
text = re.sub(metadata_pattern, "", text)
|
||||||
|
|
||||||
# --- Chapter splitting logic ---
|
# --- Chapter splitting logic ---
|
||||||
chapter_pattern = r"<<CHAPTER_MARKER:(.*?)>>"
|
chapter_pattern = r"<<CHAPTER_MARKER:(.*?)>>"
|
||||||
chapter_splits = list(re.finditer(chapter_pattern, text))
|
chapter_splits = list(re.finditer(chapter_pattern, text))
|
||||||
chapters = []
|
chapters = []
|
||||||
if chapter_splits:
|
if chapter_splits:
|
||||||
|
# prepend Introduction for content before first marker
|
||||||
|
first_start = chapter_splits[0].start()
|
||||||
|
if first_start > 0:
|
||||||
|
intro_text = text[:first_start].strip()
|
||||||
|
if intro_text:
|
||||||
|
chapters.append(("Introduction", intro_text))
|
||||||
for idx, match in enumerate(chapter_splits):
|
for idx, match in enumerate(chapter_splits):
|
||||||
start = match.end()
|
start = match.end()
|
||||||
end = (
|
end = (
|
||||||
@@ -224,18 +272,12 @@ class ConversionThread(QThread):
|
|||||||
and not self.chapter_options_set
|
and not self.chapter_options_set
|
||||||
):
|
):
|
||||||
|
|
||||||
self.waiting_for_user_input = True
|
# Emit signal to main thread and wait
|
||||||
# Emit signal to main thread to show dialog
|
|
||||||
self.chapters_detected.emit(total_chapters)
|
self.chapters_detected.emit(total_chapters)
|
||||||
|
self._chapter_options_event.wait()
|
||||||
# Wait for chapter options to be set
|
|
||||||
while self.waiting_for_user_input and not self.cancel_requested:
|
|
||||||
time.sleep(0.1)
|
|
||||||
|
|
||||||
if self.cancel_requested:
|
if self.cancel_requested:
|
||||||
self.conversion_finished.emit("Cancelled", None)
|
self.conversion_finished.emit("Cancelled", None)
|
||||||
return
|
return
|
||||||
|
|
||||||
self.chapter_options_set = True
|
self.chapter_options_set = True
|
||||||
|
|
||||||
# Log all detected chapters at the beginning
|
# Log all detected chapters at the beginning
|
||||||
@@ -256,11 +298,19 @@ class ConversionThread(QThread):
|
|||||||
base_path = self.display_path if self.display_path else self.file_name
|
base_path = self.display_path if self.display_path else self.file_name
|
||||||
base_name = os.path.splitext(os.path.basename(base_path))[0]
|
base_name = os.path.splitext(os.path.basename(base_path))[0]
|
||||||
if self.save_option == "Save to Desktop":
|
if self.save_option == "Save to Desktop":
|
||||||
parent_dir = os.path.join(os.path.expanduser("~"), "Desktop")
|
parent_dir = user_desktop_dir()
|
||||||
elif self.save_option == "Save next to input file":
|
elif self.save_option == "Save next to input file":
|
||||||
parent_dir = os.path.dirname(base_path)
|
parent_dir = os.path.dirname(base_path)
|
||||||
else:
|
else:
|
||||||
parent_dir = self.output_folder or os.getcwd()
|
parent_dir = self.output_folder or os.getcwd()
|
||||||
|
# Ensure the output folder exists, error if it doesn't
|
||||||
|
if not os.path.exists(parent_dir):
|
||||||
|
self.log_updated.emit(
|
||||||
|
(
|
||||||
|
f"Output folder does not exist: {parent_dir}",
|
||||||
|
"red",
|
||||||
|
)
|
||||||
|
)
|
||||||
# Find a unique suffix for both folder and merged file, always
|
# Find a unique suffix for both folder and merged file, always
|
||||||
counter = 1
|
counter = 1
|
||||||
while True:
|
while True:
|
||||||
@@ -303,6 +353,12 @@ class ConversionThread(QThread):
|
|||||||
# Initialize current segment counter
|
# Initialize current segment counter
|
||||||
current_segment = 0
|
current_segment = 0
|
||||||
|
|
||||||
|
# Initialize chapter times
|
||||||
|
chapters_time = [
|
||||||
|
{"chapter": chapter[0], "start": 0.0, "end": 0.0}
|
||||||
|
for chapter in chapters
|
||||||
|
]
|
||||||
|
|
||||||
# Instead of processing the whole text, process by chapter
|
# Instead of processing the whole text, process by chapter
|
||||||
for chapter_idx, (chapter_name, chapter_text) in enumerate(chapters, 1):
|
for chapter_idx, (chapter_name, chapter_text) in enumerate(chapters, 1):
|
||||||
if total_chapters > 1:
|
if total_chapters > 1:
|
||||||
@@ -318,11 +374,22 @@ class ConversionThread(QThread):
|
|||||||
chapter_subtitle_entries = []
|
chapter_subtitle_entries = []
|
||||||
chapter_current_time = 0.0
|
chapter_current_time = 0.0
|
||||||
|
|
||||||
|
# chapter start time
|
||||||
|
chapter_time = chapters_time[chapter_idx - 1]
|
||||||
|
chapter_time["start"] = current_time
|
||||||
|
|
||||||
# Set split_pattern to \n+ which will split on one or more newlines
|
# Set split_pattern to \n+ which will split on one or more newlines
|
||||||
split_pattern = r"\n+"
|
split_pattern = r"\n+"
|
||||||
|
|
||||||
|
# Check if the voice is a formula and load it if necessary
|
||||||
|
if "*" in self.voice:
|
||||||
|
loaded_voice = get_new_voice(tts, self.voice, self.use_gpu)
|
||||||
|
else:
|
||||||
|
loaded_voice = self.voice
|
||||||
|
|
||||||
for result in tts(
|
for result in tts(
|
||||||
chapter_text,
|
chapter_text,
|
||||||
voice=self.voice,
|
voice=loaded_voice,
|
||||||
speed=self.speed,
|
speed=self.speed,
|
||||||
split_pattern=split_pattern,
|
split_pattern=split_pattern,
|
||||||
):
|
):
|
||||||
@@ -424,6 +491,9 @@ class ConversionThread(QThread):
|
|||||||
# Update progress more frequently (after each result)
|
# Update progress more frequently (after each result)
|
||||||
self.progress_updated.emit(percent, etr_str)
|
self.progress_updated.emit(percent, etr_str)
|
||||||
|
|
||||||
|
# Update chapter end time
|
||||||
|
chapter_time["end"] = current_time
|
||||||
|
|
||||||
# Save the individual chapter output if save_chapters_separately is enabled
|
# Save the individual chapter output if save_chapters_separately is enabled
|
||||||
if (
|
if (
|
||||||
save_chapters_separately
|
save_chapters_separately
|
||||||
@@ -440,22 +510,56 @@ class ConversionThread(QThread):
|
|||||||
|
|
||||||
# Concatenate chapter audio and save
|
# Concatenate chapter audio and save
|
||||||
chapter_audio = self.np.concatenate(chapter_audio_segments)
|
chapter_audio = self.np.concatenate(chapter_audio_segments)
|
||||||
|
|
||||||
|
# Use separate_chapters_format
|
||||||
|
separate_format = getattr(self, 'separate_chapters_format', 'wav')
|
||||||
|
|
||||||
chapter_out_path = os.path.join(
|
chapter_out_path = os.path.join(
|
||||||
chapters_out_dir, f"{chapter_filename}.{self.output_format}"
|
chapters_out_dir, f"{chapter_filename}.{separate_format}"
|
||||||
)
|
|
||||||
sf.write(
|
|
||||||
chapter_out_path,
|
|
||||||
chapter_audio,
|
|
||||||
24000,
|
|
||||||
format=self.output_format,
|
|
||||||
)
|
)
|
||||||
|
if separate_format == "opus":
|
||||||
|
static_ffmpeg.add_paths()
|
||||||
|
proc = create_process(
|
||||||
|
[
|
||||||
|
"ffmpeg",
|
||||||
|
"-y",
|
||||||
|
"-thread_queue_size", "32768", # Increased thread_queue_size for chapter opus
|
||||||
|
"-f",
|
||||||
|
"f32le",
|
||||||
|
"-ar",
|
||||||
|
"24000",
|
||||||
|
"-ac",
|
||||||
|
"1",
|
||||||
|
"-i",
|
||||||
|
"pipe:0",
|
||||||
|
"-c:a",
|
||||||
|
"libopus",
|
||||||
|
"-b:a",
|
||||||
|
"24000",
|
||||||
|
chapter_out_path,
|
||||||
|
],
|
||||||
|
stdin=subprocess.PIPE,
|
||||||
|
text=False # Ensure binary stdin for audio data
|
||||||
|
)
|
||||||
|
proc.stdin.write(chapter_audio.astype("float32").tobytes())
|
||||||
|
proc.stdin.close()
|
||||||
|
proc.wait()
|
||||||
|
else:
|
||||||
|
sf.write(
|
||||||
|
chapter_out_path,
|
||||||
|
chapter_audio,
|
||||||
|
24000,
|
||||||
|
format=separate_format,
|
||||||
|
)
|
||||||
|
|
||||||
# Generate .srt subtitle file for chapter if not Disabled
|
# Generate .srt subtitle file for chapter if not Disabled
|
||||||
if self.subtitle_mode != "Disabled" and chapter_subtitle_entries:
|
if self.subtitle_mode != "Disabled" and chapter_subtitle_entries:
|
||||||
chapter_srt_path = os.path.join(
|
chapter_srt_path = os.path.join(
|
||||||
chapters_out_dir, f"{chapter_filename}.srt"
|
chapters_out_dir, f"{chapter_filename}.srt"
|
||||||
)
|
)
|
||||||
with open(chapter_srt_path, "w", encoding="utf-8") as srt_file:
|
with open(
|
||||||
|
chapter_srt_path, "w", encoding="utf-8", errors="replace"
|
||||||
|
) as srt_file:
|
||||||
for i, (start, end, text) in enumerate(
|
for i, (start, end, text) in enumerate(
|
||||||
chapter_subtitle_entries, 1
|
chapter_subtitle_entries, 1
|
||||||
):
|
):
|
||||||
@@ -486,32 +590,74 @@ class ConversionThread(QThread):
|
|||||||
or not self.save_chapters_separately
|
or not self.save_chapters_separately
|
||||||
or getattr(self, "merge_chapters_at_end", True)
|
or getattr(self, "merge_chapters_at_end", True)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
intended_output_format = self.output_format # Store the original choice
|
||||||
|
|
||||||
if audio_segments and merge_chapters:
|
if audio_segments and merge_chapters:
|
||||||
self.log_updated.emit("\nFinalizing audio file...\n")
|
self.log_updated.emit("\nFinalizing audio file...\n")
|
||||||
audio = self.np.concatenate(audio_segments)
|
audio_data_np = self.np.concatenate(audio_segments)
|
||||||
|
|
||||||
out_dir = parent_dir
|
out_dir = parent_dir
|
||||||
# Use the same suffix as above
|
base_filepath_no_ext = os.path.join(out_dir, f"{base_name}{suffix}")
|
||||||
out_filename = f"{base_name}{suffix}.{self.output_format}"
|
|
||||||
out_path = os.path.join(out_dir, out_filename)
|
final_out_path = None
|
||||||
srt_path = os.path.splitext(out_path)[0] + ".srt"
|
|
||||||
sf.write(out_path, audio, 24000, format=self.output_format)
|
if intended_output_format == "m4b":
|
||||||
if self.subtitle_mode != "Disabled":
|
final_out_path = self._generate_m4b_with_chapters(audio_data_np, chapters_time, base_filepath_no_ext)
|
||||||
with open(srt_path, "w", encoding="utf-8") as srt_file:
|
elif intended_output_format == "opus":
|
||||||
for i, (start, end, text) in enumerate(subtitle_entries, 1):
|
static_ffmpeg.add_paths()
|
||||||
srt_file.write(
|
opus_out_path = f"{base_filepath_no_ext}.opus"
|
||||||
f"{i}\n{self._srt_time(start)} --> {self._srt_time(end)}\n{text}\n\n"
|
ffmpeg_cmd_opus = [
|
||||||
)
|
"ffmpeg", "-y",
|
||||||
self.conversion_finished.emit(
|
"-thread_queue_size", "32768", # Increased thread_queue_size
|
||||||
(
|
"-f", "f32le", "-ar", "24000", "-ac", "1", "-i", "pipe:0",
|
||||||
f"Audiobook saved to: {out_path}\n\nSubtitle saved to: {srt_path}",
|
"-c:a", "libopus", "-b:a", "24000", # Original bitrate
|
||||||
"green",
|
opus_out_path,
|
||||||
),
|
]
|
||||||
out_path,
|
try:
|
||||||
)
|
process = create_process(ffmpeg_cmd_opus, stdin=subprocess.PIPE, text=False) # Ensure binary stdin
|
||||||
|
process.stdin.write(audio_data_np.astype("float32").tobytes())
|
||||||
|
process.stdin.close()
|
||||||
|
if process.wait() == 0:
|
||||||
|
final_out_path = opus_out_path
|
||||||
|
else:
|
||||||
|
self.log_updated.emit(("Opus conversion failed.", "red"))
|
||||||
|
final_out_path = None
|
||||||
|
except Exception as e_opus:
|
||||||
|
self.log_updated.emit((f"Error during Opus conversion: {str(e_opus)}", "red"))
|
||||||
|
final_out_path = None
|
||||||
|
else: # For other formats like wav
|
||||||
|
standard_out_path = f"{base_filepath_no_ext}.{intended_output_format}"
|
||||||
|
try:
|
||||||
|
sf.write(standard_out_path, audio_data_np, 24000, format=intended_output_format)
|
||||||
|
final_out_path = standard_out_path
|
||||||
|
except Exception as e_sf:
|
||||||
|
self.log_updated.emit((f"Failed to write audio file {standard_out_path}: {str(e_sf)}", "red"))
|
||||||
|
final_out_path = None
|
||||||
|
|
||||||
|
# Subtitle and final message logic
|
||||||
|
if final_out_path:
|
||||||
|
if self.subtitle_mode != "Disabled":
|
||||||
|
srt_path = os.path.splitext(final_out_path)[0] + ".srt"
|
||||||
|
with open(srt_path, "w", encoding="utf-8", errors="replace") as srt_file:
|
||||||
|
for i, (start, end, text) in enumerate(subtitle_entries, 1):
|
||||||
|
srt_file.write(
|
||||||
|
f"{i}\n{self._srt_time(start)} --> {self._srt_time(end)}\n{text}\n\n"
|
||||||
|
)
|
||||||
|
self.conversion_finished.emit(
|
||||||
|
(
|
||||||
|
f"Audiobook saved to: {final_out_path}\n\nSubtitle saved to: {srt_path}",
|
||||||
|
"green",
|
||||||
|
),
|
||||||
|
final_out_path,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
self.conversion_finished.emit(
|
||||||
|
(f"Audiobook saved to: {final_out_path}", "green"), final_out_path
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
self.conversion_finished.emit(
|
self.log_updated.emit(("Audio generation failed (final_out_path was not set).", "red"))
|
||||||
(f"Audiobook saved to: {out_path}", "green"), out_path
|
self.conversion_finished.emit(("Audio generation failed.", "red"), None)
|
||||||
)
|
|
||||||
elif audio_segments and not merge_chapters:
|
elif audio_segments and not merge_chapters:
|
||||||
self.conversion_finished.emit(
|
self.conversion_finished.emit(
|
||||||
(
|
(
|
||||||
@@ -532,6 +678,151 @@ class ConversionThread(QThread):
|
|||||||
self.save_chapters_separately = options["save_chapters_separately"]
|
self.save_chapters_separately = options["save_chapters_separately"]
|
||||||
self.merge_chapters_at_end = options["merge_chapters_at_end"]
|
self.merge_chapters_at_end = options["merge_chapters_at_end"]
|
||||||
self.waiting_for_user_input = False
|
self.waiting_for_user_input = False
|
||||||
|
self._chapter_options_event.set()
|
||||||
|
|
||||||
|
def _generate_m4b_with_chapters(self, audio_data_np, chapters_time, base_filepath_no_ext):
|
||||||
|
final_wav_path = f"{base_filepath_no_ext}.wav"
|
||||||
|
|
||||||
|
if not chapters_time or len(chapters_time) <= 1:
|
||||||
|
self.log_updated.emit(
|
||||||
|
(
|
||||||
|
"File contains only one chapter or no chapters were detected. Audio will be saved as a standard .wav file instead.\n",
|
||||||
|
"red",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
sf.write(final_wav_path, audio_data_np, 24000, format="wav")
|
||||||
|
return final_wav_path
|
||||||
|
except Exception as e_wav_single:
|
||||||
|
self.log_updated.emit((f"Failed to save single/no chapter audio as WAV: {str(e_wav_single)}", "red"))
|
||||||
|
return None
|
||||||
|
|
||||||
|
output_m4b_path = f"{base_filepath_no_ext}.m4b"
|
||||||
|
chapters_info_path = f"{base_filepath_no_ext}_chapters.txt"
|
||||||
|
|
||||||
|
try:
|
||||||
|
with open(chapters_info_path, "w", encoding="utf-8") as f:
|
||||||
|
f.write(";FFMETADATA1\n")
|
||||||
|
for chapter in chapters_time:
|
||||||
|
f.write(f"[CHAPTER]\n")
|
||||||
|
f.write(f"TIMEBASE=1/1000\n") # Using milliseconds for precision
|
||||||
|
f.write(f"START={int(chapter['start']*1000)}\n")
|
||||||
|
f.write(f"END={int(chapter['end']*1000)}\n")
|
||||||
|
f.write(f"title={chapter['chapter']}\n\n")
|
||||||
|
|
||||||
|
static_ffmpeg.add_paths()
|
||||||
|
ffmpeg_cmd = [
|
||||||
|
"ffmpeg", "-y",
|
||||||
|
"-thread_queue_size", "32768", # Increased thread_queue_size
|
||||||
|
"-f", "f32le", "-ar", "24000", "-ac", "1", "-i", "pipe:0",
|
||||||
|
"-i", chapters_info_path,
|
||||||
|
"-map", "0:a",
|
||||||
|
"-map_metadata", "1",
|
||||||
|
"-map_chapters", "1",
|
||||||
|
# Add metadata tags
|
||||||
|
"-metadata", "composer=Narrator",
|
||||||
|
"-metadata", "genre=Audiobook",
|
||||||
|
# Extract metadata from text markers if available
|
||||||
|
*self._extract_and_add_metadata_tags_to_ffmpeg_cmd(),
|
||||||
|
"-c:a", "aac",
|
||||||
|
"-q:a", "2", # Quality-based VBR for better quality control (lower = better, range 0.1-2)
|
||||||
|
"-movflags", "+faststart+use_metadata_tags", # Added for better compatibility
|
||||||
|
output_m4b_path,
|
||||||
|
]
|
||||||
|
|
||||||
|
self.log_updated.emit(f"Generating audio with chapters...\n")
|
||||||
|
|
||||||
|
process = create_process(ffmpeg_cmd, stdin=subprocess.PIPE, text=False) # Ensure binary stdin
|
||||||
|
process.stdin.write(audio_data_np.astype("float32").tobytes())
|
||||||
|
process.stdin.close()
|
||||||
|
return_code = process.wait()
|
||||||
|
|
||||||
|
if return_code == 0:
|
||||||
|
return output_m4b_path
|
||||||
|
else:
|
||||||
|
self.log_updated.emit(
|
||||||
|
(f"FFmpeg failed to create M4B (return code {return_code}).\n\nFalling back to WAV.\n", "red")
|
||||||
|
)
|
||||||
|
sf.write(final_wav_path, audio_data_np, 24000, format="wav")
|
||||||
|
return final_wav_path
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
self.log_updated.emit((f"Error during M4B generation: {str(e)}.\n\nFalling back to WAV.\n", "red"))
|
||||||
|
try:
|
||||||
|
sf.write(final_wav_path, audio_data_np, 24000, format="wav")
|
||||||
|
return final_wav_path
|
||||||
|
except Exception as e_wav_fallback:
|
||||||
|
self.log_updated.emit((f"Critical error: Failed to save fallback WAV: {str(e_wav_fallback)}\n", "red"))
|
||||||
|
return None
|
||||||
|
finally:
|
||||||
|
if os.path.exists(chapters_info_path):
|
||||||
|
try:
|
||||||
|
os.remove(chapters_info_path)
|
||||||
|
except Exception as e_clean:
|
||||||
|
self.log_updated.emit((f"Warning: Could not delete temporary chapter file {chapters_info_path}: {e_clean}", "orange"))
|
||||||
|
|
||||||
|
def _extract_and_add_metadata_tags_to_ffmpeg_cmd(self):
|
||||||
|
"""Extract metadata tags from text content and add them to ffmpeg command"""
|
||||||
|
metadata_options = []
|
||||||
|
|
||||||
|
# Get the input text (either direct or from file)
|
||||||
|
text = ""
|
||||||
|
if self.is_direct_text:
|
||||||
|
text = self.file_name
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
encoding = detect_encoding(self.file_name)
|
||||||
|
with open(self.file_name, "r", encoding=encoding, errors="replace") as file:
|
||||||
|
text = file.read()
|
||||||
|
except Exception as e:
|
||||||
|
self.log_updated.emit(f"Warning: Could not read file for metadata extraction: {e}")
|
||||||
|
return []
|
||||||
|
|
||||||
|
# Extract metadata tags using regex
|
||||||
|
title_match = re.search(r"<<METADATA_TITLE:([^>]*)>>", text)
|
||||||
|
artist_match = re.search(r"<<METADATA_ARTIST:([^>]*)>>", text)
|
||||||
|
album_match = re.search(r"<<METADATA_ALBUM:([^>]*)>>", text)
|
||||||
|
year_match = re.search(r"<<METADATA_YEAR:([^>]*)>>", text)
|
||||||
|
album_artist_match = re.search(r"<<METADATA_ALBUM_ARTIST:([^>]*)>>", text)
|
||||||
|
|
||||||
|
# Use display path or filename as fallback for title
|
||||||
|
filename = os.path.splitext(os.path.basename(self.display_path if self.display_path else self.file_name))[0]
|
||||||
|
|
||||||
|
# Add title metadata
|
||||||
|
if title_match:
|
||||||
|
metadata_options.extend(["-metadata", f"title=Dinginliğin Gücü"])
|
||||||
|
else:
|
||||||
|
metadata_options.extend(["-metadata", f"title={filename}"])
|
||||||
|
|
||||||
|
# Add artist metadata
|
||||||
|
if artist_match:
|
||||||
|
metadata_options.extend(["-metadata", f"artist={artist_match.group(1)}"])
|
||||||
|
else:
|
||||||
|
metadata_options.extend(["-metadata", f"artist=Unknown"])
|
||||||
|
|
||||||
|
# Add album metadata
|
||||||
|
if album_match:
|
||||||
|
metadata_options.extend(["-metadata", f"album={album_match.group(1)}"])
|
||||||
|
else:
|
||||||
|
metadata_options.extend(["-metadata", f"album={filename}"])
|
||||||
|
|
||||||
|
# Add year metadata
|
||||||
|
if year_match:
|
||||||
|
metadata_options.extend(["-metadata", f"date={year_match.group(1)}"])
|
||||||
|
else:
|
||||||
|
# Use current year if year is not specified
|
||||||
|
import datetime
|
||||||
|
current_year = datetime.datetime.now().year
|
||||||
|
metadata_options.extend(["-metadata", f"date={current_year}"])
|
||||||
|
|
||||||
|
# Add album artist metadata
|
||||||
|
if album_artist_match:
|
||||||
|
metadata_options.extend(["-metadata", f"album_artist={album_artist_match.group(1)}"])
|
||||||
|
else:
|
||||||
|
metadata_options.extend(["-metadata", f"album_artist=Unknown"])
|
||||||
|
|
||||||
|
# Add these to ffmpeg command
|
||||||
|
return metadata_options
|
||||||
|
|
||||||
def _srt_time(self, t):
|
def _srt_time(self, t):
|
||||||
"""Helper function to format time for SRT files"""
|
"""Helper function to format time for SRT files"""
|
||||||
@@ -639,7 +930,14 @@ class VoicePreviewThread(QThread):
|
|||||||
error = pyqtSignal(str)
|
error = pyqtSignal(str)
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self, np_module, kpipeline_class, lang_code, voice, speed, parent=None
|
self,
|
||||||
|
np_module,
|
||||||
|
kpipeline_class,
|
||||||
|
lang_code,
|
||||||
|
voice,
|
||||||
|
speed,
|
||||||
|
use_gpu=False,
|
||||||
|
parent=None,
|
||||||
):
|
):
|
||||||
super().__init__(parent)
|
super().__init__(parent)
|
||||||
self.np_module = np_module
|
self.np_module = np_module
|
||||||
@@ -647,30 +945,54 @@ class VoicePreviewThread(QThread):
|
|||||||
self.lang_code = lang_code
|
self.lang_code = lang_code
|
||||||
self.voice = voice
|
self.voice = voice
|
||||||
self.speed = speed
|
self.speed = speed
|
||||||
self.temp_wav = None
|
self.use_gpu = use_gpu
|
||||||
|
|
||||||
|
# Cache location for preview audio
|
||||||
|
self.cache_dir = os.path.join(tempfile.gettempdir(), PROGRAM_NAME, "preview_cache")
|
||||||
|
os.makedirs(self.cache_dir, exist_ok=True)
|
||||||
|
|
||||||
|
# Calculate cache path
|
||||||
|
self.cache_path = self._get_cache_path()
|
||||||
|
|
||||||
|
def _get_cache_path(self):
|
||||||
|
"""Generate a unique filename for the voice with its parameters"""
|
||||||
|
# For a voice formula, use a hash of the formula
|
||||||
|
if "*" in self.voice:
|
||||||
|
voice_id = f"voice_formula_{hashlib.md5(self.voice.encode()).hexdigest()[:8]}"
|
||||||
|
else:
|
||||||
|
voice_id = self.voice
|
||||||
|
|
||||||
|
# Create a unique filename based on voice_id, language, and speed
|
||||||
|
filename = f"{voice_id}_{self.lang_code}_{self.speed:.2f}.wav"
|
||||||
|
return os.path.join(self.cache_dir, filename)
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
|
print(
|
||||||
|
f"\nVoice: {self.voice}\nLanguage: {self.lang_code}\nSpeed: {self.speed}\nGPU: {self.use_gpu}\n"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Generate the preview and save to cache
|
||||||
try:
|
try:
|
||||||
|
device = "cuda" if self.use_gpu else "cpu"
|
||||||
tts = self.kpipeline_class(
|
tts = self.kpipeline_class(
|
||||||
lang_code=self.lang_code, repo_id="hexgrad/Kokoro-82M"
|
lang_code=self.lang_code, repo_id="hexgrad/Kokoro-82M", device=device
|
||||||
)
|
)
|
||||||
|
# Enable voice formula support for preview
|
||||||
|
if "*" in self.voice:
|
||||||
|
loaded_voice = get_new_voice(tts, self.voice, self.use_gpu)
|
||||||
|
else:
|
||||||
|
loaded_voice = self.voice
|
||||||
sample_text = get_sample_voice_text(self.lang_code)
|
sample_text = get_sample_voice_text(self.lang_code)
|
||||||
audio_segments = []
|
audio_segments = []
|
||||||
for result in tts(
|
for result in tts(
|
||||||
sample_text, voice=self.voice, speed=self.speed, split_pattern=None
|
sample_text, voice=loaded_voice, speed=self.speed, split_pattern=None
|
||||||
):
|
):
|
||||||
audio_segments.append(result.audio)
|
audio_segments.append(result.audio)
|
||||||
if audio_segments:
|
if audio_segments:
|
||||||
audio = self.np_module.concatenate(audio_segments)
|
audio = self.np_module.concatenate(audio_segments)
|
||||||
# Create temp wav file in a folder in the system temp directory
|
# Save directly to the cache path
|
||||||
temp_dir = os.path.join(tempfile.gettempdir(), PROGRAM_NAME)
|
sf.write(self.cache_path, audio, 24000)
|
||||||
os.makedirs(temp_dir, exist_ok=True)
|
self.temp_wav = self.cache_path
|
||||||
fd, temp_path = tempfile.mkstemp(
|
|
||||||
prefix="abogen_", suffix=".wav", dir=temp_dir
|
|
||||||
)
|
|
||||||
os.close(fd)
|
|
||||||
sf.write(temp_path, audio, 24000)
|
|
||||||
self.temp_wav = temp_path
|
|
||||||
self.finished.emit()
|
self.finished.emit()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.error.emit(f"Voice preview error: {str(e)}")
|
self.error.emit(f"Voice preview error: {str(e)}")
|
||||||
@@ -683,6 +1005,7 @@ class PlayAudioThread(QThread):
|
|||||||
def __init__(self, wav_path, parent=None):
|
def __init__(self, wav_path, parent=None):
|
||||||
super().__init__(parent)
|
super().__init__(parent)
|
||||||
self.wav_path = wav_path
|
self.wav_path = wav_path
|
||||||
|
self.is_canceled = False
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
try:
|
try:
|
||||||
@@ -692,10 +1015,37 @@ class PlayAudioThread(QThread):
|
|||||||
pygame.mixer.init()
|
pygame.mixer.init()
|
||||||
pygame.mixer.music.load(self.wav_path)
|
pygame.mixer.music.load(self.wav_path)
|
||||||
pygame.mixer.music.play()
|
pygame.mixer.music.play()
|
||||||
# Wait until playback is finished
|
# Wait until playback is finished or canceled
|
||||||
while pygame.mixer.music.get_busy():
|
while pygame.mixer.music.get_busy() and not self.is_canceled:
|
||||||
_time.sleep(0.1)
|
_time.sleep(0.2)
|
||||||
pygame.mixer.music.unload()
|
|
||||||
|
# Make sure to clean up regardless of how we exited the loop
|
||||||
|
try:
|
||||||
|
pygame.mixer.music.stop()
|
||||||
|
pygame.mixer.music.unload()
|
||||||
|
pygame.mixer.quit() # Quit the mixer
|
||||||
|
except Exception:
|
||||||
|
# Ignore any errors during cleanup
|
||||||
|
pass
|
||||||
|
|
||||||
self.finished.emit()
|
self.finished.emit()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.error.emit(f"Audio playback error: {str(e)}")
|
# Handle initialization errors separately to give better error messages
|
||||||
|
if "mixer not initialized" in str(e):
|
||||||
|
self.error.emit("Audio playback error: The audio system was not properly initialized")
|
||||||
|
else:
|
||||||
|
self.error.emit(f"Audio playback error: {str(e)}")
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
"""Safely stop playback"""
|
||||||
|
self.is_canceled = True
|
||||||
|
# Try to stop pygame if it's running, but catch all exceptions
|
||||||
|
try:
|
||||||
|
import pygame
|
||||||
|
if pygame.mixer.get_init():
|
||||||
|
if pygame.mixer.music.get_busy():
|
||||||
|
pygame.mixer.music.stop()
|
||||||
|
pygame.mixer.music.unload()
|
||||||
|
except Exception:
|
||||||
|
# Ignore all errors when stopping since mixer might not be initialized
|
||||||
|
pass
|
||||||
|
|||||||
@@ -0,0 +1,36 @@
|
|||||||
|
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,16 @@
|
|||||||
|
import gpustat
|
||||||
|
|
||||||
|
def check():
|
||||||
|
try:
|
||||||
|
stats = gpustat.new_query()
|
||||||
|
except Exception:
|
||||||
|
return False
|
||||||
|
|
||||||
|
for gpu in stats.gpus:
|
||||||
|
print(gpu.name)
|
||||||
|
if 'nvidia' in gpu.name.lower():
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
print(check())
|
||||||
@@ -10,6 +10,10 @@ from gui import abogen
|
|||||||
from utils import get_resource_path
|
from utils import get_resource_path
|
||||||
from constants import PROGRAM_NAME, VERSION
|
from 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"
|
||||||
|
|
||||||
# Ensure sys.stdout and sys.stderr are valid in GUI mode
|
# Ensure sys.stdout and sys.stderr are valid in GUI mode
|
||||||
if sys.stdout is None:
|
if sys.stdout is None:
|
||||||
sys.stdout = open(os.devnull, "w")
|
sys.stdout = open(os.devnull, "w")
|
||||||
@@ -48,6 +52,13 @@ def main():
|
|||||||
if icon_path:
|
if icon_path:
|
||||||
app.setWindowIcon(QIcon(icon_path))
|
app.setWindowIcon(QIcon(icon_path))
|
||||||
|
|
||||||
|
# Set the .desktop name on Linux
|
||||||
|
if platform.system() == "Linux":
|
||||||
|
try:
|
||||||
|
app.setDesktopFileName("abogen.desktop")
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
|
||||||
ex = abogen()
|
ex = abogen()
|
||||||
ex.show()
|
ex.show()
|
||||||
sys.exit(app.exec_())
|
sys.exit(app.exec_())
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
import os
|
import os
|
||||||
|
import sys
|
||||||
import json
|
import json
|
||||||
import warnings
|
import warnings
|
||||||
import platform
|
import platform
|
||||||
@@ -32,6 +33,14 @@ def get_resource_path(package, resource):
|
|||||||
except (ImportError, FileNotFoundError):
|
except (ImportError, FileNotFoundError):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
# Always try to resolve as a relative path from this file
|
||||||
|
parts = package.split(".")
|
||||||
|
rel_path = os.path.join(
|
||||||
|
os.path.dirname(os.path.abspath(__file__)), *parts[1:], resource
|
||||||
|
)
|
||||||
|
if os.path.exists(rel_path):
|
||||||
|
return rel_path
|
||||||
|
|
||||||
# Fallback to local file system
|
# Fallback to local file system
|
||||||
try:
|
try:
|
||||||
# Extract the subdirectory from package name (e.g., 'assets' from 'abogen.assets')
|
# Extract the subdirectory from package name (e.g., 'assets' from 'abogen.assets')
|
||||||
@@ -50,7 +59,7 @@ def get_resource_path(package, resource):
|
|||||||
def get_version():
|
def get_version():
|
||||||
"""Return the current version of the application."""
|
"""Return the current version of the application."""
|
||||||
try:
|
try:
|
||||||
with open(get_resource_path("abogen", "VERSION"), "r") as f:
|
with open(get_resource_path("/", "VERSION"), "r") as f:
|
||||||
return f.read().strip()
|
return f.read().strip()
|
||||||
except Exception:
|
except Exception:
|
||||||
return "Unknown"
|
return "Unknown"
|
||||||
@@ -58,10 +67,20 @@ def get_version():
|
|||||||
|
|
||||||
# Define config path
|
# Define config path
|
||||||
def get_user_config_path():
|
def get_user_config_path():
|
||||||
if os.name == "nt":
|
from platformdirs import user_config_dir
|
||||||
config_dir = os.path.join(os.environ["APPDATA"], "abogen")
|
|
||||||
|
# TODO Config directory is changed for Linux and MacOS. But if old config exists, it will be used.
|
||||||
|
# On non‑Windows, prefer ~/.config/abogen if it already exists
|
||||||
|
if platform.system() != "Windows":
|
||||||
|
custom_dir = os.path.join(os.path.expanduser("~"), ".config", "abogen")
|
||||||
|
if os.path.exists(custom_dir):
|
||||||
|
config_dir = custom_dir
|
||||||
|
else:
|
||||||
|
config_dir = user_config_dir("abogen", appauthor=False, roaming=True)
|
||||||
else:
|
else:
|
||||||
config_dir = os.path.join(os.path.expanduser("~"), ".config", "abogen")
|
# Windows and fallback case
|
||||||
|
config_dir = user_config_dir("abogen", appauthor=False, roaming=True)
|
||||||
|
|
||||||
os.makedirs(config_dir, exist_ok=True)
|
os.makedirs(config_dir, exist_ok=True)
|
||||||
return os.path.join(config_dir, "config.json")
|
return os.path.join(config_dir, "config.json")
|
||||||
|
|
||||||
@@ -69,18 +88,104 @@ def get_user_config_path():
|
|||||||
_sleep_procs = {"Darwin": None, "Linux": None} # Store sleep prevention processes
|
_sleep_procs = {"Darwin": None, "Linux": None} # Store sleep prevention processes
|
||||||
|
|
||||||
|
|
||||||
def clean_text(text):
|
def clean_text(text, *args, **kwargs):
|
||||||
# Trim spaces and tabs at the start and end of each line, preserving blank lines
|
# Load replace_single_newlines from config
|
||||||
text = "\n".join(line.strip() for line in text.splitlines())
|
cfg = load_config()
|
||||||
|
replace_single_newlines = cfg.get("replace_single_newlines", False)
|
||||||
|
# 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()]
|
||||||
|
text = "\n".join(lines)
|
||||||
# Standardize paragraph breaks (multiple newlines become exactly two) and trim overall whitespace
|
# Standardize paragraph breaks (multiple newlines become exactly two) and trim overall whitespace
|
||||||
text = re.sub(r"\n{3,}", "\n\n", text).strip()
|
text = re.sub(r"\n{3,}", "\n\n", text).strip()
|
||||||
# Replace single newlines with spaces, but preserve double newlines
|
# Optionally replace single newlines with spaces, but preserve double newlines
|
||||||
# text = re.sub(r"(?<!\n)\n(?!\n)", " ", text)
|
if replace_single_newlines:
|
||||||
# Collapse multiple spaces and tabs into a single space
|
text = re.sub(r"(?<!\n)\n(?!\n)", " ", text)
|
||||||
text = re.sub(r"[ \t]+", " ", text)
|
|
||||||
return 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():
|
def load_config():
|
||||||
try:
|
try:
|
||||||
with open(get_user_config_path(), "r", encoding="utf-8") as f:
|
with open(get_user_config_path(), "r", encoding="utf-8") as f:
|
||||||
@@ -98,10 +203,16 @@ def save_config(config):
|
|||||||
|
|
||||||
|
|
||||||
def calculate_text_length(text):
|
def calculate_text_length(text):
|
||||||
# Remove double newlines (replace them with single newlines)
|
# Ignore chapter markers
|
||||||
cleaned_text = text.replace("\n\n", "")
|
text = re.sub(r"<<CHAPTER_MARKER:.*?>>", "", text)
|
||||||
|
# Ignore metadata patterns
|
||||||
|
text = re.sub(r"<<METADATA_[^:]+:[^>]*>>", "", text)
|
||||||
|
# Ignore newlines
|
||||||
|
text = text.replace("\n", "")
|
||||||
|
# Ignore leading/trailing spaces
|
||||||
|
text = text.strip()
|
||||||
# Calculate character count
|
# Calculate character count
|
||||||
char_count = len(cleaned_text)
|
char_count = len(text)
|
||||||
return char_count
|
return char_count
|
||||||
|
|
||||||
|
|
||||||
@@ -120,35 +231,26 @@ def prevent_sleep_start():
|
|||||||
system = platform.system()
|
system = platform.system()
|
||||||
if system == "Windows":
|
if system == "Windows":
|
||||||
import ctypes
|
import ctypes
|
||||||
|
|
||||||
ctypes.windll.kernel32.SetThreadExecutionState(
|
ctypes.windll.kernel32.SetThreadExecutionState(
|
||||||
0x80000000 | 0x00000001 | 0x00000040
|
0x80000000 | 0x00000001 | 0x00000040
|
||||||
) # ES_CONTINUOUS | ES_SYSTEM_REQUIRED | ES_AWAYMODE_REQUIRED
|
)
|
||||||
elif system == "Darwin":
|
elif system == "Darwin":
|
||||||
_sleep_procs["Darwin"] = subprocess.Popen(["caffeinate"])
|
_sleep_procs["Darwin"] = create_process(["caffeinate"])
|
||||||
elif system == "Linux":
|
elif system == "Linux":
|
||||||
try:
|
# use a sleep that never exits so inhibition stays active
|
||||||
_sleep_procs["Linux"] = subprocess.Popen(
|
_sleep_procs["Linux"] = create_process([
|
||||||
[
|
"systemd-inhibit",
|
||||||
"systemd-inhibit",
|
"--what=handle-lid-switch:sleep",
|
||||||
"--what=sleep",
|
"--mode=block",
|
||||||
"--why=TextToAudiobook conversion",
|
"sleep",
|
||||||
"sleep",
|
"infinity",
|
||||||
"999999",
|
])
|
||||||
]
|
|
||||||
)
|
|
||||||
except Exception:
|
|
||||||
try:
|
|
||||||
subprocess.Popen(["xdg-screensaver", "reset"])
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
def prevent_sleep_end():
|
def prevent_sleep_end():
|
||||||
system = platform.system()
|
system = platform.system()
|
||||||
if system == "Windows":
|
if system == "Windows":
|
||||||
import ctypes
|
import ctypes
|
||||||
|
|
||||||
ctypes.windll.kernel32.SetThreadExecutionState(0x80000000) # ES_CONTINUOUS
|
ctypes.windll.kernel32.SetThreadExecutionState(0x80000000) # ES_CONTINUOUS
|
||||||
elif system in ("Darwin", "Linux") and _sleep_procs[system]:
|
elif system in ("Darwin", "Linux") and _sleep_procs[system]:
|
||||||
try:
|
try:
|
||||||
|
|||||||
@@ -0,0 +1,57 @@
|
|||||||
|
import re
|
||||||
|
from constants import VOICES_INTERNAL
|
||||||
|
|
||||||
|
|
||||||
|
# Calls parsing and loads the voice to gpu or cpu
|
||||||
|
def get_new_voice(pipeline, formula, use_gpu):
|
||||||
|
try:
|
||||||
|
weighted_voice = parse_voice_formula(pipeline, formula)
|
||||||
|
# device = "cuda" if use_gpu else "cpu"
|
||||||
|
# Setting the device "cuda" gives "Error occurred: split_with_sizes(): argument 'split_sizes' (position 2)"
|
||||||
|
# error when the device is gpu. So disabling this for now.
|
||||||
|
device = "cpu"
|
||||||
|
return weighted_voice.to(device)
|
||||||
|
except Exception as e:
|
||||||
|
raise ValueError(f"Failed to create voice: {str(e)}")
|
||||||
|
|
||||||
|
|
||||||
|
# Parse the formula and get the combined voice tensor
|
||||||
|
def parse_voice_formula(pipeline, formula):
|
||||||
|
if not formula.strip():
|
||||||
|
raise ValueError("Empty voice formula")
|
||||||
|
|
||||||
|
# Initialize the weighted sum
|
||||||
|
weighted_sum = None
|
||||||
|
|
||||||
|
total_weight = calculate_sum_from_formula(formula)
|
||||||
|
|
||||||
|
# Split the formula into terms
|
||||||
|
voices = formula.split("+")
|
||||||
|
|
||||||
|
for term in voices:
|
||||||
|
# Parse each term (format: "voice_name*0.333")
|
||||||
|
voice_name, weight = term.strip().split("*")
|
||||||
|
weight = float(weight.strip())
|
||||||
|
# normalize the weight
|
||||||
|
weight /= total_weight if total_weight > 0 else 1.0
|
||||||
|
voice_name = voice_name.strip()
|
||||||
|
|
||||||
|
# Get the voice tensor
|
||||||
|
if voice_name not in VOICES_INTERNAL:
|
||||||
|
raise ValueError(f"Unknown voice: {voice_name}")
|
||||||
|
|
||||||
|
voice_tensor = pipeline.load_single_voice(voice_name)
|
||||||
|
|
||||||
|
# Add to weighted sum
|
||||||
|
if weighted_sum is None:
|
||||||
|
weighted_sum = weight * voice_tensor
|
||||||
|
else:
|
||||||
|
weighted_sum += weight * voice_tensor
|
||||||
|
|
||||||
|
return weighted_sum
|
||||||
|
|
||||||
|
|
||||||
|
def calculate_sum_from_formula(formula):
|
||||||
|
weights = re.findall(r"\* *([\d.]+)", formula)
|
||||||
|
total_sum = sum(float(weight) for weight in weights)
|
||||||
|
return total_sum
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
import os
|
||||||
|
import json
|
||||||
|
from utils import get_user_config_path
|
||||||
|
|
||||||
|
|
||||||
|
def _get_profiles_path():
|
||||||
|
config_path = get_user_config_path()
|
||||||
|
config_dir = os.path.dirname(config_path)
|
||||||
|
return os.path.join(config_dir, "voice_profiles.json")
|
||||||
|
|
||||||
|
|
||||||
|
def load_profiles():
|
||||||
|
"""Load all voice profiles from JSON file."""
|
||||||
|
path = _get_profiles_path()
|
||||||
|
if os.path.exists(path):
|
||||||
|
try:
|
||||||
|
with open(path, "r", encoding="utf-8") as f:
|
||||||
|
data = json.load(f)
|
||||||
|
# always expect abogen_voice_profiles wrapper
|
||||||
|
if isinstance(data, dict) and "abogen_voice_profiles" in data:
|
||||||
|
return data["abogen_voice_profiles"]
|
||||||
|
# fallback: treat as profiles dict
|
||||||
|
if isinstance(data, dict):
|
||||||
|
return data
|
||||||
|
except Exception:
|
||||||
|
return {}
|
||||||
|
return {}
|
||||||
|
|
||||||
|
|
||||||
|
def save_profiles(profiles):
|
||||||
|
"""Save all voice profiles to JSON file."""
|
||||||
|
path = _get_profiles_path()
|
||||||
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||||||
|
with open(path, "w", encoding="utf-8") as f:
|
||||||
|
# always save with abogen_voice_profiles wrapper
|
||||||
|
json.dump({"abogen_voice_profiles": profiles}, f, indent=2)
|
||||||
|
|
||||||
|
|
||||||
|
def delete_profile(name):
|
||||||
|
"""Remove a profile by name."""
|
||||||
|
profiles = load_profiles()
|
||||||
|
if name in profiles:
|
||||||
|
del profiles[name]
|
||||||
|
save_profiles(profiles)
|
||||||
|
|
||||||
|
|
||||||
|
def duplicate_profile(src, dest):
|
||||||
|
"""Duplicate an existing profile."""
|
||||||
|
profiles = load_profiles()
|
||||||
|
if src in profiles and dest:
|
||||||
|
profiles[dest] = profiles[src]
|
||||||
|
save_profiles(profiles)
|
||||||
|
|
||||||
|
|
||||||
|
def export_profiles(export_path):
|
||||||
|
"""Export all profiles to specified JSON file."""
|
||||||
|
profiles = load_profiles()
|
||||||
|
with open(export_path, "w", encoding="utf-8") as f:
|
||||||
|
json.dump({"abogen_voice_profiles": profiles}, f, indent=2)
|
||||||
@@ -40,11 +40,6 @@ Run this FFmpeg command to create the tiny video:
|
|||||||
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 audio.wav -vf "ass=your_subtitle_demo.ass" -c:v libvpx-vp9 -b:v 0 -crf 30 -c:a libopus -shortest demo.webm
|
||||||
```
|
```
|
||||||
|
|
||||||
That's it! The magic happens because:
|
|
||||||
- We use a single static background image instead of many frames
|
|
||||||
- The subtitles are stored as text (vector data), not as pixels
|
|
||||||
- VP9 video codec with Opus audio provides excellent compression
|
|
||||||
|
|
||||||
## For Higher Quality (But Larger) Video (.mp4)
|
## For Higher Quality (But Larger) Video (.mp4)
|
||||||
|
|
||||||
If you need better quality for distribution, use this command instead:
|
If you need better quality for distribution, use this command instead:
|
||||||
|
|||||||
|
Before Width: | Height: | Size: 22 KiB After Width: | Height: | Size: 23 KiB |
|
After Width: | Height: | Size: 11 KiB |
|
After Width: | Height: | Size: 28 KiB |
@@ -15,11 +15,15 @@ keywords = ["audiobook", "epub", "pdf", "text-to-speech", "subtitle", "tts", "ko
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"PyQt5>=5.15.11",
|
"PyQt5>=5.15.11",
|
||||||
"kokoro>=0.9.4",
|
"kokoro>=0.9.4",
|
||||||
"ebooklib>=0.18",
|
"ebooklib>=0.19",
|
||||||
"beautifulsoup4>=4.13.4",
|
"beautifulsoup4>=4.13.4",
|
||||||
"PyMuPDF>=1.25.5",
|
"PyMuPDF>=1.25.5",
|
||||||
|
"platformdirs>=4.3.7",
|
||||||
"soundfile>=0.13.1",
|
"soundfile>=0.13.1",
|
||||||
"pygame>=2.6.1"
|
"pygame>=2.6.1",
|
||||||
|
"charset_normalizer>=3.4.1",
|
||||||
|
"chardet>=5.2.0",
|
||||||
|
"static_ffmpeg>=2.13"
|
||||||
]
|
]
|
||||||
|
|
||||||
classifiers = [
|
classifiers = [
|
||||||
@@ -52,10 +56,9 @@ abogen-cli = "abogen.main:main"
|
|||||||
exclude = [
|
exclude = [
|
||||||
"/.github",
|
"/.github",
|
||||||
"/demo",
|
"/demo",
|
||||||
"/demo",
|
"/abogen/resources",
|
||||||
"/abogen/assets/create_shortcuts.bat",
|
"/abogen/assets/create_shortcuts.bat",
|
||||||
"WINDOWS_INSTALL.bat",
|
"WINDOWS_INSTALL.bat",
|
||||||
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[tool.hatch.build.targets.wheel]
|
[tool.hatch.build.targets.wheel]
|
||||||
|
|||||||