59 Commits
Author SHA1 Message Date
Deniz ŞafakandGitHub f611512747 Update CHANGELOG.md 2025-05-10 01:25:48 +03:00
Deniz Şafak accb3efc54 v1.0.7 2025-05-10 01:24:22 +03:00
Deniz Şafak f415c0d4d3 Fix ffmpeg error in Linux 2025-05-10 01:12:37 +03:00
Deniz Şafak 7fff2f154c Fix [Errno 32] in Linux 2025-05-10 00:44:26 +03:00
Deniz Şafak 85481ad119 Update changelog and shortcut text 2025-05-10 00:28:15 +03:00
Deniz Şafak 143c9d9b30 Offer installation for linux 2025-05-10 00:21:57 +03:00
Deniz Şafak 5cfb8b4be7 Update readme, fix desktop shortcut in linux 2025-05-10 00:08:38 +03:00
Deniz Şafak 1b5b8e0e31 Add icon.png 2025-05-09 20:40:54 +03:00
Deniz Şafak 11caeedef5 Added Add desktop shortcut option to Linux version 2025-05-09 20:39:28 +03:00
Deniz Şafak 9c997c9616 Added Save in a project folder with metadata option, added Go to folder button 2025-05-09 19:34:32 +03:00
Deniz Şafak ac8907041a Improved PDF handling 2025-05-09 17:36:49 +03:00
Deniz Şafak 22086793ae Improved process handling for subpprocess calls 2025-05-09 10:58:05 +03:00
Deniz Şafak 101012d9aa Prevent cancellation if process is at 99% 2025-05-09 01:50:09 +03:00
Deniz Şafak cab0a7631f Improve chaptered audio generation by outputting directly as m4b instead of converting from wav 2025-05-09 01:29:14 +03:00
Deniz Şafak 375a185aef Added .opus support as output format for generated audio files. 2025-05-08 20:43:44 +03:00
Deniz Şafak a691b9af58 Added "Playing..." indicator for "Preview" button in the voice mixer 2025-05-08 16:38:24 +03:00
Deniz Şafak 311cdaeae1 Ignore chapter markers and single newlines when calculating text length 2025-05-08 16:29:43 +03:00
Deniz Şafak 4d0aabc13e Update changelog 2025-05-08 15:51:04 +03:00
Deniz Şafak b644da8556 Fix "No module named 'progress'" in Windows 10 2025-05-05 11:41:09 +03:00
Deniz Şafak fa6e3fce1a Fix "No module named 'progress'" in Windows 10 2025-05-05 11:26:01 +03:00
Deniz Şafak 8f8ad3ec26 v1.0.6 2025-05-04 15:23:25 +03:00
Deniz Şafak 67b35d1dfb Add preview button to voice mixer 2025-05-04 15:10:08 +03:00
Deniz Şafak 15f1eb7f1e Add chapter marker button, fix f-string, fix contnet before chapter not including 2025-05-04 13:53:03 +03:00
Deniz Şafak 146fcb3c1f Restyle using black, improve m4b process handling 2025-05-04 11:39:54 +03:00
Deniz Şafak 412fb3bc7e Update readme 2025-05-04 11:10:07 +03:00
Deniz ŞafakandGitHub d9b6081a06 Merge pull request #13 from denizsafak/beta
- 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.
2025-05-04 11:00:44 +03:00
Deniz ŞafakandGitHub 674fc14813 Merge branch 'main' into beta 2025-05-04 11:00:16 +03:00
Deniz Şafak fa144561fb Update changelog 2025-05-04 10:58:37 +03:00
Deniz Şafak 1d27f6fd2e Improvements in documentation 2025-05-04 10:50:36 +03:00
Deniz Şafak a03120d66f Fix "cannot access local variable 'm4b_picked'", improve m4b handling 2025-05-04 02:07:34 +03:00
Deniz Şafak 68b44a3516 Add label for m4b output format, improvements 2025-05-04 00:51:56 +03:00
Deniz ŞafakandGitHub 3bbb88d446 Update Dockerfile 2025-05-04 00:14:01 +03:00
Deniz Şafak a1ccd51ea0 Fix dockerfile 2025-05-04 00:13:33 +03:00
Deniz Şafak 13cf6f20a4 Added informative messages and improved error handling for ffmpeg 2025-05-04 00:03:53 +03:00
Deniz Şafak 4311ed51d4 Fix m4b error in seperate chapters 2025-05-03 23:40:01 +03:00
Deniz Şafak 9170d9ae3e Update readme 2025-05-03 23:15:38 +03:00
Deniz Şafak 593d21675e Update readme 2025-05-03 23:15:06 +03:00
Deniz Şafak 9aebdecc60 Fix progress error 2025-05-03 23:13:43 +03:00
Deniz ŞafakandGitHub 290d246e80 Merge pull request #10 from jborza/main
Add chapter metadata
2025-05-03 23:01:00 +03:00
Deniz ŞafakandGitHub 5ac6e3bb2a Merge pull request #11 from denizsafak/main
Update beta branch
2025-05-03 22:57:27 +03:00
Deniz Şafak 6bf189312a Add docker support, fix config dir 2025-05-03 22:55:41 +03:00
Juraj Borza 22fdccd78d Merge branch 'main' of https://github.com/jborza/abogen 2025-05-03 19:34:39 +02:00
Juraj Borza 62aeaecf89 Removed a variable that was unused in m4b generation 2025-05-03 19:34:35 +02:00
Juraj BorzaandGitHub a7a41fa4f6 Merge branch 'denizsafak:main' into main 2025-05-03 19:27:33 +02:00
Juraj Borza 1ca4d0d5f9 Merge branch 'main' of https://github.com/jborza/abogen 2025-05-03 19:26:46 +02:00
Juraj Borza e09535fa83 Add chapter metadata - added m4b format, generating chapters info and converting via ffmpeg 2025-05-03 19:26:19 +02:00
Deniz Şafak 0863b25d82 Better approach for detemining the correct configuration folder for Linux and MacOS, using platformdirs 2025-05-03 18:31:45 +03:00
Deniz Şafak b02416e068 Fix loading movie not playing again after first time 2025-05-03 16:00:12 +03:00
Deniz Şafak 8daa3cb3b5 Update readme about chapter markers 2025-05-03 12:09:26 +03:00
Deniz Şafak 20f31f82fd v1.0.4 2025-05-02 21:18:07 +03:00
Deniz Şafak d5388c62be Fix enable/disable subtitle_combo fom voice mixer 2025-05-02 21:16:51 +03:00
Deniz Şafak 07a2fe2d5b Fix new profile duplicating 2025-05-02 20:59:23 +03:00
Deniz ŞafakandGitHub 58d9073ace Merge pull request #7 from jborza/main
Updated development instructions in README
2025-05-02 20:17:17 +03:00
Deniz ŞafakandGitHub 2cbb2f8aa7 Merge pull request #8 from denizsafak/beta
Imrpovements for voice mixer
2025-05-02 20:12:47 +03:00
Deniz Şafak f7d8d5ccd8 Imrpovements for voice mixer 2025-05-02 20:10:28 +03:00
Juraj Borza e7d479d846 use CPU conversion if GPU conversion is not available also in preview 2025-05-02 19:00:01 +02:00
Juraj Borza 036eed3142 also stopping the loading movie on preview cleanup 2025-05-02 18:58:38 +02:00
Juraj Borza 0d86d107e9 remove the animation from preview button on cleanup 2025-05-02 18:50:33 +02:00
Juraj Borza 56b336a529 Added 'pip install build' to development instructions, otherwise python -m build won't work 2025-05-02 18:28:17 +02:00
15 changed files with 985 additions and 242 deletions
+49 -1
View File
@@ -1,3 +1,36 @@
# 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 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. - 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. - Improvements in the voice mixer, mostly for organizing controls and enhancing user experience.
@@ -5,4 +38,19 @@
- Improved the content and chapter extraction process for EPUB files, ensuring better handling of various structures. - 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. - 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. - Fixed preview voices was not using GPU acceleration, which was causing performance issues.
- Improvements in code and documentation. - Improvements in code and documentation.
# v1.0.2
- Enhanced EPUB handling by treating all items in chapter list (including anchors) as chapters, improving navigation and organization for poorly structured books, mentioned by @Darthagnon in #4
- Fixed the issue with some chapters in EPUB files had missing content.
- Fixed the issue with some EPUB files only having one chapter caused the program to ignore the entire book.
- Fixed "utf-8' codec can't decode byte" error, mentioned by @nigelp in #3
- Added "Replace single newlines with spaces" option in the menu. This can be useful for texts that have imaginary line breaks.
- Improvements in code and documentation.
# v1.0.1
- Added abogen-cli command for better troubleshooting and error handling.
- Switched from setuptools to hatchling for packaging.
- Added classifiers to the package metadata.
- Fixed "No module named 'docopt'" and "setuptools.build_meta" import errors while using .bat installer in Windows, mentioned by @nigelp in #2
- Improvements in code and documentation.
+79 -10
View File
@@ -2,7 +2,7 @@
[![Build Status](https://github.com/denizsafak/abogen/actions/workflows/test_pip.yml/badge.svg)](https://github.com/denizsafak/abogen/actions) [![Build Status](https://github.com/denizsafak/abogen/actions/workflows/test_pip.yml/badge.svg)](https://github.com/denizsafak/abogen/actions)
[![GitHub Release](https://img.shields.io/github/v/release/denizsafak/abogen)](https://github.com/denizsafak/abogen/releases/latest) [![GitHub Release](https://img.shields.io/github/v/release/denizsafak/abogen)](https://github.com/denizsafak/abogen/releases/latest)
![abogen PyPi Python Versions](https://img.shields.io/pypi/pyversions/abogen) [![Abogen PyPi Python Versions](https://img.shields.io/pypi/pyversions/abogen)](https://pypi.org/project/abogen/)
[![Operating Systems](https://img.shields.io/badge/os-windows%20%7C%20linux%20%7C%20macos%20-blue)](https://github.com/denizsafak/abogen/releases/latest) [![Operating Systems](https://img.shields.io/badge/os-windows%20%7C%20linux%20%7C%20macos%20-blue)](https://github.com/denizsafak/abogen/releases/latest)
[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
[![License: MIT](https://img.shields.io/badge/License-MIT-maroon.svg)](https://opensource.org/licenses/MIT) [![License: MIT](https://img.shields.io/badge/License-MIT-maroon.svg)](https://opensource.org/licenses/MIT)
@@ -14,10 +14,9 @@ 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
@@ -56,13 +55,20 @@ sudo dnf install espeak-ng
# Install abogen # Install abogen
pip install abogen pip install abogen
``` ```
> 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. > If you get "No matching distribution found" error, try installing it on supported Python (3.10 to 3.12). You can use [pyenv](https://github.com/pyenv/pyenv) to manage multiple Python versions easily in Linux. Watch this [video](https://www.youtube.com/watch?v=MVyb-nI4KyI) by NetworkChuck for a quick guide.
Then simply run by typing: ## `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)
@@ -85,12 +91,14 @@ Heres Abogen in action: in this demo, it processes 3,000 characters of tex
- **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. - **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**:
- **Replace single newlines with spaces**: Replaces single newlines with spaces in the text. This is useful for texts that have imaginary line breaks. - **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**: Automatically configures the maximum number of words per subtitle entry. - **Configure max words per subtitle**: Configures the maximum number of words per subtitle entry.
- **Create desktop shortcut**: Creates a shortcut on your desktop for easy access. - **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.
@@ -103,6 +111,30 @@ Heres Abogen in action: in this demo, it processes 3,000 characters of tex
With voice mixer, you can create custom voices by mixing different voice models. You can adjust the weight of each voice and save your custom voice as a profile for future use. The voice mixer allows you to create unique and personalized voices. (Huge thanks to [@jborza](https://github.com/jborza) for making this possible through his contributions in [#5](https://github.com/denizsafak/abogen/pull/5)) With voice mixer, you can create custom voices by mixing different voice models. You can adjust the weight of each voice and save your custom voice as a profile for future use. The voice mixer allows you to create unique and personalized voices. (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.
![Abogen Chapter Marker](https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/chapter_marker.png)
## `Supported Languages` ## `Supported Languages`
``` ```
# 🇺🇸 'a' => American English, 🇬🇧 'b' => British English # 🇺🇸 'a' => American English, 🇬🇧 'b' => British English
@@ -131,6 +163,42 @@ audio-samplerate=48000
volume-max=200 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)**
@@ -139,11 +207,11 @@ Abogen is a standalone project, but it is inspired by and shares some similariti
- [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
## `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.
- [x] Add voice formula feature that enables mixing different voice models. #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`
@@ -160,6 +228,7 @@ 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
pip install build # Install the build package
python -m build # Builds the package in dist folder (optional) python -m build # Builds the package in dist folder (optional)
abogen # Opens the GUI abogen # Opens the GUI
``` ```
+9
View File
@@ -219,6 +219,15 @@ 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 wheel sphinx hatchling --no-warn-script-location
+42
View File
@@ -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
View File
@@ -1 +1 @@
1.0.3 1.0.7
Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

+73 -61
View File
@@ -37,6 +37,7 @@ class HandlerDialog(QDialog):
# Class variables to remember checkbox states between dialog instances # Class variables to remember checkbox states between dialog instances
_save_chapters_separately = False _save_chapters_separately = False
_merge_chapters_at_end = True _merge_chapters_at_end = True
_save_as_project = False # New class variable for save_as_project option
def __init__(self, book_path, file_type=None, checked_chapters=None, parent=None): def __init__(self, book_path, file_type=None, checked_chapters=None, parent=None):
super().__init__(parent) super().__init__(parent)
@@ -64,6 +65,7 @@ class HandlerDialog(QDialog):
# Initialize save chapters flags from class variables # Initialize save chapters flags from class variables
self.save_chapters_separately = HandlerDialog._save_chapters_separately self.save_chapters_separately = HandlerDialog._save_chapters_separately
self.merge_chapters_at_end = HandlerDialog._merge_chapters_at_end self.merge_chapters_at_end = HandlerDialog._merge_chapters_at_end
self.save_as_project = HandlerDialog._save_as_project
# Load the book based on file type # Load the book based on file type
try: try:
@@ -934,6 +936,7 @@ class HandlerDialog(QDialog):
def _build_pdf_tree(self): def _build_pdf_tree(self):
outline = self.pdf_doc.get_toc() outline = self.pdf_doc.get_toc()
self.has_pdf_bookmarks = bool(outline) self.has_pdf_bookmarks = bool(outline)
self.bookmark_items_map = {}
if not outline: if not outline:
self._build_pdf_pages_tree() self._build_pdf_pages_tree()
@@ -976,20 +979,23 @@ class HandlerDialog(QDialog):
else self.pdf_doc.resolve_link(page)[0] else self.pdf_doc.resolve_link(page)[0]
) )
page_id = f"page_{page_num+1}" page_id = f"page_{page_num+1}"
# attach chapters on same page under original
if page_num in self.bookmark_items_map:
orig = self.bookmark_items_map[page_num]
child = QTreeWidgetItem(orig, [f"{title} (Same page)"])
child.setData(0, Qt.UserRole, page_id)
child.setFlags(child.flags() & ~Qt.ItemIsUserCheckable)
continue
bookmark_item = QTreeWidgetItem(parent_item, [title]) bookmark_item = QTreeWidgetItem(parent_item, [title])
bookmark_item.setData(0, Qt.UserRole, page_id) bookmark_item.setData(0, Qt.UserRole, page_id)
bookmark_item.setFlags( # only allow checking if this chapter has content
bookmark_item.flags() | Qt.ItemIsUserCheckable if self.content_lengths.get(page_id, 0) > 0:
) bookmark_item.setFlags(bookmark_item.flags() | Qt.ItemIsUserCheckable)
bookmark_item.setCheckState( bookmark_item.setCheckState(0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked)
0, else:
( bookmark_item.setFlags(bookmark_item.flags() & ~Qt.ItemIsUserCheckable)
Qt.Checked # map for uncategorized pages
if page_id in self.checked_chapters self.bookmark_items_map[page_num] = bookmark_item
else Qt.Unchecked
),
)
added_pages.add(page_num) added_pages.add(page_num)
@@ -1012,30 +1018,39 @@ class HandlerDialog(QDialog):
page_item = QTreeWidgetItem(bookmark_item, [page_title]) page_item = QTreeWidgetItem(bookmark_item, [page_title])
page_item.setData(0, Qt.UserRole, page_id) page_item.setData(0, Qt.UserRole, page_id)
page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable) # only allow checking if this sub-page has content
page_item.setCheckState( if self.content_lengths.get(page_id, 0) > 0:
0, page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable)
( page_item.setCheckState(0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked)
Qt.Checked else:
if page_id in self.checked_chapters page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable)
else Qt.Unchecked
),
)
added_pages.add(sub_page_num) added_pages.add(sub_page_num)
if len(entry) > 3 and isinstance(entry[3], list):
build_outline_tree(entry[3], bookmark_item)
build_outline_tree(outline, self.treeWidget) build_outline_tree(outline, self.treeWidget)
covered_pages = set(added_pages) covered_pages = set(added_pages)
# attach any pages without direct bookmarks under nearest preceding chapter
uncategorized_pages = [ uncategorized_pages = [i for i in range(len(self.pdf_doc)) if i not in covered_pages]
i for i in range(len(self.pdf_doc)) if i not in covered_pages for page_num in uncategorized_pages:
] # find nearest previous bookmark
if uncategorized_pages: prev_nums = [n for n in sorted(self.bookmark_items_map) if n < page_num]
self._add_other_pages(uncategorized_pages) parent_item = self.bookmark_items_map[prev_nums[-1]] if prev_nums else self.treeWidget
page_id = f"page_{page_num+1}"
title = f"Page {page_num+1}"
text = self.content_texts.get(page_id, "").strip()
if text:
first = text.split("\n", 1)[0].strip()
if first and len(first) < 100:
title += f" - {first}"
page_item = QTreeWidgetItem(parent_item, [title])
page_item.setData(0, Qt.UserRole, page_id)
# only allow checking if uncategorized page has content
if self.content_lengths.get(page_id, 0) > 0:
page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable)
page_item.setCheckState(0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked)
else:
page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable)
def _build_pdf_pages_tree(self): def _build_pdf_pages_tree(self):
pages_item = QTreeWidgetItem(self.treeWidget, ["Pages"]) pages_item = QTreeWidgetItem(self.treeWidget, ["Pages"])
@@ -1056,34 +1071,12 @@ class HandlerDialog(QDialog):
page_item = QTreeWidgetItem(pages_item, [page_title]) page_item = QTreeWidgetItem(pages_item, [page_title])
page_item.setData(0, Qt.UserRole, page_id) page_item.setData(0, Qt.UserRole, page_id)
page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable) # only allow checking if standalone page has content
page_item.setCheckState( if self.content_lengths.get(page_id, 0) > 0:
0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable)
) page_item.setCheckState(0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked)
else:
def _add_other_pages(self, uncategorized_pages): page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable)
other_pages = QTreeWidgetItem(self.treeWidget, ["Other Pages"])
other_pages.setFlags(other_pages.flags() & ~Qt.ItemIsUserCheckable)
font = other_pages.font(0)
font.setBold(True)
other_pages.setFont(0, font)
for page_num in uncategorized_pages:
page_id = f"page_{page_num+1}"
page_title = f"Page {page_num+1}"
page_text = self.content_texts.get(page_id, "").strip()
if page_text:
first_line = page_text.split("\n", 1)[0].strip()
if first_line and len(first_line) < 100:
page_title += f" - {first_line}"
page_item = QTreeWidgetItem(other_pages, [page_title])
page_item.setData(0, Qt.UserRole, page_id)
page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable)
page_item.setCheckState(
0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked
)
def _are_provided_checks_relevant(self): def _are_provided_checks_relevant(self):
if not self.checked_chapters: if not self.checked_chapters:
@@ -1108,7 +1101,7 @@ class HandlerDialog(QDialog):
self.previewEdit.setStyleSheet("QTextEdit { border: none; }") self.previewEdit.setStyleSheet("QTextEdit { border: none; }")
self.previewInfoLabel = QLabel( self.previewInfoLabel = QLabel(
'*Note: You can modify the content later using the "Edit" button in the input box or by accessing the temporary files directory through settings.', '*Note: You can modify the content later using the "Edit" button in the input box or by accessing the temporary files directory through settings (if not saved in a project folder).',
self, self,
) )
self.previewInfoLabel.setWordWrap(True) self.previewInfoLabel.setWordWrap(True)
@@ -1182,8 +1175,7 @@ class HandlerDialog(QDialog):
self.save_chapters_checkbox = QCheckBox(checkbox_text, self) self.save_chapters_checkbox = QCheckBox(checkbox_text, self)
self.save_chapters_checkbox.setChecked(self.save_chapters_separately) self.save_chapters_checkbox.setChecked(self.save_chapters_separately)
self.save_chapters_checkbox.stateChanged.connect(self.on_save_chapters_changed) self.save_chapters_checkbox.stateChanged.connect(self.on_save_chapters_changed)
leftLayout.addWidget(self.save_chapters_checkbox) leftLayout.addWidget(self.save_chapters_checkbox)
self.merge_chapters_checkbox = QCheckBox( self.merge_chapters_checkbox = QCheckBox(
"Create a merged version at the end", self "Create a merged version at the end", self
) )
@@ -1192,6 +1184,19 @@ class HandlerDialog(QDialog):
self.on_merge_chapters_changed self.on_merge_chapters_changed
) )
leftLayout.addWidget(self.merge_chapters_checkbox) leftLayout.addWidget(self.merge_chapters_checkbox)
self.save_as_project_checkbox = QCheckBox(
"Save in a project folder with metadata", self
)
self.save_as_project_checkbox.setToolTip(
"Save the converted item in a project folder with metadata files. "
"(Useful if you want to work with converted items in the future.)"
)
self.save_as_project_checkbox.setChecked(self.save_as_project)
self.save_as_project_checkbox.stateChanged.connect(
self.on_save_as_project_changed
)
leftLayout.addWidget(self.save_as_project_checkbox)
leftLayout.addWidget(buttons) leftLayout.addWidget(buttons)
@@ -1705,6 +1710,10 @@ class HandlerDialog(QDialog):
self.merge_chapters_at_end = bool(state) self.merge_chapters_at_end = bool(state)
HandlerDialog._merge_chapters_at_end = self.merge_chapters_at_end HandlerDialog._merge_chapters_at_end = self.merge_chapters_at_end
def on_save_as_project_changed(self, state):
self.save_as_project = bool(state)
HandlerDialog._save_as_project = self.save_as_project
def get_save_chapters_separately(self): def get_save_chapters_separately(self):
return ( return (
self.save_chapters_separately self.save_chapters_separately
@@ -1715,6 +1724,9 @@ class HandlerDialog(QDialog):
def get_merge_chapters_at_end(self): def get_merge_chapters_at_end(self):
return self.merge_chapters_at_end return self.merge_chapters_at_end
def get_save_as_project(self):
return self.save_as_project
def on_tree_context_menu(self, pos): def on_tree_context_menu(self, pos):
item = self.treeWidget.itemAt(pos) item = self.treeWidget.itemAt(pos)
if ( if (
+202 -40
View File
@@ -8,9 +8,12 @@ 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 from voice_formulas import get_new_voice
import static_ffmpeg
import threading # for efficient waiting
import subprocess
def get_sample_voice_text(lang_code): def get_sample_voice_text(lang_code):
@@ -128,6 +131,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
@@ -222,6 +226,12 @@ class ConversionThread(QThread):
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 = (
@@ -252,18 +262,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
@@ -339,6 +343,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:
@@ -354,6 +364,10 @@ 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+"
@@ -467,6 +481,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
@@ -483,15 +500,45 @@ 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)
# Determine chapter extension (.wav for m4b output)
chapter_ext = 'wav' if self.output_format == 'm4b' else self.output_format
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}.{chapter_ext}"
)
sf.write(
chapter_out_path,
chapter_audio,
24000,
format=self.output_format,
) )
if self.output_format == "opus":
static_ffmpeg.add_paths()
proc = create_process(
[
"ffmpeg",
"-y",
"-thread_queue_size", "1024", # 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='wav' if self.output_format == 'm4b' else self.output_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:
@@ -531,34 +578,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( elif intended_output_format == "opus":
srt_path, "w", encoding="utf-8", errors="replace" static_ffmpeg.add_paths()
) as srt_file: opus_out_path = f"{base_filepath_no_ext}.opus"
for i, (start, end, text) in enumerate(subtitle_entries, 1): ffmpeg_cmd_opus = [
srt_file.write( "ffmpeg", "-y",
f"{i}\n{self._srt_time(start)} --> {self._srt_time(end)}\n{text}\n\n" "-thread_queue_size", "1024", # Increased thread_queue_size
) "-f", "f32le", "-ar", "24000", "-ac", "1", "-i", "pipe:0",
self.conversion_finished.emit( "-c:a", "libopus", "-b:a", "24000", # Original bitrate
( opus_out_path,
f"Audiobook saved to: {out_path}\n\nSubtitle saved to: {srt_path}", ]
"green", try:
), process = create_process(ffmpeg_cmd_opus, stdin=subprocess.PIPE, text=False) # Ensure binary stdin
out_path, 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(
( (
@@ -579,6 +666,81 @@ 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", "1024", # 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",
"-c:a", "aac", "-b:a", "128k", # Explicitly AAC with a common bitrate
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 _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"""
+290 -77
View File
@@ -71,7 +71,6 @@ from constants import (
) )
from threading import Thread from threading import Thread
from voice_formula_gui import VoiceFormulaDialog from voice_formula_gui import VoiceFormulaDialog
from platformdirs import user_desktop_dir
from voice_profiles import load_profiles from voice_profiles import load_profiles
# Import ctypes for Windows-specific taskbar icon # Import ctypes for Windows-specific taskbar icon
@@ -129,6 +128,13 @@ class InputBox(QLabel):
self.edit_btn.clicked.connect(self.on_edit_clicked) self.edit_btn.clicked.connect(self.on_edit_clicked)
self.edit_btn.hide() self.edit_btn.hide()
# Add Go to folder button
self.go_to_folder_btn = QPushButton("Go to folder", self)
self.go_to_folder_btn.setStyleSheet("QPushButton { padding: 6px 10px; }")
self.go_to_folder_btn.setToolTip("Open the folder that contains the converted file")
self.go_to_folder_btn.clicked.connect(self.on_go_to_folder_clicked)
self.go_to_folder_btn.hide()
def resizeEvent(self, event): def resizeEvent(self, event):
super().resizeEvent(event) super().resizeEvent(event)
margin = 12 margin = 12
@@ -139,6 +145,11 @@ class InputBox(QLabel):
# Position textbox button at top left # Position textbox button at top left
self.textbox_btn.move(margin, margin) self.textbox_btn.move(margin, margin)
self.edit_btn.move(margin, margin) self.edit_btn.move(margin, margin)
# Position go to folder button at bottom right
self.go_to_folder_btn.move(
self.width() - self.go_to_folder_btn.width(),
self.height() - self.go_to_folder_btn.height() - margin,
)
def set_file_info(self, file_path): def set_file_info(self, file_path):
# get icon without resizing using custom provider # get icon without resizing using custom provider
@@ -228,6 +239,7 @@ class InputBox(QLabel):
should_show_edit = True should_show_edit = True
self.edit_btn.setVisible(should_show_edit) self.edit_btn.setVisible(should_show_edit)
self.go_to_folder_btn.show()
def set_error(self, message): def set_error(self, message):
self.setText(message) self.setText(message)
@@ -250,6 +262,7 @@ class InputBox(QLabel):
# Show textbox and hide edit when input is cleared # Show textbox and hide edit when input is cleared
self.textbox_btn.show() self.textbox_btn.show()
self.edit_btn.hide() self.edit_btn.hide()
self.go_to_folder_btn.hide()
def _human_readable_size(self, size, decimal_places=2): def _human_readable_size(self, size, decimal_places=2):
for unit in ["B", "KB", "MB", "GB", "TB"]: for unit in ["B", "KB", "MB", "GB", "TB"]:
@@ -290,8 +303,8 @@ class InputBox(QLabel):
) )
win.selected_book_path = file_path win.selected_book_path = file_path
win.open_book_file( win.open_book_file(
file_path file_path # This will handle the dialog and setting file info
) # This will handle the dialog and setting file info )
event.acceptProposedAction() event.acceptProposedAction()
else: else:
self.set_error("Please drop a .txt, .epub, or .pdf file.") self.set_error("Please drop a .txt, .epub, or .pdf file.")
@@ -320,6 +333,17 @@ class InputBox(QLabel):
# For regular txt files # For regular txt files
win.open_textbox_dialog() win.open_textbox_dialog()
def on_go_to_folder_clicked(self):
win = self.window()
# win.selected_file holds the path to the text that is converted.
file_to_check = win.selected_file
if file_to_check and os.path.exists(file_to_check) and os.path.isfile(file_to_check):
folder_path = os.path.dirname(file_to_check)
QDesktopServices.openUrl(QUrl.fromLocalFile(folder_path))
else:
QMessageBox.warning(win, "Error", "Converted file not found.")
class TextboxDialog(QDialog): class TextboxDialog(QDialog):
def __init__(self, parent=None): def __init__(self, parent=None):
@@ -328,7 +352,7 @@ class TextboxDialog(QDialog):
self.setWindowFlags( self.setWindowFlags(
Qt.Window | Qt.WindowCloseButtonHint | Qt.WindowMaximizeButtonHint Qt.Window | Qt.WindowCloseButtonHint | Qt.WindowMaximizeButtonHint
) )
self.resize(600, 400) self.resize(700, 500)
layout = QVBoxLayout(self) layout = QVBoxLayout(self)
@@ -358,6 +382,11 @@ class TextboxDialog(QDialog):
self.save_as_button.clicked.connect(self.save_as_text) self.save_as_button.clicked.connect(self.save_as_text)
self.save_as_button.setToolTip("Save the current text to a file") self.save_as_button.setToolTip("Save the current text to a file")
self.insert_chapter_btn = QPushButton("Insert Chapter Marker", self)
self.insert_chapter_btn.setToolTip("Insert a chapter marker at the cursor")
self.insert_chapter_btn.clicked.connect(self.insert_chapter_marker)
button_layout.addWidget(self.insert_chapter_btn)
self.cancel_button = QPushButton("Cancel", self) self.cancel_button = QPushButton("Cancel", self)
self.cancel_button.clicked.connect(self.reject) self.cancel_button.clicked.connect(self.reject)
@@ -382,8 +411,14 @@ class TextboxDialog(QDialog):
return self.text_edit.toPlainText() return self.text_edit.toPlainText()
def handle_ok(self): def handle_ok(self):
text = self.text_edit.toPlainText()
# Check if text is empty based on character count
if calculate_text_length(text) == 0:
QMessageBox.warning(self, "Textbox Error", "Text cannot be empty.")
return
# If the text hasn't changed, treat as cancel # If the text hasn't changed, treat as cancel
if self.text_edit.toPlainText() == self.original_text: if text == self.original_text:
self.reject() self.reject()
else: else:
# Check if we need to warn about overwriting a non-temporary file # Check if we need to warn about overwriting a non-temporary file
@@ -444,6 +479,14 @@ class TextboxDialog(QDialog):
except Exception as e: except Exception as e:
QMessageBox.critical(self, "Save Error", f"Could not save file:\n{e}") QMessageBox.critical(self, "Save Error", f"Could not save file:\n{e}")
def insert_chapter_marker(self):
# Insert a fixed chapter marker without prompting
cursor = self.text_edit.textCursor()
cursor.insertText("<<CHAPTER_MARKER:Title>>")
self.text_edit.setTextCursor(cursor)
self.update_char_count()
self.text_edit.setFocus()
class abogen(QWidget): class abogen(QWidget):
def __init__(self): def __init__(self):
@@ -673,10 +716,23 @@ class abogen(QWidget):
self.format_combo.setStyleSheet( self.format_combo.setStyleSheet(
"QComboBox { min-height: 20px; padding: 6px 12px; }" "QComboBox { min-height: 20px; padding: 6px 12px; }"
) )
format_options = ["wav", "flac", "mp3"] # Add items with display labels and underlying keys
self.format_combo.addItems(format_options) for key, label in [
self.format_combo.setCurrentText(self.selected_format) ("wav", "wav"),
self.format_combo.currentTextChanged.connect(self.on_format_changed) ("flac", "flac"),
("mp3", "mp3"),
("opus", "opus (best compression)"),
("m4b", "m4b (with chapters)"),
]:
self.format_combo.addItem(label, key)
# Initialize selection by matching saved key
idx = self.format_combo.findData(self.selected_format)
if idx >= 0:
self.format_combo.setCurrentIndex(idx)
# Map selection back to key on change
self.format_combo.currentIndexChanged.connect(
lambda i: self.on_format_changed(self.format_combo.itemData(i))
)
format_layout.addWidget(self.format_combo) format_layout.addWidget(self.format_combo)
controls_layout.addLayout(format_layout) controls_layout.addLayout(format_layout)
# Save location # Save location
@@ -831,9 +887,7 @@ class abogen(QWidget):
) )
dialog.setWindowModality(Qt.NonModal) dialog.setWindowModality(Qt.NonModal)
dialog.setModal(False) dialog.setModal(False)
dialog.show() dialog.show() # We'll handle the dialog result asynchronously
# We'll handle the dialog result asynchronously
def on_dialog_finished(result): def on_dialog_finished(result):
if result != QDialog.Accepted: if result != QDialog.Accepted:
return False return False
@@ -848,6 +902,7 @@ class abogen(QWidget):
self.selected_chapters = all_checked_hrefs self.selected_chapters = all_checked_hrefs
self.save_chapters_separately = dialog.get_save_chapters_separately() self.save_chapters_separately = dialog.get_save_chapters_separately()
self.merge_chapters_at_end = dialog.get_merge_chapters_at_end() self.merge_chapters_at_end = dialog.get_merge_chapters_at_end()
self.save_as_project = dialog.get_save_as_project()
# Store if the PDF has bookmarks for button text display # Store if the PDF has bookmarks for button text display
if book_path.lower().endswith(".pdf"): if book_path.lower().endswith(".pdf"):
@@ -856,8 +911,50 @@ class abogen(QWidget):
# Use "abogen" prefix for temporary files # Use "abogen" prefix for temporary files
# Extract base name without extension # Extract base name without extension
base_name = os.path.splitext(os.path.basename(book_path))[0] base_name = os.path.splitext(os.path.basename(book_path))[0]
temp_dir = os.path.join(tempfile.gettempdir(), PROGRAM_NAME)
os.makedirs(temp_dir, exist_ok=True) if self.save_as_project:
# Get project directory from user
project_dir = QFileDialog.getExistingDirectory(
self, "Select Project Folder", "", QFileDialog.ShowDirsOnly
)
if not project_dir:
# User cancelled, fallback to temp
self.save_as_project = False
temp_dir = os.path.join(tempfile.gettempdir(), PROGRAM_NAME)
else:
# Create project folder structure
project_name = f"{base_name}_project"
project_dir = os.path.join(project_dir, project_name)
temp_dir = os.path.join(project_dir, "text")
os.makedirs(temp_dir, exist_ok=True)
# Save metadata if available
meta_dir = os.path.join(project_dir, "metadata")
os.makedirs(meta_dir, exist_ok=True) # Save book metadata if available
if hasattr(dialog, "book_metadata"):
meta_path = os.path.join(meta_dir, "book_info.txt")
with open(meta_path, "w", encoding="utf-8") as f:
# Clean HTML tags from metadata
title = re.sub(r'<[^>]+>', '', str(dialog.book_metadata.get('title', 'Unknown')))
publisher = re.sub(r'<[^>]+>', '', str(dialog.book_metadata.get('publisher', 'Unknown')))
authors = [re.sub(r'<[^>]+>', '', str(author)) for author in dialog.book_metadata.get('authors', ['Unknown'])]
f.write(f"Title: {title}\n")
f.write(f"Authors: {', '.join(authors)}\n")
f.write(f"Publisher: {publisher}\n")
if dialog.book_metadata.get('description'):
description = re.sub(r'<[^>]+>', '', str(dialog.book_metadata.get('description')))
f.write(f"\nDescription:\n{description}\n")
# Save cover image if available
if dialog.book_metadata.get("cover_image"):
cover_path = os.path.join(meta_dir, "cover.png")
with open(cover_path, "wb") as f:
f.write(dialog.book_metadata["cover_image"])
else:
temp_dir = os.path.join(tempfile.gettempdir(), PROGRAM_NAME)
os.makedirs(temp_dir, exist_ok=True)
fd, tmp = tempfile.mkstemp( fd, tmp = tempfile.mkstemp(
prefix=f"abogen_{base_name}_", suffix=".txt", dir=temp_dir prefix=f"abogen_{base_name}_", suffix=".txt", dir=temp_dir
) )
@@ -1097,12 +1194,20 @@ class abogen(QWidget):
QApplication.processEvents() QApplication.processEvents()
def update_progress(self, value, etr_str): # Add etr_str parameter def update_progress(self, value, etr_str): # Add etr_str parameter
# Ensure progress doesn't exceed 99%
if value >= 100:
value = 99
self.progress_bar.setValue(value) self.progress_bar.setValue(value)
self.progress_bar.setFormat("%p%") # Keep format as percentage only self.progress_bar.setFormat("%p%") # Keep format as percentage only
self.etr_label.setText( self.etr_label.setText(
f"Estimated time remaining: {etr_str}" f"Estimated time remaining: {etr_str}"
) # Update ETR label ) # Update ETR label
self.etr_label.show() # Show only when estimate is ready self.etr_label.show() # Show only when estimate is ready
# Disable cancel button if progress is >= 98%
if value >= 98:
self.btn_cancel.setEnabled(False)
self.progress_bar.repaint() self.progress_bar.repaint()
QApplication.processEvents() QApplication.processEvents()
@@ -1432,8 +1537,22 @@ class abogen(QWidget):
self.voice_combo.setEnabled(False) self.voice_combo.setEnabled(False)
self.btn_voice_formula_mixer.setEnabled(False) # Disable mixer button self.btn_voice_formula_mixer.setEnabled(False) # Disable mixer button
self.btn_start.setEnabled(False) # Disable start button during preview self.btn_start.setEnabled(False) # Disable start button during preview
# start loading animation
self.loading_movie.start() # Start loading animation - ensure signal connection is always active
if hasattr(self, "loading_movie"):
# Disconnect previous connections to avoid multiple connections
try:
self.loading_movie.frameChanged.disconnect()
except TypeError:
pass # Ignore error if not connected
# Reconnect the signal
self.loading_movie.frameChanged.connect(
lambda: self.btn_preview.setIcon(
QIcon(self.loading_movie.currentPixmap())
)
)
self.loading_movie.start()
def pipeline_loaded_callback(np_module, kpipeline_class, error): def pipeline_loaded_callback(np_module, kpipeline_class, error):
self._on_pipeline_loaded_for_preview(np_module, kpipeline_class, error) self._on_pipeline_loaded_for_preview(np_module, kpipeline_class, error)
@@ -1462,14 +1581,14 @@ class abogen(QWidget):
# Build voice formula string # Build voice formula string
components = [f"{name}*{weight}" for name, weight in self.mixed_voice_state] components = [f"{name}*{weight}" for name, weight in self.mixed_voice_state]
voice = " + ".join(filter(None, components)) voice = " + ".join(filter(None, components))
# determine language: use profile setting if available, else first voice code # determine language: use profile setting, else explicit mixer selection, else fallback to first voice code
if self.selected_profile_name: if self.selected_profile_name:
from voice_profiles import load_profiles from voice_profiles import load_profiles
entry = load_profiles().get(self.selected_profile_name, {}) entry = load_profiles().get(self.selected_profile_name, {})
lang = entry.get("language") lang = entry.get("language")
else: else:
lang = None lang = self.selected_lang
if not lang and self.mixed_voice_state: if not lang and self.mixed_voice_state:
lang = ( lang = (
self.mixed_voice_state[0][0][0] self.mixed_voice_state[0][0][0]
@@ -1480,8 +1599,11 @@ class abogen(QWidget):
lang = self.selected_voice[0] lang = self.selected_voice[0]
voice = self.selected_voice voice = self.selected_voice
# use same gpu/cpu logic as in conversion
gpu_msg, gpu_ok = get_gpu_acceleration(self.use_gpu)
self.preview_thread = VoicePreviewThread( self.preview_thread = VoicePreviewThread(
np_module, kpipeline_class, lang, voice, speed, self.use_gpu np_module, kpipeline_class, lang, voice, speed, gpu_ok
) )
self.preview_thread.finished.connect(self._play_preview_audio) self.preview_thread.finished.connect(self._play_preview_audio)
self.preview_thread.error.connect(self._preview_error) self.preview_thread.error.connect(self._preview_error)
@@ -1545,6 +1667,11 @@ class abogen(QWidget):
def _preview_cleanup(self): def _preview_cleanup(self):
self.preview_playing = False self.preview_playing = False
self.loading_movie.stop()
try:
self.loading_movie.frameChanged.disconnect()
except Exception:
pass # Ignore error if not connected
self.btn_preview.setIcon(self.play_icon) self.btn_preview.setIcon(self.play_icon)
self.btn_preview.setToolTip("Preview selected voice") self.btn_preview.setToolTip("Preview selected voice")
self.btn_preview.setEnabled(True) self.btn_preview.setEnabled(True)
@@ -1679,9 +1806,11 @@ class abogen(QWidget):
# Add seperator # Add seperator
menu.addSeparator() menu.addSeparator()
# Add shortcut to desktop (Windows only) # Add shortcut to desktop (Windows or Linux)
if platform.system() == "Windows": if platform.system() == "Windows" or platform.system() == "Linux":
add_shortcut_action = QAction("Create desktop shortcut", self) # Use extended label on Linux
label = "Create desktop shortcut and install" if platform.system() == "Linux" else "Create desktop shortcut"
add_shortcut_action = QAction(label, self)
add_shortcut_action.triggered.connect(self.add_shortcut_to_desktop) add_shortcut_action.triggered.connect(self.add_shortcut_to_desktop)
menu.addAction(add_shortcut_action) menu.addAction(add_shortcut_action)
@@ -1754,62 +1883,133 @@ class abogen(QWidget):
def add_shortcut_to_desktop(self): def add_shortcut_to_desktop(self):
"""Create a desktop shortcut to this program using PowerShell.""" """Create a desktop shortcut to this program using PowerShell."""
import sys, subprocess import sys
from PyQt5.QtWidgets import QMessageBox from platformdirs import user_desktop_dir
from utils import create_process
try: try:
# where to put the .lnk if platform.system() == "Windows":
desktop = user_desktop_dir() # where to put the .lnk
shortcut_path = os.path.join(desktop, "abogen.lnk") desktop = user_desktop_dir()
shortcut_path = os.path.join(desktop, "abogen.lnk")
# target exe # target exe
python_dir = os.path.dirname(sys.executable) python_dir = os.path.dirname(sys.executable)
target = os.path.join(python_dir, "Scripts", "abogen.exe") target = os.path.join(python_dir, "Scripts", "abogen.exe")
if not os.path.exists(target): if not os.path.exists(target):
QMessageBox.critical( QMessageBox.critical(
self, "Shortcut Error", f"Could not find abogen.exe at:\n{target}" self, "Shortcut Error", f"Could not find abogen.exe at:\n{target}"
)
return
# icon (fallback to exe if missing)
icon = get_resource_path("abogen.assets", "icon.ico")
if not icon or not os.path.exists(icon):
icon = target # Create a more direct PowerShell command
shortcut_ps = shortcut_path.replace("'", "''").replace("\\", "\\\\")
target_ps = target.replace("'", "''").replace("\\", "\\\\")
workdir_ps = os.path.dirname(target).replace("'", "''").replace("\\", "\\\\")
icon_ps = icon.replace("'", "''").replace("\\", "\\\\")
# Create PowerShell script as a single line with no line breaks (more reliable)
ps_cmd = f"$s=New-Object -ComObject WScript.Shell; $lnk=$s.CreateShortcut('{shortcut_ps}'); $lnk.TargetPath='{target_ps}'; $lnk.WorkingDirectory='{workdir_ps}'; $lnk.IconLocation='{icon_ps}'; $lnk.Save()"
# Run PowerShell with the command directly
proc = create_process("powershell -NoProfile -ExecutionPolicy Bypass -Command \"" + ps_cmd + "\"")
proc.wait()
if proc.returncode == 0:
QMessageBox.information(
self,
"Shortcut Created",
f"Shortcut created on desktop:\n{shortcut_path}",
)
else:
QMessageBox.critical(
self,
"Shortcut Error",
f"PowerShell failed with exit code: {proc.returncode}"
)
elif platform.system() == "Linux":
desktop = user_desktop_dir()
if not desktop or not os.path.isdir(desktop):
QMessageBox.critical(self, "Shortcut Error", "Could not determine desktop directory.")
return
shortcut_path = os.path.join(desktop, "abogen.desktop")
import shutil
found = shutil.which("abogen")
if found:
target = found
else:
local_bin = os.path.expanduser("~/.local/bin/abogen")
if os.path.exists(local_bin):
target = local_bin
else:
python_dir = os.path.dirname(sys.executable)
target = os.path.join(python_dir, "bin", "abogen")
if not os.path.exists(target):
target_fallback = os.path.join(python_dir, "abogen")
if os.path.exists(target_fallback):
target = target_fallback
else:
QMessageBox.critical(self, "Shortcut Error", "Could not find abogen executable in PATH or common installation directories.")
return
icon_path = get_resource_path("abogen.assets", "icon.png")
desktop_entry_content = f"""[Desktop Entry]
Version={VERSION}
Name={PROGRAM_NAME}
Comment={PROGRAM_DESCRIPTION}
Exec={target}
Icon={icon_path}
Terminal=false
Type=Application
Categories=AudioVideo;Audio;Utility;
"""
with open(shortcut_path, "w", encoding="utf-8") as f:
f.write(desktop_entry_content)
os.chmod(shortcut_path, 0o755)
QMessageBox.information(
self,
"Shortcut Created",
f"Shortcut created on desktop:\n{shortcut_path}",
) )
return
# icon (fallback to exe if missing) # Offer installation for current user under ~/.local/share/applications
icon = get_resource_path("abogen.assets", "icon.ico") reply = QMessageBox.question(
if not icon or not os.path.exists(icon): self,
icon = target "Install Application Entry",
"Install application entry for current user?",
QMessageBox.Yes | QMessageBox.No,
)
if reply == QMessageBox.Yes:
import shutil
user_app_dir = os.path.expanduser("~/.local/share/applications")
os.makedirs(user_app_dir, exist_ok=True)
user_entry = os.path.join(user_app_dir, "abogen.desktop")
try:
shutil.copyfile(shortcut_path, user_entry)
os.chmod(user_entry, 0o644)
QMessageBox.information(
self,
"Installation Complete",
f"Desktop entry installed to {user_entry}",
)
except Exception as e:
QMessageBox.warning(
self,
"Install Error",
f"Could not install entry:\n{e}",
)
else:
QMessageBox.information(
self, "Unsupported OS", "Desktop shortcut creation is not supported on this operating system."
)
# PowerShell command to make the shortcut
ps_cmd = r"""
$s = New-Object -ComObject WScript.Shell
$lnk = $s.CreateShortcut('{lnk}')
$lnk.TargetPath = '{tgt}'
$lnk.WorkingDirectory = '{cwd}'
$lnk.IconLocation = '{ico}'
$lnk.Save()
""".format(
lnk=shortcut_path.replace("'", "''"),
tgt=target.replace("'", "''"),
cwd=os.path.dirname(target).replace("'", "''"),
ico=icon.replace("'", "''"),
).strip()
subprocess.run(
["powershell", "-NoProfile", "-Command", ps_cmd],
check=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
)
QMessageBox.information(
self,
"Shortcut Created",
f"Shortcut created on desktop:\n{shortcut_path}",
)
except subprocess.CalledProcessError as e:
QMessageBox.critical(
self,
"Shortcut Error",
f"PowerShell failed:\n{e.stderr.decode(errors='ignore')}",
)
except Exception as e: except Exception as e:
QMessageBox.critical( QMessageBox.critical(
self, "Shortcut Error", f"Could not create shortcut:\n{e}" self, "Shortcut Error", f"Could not create shortcut:\n{e}"
@@ -1820,25 +2020,37 @@ class abogen(QWidget):
save_config(self.config) save_config(self.config)
def show_voice_formula_dialog(self): def show_voice_formula_dialog(self):
# If a profile is selected in combo, open mixer with that profile from voice_profiles import load_profiles
profiles = load_profiles()
initial_state = None initial_state = None
selected_profile = self.selected_profile_name selected_profile = self.selected_profile_name
if selected_profile: if selected_profile:
entry = load_profiles().get(selected_profile, {}) entry = profiles.get(selected_profile, {})
# support new dict format
if isinstance(entry, dict): if isinstance(entry, dict):
initial_state = entry.get("voices", []) initial_state = entry.get("voices", [])
else: else:
initial_state = entry initial_state = entry
elif self.mixed_voice_state is not None: elif self.mixed_voice_state is not None:
initial_state = self.mixed_voice_state initial_state = self.mixed_voice_state
elif self.selected_voice:
# If a single voice is selected, default to first profile if available
if profiles:
first_profile = next(iter(profiles))
entry = profiles[first_profile]
selected_profile = first_profile
if isinstance(entry, dict):
initial_state = entry.get("voices", [])
else:
initial_state = entry
else:
initial_state = []
else: else:
initial_state = [(self.selected_voice, 1.0)] if self.selected_voice else [] initial_state = []
dialog = VoiceFormulaDialog( dialog = VoiceFormulaDialog(
self, initial_state=initial_state, selected_profile=selected_profile self, initial_state=initial_state, selected_profile=selected_profile
) )
if dialog.exec_() == QDialog.Accepted: if dialog.exec_() == QDialog.Accepted:
# After OK, select the profile in combo and update config
if dialog.current_profile: if dialog.current_profile:
self.selected_profile_name = dialog.current_profile self.selected_profile_name = dialog.current_profile
self.config["selected_profile_name"] = dialog.current_profile self.config["selected_profile_name"] = dialog.current_profile
@@ -2080,3 +2292,4 @@ class abogen(QWidget):
"Setting Saved", "Setting Saved",
f"Maximum words per subtitle set to {value}.", f"Maximum words per subtitle set to {value}.",
) )
Binary file not shown.
+105 -15
View File
@@ -1,4 +1,5 @@
import os import os
import sys
import json import json
import warnings import warnings
import platform import platform
@@ -66,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 nonWindows, 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")
@@ -92,6 +103,89 @@ def clean_text(text, *args, **kwargs):
return text return text
default_encoding = sys.getfilesystemencoding()
def create_process(cmd, stdin=None, text=True):
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 proc.stdout:
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:
@@ -109,8 +203,10 @@ 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_no_markers = re.sub(r"<<CHAPTER_MARKER:.*?>>", "", text)
# Ignore newlines
cleaned_text = text_no_markers.replace("\n", "")
# Calculate character count # Calculate character count
char_count = len(cleaned_text) char_count = len(cleaned_text)
return char_count return char_count
@@ -136,21 +232,15 @@ def prevent_sleep_start():
0x80000000 | 0x00000001 | 0x00000040 0x80000000 | 0x00000001 | 0x00000040
) # ES_CONTINUOUS | ES_SYSTEM_REQUIRED | ES_AWAYMODE_REQUIRED ) # 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: try:
_sleep_procs["Linux"] = subprocess.Popen( _sleep_procs["Linux"] = create_process(
[ "systemd-inhibit --what=sleep --why=TextToAudiobook conversion sleep 999999"
"systemd-inhibit",
"--what=sleep",
"--why=TextToAudiobook conversion",
"sleep",
"999999",
]
) )
except Exception: except Exception:
try: try:
subprocess.Popen(["xdg-screensaver", "reset"]) create_process("xdg-screensaver reset")
except Exception: except Exception:
pass pass
+132 -35
View File
@@ -47,7 +47,7 @@ VOICE_MIXER_WIDTH = 100
SLIDER_WIDTH = 32 SLIDER_WIDTH = 32
MIN_WINDOW_WIDTH = 600 MIN_WINDOW_WIDTH = 600
MIN_WINDOW_HEIGHT = 400 MIN_WINDOW_HEIGHT = 400
INITIAL_WINDOW_WIDTH = 1000 INITIAL_WINDOW_WIDTH = 1200
INITIAL_WINDOW_HEIGHT = 500 INITIAL_WINDOW_HEIGHT = 500
# Language options for the language selector loaded from constants # Language options for the language selector loaded from constants
@@ -413,6 +413,11 @@ class VoiceFormulaDialog(QDialog):
self.language_combo.setCurrentIndex(idx) self.language_combo.setCurrentIndex(idx)
self.language_combo.currentIndexChanged.connect(self.mark_profile_modified) self.language_combo.currentIndexChanged.connect(self.mark_profile_modified)
header_row.addWidget(self.language_combo) header_row.addWidget(self.language_combo)
# Preview current voice mix using main window's preview
self.btn_preview_mix = QPushButton("Preview", self)
self.btn_preview_mix.setToolTip("Preview current voice mix")
self.btn_preview_mix.clicked.connect(self.preview_current_mix)
header_row.addWidget(self.btn_preview_mix)
mixer_layout.addLayout(header_row) mixer_layout.addLayout(header_row)
# Error message # Error message
@@ -475,7 +480,6 @@ class VoiceFormulaDialog(QDialog):
self.add_voices(initial_state or []) self.add_voices(initial_state or [])
self.update_weighted_sums() self.update_weighted_sums()
self.update_subtitle_combo_enabled()
# assemble splitter # assemble splitter
splitter.addWidget(profile_widget) splitter.addWidget(profile_widget)
@@ -496,8 +500,6 @@ class VoiceFormulaDialog(QDialog):
for vm in self.voice_mixers: for vm in self.voice_mixers:
vm.spin_box.valueChanged.connect(self.mark_profile_modified) vm.spin_box.valueChanged.connect(self.mark_profile_modified)
vm.checkbox.stateChanged.connect(lambda *_: self.mark_profile_modified()) vm.checkbox.stateChanged.connect(lambda *_: self.mark_profile_modified())
vm.spin_box.valueChanged.connect(self.update_subtitle_combo_enabled)
vm.checkbox.stateChanged.connect(self.update_subtitle_combo_enabled)
def keyPressEvent(self, event): def keyPressEvent(self, event):
# Bind Delete key to delete_profile when a profile is selected # Bind Delete key to delete_profile when a profile is selected
@@ -682,8 +684,6 @@ class VoiceFormulaDialog(QDialog):
voice_mixer.checkbox.stateChanged.connect( voice_mixer.checkbox.stateChanged.connect(
lambda *_: self.mark_profile_modified() lambda *_: self.mark_profile_modified()
) )
voice_mixer.spin_box.valueChanged.connect(self.update_subtitle_combo_enabled)
voice_mixer.checkbox.stateChanged.connect(self.update_subtitle_combo_enabled)
return voice_mixer return voice_mixer
def handle_voice_checkbox(self, voice_mixer, state): def handle_voice_checkbox(self, voice_mixer, state):
@@ -713,6 +713,9 @@ class VoiceFormulaDialog(QDialog):
] ]
total = sum(w for _, w in selected) total = sum(w for _, w in selected)
# disable Preview if no voices selected, but don't enable while loading
if not getattr(self, "_loading", False):
self.btn_preview_mix.setEnabled(total > 0)
if total > 0: if total > 0:
self.error_label.hide() self.error_label.hide()
@@ -743,20 +746,6 @@ class VoiceFormulaDialog(QDialog):
self.error_label.show() self.error_label.show()
self.weighted_sums_container.hide() self.weighted_sums_container.hide()
def update_subtitle_combo_enabled(self):
# Only enable subtitle_combo if at least one selected voice is from supported languages
selected_langs = set()
for vm in self.voice_mixers:
if vm.checkbox.isChecked() and vm.spin_box.value() > 0:
lang_code = vm.voice_name[0]
selected_langs.add(lang_code)
enable = any(
lang in SUPPORTED_LANGUAGES_FOR_SUBTITLE_GENERATION
for lang in selected_langs
)
if self.subtitle_combo:
self.subtitle_combo.setEnabled(enable)
def disable_voice_by_name(self, voice_name): def disable_voice_by_name(self, voice_name):
for mixer in self.voice_mixers: for mixer in self.voice_mixers:
if mixer.voice_name == voice_name: if mixer.voice_name == voice_name:
@@ -819,7 +808,6 @@ class VoiceFormulaDialog(QDialog):
# sync enabled state # sync enabled state
vm.toggle_inputs() vm.toggle_inputs()
self.update_weighted_sums() self.update_weighted_sums()
self.update_subtitle_combo_enabled()
def save_profile_by_name(self, name): def save_profile_by_name(self, name):
profiles = load_profiles() profiles = load_profiles()
@@ -846,10 +834,10 @@ class VoiceFormulaDialog(QDialog):
self.update_weighted_sums() self.update_weighted_sums()
def _handle_zero_weight_profiles(self): def _handle_zero_weight_profiles(self):
if self.profile_list.count() <= 1: profiles = load_profiles()
if len(profiles) < 1:
return False return False
zero = [] zero = []
profiles = load_profiles()
for i in range(self.profile_list.count()): for i in range(self.profile_list.count()):
item = self.profile_list.item(i) item = self.profile_list.item(i)
name = item.text().lstrip("*") name = item.text().lstrip("*")
@@ -868,7 +856,6 @@ class VoiceFormulaDialog(QDialog):
if not zero: if not zero:
return False return False
msg = f"{len(zero)} invalid profile(s) with no voices selected or their total weights are 0. They will be ignored and deleted. Do you want to delete?" msg = f"{len(zero)} invalid profile(s) with no voices selected or their total weights are 0. They will be ignored and deleted. Do you want to delete?"
# Use Delete instead of Ignore
reply = QMessageBox.question( reply = QMessageBox.question(
self, self,
"Invalid Profiles", "Invalid Profiles",
@@ -963,8 +950,22 @@ class VoiceFormulaDialog(QDialog):
self.update_profile_list_colors() self.update_profile_list_colors()
def new_profile(self): def new_profile(self):
name, ok = QInputDialog.getText(self, "New Profile", "Enter profile name:") import re
if ok and name:
while True:
name, ok = QInputDialog.getText(self, "New Profile", "Enter profile name:")
if not ok or not name:
break
name = name.strip() # Remove leading/trailing spaces
if not name:
continue
if not re.match(r"^[\w\- ]+$", name):
QMessageBox.warning(
self,
"Invalid Name",
"Profile name can only contain letters, numbers, spaces, underscores, and hyphens.",
)
continue
profiles = load_profiles() profiles = load_profiles()
# Remove 'New profile' placeholder if not persisted in JSON # Remove 'New profile' placeholder if not persisted in JSON
if ( if (
@@ -977,7 +978,7 @@ class VoiceFormulaDialog(QDialog):
self._profile_dirty.pop("New profile", None) self._profile_dirty.pop("New profile", None)
if name in profiles: if name in profiles:
QMessageBox.warning(self, "Duplicate Name", "Profile already exists.") QMessageBox.warning(self, "Duplicate Name", "Profile already exists.")
return continue
profiles[name] = { profiles[name] = {
"voices": [], "voices": [],
"language": self.language_combo.currentData(), "language": self.language_combo.currentData(),
@@ -996,6 +997,7 @@ class VoiceFormulaDialog(QDialog):
parent = self.parent() parent = self.parent()
if hasattr(parent, "populate_profiles_in_voice_combo"): if hasattr(parent, "populate_profiles_in_voice_combo"):
parent.populate_profiles_in_voice_combo() parent.populate_profiles_in_voice_combo()
break
self.update_profile_save_buttons() self.update_profile_save_buttons()
self.update_profile_list_colors() self.update_profile_list_colors()
self.update_weighted_sums() self.update_weighted_sums()
@@ -1182,21 +1184,71 @@ class VoiceFormulaDialog(QDialog):
) )
def rename_profile(self, item): def rename_profile(self, item):
name = item.text().lstrip("*")
# block if profile has unsaved changes and it's not a virtual New profile
if self._profile_dirty.get(name, False) and not (
self._virtual_new_profile and name == "New profile"
):
QMessageBox.warning(
self, "Unsaved Changes", "Please save the profile before renaming."
)
return
old = item.text().lstrip("*") old = item.text().lstrip("*")
new, ok = QInputDialog.getText( import re
self, "Rename Profile", f"Profile name:", text=old
) while True:
if ok and new and new != old: new, ok = QInputDialog.getText(
self, "Rename Profile", f"Profile name:", text=old
)
if not ok or not new or new == old:
break
new = new.strip() # Remove leading/trailing spaces
if not new:
continue
if not re.match(r"^[\w\- ]+$", new):
QMessageBox.warning(
self,
"Invalid Name",
"Profile name can only contain letters, numbers, spaces, underscores, and hyphens.",
)
continue
profiles = load_profiles() profiles = load_profiles()
if new in profiles: if new in profiles:
QMessageBox.warning(self, "Duplicate Name", "Profile already exists.") QMessageBox.warning(self, "Duplicate Name", "Profile already exists.")
return continue
profiles[new] = profiles.pop(old)
save_profiles(profiles) # Special case for renaming the virtual "New profile"
item.setText(new) if self._virtual_new_profile and name == "New profile":
# Create the profile with the new name
profiles[new] = {
"voices": self.get_selected_voices(),
"language": self.language_combo.currentData(),
}
save_profiles(profiles)
# Update tracking properties
self._virtual_new_profile = False
self._profile_dirty.pop("New profile", None)
self._profile_dirty[new] = False
# Update the current profile name
self.current_profile = new
item.setText(new)
else:
# Standard renaming for regular profiles
profiles[new] = profiles.pop(old)
save_profiles(profiles)
item.setText(new)
# Update the current profile name if it was renamed
if self.current_profile == old:
self.current_profile = new
parent = self.parent() parent = self.parent()
if hasattr(parent, "populate_profiles_in_voice_combo"): if hasattr(parent, "populate_profiles_in_voice_combo"):
parent.populate_profiles_in_voice_combo() parent.populate_profiles_in_voice_combo()
break
self.update_profile_save_buttons() self.update_profile_save_buttons()
self.update_profile_list_colors() self.update_profile_list_colors()
@@ -1227,6 +1279,13 @@ class VoiceFormulaDialog(QDialog):
self.update_profile_list_colors() self.update_profile_list_colors()
def duplicate_profile(self, item): def duplicate_profile(self, item):
name = item.text().lstrip("*")
# block duplicating if profile has unsaved changes
if self._profile_dirty.get(name, False):
QMessageBox.warning(
self, "Unsaved Changes", "Please save the profile before duplicating."
)
return
src = item.text().lstrip("*") src = item.text().lstrip("*")
profiles = load_profiles() profiles = load_profiles()
base = f"{src}_duplicate" base = f"{src}_duplicate"
@@ -1287,3 +1346,41 @@ class VoiceFormulaDialog(QDialog):
else: else:
item.setBackground(QColor("white")) item.setBackground(QColor("white"))
self.update_profile_save_buttons() self.update_profile_save_buttons()
def preview_current_mix(self):
# Disable preview until playback completes
self.btn_preview_mix.setEnabled(False)
self.btn_preview_mix.setText("Loading...")
self._loading = True
parent = self.parent()
if parent and hasattr(parent, "preview_voice"):
# Apply mixed voices and selected language
parent.mixed_voice_state = self.get_selected_voices()
parent.selected_profile_name = None
lang = self.language_combo.currentData()
parent.selected_lang = lang
parent.subtitle_combo.setEnabled(
lang in SUPPORTED_LANGUAGES_FOR_SUBTITLE_GENERATION
)
# Reset start flag and trigger preview
self._started = False
parent.preview_voice()
# Poll preview_playing: wait for start then end
self._preview_poll_timer = QTimer(self)
self._preview_poll_timer.timeout.connect(self._check_preview_done)
self._preview_poll_timer.start(200)
def _check_preview_done(self):
parent = self.parent()
if parent and hasattr(parent, "preview_playing"):
# Mark when playback starts
if parent.preview_playing:
self._started = True
# Update button text to "Playing..." when playback starts
self.btn_preview_mix.setText("Playing...")
# Once started and then stopped, re-enable
elif getattr(self, "_started", False):
self.btn_preview_mix.setEnabled(True)
self.btn_preview_mix.setText("Preview")
self._loading = False
self._preview_poll_timer.stop()
+1 -1
View File
@@ -1,6 +1,6 @@
import os import os
import json import json
from utils import get_user_config_path, get_resource_path from utils import get_user_config_path
def _get_profiles_path(): def _get_profiles_path():
Binary file not shown.

After

Width:  |  Height:  |  Size: 11 KiB

+2 -1
View File
@@ -22,7 +22,8 @@ dependencies = [
"soundfile>=0.13.1", "soundfile>=0.13.1",
"pygame>=2.6.1", "pygame>=2.6.1",
"charset_normalizer>=3.4.1", "charset_normalizer>=3.4.1",
"chardet>=5.2.0" "chardet>=5.2.0",
"static_ffmpeg>=2.13"
] ]
classifiers = [ classifiers = [