mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 13:40:27 +02:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
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 |
@@ -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 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.
|
||||
@@ -6,3 +39,18 @@
|
||||
- Switched to platformdirs for determining the correct desktop path, instead of using old methods.
|
||||
- Fixed preview voices was not using GPU acceleration, which was causing performance issues.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# v1.0.2
|
||||
- Enhanced EPUB handling by treating all items in chapter list (including anchors) as chapters, improving navigation and organization for poorly structured books, mentioned by @Darthagnon in #4
|
||||
- Fixed the issue with some chapters in EPUB files had missing content.
|
||||
- Fixed the issue with some EPUB files only having one chapter caused the program to ignore the entire book.
|
||||
- Fixed "utf-8' codec can't decode byte" error, mentioned by @nigelp in #3
|
||||
- Added "Replace single newlines with spaces" option in the menu. This can be useful for texts that have imaginary line breaks.
|
||||
- Improvements in code and documentation.
|
||||
|
||||
# v1.0.1
|
||||
- Added abogen-cli command for better troubleshooting and error handling.
|
||||
- Switched from setuptools to hatchling for packaging.
|
||||
- Added classifiers to the package metadata.
|
||||
- Fixed "No module named 'docopt'" and "setuptools.build_meta" import errors while using .bat installer in Windows, mentioned by @nigelp in #2
|
||||
- Improvements in code and documentation.
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
[](https://github.com/denizsafak/abogen/actions)
|
||||
[](https://github.com/denizsafak/abogen/releases/latest)
|
||||

|
||||
[](https://pypi.org/project/abogen/)
|
||||
[](https://github.com/denizsafak/abogen/releases/latest)
|
||||
[](https://github.com/psf/black)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
@@ -14,10 +14,9 @@ Abogen is a powerful text-to-speech conversion tool that makes it easy to turn e
|
||||
## Demo
|
||||
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).
|
||||
|
||||
## `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
|
||||
Go to [espeak-ng latest release](https://github.com/espeak-ng/espeak-ng/releases/latest) download and run the *.msi file.
|
||||
```bash
|
||||
@@ -56,13 +55,20 @@ sudo dnf install espeak-ng
|
||||
# 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.
|
||||
|
||||
Then simply run by typing:
|
||||
## `How to run?`
|
||||
If you installed using pip, you can simply run the following command to start Abogen:
|
||||
|
||||
```bash
|
||||
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?`
|
||||
1) Drag and drop any ePub, PDF, or text file (or use the built-in text editor)
|
||||
@@ -85,12 +91,14 @@ Here’s 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.
|
||||
- **Voice mixer**: Create custom voices by mixing different voice models with a profile system.
|
||||
- **Generate subtitles**: `Disabled`, `Sentence`, `Sentence + Comma`, `1 word`, `2 words`, `3 words`, etc. (Represents the number of words in each subtitle entry)
|
||||
- **Output formats**: `.WAV`, `.FLAC`, 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`
|
||||
- **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**:
|
||||
- **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.
|
||||
- **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.
|
||||
@@ -103,6 +111,30 @@ Here’s 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))
|
||||
|
||||
## `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`
|
||||
```
|
||||
# 🇺🇸 'a' => American English, 🇬🇧 'b' => British English
|
||||
@@ -131,6 +163,42 @@ 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`
|
||||
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)**
|
||||
@@ -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
|
||||
|
||||
## `Roadmap`
|
||||
- [ ] Improve PDF support for better text extraction.
|
||||
- [ ] Add chapter metadata for .m4a files using ffmpeg-bin.
|
||||
- [ ] Add OCR scan feature for PDF files using docling.
|
||||
- [x] Add chapter metadata for .m4a files. (Issue [#9](https://github.com/denizsafak/abogen/issues/9), PR [#10](https://github.com/denizsafak/abogen/pull/10))
|
||||
- [ ] Add support for different languages in GUI.
|
||||
- [x] Add voice formula feature that enables mixing different voice models. #1
|
||||
- [ ] Add support for kokoro-onnx.
|
||||
- [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 (If it's necessary).
|
||||
- [ ] Add dark mode.
|
||||
|
||||
## `Troubleshooting`
|
||||
@@ -160,6 +228,7 @@ If you'd like to modify the code and contribute to development, you can [downloa
|
||||
```bash
|
||||
# Go to the directory where you extracted the repository and run:
|
||||
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)
|
||||
abogen # Opens the GUI
|
||||
```
|
||||
|
||||
@@ -219,6 +219,15 @@ if errorlevel 1 (
|
||||
exit /b
|
||||
)
|
||||
|
||||
:: Install progress's fixed version
|
||||
echo Installing fixed version of progress...
|
||||
%PYTHON_CONSOLE_PATH% -m pip install --force-reinstall https://github.com/denizsafak/abogen/raw/refs/heads/main/abogen/resources/progress-1.6-py3-none-any.whl --no-warn-script-location
|
||||
if errorlevel 1 (
|
||||
echo Failed to install fixed version of progress.
|
||||
pause
|
||||
exit /b
|
||||
)
|
||||
|
||||
:: Install setup requirements
|
||||
echo Installing setup requirements...
|
||||
%PYTHON_CONSOLE_PATH% -m pip install --upgrade setuptools wheel sphinx hatchling --no-warn-script-location
|
||||
|
||||
@@ -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 +1 @@
|
||||
1.0.3
|
||||
1.0.7
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 18 KiB |
+70
-58
@@ -37,6 +37,7 @@ class HandlerDialog(QDialog):
|
||||
# Class variables to remember checkbox states between dialog instances
|
||||
_save_chapters_separately = False
|
||||
_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):
|
||||
super().__init__(parent)
|
||||
@@ -64,6 +65,7 @@ class HandlerDialog(QDialog):
|
||||
# Initialize save chapters flags from class variables
|
||||
self.save_chapters_separately = HandlerDialog._save_chapters_separately
|
||||
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
|
||||
try:
|
||||
@@ -934,6 +936,7 @@ class HandlerDialog(QDialog):
|
||||
def _build_pdf_tree(self):
|
||||
outline = self.pdf_doc.get_toc()
|
||||
self.has_pdf_bookmarks = bool(outline)
|
||||
self.bookmark_items_map = {}
|
||||
|
||||
if not outline:
|
||||
self._build_pdf_pages_tree()
|
||||
@@ -976,20 +979,23 @@ class HandlerDialog(QDialog):
|
||||
else self.pdf_doc.resolve_link(page)[0]
|
||||
)
|
||||
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.setData(0, Qt.UserRole, page_id)
|
||||
bookmark_item.setFlags(
|
||||
bookmark_item.flags() | Qt.ItemIsUserCheckable
|
||||
)
|
||||
bookmark_item.setCheckState(
|
||||
0,
|
||||
(
|
||||
Qt.Checked
|
||||
if page_id in self.checked_chapters
|
||||
else Qt.Unchecked
|
||||
),
|
||||
)
|
||||
# only allow checking if this chapter has content
|
||||
if self.content_lengths.get(page_id, 0) > 0:
|
||||
bookmark_item.setFlags(bookmark_item.flags() | Qt.ItemIsUserCheckable)
|
||||
bookmark_item.setCheckState(0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked)
|
||||
else:
|
||||
bookmark_item.setFlags(bookmark_item.flags() & ~Qt.ItemIsUserCheckable)
|
||||
# map for uncategorized pages
|
||||
self.bookmark_items_map[page_num] = bookmark_item
|
||||
|
||||
added_pages.add(page_num)
|
||||
|
||||
@@ -1012,30 +1018,39 @@ class HandlerDialog(QDialog):
|
||||
|
||||
page_item = QTreeWidgetItem(bookmark_item, [page_title])
|
||||
page_item.setData(0, Qt.UserRole, page_id)
|
||||
# only allow checking if this sub-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
|
||||
),
|
||||
)
|
||||
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)
|
||||
|
||||
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)
|
||||
|
||||
covered_pages = set(added_pages)
|
||||
|
||||
uncategorized_pages = [
|
||||
i for i in range(len(self.pdf_doc)) if i not in covered_pages
|
||||
]
|
||||
if uncategorized_pages:
|
||||
self._add_other_pages(uncategorized_pages)
|
||||
# attach any pages without direct bookmarks under nearest preceding chapter
|
||||
uncategorized_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
|
||||
prev_nums = [n for n in sorted(self.bookmark_items_map) if n < page_num]
|
||||
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):
|
||||
pages_item = QTreeWidgetItem(self.treeWidget, ["Pages"])
|
||||
@@ -1056,34 +1071,12 @@ class HandlerDialog(QDialog):
|
||||
|
||||
page_item = QTreeWidgetItem(pages_item, [page_title])
|
||||
page_item.setData(0, Qt.UserRole, page_id)
|
||||
# only allow checking if standalone 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
|
||||
)
|
||||
|
||||
def _add_other_pages(self, uncategorized_pages):
|
||||
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
|
||||
)
|
||||
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 _are_provided_checks_relevant(self):
|
||||
if not self.checked_chapters:
|
||||
@@ -1108,7 +1101,7 @@ class HandlerDialog(QDialog):
|
||||
self.previewEdit.setStyleSheet("QTextEdit { border: none; }")
|
||||
|
||||
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.previewInfoLabel.setWordWrap(True)
|
||||
@@ -1183,7 +1176,6 @@ class HandlerDialog(QDialog):
|
||||
self.save_chapters_checkbox.setChecked(self.save_chapters_separately)
|
||||
self.save_chapters_checkbox.stateChanged.connect(self.on_save_chapters_changed)
|
||||
leftLayout.addWidget(self.save_chapters_checkbox)
|
||||
|
||||
self.merge_chapters_checkbox = QCheckBox(
|
||||
"Create a merged version at the end", self
|
||||
)
|
||||
@@ -1193,6 +1185,19 @@ class HandlerDialog(QDialog):
|
||||
)
|
||||
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)
|
||||
|
||||
leftWidget = QWidget()
|
||||
@@ -1705,6 +1710,10 @@ class HandlerDialog(QDialog):
|
||||
self.merge_chapters_at_end = bool(state)
|
||||
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):
|
||||
return (
|
||||
self.save_chapters_separately
|
||||
@@ -1715,6 +1724,9 @@ class HandlerDialog(QDialog):
|
||||
def get_merge_chapters_at_end(self):
|
||||
return self.merge_chapters_at_end
|
||||
|
||||
def get_save_as_project(self):
|
||||
return self.save_as_project
|
||||
|
||||
def on_tree_context_menu(self, pos):
|
||||
item = self.treeWidget.itemAt(pos)
|
||||
if (
|
||||
|
||||
+185
-23
@@ -8,9 +8,12 @@ from platformdirs import user_desktop_dir
|
||||
from PyQt5.QtCore import QThread, pyqtSignal, Qt
|
||||
from PyQt5.QtWidgets import QCheckBox, QVBoxLayout, QDialog, QLabel, QDialogButtonBox
|
||||
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 voice_formulas import get_new_voice
|
||||
import static_ffmpeg
|
||||
import threading # for efficient waiting
|
||||
import subprocess
|
||||
|
||||
|
||||
def get_sample_voice_text(lang_code):
|
||||
@@ -128,6 +131,7 @@ class ConversionThread(QThread):
|
||||
use_gpu=True,
|
||||
): # Add use_gpu parameter
|
||||
super().__init__()
|
||||
self._chapter_options_event = threading.Event()
|
||||
self.np = np_module
|
||||
self.KPipeline = kpipeline_class
|
||||
self.file_name = file_name
|
||||
@@ -222,6 +226,12 @@ class ConversionThread(QThread):
|
||||
chapter_splits = list(re.finditer(chapter_pattern, text))
|
||||
chapters = []
|
||||
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):
|
||||
start = match.end()
|
||||
end = (
|
||||
@@ -252,18 +262,12 @@ class ConversionThread(QThread):
|
||||
and not self.chapter_options_set
|
||||
):
|
||||
|
||||
self.waiting_for_user_input = True
|
||||
# Emit signal to main thread to show dialog
|
||||
# Emit signal to main thread and wait
|
||||
self.chapters_detected.emit(total_chapters)
|
||||
|
||||
# Wait for chapter options to be set
|
||||
while self.waiting_for_user_input and not self.cancel_requested:
|
||||
time.sleep(0.1)
|
||||
|
||||
self._chapter_options_event.wait()
|
||||
if self.cancel_requested:
|
||||
self.conversion_finished.emit("Cancelled", None)
|
||||
return
|
||||
|
||||
self.chapter_options_set = True
|
||||
|
||||
# Log all detected chapters at the beginning
|
||||
@@ -339,6 +343,12 @@ class ConversionThread(QThread):
|
||||
# Initialize current segment counter
|
||||
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
|
||||
for chapter_idx, (chapter_name, chapter_text) in enumerate(chapters, 1):
|
||||
if total_chapters > 1:
|
||||
@@ -354,6 +364,10 @@ class ConversionThread(QThread):
|
||||
chapter_subtitle_entries = []
|
||||
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
|
||||
split_pattern = r"\n+"
|
||||
|
||||
@@ -467,6 +481,9 @@ class ConversionThread(QThread):
|
||||
# Update progress more frequently (after each result)
|
||||
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
|
||||
if (
|
||||
save_chapters_separately
|
||||
@@ -483,14 +500,44 @@ class ConversionThread(QThread):
|
||||
|
||||
# Concatenate chapter audio and save
|
||||
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(
|
||||
chapters_out_dir, f"{chapter_filename}.{self.output_format}"
|
||||
chapters_out_dir, f"{chapter_filename}.{chapter_ext}"
|
||||
)
|
||||
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=self.output_format,
|
||||
format='wav' if self.output_format == 'm4b' else self.output_format,
|
||||
)
|
||||
|
||||
# Generate .srt subtitle file for chapter if not Disabled
|
||||
@@ -531,34 +578,74 @@ class ConversionThread(QThread):
|
||||
or not self.save_chapters_separately
|
||||
or getattr(self, "merge_chapters_at_end", True)
|
||||
)
|
||||
|
||||
intended_output_format = self.output_format # Store the original choice
|
||||
|
||||
if audio_segments and merge_chapters:
|
||||
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
|
||||
# Use the same suffix as above
|
||||
out_filename = f"{base_name}{suffix}.{self.output_format}"
|
||||
out_path = os.path.join(out_dir, out_filename)
|
||||
srt_path = os.path.splitext(out_path)[0] + ".srt"
|
||||
sf.write(out_path, audio, 24000, format=self.output_format)
|
||||
base_filepath_no_ext = os.path.join(out_dir, f"{base_name}{suffix}")
|
||||
|
||||
final_out_path = None
|
||||
|
||||
if intended_output_format == "m4b":
|
||||
final_out_path = self._generate_m4b_with_chapters(audio_data_np, chapters_time, base_filepath_no_ext)
|
||||
elif intended_output_format == "opus":
|
||||
static_ffmpeg.add_paths()
|
||||
opus_out_path = f"{base_filepath_no_ext}.opus"
|
||||
ffmpeg_cmd_opus = [
|
||||
"ffmpeg", "-y",
|
||||
"-thread_queue_size", "1024", # Increased thread_queue_size
|
||||
"-f", "f32le", "-ar", "24000", "-ac", "1", "-i", "pipe:0",
|
||||
"-c:a", "libopus", "-b:a", "24000", # Original bitrate
|
||||
opus_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":
|
||||
with open(
|
||||
srt_path, "w", encoding="utf-8", errors="replace"
|
||||
) as srt_file:
|
||||
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: {out_path}\n\nSubtitle saved to: {srt_path}",
|
||||
f"Audiobook saved to: {final_out_path}\n\nSubtitle saved to: {srt_path}",
|
||||
"green",
|
||||
),
|
||||
out_path,
|
||||
final_out_path,
|
||||
)
|
||||
else:
|
||||
self.conversion_finished.emit(
|
||||
(f"Audiobook saved to: {out_path}", "green"), out_path
|
||||
(f"Audiobook saved to: {final_out_path}", "green"), final_out_path
|
||||
)
|
||||
else:
|
||||
self.log_updated.emit(("Audio generation failed (final_out_path was not set).", "red"))
|
||||
self.conversion_finished.emit(("Audio generation failed.", "red"), None)
|
||||
elif audio_segments and not merge_chapters:
|
||||
self.conversion_finished.emit(
|
||||
(
|
||||
@@ -579,6 +666,81 @@ class ConversionThread(QThread):
|
||||
self.save_chapters_separately = options["save_chapters_separately"]
|
||||
self.merge_chapters_at_end = options["merge_chapters_at_end"]
|
||||
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):
|
||||
"""Helper function to format time for SRT files"""
|
||||
|
||||
+263
-50
@@ -71,7 +71,6 @@ from constants import (
|
||||
)
|
||||
from threading import Thread
|
||||
from voice_formula_gui import VoiceFormulaDialog
|
||||
from platformdirs import user_desktop_dir
|
||||
from voice_profiles import load_profiles
|
||||
|
||||
# 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.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):
|
||||
super().resizeEvent(event)
|
||||
margin = 12
|
||||
@@ -139,6 +145,11 @@ class InputBox(QLabel):
|
||||
# Position textbox button at top left
|
||||
self.textbox_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):
|
||||
# get icon without resizing using custom provider
|
||||
@@ -228,6 +239,7 @@ class InputBox(QLabel):
|
||||
should_show_edit = True
|
||||
|
||||
self.edit_btn.setVisible(should_show_edit)
|
||||
self.go_to_folder_btn.show()
|
||||
|
||||
def set_error(self, message):
|
||||
self.setText(message)
|
||||
@@ -250,6 +262,7 @@ class InputBox(QLabel):
|
||||
# Show textbox and hide edit when input is cleared
|
||||
self.textbox_btn.show()
|
||||
self.edit_btn.hide()
|
||||
self.go_to_folder_btn.hide()
|
||||
|
||||
def _human_readable_size(self, size, decimal_places=2):
|
||||
for unit in ["B", "KB", "MB", "GB", "TB"]:
|
||||
@@ -290,8 +303,8 @@ class InputBox(QLabel):
|
||||
)
|
||||
win.selected_book_path = file_path
|
||||
win.open_book_file(
|
||||
file_path
|
||||
) # This will handle the dialog and setting file info
|
||||
file_path # This will handle the dialog and setting file info
|
||||
)
|
||||
event.acceptProposedAction()
|
||||
else:
|
||||
self.set_error("Please drop a .txt, .epub, or .pdf file.")
|
||||
@@ -320,6 +333,17 @@ class InputBox(QLabel):
|
||||
# For regular txt files
|
||||
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):
|
||||
def __init__(self, parent=None):
|
||||
@@ -328,7 +352,7 @@ class TextboxDialog(QDialog):
|
||||
self.setWindowFlags(
|
||||
Qt.Window | Qt.WindowCloseButtonHint | Qt.WindowMaximizeButtonHint
|
||||
)
|
||||
self.resize(600, 400)
|
||||
self.resize(700, 500)
|
||||
|
||||
layout = QVBoxLayout(self)
|
||||
|
||||
@@ -358,6 +382,11 @@ class TextboxDialog(QDialog):
|
||||
self.save_as_button.clicked.connect(self.save_as_text)
|
||||
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.clicked.connect(self.reject)
|
||||
|
||||
@@ -382,8 +411,14 @@ class TextboxDialog(QDialog):
|
||||
return self.text_edit.toPlainText()
|
||||
|
||||
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 self.text_edit.toPlainText() == self.original_text:
|
||||
if text == self.original_text:
|
||||
self.reject()
|
||||
else:
|
||||
# Check if we need to warn about overwriting a non-temporary file
|
||||
@@ -444,6 +479,14 @@ class TextboxDialog(QDialog):
|
||||
except Exception as 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):
|
||||
def __init__(self):
|
||||
@@ -673,10 +716,23 @@ class abogen(QWidget):
|
||||
self.format_combo.setStyleSheet(
|
||||
"QComboBox { min-height: 20px; padding: 6px 12px; }"
|
||||
)
|
||||
format_options = ["wav", "flac", "mp3"]
|
||||
self.format_combo.addItems(format_options)
|
||||
self.format_combo.setCurrentText(self.selected_format)
|
||||
self.format_combo.currentTextChanged.connect(self.on_format_changed)
|
||||
# Add items with display labels and underlying keys
|
||||
for key, label in [
|
||||
("wav", "wav"),
|
||||
("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)
|
||||
controls_layout.addLayout(format_layout)
|
||||
# Save location
|
||||
@@ -831,9 +887,7 @@ class abogen(QWidget):
|
||||
)
|
||||
dialog.setWindowModality(Qt.NonModal)
|
||||
dialog.setModal(False)
|
||||
dialog.show()
|
||||
|
||||
# We'll handle the dialog result asynchronously
|
||||
dialog.show() # We'll handle the dialog result asynchronously
|
||||
def on_dialog_finished(result):
|
||||
if result != QDialog.Accepted:
|
||||
return False
|
||||
@@ -848,6 +902,7 @@ class abogen(QWidget):
|
||||
self.selected_chapters = all_checked_hrefs
|
||||
self.save_chapters_separately = dialog.get_save_chapters_separately()
|
||||
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
|
||||
if book_path.lower().endswith(".pdf"):
|
||||
@@ -856,8 +911,50 @@ class abogen(QWidget):
|
||||
# Use "abogen" prefix for temporary files
|
||||
# Extract base name without extension
|
||||
base_name = os.path.splitext(os.path.basename(book_path))[0]
|
||||
|
||||
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(
|
||||
prefix=f"abogen_{base_name}_", suffix=".txt", dir=temp_dir
|
||||
)
|
||||
@@ -1097,12 +1194,20 @@ class abogen(QWidget):
|
||||
QApplication.processEvents()
|
||||
|
||||
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.setFormat("%p%") # Keep format as percentage only
|
||||
self.etr_label.setText(
|
||||
f"Estimated time remaining: {etr_str}"
|
||||
) # Update ETR label
|
||||
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()
|
||||
QApplication.processEvents()
|
||||
|
||||
@@ -1432,7 +1537,21 @@ class abogen(QWidget):
|
||||
self.voice_combo.setEnabled(False)
|
||||
self.btn_voice_formula_mixer.setEnabled(False) # Disable mixer button
|
||||
self.btn_start.setEnabled(False) # Disable start button during preview
|
||||
# start loading animation
|
||||
|
||||
# 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):
|
||||
@@ -1462,14 +1581,14 @@ class abogen(QWidget):
|
||||
# Build voice formula string
|
||||
components = [f"{name}*{weight}" for name, weight in self.mixed_voice_state]
|
||||
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:
|
||||
from voice_profiles import load_profiles
|
||||
|
||||
entry = load_profiles().get(self.selected_profile_name, {})
|
||||
lang = entry.get("language")
|
||||
else:
|
||||
lang = None
|
||||
lang = self.selected_lang
|
||||
if not lang and self.mixed_voice_state:
|
||||
lang = (
|
||||
self.mixed_voice_state[0][0][0]
|
||||
@@ -1480,8 +1599,11 @@ class abogen(QWidget):
|
||||
lang = self.selected_voice[0]
|
||||
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(
|
||||
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.error.connect(self._preview_error)
|
||||
@@ -1545,6 +1667,11 @@ class abogen(QWidget):
|
||||
|
||||
def _preview_cleanup(self):
|
||||
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.setToolTip("Preview selected voice")
|
||||
self.btn_preview.setEnabled(True)
|
||||
@@ -1679,9 +1806,11 @@ class abogen(QWidget):
|
||||
# Add seperator
|
||||
menu.addSeparator()
|
||||
|
||||
# Add shortcut to desktop (Windows only)
|
||||
if platform.system() == "Windows":
|
||||
add_shortcut_action = QAction("Create desktop shortcut", self)
|
||||
# Add shortcut to desktop (Windows or Linux)
|
||||
if platform.system() == "Windows" or platform.system() == "Linux":
|
||||
# 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)
|
||||
menu.addAction(add_shortcut_action)
|
||||
|
||||
@@ -1754,10 +1883,12 @@ class abogen(QWidget):
|
||||
|
||||
def add_shortcut_to_desktop(self):
|
||||
"""Create a desktop shortcut to this program using PowerShell."""
|
||||
import sys, subprocess
|
||||
from PyQt5.QtWidgets import QMessageBox
|
||||
import sys
|
||||
from platformdirs import user_desktop_dir
|
||||
from utils import create_process
|
||||
|
||||
try:
|
||||
if platform.system() == "Windows":
|
||||
# where to put the .lnk
|
||||
desktop = user_desktop_dir()
|
||||
shortcut_path = os.path.join(desktop, "abogen.lnk")
|
||||
@@ -1774,29 +1905,73 @@ class abogen(QWidget):
|
||||
# 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
|
||||
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()"
|
||||
|
||||
# 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()
|
||||
# Run PowerShell with the command directly
|
||||
proc = create_process("powershell -NoProfile -ExecutionPolicy Bypass -Command \"" + ps_cmd + "\"")
|
||||
proc.wait()
|
||||
|
||||
subprocess.run(
|
||||
["powershell", "-NoProfile", "-Command", ps_cmd],
|
||||
check=True,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
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,
|
||||
@@ -1804,12 +1979,37 @@ class abogen(QWidget):
|
||||
f"Shortcut created on desktop:\n{shortcut_path}",
|
||||
)
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
QMessageBox.critical(
|
||||
# Offer installation for current user under ~/.local/share/applications
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
"Shortcut Error",
|
||||
f"PowerShell failed:\n{e.stderr.decode(errors='ignore')}",
|
||||
"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."
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
QMessageBox.critical(
|
||||
self, "Shortcut Error", f"Could not create shortcut:\n{e}"
|
||||
@@ -1820,25 +2020,37 @@ class abogen(QWidget):
|
||||
save_config(self.config)
|
||||
|
||||
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
|
||||
selected_profile = self.selected_profile_name
|
||||
if selected_profile:
|
||||
entry = load_profiles().get(selected_profile, {})
|
||||
# support new dict format
|
||||
entry = profiles.get(selected_profile, {})
|
||||
if isinstance(entry, dict):
|
||||
initial_state = entry.get("voices", [])
|
||||
else:
|
||||
initial_state = entry
|
||||
elif self.mixed_voice_state is not None:
|
||||
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 = [(self.selected_voice, 1.0)] if self.selected_voice else []
|
||||
initial_state = entry
|
||||
else:
|
||||
initial_state = []
|
||||
else:
|
||||
initial_state = []
|
||||
dialog = VoiceFormulaDialog(
|
||||
self, initial_state=initial_state, selected_profile=selected_profile
|
||||
)
|
||||
if dialog.exec_() == QDialog.Accepted:
|
||||
# After OK, select the profile in combo and update config
|
||||
if dialog.current_profile:
|
||||
self.selected_profile_name = dialog.current_profile
|
||||
self.config["selected_profile_name"] = dialog.current_profile
|
||||
@@ -2080,3 +2292,4 @@ class abogen(QWidget):
|
||||
"Setting Saved",
|
||||
f"Maximum words per subtitle set to {value}.",
|
||||
)
|
||||
|
||||
|
||||
Binary file not shown.
+105
-15
@@ -1,4 +1,5 @@
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
import warnings
|
||||
import platform
|
||||
@@ -66,10 +67,20 @@ def get_version():
|
||||
|
||||
# Define config path
|
||||
def get_user_config_path():
|
||||
if os.name == "nt":
|
||||
config_dir = os.path.join(os.environ["APPDATA"], "abogen")
|
||||
from platformdirs import user_config_dir
|
||||
|
||||
# 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 = os.path.join(os.path.expanduser("~"), ".config", "abogen")
|
||||
config_dir = user_config_dir("abogen", appauthor=False, roaming=True)
|
||||
else:
|
||||
# Windows and fallback case
|
||||
config_dir = user_config_dir("abogen", appauthor=False, roaming=True)
|
||||
|
||||
os.makedirs(config_dir, exist_ok=True)
|
||||
return os.path.join(config_dir, "config.json")
|
||||
|
||||
@@ -92,6 +103,89 @@ def clean_text(text, *args, **kwargs):
|
||||
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():
|
||||
try:
|
||||
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):
|
||||
# Remove double newlines (replace them with single newlines)
|
||||
cleaned_text = text.replace("\n\n", "")
|
||||
# Ignore chapter markers
|
||||
text_no_markers = re.sub(r"<<CHAPTER_MARKER:.*?>>", "", text)
|
||||
# Ignore newlines
|
||||
cleaned_text = text_no_markers.replace("\n", "")
|
||||
# Calculate character count
|
||||
char_count = len(cleaned_text)
|
||||
return char_count
|
||||
@@ -136,21 +232,15 @@ def prevent_sleep_start():
|
||||
0x80000000 | 0x00000001 | 0x00000040
|
||||
) # ES_CONTINUOUS | ES_SYSTEM_REQUIRED | ES_AWAYMODE_REQUIRED
|
||||
elif system == "Darwin":
|
||||
_sleep_procs["Darwin"] = subprocess.Popen(["caffeinate"])
|
||||
_sleep_procs["Darwin"] = create_process("caffeinate")
|
||||
elif system == "Linux":
|
||||
try:
|
||||
_sleep_procs["Linux"] = subprocess.Popen(
|
||||
[
|
||||
"systemd-inhibit",
|
||||
"--what=sleep",
|
||||
"--why=TextToAudiobook conversion",
|
||||
"sleep",
|
||||
"999999",
|
||||
]
|
||||
_sleep_procs["Linux"] = create_process(
|
||||
"systemd-inhibit --what=sleep --why=TextToAudiobook conversion sleep 999999"
|
||||
)
|
||||
except Exception:
|
||||
try:
|
||||
subprocess.Popen(["xdg-screensaver", "reset"])
|
||||
create_process("xdg-screensaver reset")
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
|
||||
+125
-28
@@ -47,7 +47,7 @@ VOICE_MIXER_WIDTH = 100
|
||||
SLIDER_WIDTH = 32
|
||||
MIN_WINDOW_WIDTH = 600
|
||||
MIN_WINDOW_HEIGHT = 400
|
||||
INITIAL_WINDOW_WIDTH = 1000
|
||||
INITIAL_WINDOW_WIDTH = 1200
|
||||
INITIAL_WINDOW_HEIGHT = 500
|
||||
|
||||
# Language options for the language selector loaded from constants
|
||||
@@ -413,6 +413,11 @@ class VoiceFormulaDialog(QDialog):
|
||||
self.language_combo.setCurrentIndex(idx)
|
||||
self.language_combo.currentIndexChanged.connect(self.mark_profile_modified)
|
||||
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)
|
||||
|
||||
# Error message
|
||||
@@ -475,7 +480,6 @@ class VoiceFormulaDialog(QDialog):
|
||||
|
||||
self.add_voices(initial_state or [])
|
||||
self.update_weighted_sums()
|
||||
self.update_subtitle_combo_enabled()
|
||||
|
||||
# assemble splitter
|
||||
splitter.addWidget(profile_widget)
|
||||
@@ -496,8 +500,6 @@ class VoiceFormulaDialog(QDialog):
|
||||
for vm in self.voice_mixers:
|
||||
vm.spin_box.valueChanged.connect(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):
|
||||
# Bind Delete key to delete_profile when a profile is selected
|
||||
@@ -682,8 +684,6 @@ class VoiceFormulaDialog(QDialog):
|
||||
voice_mixer.checkbox.stateChanged.connect(
|
||||
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
|
||||
|
||||
def handle_voice_checkbox(self, voice_mixer, state):
|
||||
@@ -713,6 +713,9 @@ class VoiceFormulaDialog(QDialog):
|
||||
]
|
||||
|
||||
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:
|
||||
self.error_label.hide()
|
||||
@@ -743,20 +746,6 @@ class VoiceFormulaDialog(QDialog):
|
||||
self.error_label.show()
|
||||
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):
|
||||
for mixer in self.voice_mixers:
|
||||
if mixer.voice_name == voice_name:
|
||||
@@ -819,7 +808,6 @@ class VoiceFormulaDialog(QDialog):
|
||||
# sync enabled state
|
||||
vm.toggle_inputs()
|
||||
self.update_weighted_sums()
|
||||
self.update_subtitle_combo_enabled()
|
||||
|
||||
def save_profile_by_name(self, name):
|
||||
profiles = load_profiles()
|
||||
@@ -846,10 +834,10 @@ class VoiceFormulaDialog(QDialog):
|
||||
self.update_weighted_sums()
|
||||
|
||||
def _handle_zero_weight_profiles(self):
|
||||
if self.profile_list.count() <= 1:
|
||||
profiles = load_profiles()
|
||||
if len(profiles) < 1:
|
||||
return False
|
||||
zero = []
|
||||
profiles = load_profiles()
|
||||
for i in range(self.profile_list.count()):
|
||||
item = self.profile_list.item(i)
|
||||
name = item.text().lstrip("*")
|
||||
@@ -868,7 +856,6 @@ class VoiceFormulaDialog(QDialog):
|
||||
if not zero:
|
||||
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?"
|
||||
# Use Delete instead of Ignore
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
"Invalid Profiles",
|
||||
@@ -963,8 +950,22 @@ class VoiceFormulaDialog(QDialog):
|
||||
self.update_profile_list_colors()
|
||||
|
||||
def new_profile(self):
|
||||
import re
|
||||
|
||||
while True:
|
||||
name, ok = QInputDialog.getText(self, "New Profile", "Enter profile name:")
|
||||
if ok and 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()
|
||||
# Remove 'New profile' placeholder if not persisted in JSON
|
||||
if (
|
||||
@@ -977,7 +978,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
self._profile_dirty.pop("New profile", None)
|
||||
if name in profiles:
|
||||
QMessageBox.warning(self, "Duplicate Name", "Profile already exists.")
|
||||
return
|
||||
continue
|
||||
profiles[name] = {
|
||||
"voices": [],
|
||||
"language": self.language_combo.currentData(),
|
||||
@@ -996,6 +997,7 @@ class VoiceFormulaDialog(QDialog):
|
||||
parent = self.parent()
|
||||
if hasattr(parent, "populate_profiles_in_voice_combo"):
|
||||
parent.populate_profiles_in_voice_combo()
|
||||
break
|
||||
self.update_profile_save_buttons()
|
||||
self.update_profile_list_colors()
|
||||
self.update_weighted_sums()
|
||||
@@ -1182,21 +1184,71 @@ class VoiceFormulaDialog(QDialog):
|
||||
)
|
||||
|
||||
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("*")
|
||||
import re
|
||||
|
||||
while True:
|
||||
new, ok = QInputDialog.getText(
|
||||
self, "Rename Profile", f"Profile name:", text=old
|
||||
)
|
||||
if ok and new and new != 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()
|
||||
if new in profiles:
|
||||
QMessageBox.warning(self, "Duplicate Name", "Profile already exists.")
|
||||
return
|
||||
continue
|
||||
|
||||
# Special case for renaming the virtual "New profile"
|
||||
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()
|
||||
if hasattr(parent, "populate_profiles_in_voice_combo"):
|
||||
parent.populate_profiles_in_voice_combo()
|
||||
break
|
||||
self.update_profile_save_buttons()
|
||||
self.update_profile_list_colors()
|
||||
|
||||
@@ -1227,6 +1279,13 @@ class VoiceFormulaDialog(QDialog):
|
||||
self.update_profile_list_colors()
|
||||
|
||||
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("*")
|
||||
profiles = load_profiles()
|
||||
base = f"{src}_duplicate"
|
||||
@@ -1287,3 +1346,41 @@ class VoiceFormulaDialog(QDialog):
|
||||
else:
|
||||
item.setBackground(QColor("white"))
|
||||
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,6 +1,6 @@
|
||||
import os
|
||||
import json
|
||||
from utils import get_user_config_path, get_resource_path
|
||||
from utils import get_user_config_path
|
||||
|
||||
|
||||
def _get_profiles_path():
|
||||
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 11 KiB |
+2
-1
@@ -22,7 +22,8 @@ dependencies = [
|
||||
"soundfile>=0.13.1",
|
||||
"pygame>=2.6.1",
|
||||
"charset_normalizer>=3.4.1",
|
||||
"chardet>=5.2.0"
|
||||
"chardet>=5.2.0",
|
||||
"static_ffmpeg>=2.13"
|
||||
]
|
||||
|
||||
classifiers = [
|
||||
|
||||
Reference in New Issue
Block a user