50 Commits
Author SHA1 Message Date
Deniz Şafak 0b15633995 v1.1.3 2025-07-16 06:27:22 +03:00
Deniz Şafak 8bc51dd270 App window now tries to fit the screen 2025-07-16 06:26:59 +03:00
Deniz Şafak d16c7603ce Better logic for M4B generation that makes it very fast 2025-07-16 05:15:08 +03:00
Deniz Şafak 970fc6c9c0 Better sleep state handling for Linux 2025-07-16 01:14:14 +03:00
Deniz Şafak d2f1c12de9 Fixed last sentence/subtitle entry timing in generated subtitles 2025-07-16 01:00:27 +03:00
Deniz Şafak 5d9409d391 Fix restart app in Windows 2025-07-15 12:23:15 +03:00
Deniz Şafak 0195788a54 v1.1.2 2025-07-15 11:59:27 +03:00
Deniz Şafak 1fd8fe21b7 Reformat using black 2025-07-15 11:57:43 +03:00
Deniz Şafak 89a7116c87 Generate subtitles dynamically like sound files, improvements in code 2025-07-15 11:50:22 +03:00
Deniz Şafak 6b85a08ccb Add new options and improvements for hfhub 2025-07-15 10:24:05 +03:00
Deniz Şafak b9c3fd7c6c Added a better logic for detecting chapters from the epub 2025-07-15 09:05:18 +03:00
Deniz Şafak a35bea9053 Directly write to audio file instead of RAM, improvements 2025-07-15 08:50:26 +03:00
Deniz Şafak e2802a2737 Potential fix for #37 and #38 2025-07-15 02:11:31 +03:00
Deniz Şafak e86ad435e5 Fixed Open folder and Open file buttons in the queue manager GUI 2025-07-14 12:09:44 +03:00
Deniz Şafak f2b42ef00e Windows installer build in editable mode 2025-07-12 01:52:17 +03:00
Deniz Şafak 7a8df9b34e v1.1.1 2025-07-11 11:51:41 +03:00
Deniz ŞafakandGitHub c9450d2ca5 v1.1.0
v1.1.0
2025-07-11 11:09:24 +03:00
Deniz Şafak efe0c48877 v1.1.0 2025-07-11 11:06:12 +03:00
Deniz Şafak 33361275fe Add new lines around chapter markers 2025-07-11 11:00:45 +03:00
Deniz Şafak 05fcf00a9c Beter canceling logic, improve quque summary and overall code 2025-07-11 10:26:33 +03:00
Deniz Şafak ef1063018f Add queue summary window, check GPU errors 2025-07-11 05:23:03 +03:00
Deniz Şafak 9b88a1ba55 Added chapter options countdown and new option "Configure max lines in log window", improvements in code 2025-07-11 03:31:47 +03:00
Deniz Şafak 6ede337658 Various improvements and modifications in code and documentation 2025-07-10 23:27:19 +03:00
Deniz Şafak 0c7a52fd86 Add drag & drop for gui manager, improvements 2025-07-10 06:13:15 +03:00
Deniz Şafak 5babc936b4 Improvements in gui and overall code 2025-07-10 05:12:32 +03:00
Deniz Şafak 742a77234f Imrpovements in queue manager gui and code 2025-07-10 04:02:39 +03:00
Deniz Şafak f2c558c8f4 Imrpovements in queue system and buttons 2025-07-09 22:17:08 +03:00
Deniz ŞafakandGitHub c540361160 Merge pull request #35 from jborza/main
Batch processing (multiple books in sequence)
2025-07-08 15:10:33 +03:00
Juraj BorzaandGitHub 26dcf9368b Merge branch 'denizsafak:main' into main 2025-07-08 13:02:47 +02:00
Juraj Borza 95f647fb54 disabling add to queue button on start / after queuing an item - fixes #30 2025-07-08 13:02:16 +02:00
Juraj Borza d0865015f9 refactored setting selected_lang, actual_subtitle_mode, voice_formula in non-queued conversion - #30 2025-07-08 11:49:57 +02:00
Deniz Şafak 3a87fb032f Better GPU detection 2025-07-06 05:17:45 +03:00
Deniz Şafak 6b997d2909 Revert defaults, disable workflow 2025-07-06 04:50:24 +03:00
Deniz Şafak c1fb4541dc Replace single newlines by default 2025-07-06 04:45:34 +03:00
Deniz Şafak 0f025da06b Fix "chapter_subtitle_path" error 2025-07-06 04:41:55 +03:00
Deniz Şafak 60b22c7731 Add dark theme support, improvements 2025-07-05 02:48:34 +03:00
Deniz Şafak f6c813602c Use values instad of plain text in subtitle formats 2025-07-04 20:44:58 +03:00
Juraj Borza bc474fecff filled in lang, voice, subtitle mode for queued items, converting the queued items in sequence - #30 2025-07-02 17:53:42 +02:00
Juraj Borza ad91e0ec34 queue manager - can remove items now - #30 2025-07-02 17:15:45 +02:00
Juraj Borza 17936a868e queue manager UI - a simple listwidget - #30 2025-07-01 13:31:47 +02:00
Juraj Borza d08ab54d2b list of queued items, unfinished adding of queued items - #30 2025-06-30 12:10:59 +02:00
Deniz Şafak bc4ae106c1 Update changelog 2025-06-23 05:12:49 +03:00
Deniz Şafak ec7656efb3 Update changelog 2025-06-23 05:12:17 +03:00
Deniz ŞafakandGitHub 8fac957f1b Merge pull request #31 from jborza/main
selecting multiple items implemented in chapter selection - #28
2025-06-23 05:00:05 +03:00
Juraj Borza c848ed473b selecting multiple items implemented in chapter selection - #28 2025-06-21 12:09:09 +02:00
Deniz ŞafakandGitHub 9ddd30fe1f Update test_publish.yml 2025-06-07 22:57:56 +03:00
Deniz ŞafakandGitHub 627240ad00 Update test_publish.yml 2025-06-07 22:52:56 +03:00
Deniz ŞafakandGitHub 2c7fc8a6ee Merge pull request #29 from KyleAure/ghcr-published-image
feat: publish image to ghcr
2025-06-07 22:42:09 +03:00
Kyle Aure b70c4ba57e feat: publish image to ghcr 2025-06-06 16:22:03 -05:00
Deniz Şafak 89d5fb8116 Updare Readmes 2025-05-30 11:50:19 +02:00
26 changed files with 2875 additions and 888 deletions
+61
View File
@@ -0,0 +1,61 @@
name: Build multi-arch Docker Image
on:
# Build and push
#release:
# types: [published]
# Build only
#push:
# branches: [main]
# TODO - enable build on pull requests if build times can be reduced
# pull_request:
workflow_dispatch:
env:
IMAGE_REPOSITORY: ghcr.io/denizsafak/abogen
jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Login to Github Container Registry
# Only if we need to push an image
if: ${{ github.event_name == 'release' && github.event.action == 'published' }}
uses: docker/login-action@v3
with:
registry: ghcr.io
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}
# Setup for buildx
- name: Set up QEMU
uses: docker/setup-qemu-action@v3
- name: Set up Docker Buildx
id: buildx
uses: docker/setup-buildx-action@v3
# Debugging information
- name: Docker info
run: docker info
- name: Buildx inspect
run: docker buildx inspect
# Build and (optionally) push the image
- name: Build image
uses: docker/build-push-action@v6
with:
context: ./abogen
file: ./abogen/Dockerfile
# platforms: linux/amd64,linux/arm/v7,linux/arm64,linux/ppc64le,linux/s390x
platforms: linux/amd64,linux/arm64
# Only push if we are publishing a release
push: ${{ github.event_name == 'release' && github.event.action == 'published' }}
# Use a 'temp' tag, that won't be pushed, for non-release builds
tags: ${{ env.IMAGE_REPOSITORY }}:${{ github.event.release.tag_name || 'temp' }}
# Use a cache to reduce build times
cache-to: type=gha,mode=max
cache-from: type=gha
+30
View File
@@ -1,3 +1,33 @@
# 1.1.3
- `M4B (with chapters)` generation is faster now, as it directly generates `m4b` files instead of converting from `wav`, which significantly reduces processing time.
- Better sleep state handling for Linux.
- The app window now tries to fit the screen if its height would exceed the available display area.
- Fixed issue where the app would not restart properly on Windows.
- Fixed last sentence/subtitle entry timing in generated subtitles, the end time of the final subtitle entry now correctly matches the end of the audio chunk, preventing zero or invalid timings at the end.
# v1.1.2
- Now you can play the audio files while they are processing.
- Audio and subtitle files are now written directly to disk during generation, which significantly reduces memory usage.
- Added a better logic for detecting chapters from the epub, mentioned by @jefro108 in #33
- Added a new option: `Reset to default settings`, allowing users to reset all settings to their default values.
- Added a new option: `Disable Kokoro's internet access`. This lets you prevent Kokoro from downloading models or voices from HuggingFace Hub, which can help avoid long waiting times if your computer is offline.
- HuggingFace Hub telemetry is now disabled by default for improved privacy. (HuggingFace Hub is used by Kokoro to download its models)
- Potential fix for #37 and #38, where the program was becoming slow while processing large files.
- Fixed `Open folder` and `Open file` buttons in the queue manager GUI.
- Improvements in code structure.
# v1.1.1
- Fixed adding wrong file in queue for EPUB and PDF files, ensuring the correct file is added to the queue.
- Reformatted the code using Black.
# v1.1.0
- Added queue system for processing multiple items, allowing users to add multiple files and process them in a queue, mentioned by @jborza in #30 (Special thanks to @jborza for implementing this feature in PR #35)
- Added a feature that allows selecting multiple items in book handler (in right click menu) by @jborza in #31, that fixes #28
- Added dark theme support, allowing users to switch between light and dark themes in the settings.
- Added auto-accept system to the chapter options dialog in conversion process, allowing the dialog to auto-accept after a certain time if no action is taken.
- Added new option: `Configure max lines in log window` that allows configuring the maximum number of lines to display in the log window.
- Improvements in documentation and code.
# v1.0.9 # v1.0.9
- Added chunking/segmenting system that fixes memory outage issues when processing large audio files. - Added chunking/segmenting system that fixes memory outage issues when processing large audio files.
- Added new option: `Subtitle format`, allowing users to choose between `srt` , `ass (wide)`, `ass (narrow)`, and `ass (centered wide)` and `ass (centered narrow)` - Added new option: `Subtitle format`, allowing users to choose between `srt` , `ass (wide)`, `ass (narrow)`, and `ass (centered wide)` and `ass (centered narrow)`
+54 -25
View File
@@ -34,6 +34,7 @@ This method handles everything automatically - installing all dependencies inclu
#### OPTION 2: Install using pip #### OPTION 2: Install using pip
```bash ```bash
# Create a virtual environment (optional) # Create a virtual environment (optional)
mkdir abogen && cd abogen
python -m venv venv python -m venv venv
venv\Scripts\activate venv\Scripts\activate
@@ -53,6 +54,7 @@ pip install abogen
brew install espeak-ng brew install espeak-ng
# Create a virtual environment (recommended) # Create a virtual environment (recommended)
mkdir abogen && cd abogen
python3 -m venv venv python3 -m venv venv
source venv/bin/activate source venv/bin/activate
@@ -67,6 +69,7 @@ sudo pacman -S espeak-ng # Arch Linux
sudo dnf install espeak-ng # Fedora sudo dnf install espeak-ng # Fedora
# Create a virtual environment (recommended) # Create a virtual environment (recommended)
mkdir abogen && cd abogen
python3 -m venv venv python3 -m venv venv
source venv/bin/activate source venv/bin/activate
@@ -116,34 +119,60 @@ abogen
Heres Abogen in action: in this demo, it processes 3,000 characters of text in just 11 seconds and turns it into 3 minutes and 28 seconds of audio, and I have a low-end **RTX 2060 Mobile laptop GPU**. Your results may vary depending on your hardware. Heres Abogen in action: in this demo, it processes 3,000 characters of text in just 11 seconds and turns it into 3 minutes and 28 seconds of audio, and I have a low-end **RTX 2060 Mobile laptop GPU**. Your results may vary depending on your hardware.
## `Key Features` ## `Configuration`
- **Supported formats**: `ePub`, `PDF`, or `.TXT` files (or use built-in text editor)
- **Speed**: Adjust speech rate from `0.1x` to `2.0x` | Options | Description |
- **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. | **Input Box** | Drag and drop `ePub`, `PDF`, or `.TXT` files (or use built-in text editor) |
- **Generate subtitles**: `Disabled`, `Sentence`, `Sentence + Comma`, `1 word`, `2 words`, `3 words`, etc. (Represents the number of words in each subtitle entry) | **Queue options** | Add multiple files to a queue and process them in batch, with individual settings for each file. See [Queue mode](#queue-mode) for more details. |
- **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)) | **Speed** | Adjust speech rate from `0.1x` to `2.0x` |
- **Save location**: `Save next to input file`, `Save to desktop`, or `Choose output folder` | **Select Voice** | First letter of the language code (e.g., `a` for American English, `b` for British English, etc.), second letter is for `m` for male and `f` for female. |
- **Chapter Control**: Select specific `chapters` from ePUBs or `chapters + pages` from PDFs. | **Voice mixer** | Create custom voices by mixing different voice models with a profile system. See [Voice Mixer](#voice-mixer) for more details. |
- **Save each chapter separately**: Save each chapter in e-books as a separate audio file. | **Voice preview** | Listen to the selected voice before processing. |
- **Project folder**: Save the converted items in a project folder with available metadata files. | **Generate subtitles** | `Disabled`, `Sentence`, `Sentence + Comma`, `1 word`, `2 words`, `3 words`, etc. (Represents the number of words in each subtitle entry) |
- **Options**: | **Output voice format** | `.WAV`, `.FLAC`, `.MP3`, `.OPUS (best compression)` 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)) |
- **Replace single newlines with spaces**: Replaces single newlines with spaces in the text. This is useful for texts that have imaginary line breaks. | **Output subtitle format** | Configures the subtitle format as `SRT (standard)`, `ASS (wide)`, `ASS (narrow)`, `ASS (centered wide)`, or `ASS (centered narrow)`. |
- **Configure max words per subtitle**: Configures the maximum number of words per subtitle entry. | **Replace single newlines with spaces** | Replaces single newlines with spaces in the text. This is useful for texts that have imaginary line breaks. |
- **Subtitle format**: Configures the subtitle format as `srt`, `ass (wide)`, `ass (narrow)`, `ass (centered wide)`, or `ass (centered narrow)`. | **Save location** | `Save next to input file`, `Save to desktop`, or `Choose output folder` |
- **Separate chapters audio format**: Configures the audio format for separate chapters as `wav`, `flac`, `mp3`, or `opus`.
- **Create desktop shortcut**: Creates a shortcut on your desktop for easy access. | Book handler options | Description |
- **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. | **Chapter Control** | Select specific `chapters` from ePUBs or `chapters + pages` from PDFs. |
- **Clear temporary files**: Deletes temporary files created during the conversion or preview. | **Save each chapter separately** | Save each chapter in e-books as a separate audio file. |
- **Check for updates at startup**: Automatically checks for updates when the program starts. | **Create a merged version** | Create a single audio file that combines all chapters. (If `Save each chapter separately` is disabled, this option will be the default behavior.) |
- **After conversion**: `Open file`, `Go to folder`, `New conversion`, or `Go back`. | **Save in a project folder with metadata** | Save the converted items in a project folder with available metadata files. |
| Menu options | Description |
|---------|-------------|
| **Theme** | Change the application's theme using `System`, `Light`, or `Dark` options. |
| **Configure max words per subtitle** | Configures the maximum number of words per subtitle entry. |
| **Configure max lines in log window** | Configures the maximum number of lines to display in the log window. |
| **Separate chapters audio format** | Configures the audio format for separate chapters as `wav`, `flac`, `mp3`, or `opus`. |
| **Create desktop shortcut** | Creates a shortcut on your desktop for easy access. |
| **Open config.json directory** | Opens the directory where the configuration file is stored. |
| **Open temp directory** | Opens the temporary directory where converted text files are stored. |
| **Clear temporary files** | Deletes temporary files created during the conversion or preview. |
| **Check for updates at startup** | Automatically checks for updates when the program starts. |
| **Disable Kokoro's internet access** | Prevents Kokoro from downloading models or voices from HuggingFace Hub, useful for offline use. |
| **Reset to default settings** | Resets all settings to their default values. |
## `Voice Mixer` ## `Voice Mixer`
<img title="Abogen Voice Mixer" src='https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/voice_mixer.png'> <img title="Abogen Voice Mixer" src='https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/voice_mixer.png'>
With voice mixer, you can create custom voices by mixing different voice models. You can adjust the weight of each voice and save your custom voice as a profile for future use. The voice mixer allows you to create unique and personalized voices. (Huge thanks to [@jborza](https://github.com/jborza) for making this possible through his contributions in [#5](https://github.com/denizsafak/abogen/pull/5)) With voice mixer, you can create custom voices by mixing different voice models. You can adjust the weight of each voice and save your custom voice as a profile for future use. The voice mixer allows you to create unique and personalized voices. (Huge thanks to [@jborza](https://github.com/jborza) for making this possible through his contributions in [#5](https://github.com/denizsafak/abogen/pull/5))
## `Queue Mode`
<img title="Abogen queue mode" src='https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/queue.png'>
Abogen supports **queue mode**, allowing you to add multiple files to a processing queue. This is useful if you want to convert several files in one batch.
- You can add text files (`.txt`) directly using the **Add files** button in the Queue Manager. To add PDF or EPUB files, use the input box in the main window and click the **Add to Queue** button.
- Each file in the queue keeps the configuration settings that were active when it was added. Changing the main window configuration afterward does **not** affect files already in the queue.
- You can view each file's configuration by hovering over them.
Abogen will process each item in the queue automatically, saving outputs as configured.
> Special thanks to [@jborza](https://github.com/jborza) for adding queue mode in PR [#35](https://github.com/denizsafak/abogen/pull/35)
## `About Chapter Markers` ## `About Chapter Markers`
When you process ePUB or PDF files, Abogen converts them into text files stored in your temporary directory. When you click "Edit," you're actually modifying these converted text files. In these text files, you'll notice tags that look like this: When you process ePUB 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:
@@ -168,7 +197,7 @@ When you process the text file, Abogen will detect these markers automatically a
![Abogen Chapter Marker](https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/chapter_marker.png) ![Abogen Chapter Marker](https://raw.githubusercontent.com/denizsafak/abogen/refs/heads/main/demo/chapter_marker.png)
## `About Metadata Tags` ## `About Metadata Tags`
Similar to chapter markers, it is possible to add metadata tags to `M4B` files. This is useful for audiobook players that support metadata, allowing you to add information like title, author, year, etc. Abogen automatically adds these tags when you process ePUB or PDF files, but you can also add them manually to your text files. Add metadata tags **at the beginning of your text file** like this: Similar to chapter markers, it is possible to add metadata tags for `M4B` files. This is useful for audiobook players that support metadata, allowing you to add information like title, author, year, etc. Abogen automatically adds these tags when you process ePUB or PDF files, but you can also add them manually to your text files. Add metadata tags **at the beginning of your text file** like this:
``` ```
<<METADATA_TITLE:Title>> <<METADATA_TITLE:Title>>
<<METADATA_ARTIST:Author>> <<METADATA_ARTIST:Author>>
@@ -254,12 +283,12 @@ Abogen is a standalone project, but it is inspired by and shares some similariti
- [ebook2audiobook](https://github.com/DrewThomasson/ebook2audiobook): Convert ebooks to audiobooks with chapters and metadata using dynamic AI models and voice cloning - [ebook2audiobook](https://github.com/DrewThomasson/ebook2audiobook): Convert ebooks to audiobooks with chapters and metadata using dynamic AI models and voice cloning
## `Roadmap` ## `Roadmap`
- [ ] Add OCR scan feature for PDF files using docling. - [ ] Add OCR scan feature for PDF files using docling/teserract.
- [x] Add chapter metadata for .m4a files. (Issue [#9](https://github.com/denizsafak/abogen/issues/9), PR [#10](https://github.com/denizsafak/abogen/pull/10)) - [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. (Issue [#1](https://github.com/denizsafak/abogen/issues/1), PR [#5](https://github.com/denizsafak/abogen/pull/5)) - [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 support for kokoro-onnx (If it's necessary).
- [ ] Add dark mode. - [x] Add dark mode.
## `Troubleshooting` ## `Troubleshooting`
If you encounter any issues while running Abogen, try launching it from the command line with: If you encounter any issues while running Abogen, try launching it from the command line with:
+2 -2
View File
@@ -230,7 +230,7 @@ if errorlevel 1 (
:: Install setup requirements :: Install setup requirements
echo Installing setup requirements... echo Installing setup requirements...
%PYTHON_CONSOLE_PATH% -m pip install --upgrade setuptools setuptools-scm wheel sphinx hatchling --no-warn-script-location %PYTHON_CONSOLE_PATH% -m pip install --upgrade setuptools setuptools-scm wheel sphinx hatchling editables --no-warn-script-location
if errorlevel 1 ( if errorlevel 1 (
echo Failed to install setup requirements. echo Failed to install setup requirements.
pause pause
@@ -250,7 +250,7 @@ if errorlevel 1 (
echo Checking and installing project dependencies... echo Checking and installing project dependencies...
if exist %PYPROJECT_FILE% ( if exist %PYPROJECT_FILE% (
echo Installing project from pyproject.toml... echo Installing project from pyproject.toml...
%PYTHON_CONSOLE_PATH% -m pip install . --no-warn-script-location %PYTHON_CONSOLE_PATH% -m pip install -e . --no-warn-script-location
if errorlevel 1 ( if errorlevel 1 (
echo Failed to install from pyproject.toml. echo Failed to install from pyproject.toml.
pause pause
+1 -1
View File
@@ -1 +1 @@
1.0.9 1.1.3
Binary file not shown.

Before

Width:  |  Height:  |  Size: 585 B

+31
View File
@@ -0,0 +1,31 @@
<?xml version="1.0" encoding="utf-8"?>
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
<!-- Uploaded to: SVG Repo, www.svgrepo.com, Generator: SVG Repo Mixer Tools -->
<svg height="800px" width="800px" version="1.1" id="_x32_" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink"
viewBox="0 0 512 512" xml:space="preserve">
<style type="text/css">
.st0{fill:#808080;}
</style>
<g>
<path class="st0" d="M502.325,307.303l-39.006-30.805c-6.215-4.908-9.665-12.429-9.668-20.348c0-0.084,0-0.168,0-0.252
c-0.014-7.936,3.44-15.478,9.667-20.396l39.007-30.806c8.933-7.055,12.093-19.185,7.737-29.701l-17.134-41.366
c-4.356-10.516-15.167-16.86-26.472-15.532l-49.366,5.8c-7.881,0.926-15.656-1.966-21.258-7.586
c-0.059-0.06-0.118-0.119-0.177-0.178c-5.597-5.602-8.476-13.36-7.552-21.225l5.799-49.363
c1.328-11.305-5.015-22.116-15.531-26.472L337.004,1.939c-10.516-4.356-22.646-1.196-29.701,7.736l-30.805,39.005
c-4.908,6.215-12.43,9.665-20.349,9.668c-0.084,0-0.168,0-0.252,0c-7.935,0.014-15.477-3.44-20.395-9.667L204.697,9.675
c-7.055-8.933-19.185-12.092-29.702-7.736L133.63,19.072c-10.516,4.356-16.86,15.167-15.532,26.473l5.799,49.366
c0.926,7.881-1.964,15.656-7.585,21.257c-0.059,0.059-0.118,0.118-0.178,0.178c-5.602,5.598-13.36,8.477-21.226,7.552
l-49.363-5.799c-11.305-1.328-22.116,5.015-26.472,15.531L1.939,174.996c-4.356,10.516-1.196,22.646,7.736,29.701l39.006,30.805
c6.215,4.908,9.665,12.429,9.668,20.348c0,0.084,0,0.167,0,0.251c0.014,7.935-3.44,15.477-9.667,20.395L9.675,307.303
c-8.933,7.055-12.092,19.185-7.736,29.701l17.134,41.365c4.356,10.516,15.168,16.86,26.472,15.532l49.366-5.799
c7.882-0.926,15.656,1.965,21.258,7.586c0.059,0.059,0.118,0.119,0.178,0.178c5.597,5.603,8.476,13.36,7.552,21.226l-5.799,49.364
c-1.328,11.305,5.015,22.116,15.532,26.472l41.366,17.134c10.516,4.356,22.646,1.196,29.701-7.736l30.804-39.005
c4.908-6.215,12.43-9.665,20.348-9.669c0.084,0,0.168,0,0.251,0c7.936-0.014,15.478,3.44,20.396,9.667l30.806,39.007
c7.055,8.933,19.185,12.093,29.701,7.736l41.366-17.134c10.516-4.356,16.86-15.168,15.532-26.472l-5.8-49.366
c-0.926-7.881,1.965-15.656,7.586-21.257c0.059-0.059,0.119-0.119,0.178-0.178c5.602-5.597,13.36-8.476,21.225-7.552l49.364,5.799
c11.305,1.328,22.117-5.015,26.472-15.531l17.134-41.365C514.418,326.488,511.258,314.358,502.325,307.303z M281.292,329.698
c-39.68,16.436-85.172-2.407-101.607-42.087c-16.436-39.68,2.407-85.171,42.087-101.608c39.68-16.436,85.172,2.407,101.608,42.088
C339.815,267.771,320.972,313.262,281.292,329.698z"/>
</g>
</svg>

After

Width:  |  Height:  |  Size: 2.6 KiB

+115 -34
View File
@@ -22,7 +22,7 @@ from PyQt5.QtWidgets import (
QLabel, QLabel,
) )
from PyQt5.QtCore import Qt from PyQt5.QtCore import Qt
from utils import clean_text, calculate_text_length from abogen.utils import clean_text, calculate_text_length
import os import os
import logging # Add logging import logging # Add logging
import urllib.parse import urllib.parse
@@ -111,7 +111,7 @@ class HandlerDialog(QDialog):
# Build treeview # Build treeview
self.treeWidget = QTreeWidget(self) self.treeWidget = QTreeWidget(self)
self.treeWidget.setHeaderHidden(True) self.treeWidget.setHeaderHidden(True)
self.treeWidget.setSelectionMode(QTreeWidget.SingleSelection) self.treeWidget.setSelectionMode(QTreeWidget.ExtendedSelection)
self.treeWidget.setContextMenuPolicy(Qt.CustomContextMenu) self.treeWidget.setContextMenuPolicy(Qt.CustomContextMenu)
self.treeWidget.customContextMenuRequested.connect(self.on_tree_context_menu) self.treeWidget.customContextMenuRequested.connect(self.on_tree_context_menu)
@@ -161,6 +161,7 @@ class HandlerDialog(QDialog):
self.treeWidget.expandAll() self.treeWidget.expandAll()
if self.treeWidget.topLevelItemCount() > 0: if self.treeWidget.topLevelItemCount() > 0:
self.treeWidget.setCurrentItem(self.treeWidget.topLevelItem(0)) self.treeWidget.setCurrentItem(self.treeWidget.topLevelItem(0))
self.treeWidget.setFocus()
# Update checkbox states # Update checkbox states
self._update_checkbox_states() self._update_checkbox_states()
@@ -239,20 +240,17 @@ class HandlerDialog(QDialog):
tag.decompose() tag.decompose()
text = clean_text(soup.get_text()).strip() text = clean_text(soup.get_text()).strip()
if text: if text:
# Use doc_href as the identifier
self.content_texts[doc_href] = text self.content_texts[doc_href] = text
self.content_lengths[doc_href] = len(text) self.content_lengths[doc_href] = len(text)
# Create a synthetic TOC entry
title = f"Chapter {i+1}: {doc_href}"
# Try to get a better title from <h1> or <title>
h1 = soup.find("h1")
if h1 and h1.get_text(strip=True):
title = h1.get_text(strip=True)
else:
title_tag = soup.find("title")
if title_tag and title_tag.get_text(strip=True):
title = title_tag.get_text(strip=True)
title = None
if soup.title and soup.title.string:
title = soup.title.string.strip()
elif (h1 := soup.find("h1")) and h1.get_text(strip=True):
title = h1.get_text(strip=True)
if not title:
title = f"Untitled Chapter {i+1}"
synthetic_toc.append( synthetic_toc.append(
(epub.Link(doc_href, title, doc_href), []) (epub.Link(doc_href, title, doc_href), [])
) # Wrap in tuple and empty list for compatibility ) # Wrap in tuple and empty list for compatibility
@@ -849,6 +847,12 @@ class HandlerDialog(QDialog):
has_parents = True has_parents = True
self.treeWidget.setRootIsDecorated(has_parents) self.treeWidget.setRootIsDecorated(has_parents)
def _update_checkbox_states(self):
"""Update the checkbox states based on the current checked chapters."""
for i in range(self.treeWidget.topLevelItemCount()):
item = self.treeWidget.topLevelItem(i)
self._update_item_checkbox_state(item)
def _build_epub_tree_from_nav( def _build_epub_tree_from_nav(
self, nav_nodes, parent_item, seen_content_hashes=None self, nav_nodes, parent_item, seen_content_hashes=None
): ):
@@ -990,10 +994,21 @@ class HandlerDialog(QDialog):
bookmark_item.setData(0, Qt.UserRole, page_id) bookmark_item.setData(0, Qt.UserRole, page_id)
# only allow checking if this chapter has content # only allow checking if this chapter has content
if self.content_lengths.get(page_id, 0) > 0: if self.content_lengths.get(page_id, 0) > 0:
bookmark_item.setFlags(bookmark_item.flags() | Qt.ItemIsUserCheckable) bookmark_item.setFlags(
bookmark_item.setCheckState(0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked) bookmark_item.flags() | Qt.ItemIsUserCheckable
)
bookmark_item.setCheckState(
0,
(
Qt.Checked
if page_id in self.checked_chapters
else Qt.Unchecked
),
)
else: else:
bookmark_item.setFlags(bookmark_item.flags() & ~Qt.ItemIsUserCheckable) bookmark_item.setFlags(
bookmark_item.flags() & ~Qt.ItemIsUserCheckable
)
# map for uncategorized pages # map for uncategorized pages
self.bookmark_items_map[page_num] = bookmark_item self.bookmark_items_map[page_num] = bookmark_item
@@ -1020,10 +1035,21 @@ class HandlerDialog(QDialog):
page_item.setData(0, Qt.UserRole, page_id) page_item.setData(0, Qt.UserRole, page_id)
# only allow checking if this sub-page has content # only allow checking if this sub-page has content
if self.content_lengths.get(page_id, 0) > 0: if self.content_lengths.get(page_id, 0) > 0:
page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable) page_item.setFlags(
page_item.setCheckState(0, Qt.Checked if page_id in self.checked_chapters else Qt.Unchecked) page_item.flags() | Qt.ItemIsUserCheckable
)
page_item.setCheckState(
0,
(
Qt.Checked
if page_id in self.checked_chapters
else Qt.Unchecked
),
)
else: else:
page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable) page_item.setFlags(
page_item.flags() & ~Qt.ItemIsUserCheckable
)
added_pages.add(sub_page_num) added_pages.add(sub_page_num)
@@ -1031,11 +1057,15 @@ class HandlerDialog(QDialog):
covered_pages = set(added_pages) covered_pages = set(added_pages)
# attach any pages without direct bookmarks under nearest preceding chapter # 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] uncategorized_pages = [
i for i in range(len(self.pdf_doc)) if i not in covered_pages
]
for page_num in uncategorized_pages: for page_num in uncategorized_pages:
# find nearest previous bookmark # find nearest previous bookmark
prev_nums = [n for n in sorted(self.bookmark_items_map) if n < page_num] 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 parent_item = (
self.bookmark_items_map[prev_nums[-1]] if prev_nums else self.treeWidget
)
page_id = f"page_{page_num+1}" page_id = f"page_{page_num+1}"
title = f"Page {page_num+1}" title = f"Page {page_num+1}"
text = self.content_texts.get(page_id, "").strip() text = self.content_texts.get(page_id, "").strip()
@@ -1048,7 +1078,9 @@ class HandlerDialog(QDialog):
# only allow checking if uncategorized page has content # only allow checking if uncategorized page has content
if self.content_lengths.get(page_id, 0) > 0: if self.content_lengths.get(page_id, 0) > 0:
page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable) 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: else:
page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable) page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable)
@@ -1074,7 +1106,9 @@ class HandlerDialog(QDialog):
# only allow checking if standalone page has content # only allow checking if standalone page has content
if self.content_lengths.get(page_id, 0) > 0: if self.content_lengths.get(page_id, 0) > 0:
page_item.setFlags(page_item.flags() | Qt.ItemIsUserCheckable) 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: else:
page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable) page_item.setFlags(page_item.flags() & ~Qt.ItemIsUserCheckable)
@@ -1494,7 +1528,9 @@ class HandlerDialog(QDialog):
authors_text = ", ".join(self.book_metadata["authors"]) authors_text = ", ".join(self.book_metadata["authors"])
html_content += f"<p style='text-align: center; font-style: italic;'>By {authors_text}</p>" html_content += f"<p style='text-align: center; font-style: italic;'>By {authors_text}</p>"
if self.book_metadata["publisher"] or self.book_metadata.get("publication_year"): if self.book_metadata["publisher"] or self.book_metadata.get(
"publication_year"
):
pub_info = [] pub_info = []
if self.book_metadata["publisher"]: if self.book_metadata["publisher"]:
pub_info.append(f"Published by {self.book_metadata['publisher']}") pub_info.append(f"Published by {self.book_metadata['publisher']}")
@@ -1536,7 +1572,9 @@ class HandlerDialog(QDialog):
try: try:
author_items = self.book.get_metadata("DC", "creator") author_items = self.book.get_metadata("DC", "creator")
if author_items: if author_items:
metadata["authors"] = [author[0] for author in author_items if len(author) > 0] metadata["authors"] = [
author[0] for author in author_items if len(author) > 0
]
except Exception as e: except Exception as e:
logging.warning(f"Error extracting author metadata: {e}") logging.warning(f"Error extracting author metadata: {e}")
@@ -1560,7 +1598,7 @@ class HandlerDialog(QDialog):
if date_items and len(date_items) > 0: if date_items and len(date_items) > 0:
date_str = date_items[0][0] date_str = date_items[0][0]
# Try to extract just the year from the date string # Try to extract just the year from the date string
year_match = re.search(r'\b(19|20)\d{2}\b', date_str) year_match = re.search(r"\b(19|20)\d{2}\b", date_str)
if year_match: if year_match:
metadata["publication_year"] = year_match.group(0) metadata["publication_year"] = year_match.group(0)
else: else:
@@ -1600,12 +1638,12 @@ class HandlerDialog(QDialog):
# Try to extract publication date from PDF metadata # Try to extract publication date from PDF metadata
if "creationDate" in pdf_info: if "creationDate" in pdf_info:
date_str = pdf_info["creationDate"] date_str = pdf_info["creationDate"]
year_match = re.search(r'D:(\d{4})', date_str) year_match = re.search(r"D:(\d{4})", date_str)
if year_match: if year_match:
metadata["publication_year"] = year_match.group(1) metadata["publication_year"] = year_match.group(1)
elif "modDate" in pdf_info: elif "modDate" in pdf_info:
date_str = pdf_info["modDate"] date_str = pdf_info["modDate"]
year_match = re.search(r'D:(\d{4})', date_str) year_match = re.search(r"D:(\d{4})", date_str)
if year_match: if year_match:
metadata["publication_year"] = year_match.group(1) metadata["publication_year"] = year_match.group(1)
@@ -1637,11 +1675,15 @@ class HandlerDialog(QDialog):
authors = metadata.get("authors") or ["Unknown"] authors = metadata.get("authors") or ["Unknown"]
authors_text = ", ".join(authors) authors_text = ", ".join(authors)
album_artist = authors_text or "Unknown" album_artist = authors_text or "Unknown"
year = metadata.get("publication_year") or current_year # Use publication year if available year = (
metadata.get("publication_year") or current_year
) # Use publication year if available
# Count chapters/pages # Count chapters/pages
total_chapters = len(self.checked_chapters) total_chapters = len(self.checked_chapters)
chapter_text = f"{total_chapters} {'Chapters' if self.file_type == 'epub' else 'Pages'}" chapter_text = (
f"{total_chapters} {'Chapters' if self.file_type == 'epub' else 'Pages'}"
)
# Format metadata tags # Format metadata tags
metadata_tags = [ metadata_tags = [
@@ -1651,7 +1693,7 @@ class HandlerDialog(QDialog):
f"<<METADATA_YEAR:{year}>>", f"<<METADATA_YEAR:{year}>>",
f"<<METADATA_ALBUM_ARTIST:{album_artist}>>", f"<<METADATA_ALBUM_ARTIST:{album_artist}>>",
f"<<METADATA_COMPOSER:Narrator>>", f"<<METADATA_COMPOSER:Narrator>>",
f"<<METADATA_GENRE:Audiobook>>" f"<<METADATA_GENRE:Audiobook>>",
] ]
return "\n".join(metadata_tags) return "\n".join(metadata_tags)
@@ -1723,7 +1765,10 @@ class HandlerDialog(QDialog):
if text: if text:
all_content.append(text) all_content.append(text)
included_text_ids.add(page_id) included_text_ids.add(page_id)
return metadata_tags + "\n\n" + "\n\n".join(all_content), all_checked_identifiers return (
metadata_tags + "\n\n" + "\n\n".join(all_content),
all_checked_identifiers,
)
iterator = QTreeWidgetItemIterator(self.treeWidget) iterator = QTreeWidgetItemIterator(self.treeWidget)
while iterator.value(): while iterator.value():
@@ -1781,7 +1826,10 @@ class HandlerDialog(QDialog):
included_text_ids.add(identifier) included_text_ids.add(identifier)
iterator += 1 iterator += 1
return metadata_tags + "\n\n" + "\n\n".join([t[1] for t in section_titles]), all_checked_identifiers return (
metadata_tags + "\n\n" + "\n\n".join([t[1] for t in section_titles]),
all_checked_identifiers,
)
def on_save_chapters_changed(self, state): def on_save_chapters_changed(self, state):
self.save_chapters_separately = bool(state) self.save_chapters_separately = bool(state)
@@ -1809,8 +1857,33 @@ class HandlerDialog(QDialog):
def get_save_as_project(self): def get_save_as_project(self):
return self.save_as_project return self.save_as_project
def check_selected_items(self):
self.set_selected_items_checked(True)
def uncheck_selected_items(self):
self.set_selected_items_checked(False)
def set_selected_items_checked(self, state: bool):
print(f"Checking selected items: {state}")
self.treeWidget.blockSignals(True)
for item in self.treeWidget.selectedItems():
if item.flags() & Qt.ItemIsUserCheckable:
item.setCheckState(0, Qt.Checked if state else Qt.Unchecked)
self.treeWidget.blockSignals(False)
self._update_checked_set_from_tree()
def on_tree_context_menu(self, pos): def on_tree_context_menu(self, pos):
item = self.treeWidget.itemAt(pos) item = self.treeWidget.itemAt(pos)
# multi-select context menu
if self.treeWidget.selectedItems() and len(self.treeWidget.selectedItems()) > 1:
menu = QMenu(self)
action = menu.addAction("Select")
action.triggered.connect(self.check_selected_items)
action = menu.addAction("Clear")
action.triggered.connect(self.uncheck_selected_items)
menu.exec_(self.treeWidget.mapToGlobal(pos))
return
if ( if (
not item not item
or item.childCount() == 0 or item.childCount() == 0
@@ -1835,5 +1908,13 @@ class HandlerDialog(QDialog):
def closeEvent(self, event): def closeEvent(self, event):
if self.pdf_doc is not None: if self.pdf_doc is not None:
self.pdf_doc.close() try:
if hasattr(self.pdf_doc, "is_closed"):
if not self.pdf_doc.is_closed:
self.pdf_doc.close()
else:
# Fallback: try/except close
self.pdf_doc.close()
except Exception:
pass
event.accept() event.accept()
+37 -1
View File
@@ -1,4 +1,4 @@
from utils import get_version from abogen.utils import get_version
# Program Information # Program Information
PROGRAM_NAME = "abogen" PROGRAM_NAME = "abogen"
@@ -8,6 +8,16 @@ PROGRAM_DESCRIPTION = (
GITHUB_URL = "https://github.com/denizsafak/abogen" GITHUB_URL = "https://github.com/denizsafak/abogen"
VERSION = get_version() VERSION = get_version()
# Settings
CHAPTER_OPTIONS_COUNTDOWN = 30 # Countdown seconds for chapter options
SUBTITLE_FORMATS = [
("srt", "SRT (standard)"),
("ass_wide", "ASS (wide)"),
("ass_narrow", "ASS (narrow)"),
("ass_centered_wide", "ASS (centered wide)"),
("ass_centered_narrow", "ASS (centered narrow)"),
]
# Language description mapping # Language description mapping
LANGUAGE_DESCRIPTIONS = { LANGUAGE_DESCRIPTIONS = {
"a": "American English", "a": "American English",
@@ -102,3 +112,29 @@ SAMPLE_VOICE_TEXTS = {
"p": "Este é um exemplo da voz selecionada.", "p": "Este é um exemplo da voz selecionada.",
"z": "这是所选语音的示例。", "z": "这是所选语音的示例。",
} }
COLORS = {
"BLUE": "#007dff",
"RED": "#c0392b",
"ORANGE": "#FFA500",
"GREEN": "#42ad4a",
"GREEN_BG": "rgba(66, 173, 73, 0.1)",
"GREEN_BG_HOVER": "rgba(66, 173, 73, 0.15)",
"GREEN_BORDER": "#42ad4a",
"BLUE_BG": "rgba(0, 102, 255, 0.05)",
"BLUE_BG_HOVER": "rgba(0, 102, 255, 0.1)",
"BLUE_BORDER_HOVER": "#6ab0de",
"YELLOW_BACKGROUND": "rgba(255, 221, 51, 0.40)",
"GREY_BACKGROUND": "rgba(128, 128, 128, 0.15)",
"RED_BACKGROUND": "rgba(232, 78, 60, 0.15)",
"RED_BG": "rgba(232, 78, 60, 0.10)",
"RED_BG_HOVER": "rgba(232, 78, 60, 0.15)",
# Theme palette colors
"DARK_BG": "#202326",
"DARK_BASE": "#141618",
"DARK_ALT": "#2c2f31",
"DARK_BUTTON": "#292c30",
"DARK_DISABLED": "#535353",
"LIGHT_BG": "#eff0f1",
"LIGHT_DISABLED": "#9a9999",
}
+605 -477
View File
File diff suppressed because it is too large Load Diff
+1119 -213
View File
File diff suppressed because it is too large Load Diff
+10 -1
View File
@@ -1,16 +1,20 @@
log_callback = None log_callback = None
show_warning_signal_emitter = None # Renamed for clarity show_warning_signal_emitter = None # Renamed for clarity
def set_log_callback(cb): def set_log_callback(cb):
global log_callback global log_callback
log_callback = cb log_callback = cb
def set_show_warning_signal_emitter(emitter): # Renamed for clarity def set_show_warning_signal_emitter(emitter): # Renamed for clarity
global show_warning_signal_emitter global show_warning_signal_emitter
show_warning_signal_emitter = emitter show_warning_signal_emitter = emitter
from huggingface_hub import hf_hub_download from huggingface_hub import hf_hub_download
def tracked_hf_hub_download(*args, **kwargs): def tracked_hf_hub_download(*args, **kwargs):
try: try:
local_kwargs = dict(kwargs) local_kwargs = dict(kwargs)
@@ -22,7 +26,10 @@ def tracked_hf_hub_download(*args, **kwargs):
if filename.endswith(".pth"): if filename.endswith(".pth"):
msg = f"\nDownloading model '{filename}' from Hugging Face ({repo_id}). This may take a while. Please wait..." msg = f"\nDownloading model '{filename}' from Hugging Face ({repo_id}). This may take a while. Please wait..."
if show_warning_signal_emitter: # Check if the emitter is set if show_warning_signal_emitter: # Check if the emitter is set
show_warning_signal_emitter.emit("Downloading Model", f"Downloading model '{filename}' from Hugging Face repository '{repo_id}'. This may take a while, please wait.") show_warning_signal_emitter.emit(
"Downloading Model",
f"Downloading model '{filename}' from Hugging Face repository '{repo_id}'. This may take a while, please wait.",
)
else: else:
msg = f"\nDownloading '{filename}' from Hugging Face ({repo_id}). Please wait..." msg = f"\nDownloading '{filename}' from Hugging Face ({repo_id}). Please wait..."
if log_callback: if log_callback:
@@ -32,5 +39,7 @@ def tracked_hf_hub_download(*args, **kwargs):
print(msg, flush=True) print(msg, flush=True)
return hf_hub_download(*args, **kwargs) return hf_hub_download(*args, **kwargs)
import huggingface_hub import huggingface_hub
huggingface_hub.hf_hub_download = tracked_hf_hub_download huggingface_hub.hf_hub_download = tracked_hf_hub_download
+8 -2
View File
@@ -1,16 +1,22 @@
import gpustat import gpustat
def check(): def check():
try: try:
stats = gpustat.new_query() stats = gpustat.new_query()
except Exception: except Exception:
return False return False
nvidia_keywords = ["nvidia", "rtx", "gtx", "quadro", "tesla", "titan", "mx"]
for gpu in stats.gpus: for gpu in stats.gpus:
print(gpu.name) name = gpu.name.lower()
if 'nvidia' in gpu.name.lower(): if any(keyword in name for keyword in nvidia_keywords):
return True return True
return False return False
if __name__ == "__main__": if __name__ == "__main__":
stats = gpustat.new_query()
for gpu in stats.gpus:
print(gpu.name)
print(check()) print(check())
+40 -3
View File
@@ -1,19 +1,36 @@
from json import load
import os import os
import sys import sys
import platform import platform
import atexit
from PyQt5.QtWidgets import QApplication from PyQt5.QtWidgets import QApplication
from PyQt5.QtGui import QIcon from PyQt5.QtGui import QIcon
from PyQt5.QtCore import qInstallMessageHandler, QtMsgType
# Add the directory to Python path # Add the directory to Python path
sys.path.insert(0, os.path.join(os.path.dirname(__file__))) sys.path.insert(0, os.path.join(os.path.dirname(__file__)))
from gui import abogen
from utils import get_resource_path from abogen.utils import get_resource_path, load_config, prevent_sleep_end
from constants import PROGRAM_NAME, VERSION
# Set Hugging Face Hub environment variables
os.environ["HF_HUB_DISABLE_TELEMETRY"] = "1" # Disable Hugging Face telemetry
os.environ["HF_HUB_ETAG_TIMEOUT"] = "10" # Metadata request timeout (seconds)
os.environ["HF_HUB_DOWNLOAD_TIMEOUT"] = "10" # File download timeout (seconds)
os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1" # Disable symlinks warning
if load_config().get("disable_kokoro_internet", False):
print("INFO: Kokoro's internet access is disabled.")
os.environ["HF_HUB_OFFLINE"] = "1" # Disable Hugging Face Hub internet access
from abogen.gui import abogen
from abogen.constants import PROGRAM_NAME, VERSION
# Set environment variables for AMD ROCm # Set environment variables for AMD ROCm
os.environ["MIOPEN_FIND_MODE"] = "FAST" os.environ["MIOPEN_FIND_MODE"] = "FAST"
os.environ["MIOPEN_CONV_PRECISE_ROCM_TUNING"] = "0" os.environ["MIOPEN_CONV_PRECISE_ROCM_TUNING"] = "0"
# Reset sleep states
atexit.register(prevent_sleep_end)
# Ensure sys.stdout and sys.stderr are valid in GUI mode # Ensure sys.stdout and sys.stderr are valid in GUI mode
if sys.stdout is None: if sys.stdout is None:
sys.stdout = open(os.devnull, "w") sys.stdout = open(os.devnull, "w")
@@ -24,6 +41,26 @@ if sys.stderr is None:
if platform.system() == "Darwin" and platform.processor() == "arm": if platform.system() == "Darwin" and platform.processor() == "arm":
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1" os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
# Custom message handler to filter out specific Qt warnings
def qt_message_handler(mode, context, message):
if "Wayland does not support QWindow::requestActivate()" in message:
return # Suppress this specific message
if "setGrabPopup called with a parent, QtWaylandClient" in message:
return
if mode == QtMsgType.QtWarningMsg:
print(f"Qt Warning: {message}")
elif mode == QtMsgType.QtCriticalMsg:
print(f"Qt Critical: {message}")
elif mode == QtMsgType.QtFatalMsg:
print(f"Qt Fatal: {message}")
elif mode == QtMsgType.QtInfoMsg:
print(f"Qt Info: {message}")
# Install the custom message handler
qInstallMessageHandler(qt_message_handler)
# Set application ID for Windows taskbar icon # Set application ID for Windows taskbar icon
if platform.system() == "Windows": if platform.system() == "Windows":
import ctypes import ctypes
+554
View File
@@ -0,0 +1,554 @@
# a simple window with a list of items in the queue, no checkboxes
# button to remove an item from the queue
# button to clear the queue
from PyQt5.QtWidgets import (
QDialog,
QVBoxLayout,
QHBoxLayout,
QDialogButtonBox,
QPushButton,
QListWidget,
QListWidgetItem,
QFileIconProvider,
QLabel,
QWidget,
QSizePolicy,
)
from PyQt5.QtCore import QFileInfo, Qt
from abogen.constants import COLORS
from copy import deepcopy
from PyQt5.QtGui import QFontMetrics
class ElidedLabel(QLabel):
def __init__(self, text, parent=None):
super().__init__(text, parent)
self._full_text = text
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Preferred)
def setText(self, text):
self._full_text = text
super().setText(text)
self.update()
def resizeEvent(self, event):
metrics = QFontMetrics(self.font())
elided = metrics.elidedText(self._full_text, Qt.ElideRight, self.width())
super().setText(elided)
super().resizeEvent(event)
def fullText(self):
return self._full_text
class QueueListItemWidget(QWidget):
def __init__(self, file_name, char_count):
super().__init__()
layout = QHBoxLayout()
layout.setContentsMargins(12, 0, 6, 0)
layout.setSpacing(0)
import os
name_label = ElidedLabel(os.path.basename(file_name))
char_label = QLabel(f"Chars: {char_count}")
char_label.setStyleSheet(f"color: {COLORS['LIGHT_DISABLED']};")
char_label.setAlignment(Qt.AlignRight | Qt.AlignVCenter)
char_label.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Preferred)
layout.addWidget(name_label, 1)
layout.addWidget(char_label, 0)
self.setLayout(layout)
class DroppableQueueListWidget(QListWidget):
def __init__(self, parent_dialog):
super().__init__()
self.parent_dialog = parent_dialog
self.setAcceptDrops(True)
# Overlay for drag hover
self.drag_overlay = QLabel("", self)
self.drag_overlay.setAlignment(Qt.AlignCenter)
self.drag_overlay.setStyleSheet(
f"border:2px dashed {COLORS['BLUE_BORDER_HOVER']}; border-radius:5px; padding:20px; background:{COLORS['BLUE_BG_HOVER']};"
)
self.drag_overlay.setVisible(False)
self.drag_overlay.setAttribute(Qt.WA_TransparentForMouseEvents, True)
def dragEnterEvent(self, event):
if event.mimeData().hasUrls():
for url in event.mimeData().urls():
if url.isLocalFile() and url.toLocalFile().lower().endswith(".txt"):
self.drag_overlay.resize(self.size())
self.drag_overlay.setVisible(True)
event.acceptProposedAction()
return
self.drag_overlay.setVisible(False)
event.ignore()
def dragMoveEvent(self, event):
if event.mimeData().hasUrls():
for url in event.mimeData().urls():
if url.isLocalFile() and url.toLocalFile().lower().endswith(".txt"):
event.acceptProposedAction()
return
event.ignore()
def dragLeaveEvent(self, event):
self.drag_overlay.setVisible(False)
event.accept()
def dropEvent(self, event):
self.drag_overlay.setVisible(False)
if event.mimeData().hasUrls():
file_paths = [
url.toLocalFile()
for url in event.mimeData().urls()
if url.isLocalFile() and url.toLocalFile().lower().endswith(".txt")
]
if file_paths:
self.parent_dialog.add_files_from_paths(file_paths)
event.acceptProposedAction()
else:
event.ignore()
else:
event.ignore()
def resizeEvent(self, event):
super().resizeEvent(event)
if hasattr(self, "drag_overlay"):
self.drag_overlay.resize(self.size())
class QueueManager(QDialog):
def __init__(self, parent, queue: list, title="Queue Manager", size=(600, 700)):
super().__init__()
self.queue = queue
self._original_queue = deepcopy(
queue
) # Store a deep copy of the original queue
self.parent = parent
layout = QVBoxLayout()
layout.setContentsMargins(15, 15, 15, 15) # set main layout margins
layout.setSpacing(12) # set spacing between widgets in main layout
# list of queued items
self.listwidget = DroppableQueueListWidget(self)
self.listwidget.setSelectionMode(QListWidget.ExtendedSelection)
self.listwidget.setAlternatingRowColors(True)
self.listwidget.setContextMenuPolicy(Qt.CustomContextMenu)
self.listwidget.customContextMenuRequested.connect(self.show_context_menu)
# Add informative instructions at the top
instructions = QLabel(
"<h2>How Queue Works?</h2>"
"You can add text files (.txt) directly using the '<b>Add files</b>' button below. "
"To add PDF or EPUB files, use the input box in the main window and click the <b>'Add to Queue'</b> button. "
"Each file in the queue keeps the configuration settings active when it was added. "
"Changing the main window configuration afterward <b>does not</b> affect files already in the queue. "
"You can view each file's configuration by hovering over them."
)
instructions.setAlignment(Qt.AlignLeft)
instructions.setWordWrap(True)
instructions.setStyleSheet("margin-bottom: 8px;")
layout.addWidget(instructions)
# Overlay label for empty queue
self.empty_overlay = QLabel(
"Drag and drop your text files here or use the 'Add files' button.",
self.listwidget,
)
self.empty_overlay.setAlignment(Qt.AlignCenter)
self.empty_overlay.setStyleSheet(
f"color: {COLORS['LIGHT_DISABLED']}; background: transparent; padding: 20px;"
)
self.empty_overlay.setWordWrap(True)
self.empty_overlay.setAttribute(Qt.WA_TransparentForMouseEvents, True)
self.empty_overlay.hide()
# add queue items to the list
self.process_queue()
button_row = QHBoxLayout()
button_row.setContentsMargins(0, 0, 0, 0) # optional: no margins for button row
button_row.setSpacing(7) # set spacing between buttons
# Add files button
add_files_button = QPushButton("Add files")
add_files_button.setFixedHeight(40)
add_files_button.clicked.connect(self.add_more_files)
button_row.addWidget(add_files_button)
# Remove button
self.remove_button = QPushButton("Remove selected")
self.remove_button.setFixedHeight(40)
self.remove_button.clicked.connect(self.remove_item)
button_row.addWidget(self.remove_button)
# Clear button
self.clear_button = QPushButton("Clear Queue")
self.clear_button.setFixedHeight(40)
self.clear_button.clicked.connect(self.clear_queue)
button_row.addWidget(self.clear_button)
layout.addLayout(button_row)
layout.addWidget(self.listwidget)
# Connect selection change to update button state
self.listwidget.currentItemChanged.connect(self.update_button_states)
self.listwidget.itemSelectionChanged.connect(self.update_button_states)
buttons = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel, self)
buttons.accepted.connect(self.accept)
buttons.rejected.connect(self.reject)
layout.addWidget(buttons)
self.setLayout(layout)
self.setWindowTitle(title)
self.resize(*size)
self.update_button_states()
def process_queue(self):
"""Process the queue items."""
self.listwidget.clear()
if not self.queue:
self.empty_overlay.resize(self.listwidget.size())
self.empty_overlay.show()
self.update_button_states()
return
else:
self.empty_overlay.hide()
icon_provider = QFileIconProvider()
for item in self.queue:
# Only show the file name, not the full path
file_name = item.file_name
display_name = file_name
import os
if os.path.sep in file_name:
display_name = os.path.basename(file_name)
# Get icon for the file
icon = icon_provider.icon(QFileInfo(file_name))
list_item = QListWidgetItem()
# Set tooltip with detailed info
output_folder = getattr(item, "output_folder", "")
tooltip = (
f"<b>Path:</b> {file_name}<br>"
f"<b>Language:</b> {getattr(item, 'lang_code', '')}<br>"
f"<b>Speed:</b> {getattr(item, 'speed', '')}<br>"
f"<b>Voice:</b> {getattr(item, 'voice', '')}<br>"
f"<b>Save Option:</b> {getattr(item, 'save_option', '')}<br>"
)
if output_folder not in (None, "", "None"):
tooltip += f"<b>Output Folder:</b> {output_folder}<br>"
tooltip += (
f"<b>Subtitle Mode:</b> {getattr(item, 'subtitle_mode', '')}<br>"
f"<b>Output Format:</b> {getattr(item, 'output_format', '')}<br>"
f"<b>Characters:</b> {getattr(item, 'total_char_count', '')}<br>"
f"<b>Replace Single Newlines:</b> {getattr(item, 'replace_single_newlines', False)}"
)
list_item.setToolTip(tooltip)
list_item.setIcon(icon)
list_item.setData(Qt.UserRole, file_name)
# Use custom widget for display
char_count = getattr(item, "total_char_count", 0)
widget = QueueListItemWidget(file_name, char_count)
self.listwidget.addItem(list_item)
self.listwidget.setItemWidget(list_item, widget)
self.update_button_states()
def remove_item(self):
items = self.listwidget.selectedItems()
if not items:
return
from PyQt5.QtWidgets import QMessageBox
# Remove by index to ensure correct mapping
rows = sorted([self.listwidget.row(item) for item in items], reverse=True)
# Warn user if removing multiple files
if len(rows) > 1:
reply = QMessageBox.question(
self,
"Confirm Remove",
f"Are you sure you want to remove {len(rows)} selected items from the queue?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No,
)
if reply != QMessageBox.Yes:
return
for row in rows:
if 0 <= row < len(self.queue):
del self.queue[row]
self.process_queue()
self.update_button_states()
def clear_queue(self):
from PyQt5.QtWidgets import QMessageBox
if len(self.queue) > 1:
reply = QMessageBox.question(
self,
"Confirm Clear Queue",
f"Are you sure you want to clear {len(self.queue)} items from the queue?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No,
)
if reply != QMessageBox.Yes:
return
self.queue.clear()
self.listwidget.clear()
self.empty_overlay.resize(
self.listwidget.size()
) # Ensure overlay is sized correctly
self.empty_overlay.show() # Show the overlay when queue is empty
self.update_button_states()
def get_queue(self):
return self.queue
def get_current_attributes(self):
# Fetch current attribute values from the parent abogen GUI
attrs = {}
parent = self.parent
if parent is not None:
# lang_code: use parent's get_voice_formula and get_selected_lang
if hasattr(parent, "get_voice_formula") and hasattr(
parent, "get_selected_lang"
):
voice_formula = parent.get_voice_formula()
attrs["lang_code"] = parent.get_selected_lang(voice_formula)
attrs["voice"] = voice_formula
else:
attrs["lang_code"] = getattr(parent, "selected_lang", "")
attrs["voice"] = getattr(parent, "selected_voice", "")
# speed
if hasattr(parent, "speed_slider"):
attrs["speed"] = parent.speed_slider.value() / 100.0
else:
attrs["speed"] = getattr(parent, "speed", 1.0)
# save_option
attrs["save_option"] = getattr(parent, "save_option", "")
# output_folder
attrs["output_folder"] = getattr(parent, "selected_output_folder", "")
# subtitle_mode
if hasattr(parent, "get_actual_subtitle_mode"):
attrs["subtitle_mode"] = parent.get_actual_subtitle_mode()
else:
attrs["subtitle_mode"] = getattr(parent, "subtitle_mode", "")
# output_format
attrs["output_format"] = getattr(parent, "selected_format", "")
# total_char_count
attrs["total_char_count"] = getattr(parent, "char_count", "")
# replace_single_newlines
attrs["replace_single_newlines"] = getattr(
parent, "replace_single_newlines", False
)
else:
# fallback: empty values
attrs = {
k: ""
for k in [
"lang_code",
"speed",
"voice",
"save_option",
"output_folder",
"subtitle_mode",
"output_format",
"total_char_count",
"replace_single_newlines",
]
}
return attrs
def add_files_from_paths(self, file_paths):
from abogen.utils import calculate_text_length
from PyQt5.QtWidgets import QMessageBox
import os
current_attrs = self.get_current_attributes()
duplicates = []
for file_path in file_paths:
class QueueItem:
pass
item = QueueItem()
item.file_name = file_path
for attr, value in current_attrs.items():
setattr(item, attr, value)
# Read file content and calculate total_char_count using calculate_text_length
try:
with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
file_content = f.read()
item.total_char_count = calculate_text_length(file_content)
except Exception:
item.total_char_count = 0
# Prevent adding duplicate items to the queue (check all attributes)
is_duplicate = False
for queued_item in self.queue:
if (
getattr(queued_item, "file_name", None)
== getattr(item, "file_name", None)
and getattr(queued_item, "lang_code", None)
== getattr(item, "lang_code", None)
and getattr(queued_item, "speed", None)
== getattr(item, "speed", None)
and getattr(queued_item, "voice", None)
== getattr(item, "voice", None)
and getattr(queued_item, "save_option", None)
== getattr(item, "save_option", None)
and getattr(queued_item, "output_folder", None)
== getattr(item, "output_folder", None)
and getattr(queued_item, "subtitle_mode", None)
== getattr(item, "subtitle_mode", None)
and getattr(queued_item, "output_format", None)
== getattr(item, "output_format", None)
and getattr(queued_item, "total_char_count", None)
== getattr(item, "total_char_count", None)
and getattr(queued_item, "replace_single_newlines", False)
== getattr(item, "replace_single_newlines", False)
):
is_duplicate = True
break
if is_duplicate:
duplicates.append(os.path.basename(file_path))
continue
self.queue.append(item)
if duplicates:
QMessageBox.warning(
self,
"Duplicate Item(s)",
f"Skipping {len(duplicates)} file(s) with the same attributes, already in the queue.",
)
self.process_queue()
self.update_button_states()
def add_more_files(self):
from PyQt5.QtWidgets import QFileDialog
from abogen.utils import calculate_text_length # import the function
# Only allow .txt files
files, _ = QFileDialog.getOpenFileNames(
self, "Select .txt files", "", "Text Files (*.txt)"
)
if not files:
return
self.add_files_from_paths(files)
def resizeEvent(self, event):
super().resizeEvent(event)
if hasattr(self, "empty_overlay"):
self.empty_overlay.resize(self.listwidget.size())
def update_button_states(self):
# Enable Remove if at least one item is selected, else disable
if hasattr(self, "remove_button"):
selected_count = len(self.listwidget.selectedItems())
self.remove_button.setEnabled(selected_count > 0)
if selected_count > 1:
self.remove_button.setText(f"Remove selected ({selected_count})")
else:
self.remove_button.setText("Remove selected")
# Disable Clear if queue is empty
if hasattr(self, "clear_button"):
self.clear_button.setEnabled(bool(self.queue))
def show_context_menu(self, pos):
from PyQt5.QtWidgets import QMenu, QAction
from PyQt5.QtGui import QDesktopServices
from PyQt5.QtCore import QUrl
import os
global_pos = self.listwidget.viewport().mapToGlobal(pos)
selected_items = self.listwidget.selectedItems()
menu = QMenu(self)
if len(selected_items) == 1:
# Add Remove action
remove_action = QAction("Remove this item", self)
remove_action.triggered.connect(self.remove_item)
menu.addAction(remove_action)
# Add Open file action
open_file_action = QAction("Open file", self)
def open_file():
from PyQt5.QtWidgets import QMessageBox
item = selected_items[0]
file_path = item.data(Qt.UserRole)
for q in self.queue:
if q.file_name == file_path:
if not os.path.exists(q.file_name):
QMessageBox.warning(
self, "File Not Found", f"The file does not exist."
)
return
QDesktopServices.openUrl(QUrl.fromLocalFile(q.file_name))
break
open_file_action.triggered.connect(open_file)
menu.addAction(open_file_action)
# Add Go to folder action
go_to_folder_action = QAction("Go to folder", self)
def go_to_folder():
from PyQt5.QtWidgets import QMessageBox
item = selected_items[0]
file_path = item.data(Qt.UserRole)
for q in self.queue:
if q.file_name == file_path:
if not os.path.exists(q.file_name):
QMessageBox.warning(
self, "File Not Found", f"The file does not exist."
)
return
folder = os.path.dirname(q.file_name)
if os.path.exists(folder):
QDesktopServices.openUrl(QUrl.fromLocalFile(folder))
break
go_to_folder_action.triggered.connect(go_to_folder)
menu.addAction(go_to_folder_action)
elif len(selected_items) > 1:
remove_action = QAction(f"Remove selected ({len(selected_items)})", self)
remove_action.triggered.connect(self.remove_item)
menu.addAction(remove_action)
# Always add Clear Queue
clear_action = QAction("Clear Queue", self)
clear_action.triggered.connect(self.clear_queue)
menu.addAction(clear_action)
menu.exec_(global_pos)
def accept(self):
# Accept: keep changes
super().accept()
def reject(self):
# Cancel: restore original queue
from PyQt5.QtWidgets import QMessageBox
# Warn if user changed a lot (e.g., more than 1 items difference)
original_count = len(self._original_queue)
current_count = len(self.queue)
if abs(original_count - current_count) > 1:
reply = QMessageBox.question(
self,
"Confirm Cancel",
f"Are you sure you want to cancel and discard all changes?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No,
)
if reply != QMessageBox.Yes:
return
self.queue.clear()
self.queue.extend(deepcopy(self._original_queue))
super().reject()
def keyPressEvent(self, event):
from PyQt5.QtCore import Qt
if event.key() == Qt.Key_Delete:
self.remove_item()
else:
super().keyPressEvent(event)
+16
View File
@@ -0,0 +1,16 @@
# represents a queued item - book, chapters, voice, etc.
from dataclasses import dataclass
@dataclass
class QueuedItem:
file_name: str
lang_code: str
speed: float
voice: str
save_option: str
output_folder: str
subtitle_mode: str
output_format: str
total_char_count: int
replace_single_newlines: bool = False
+44 -19
View File
@@ -7,8 +7,6 @@ import subprocess
import re import re
from threading import Thread from threading import Thread
# suppress warnings and disable HF hub symlink warnings
os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
@@ -105,15 +103,17 @@ def clean_text(text, *args, **kwargs):
default_encoding = sys.getfilesystemencoding() default_encoding = sys.getfilesystemencoding()
def create_process(cmd, stdin=None, text=True, capture_output=False): def create_process(cmd, stdin=None, text=True, capture_output=False):
import logging import logging
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
# Configure root logger to output to console if not already configured # Configure root logger to output to console if not already configured
root = logging.getLogger() root = logging.getLogger()
if not root.handlers: if not root.handlers:
handler = logging.StreamHandler(sys.stdout) handler = logging.StreamHandler(sys.stdout)
formatter = logging.Formatter('%(message)s') formatter = logging.Formatter("%(message)s")
handler.setFormatter(formatter) handler.setFormatter(formatter)
root.addHandler(handler) root.addHandler(handler)
root.setLevel(logging.INFO) root.setLevel(logging.INFO)
@@ -156,6 +156,7 @@ def create_process(cmd, stdin=None, text=True, capture_output=False):
# Stream output to console in real-time if not capturing # Stream output to console in real-time if not capturing
if proc.stdout and not capture_output: if proc.stdout and not capture_output:
def _stream_output(stream): def _stream_output(stream):
if text: if text:
# For text mode, read character by character for real-time output # For text mode, read character by character for real-time output
@@ -174,7 +175,9 @@ def create_process(cmd, stdin=None, text=True, capture_output=False):
break break
try: try:
# Try to decode binary data for display # Try to decode binary data for display
sys.stdout.write(chunk.decode(default_encoding, errors='replace')) sys.stdout.write(
chunk.decode(default_encoding, errors="replace")
)
sys.stdout.flush() sys.stdout.flush()
except Exception: except Exception:
pass pass
@@ -217,40 +220,62 @@ def calculate_text_length(text):
def get_gpu_acceleration(enabled): def get_gpu_acceleration(enabled):
from torch.cuda import is_available try:
import torch
from torch.cuda import is_available
if not enabled: if not enabled:
return "CUDA GPU available but using CPU.", False return "CUDA GPU available but using CPU.", False
if is_available():
if is_available(): return "CUDA GPU available and enabled.", True
return "CUDA GPU available and enabled.", True # Gather more diagnostic info if CUDA is not available
return "CUDA GPU is not available. Using CPU.", False try:
cuda_devices = torch.cuda.device_count()
cuda_error = (
torch.cuda.get_device_name(0)
if cuda_devices > 0
else "No devices found"
)
except Exception as e:
cuda_error = str(e)
return f"CUDA GPU is not available. Using CPU. ({cuda_error})", False
except Exception as e:
return f"Error checking CUDA GPU: {e}", False
def prevent_sleep_start(): def prevent_sleep_start():
from abogen.constants import PROGRAM_NAME
system = platform.system() system = platform.system()
if system == "Windows": if system == "Windows":
import ctypes import ctypes
ctypes.windll.kernel32.SetThreadExecutionState( ctypes.windll.kernel32.SetThreadExecutionState(
0x80000000 | 0x00000001 | 0x00000040 0x80000000 | 0x00000001 | 0x00000040
) )
elif system == "Darwin": elif system == "Darwin":
_sleep_procs["Darwin"] = create_process(["caffeinate"]) _sleep_procs["Darwin"] = create_process(["caffeinate"])
elif system == "Linux": elif system == "Linux":
# use a sleep that never exits so inhibition stays active # Add program name and reason for inhibition
_sleep_procs["Linux"] = create_process([ program_name = PROGRAM_NAME
"systemd-inhibit", reason = "Prevent sleep during abogen process"
"--what=handle-lid-switch:sleep", _sleep_procs["Linux"] = create_process(
"--mode=block", [
"sleep", "systemd-inhibit",
"infinity", f"--who={program_name}",
]) f"--why={reason}",
"--what=sleep",
"--mode=block",
"sleep",
"infinity",
]
)
def prevent_sleep_end(): def prevent_sleep_end():
system = platform.system() system = platform.system()
if system == "Windows": if system == "Windows":
import ctypes import ctypes
ctypes.windll.kernel32.SetThreadExecutionState(0x80000000) # ES_CONTINUOUS ctypes.windll.kernel32.SetThreadExecutionState(0x80000000) # ES_CONTINUOUS
elif system in ("Darwin", "Linux") and _sleep_procs[system]: elif system in ("Darwin", "Linux") and _sleep_procs[system]:
try: try:
+64 -26
View File
@@ -24,16 +24,20 @@ from PyQt5.QtWidgets import (
QMenu, QMenu,
QAction, QAction,
QComboBox, QComboBox,
QApplication,
) )
from PyQt5.QtCore import Qt, QTimer, QPoint, QRect, QSize from PyQt5.QtCore import Qt, QTimer, QPoint, QRect, QSize
from PyQt5.QtGui import QPixmap, QIcon, QColor from PyQt5.QtGui import QPixmap, QIcon
from constants import ( from abogen.constants import (
VOICES_INTERNAL, VOICES_INTERNAL,
SUPPORTED_LANGUAGES_FOR_SUBTITLE_GENERATION, SUPPORTED_LANGUAGES_FOR_SUBTITLE_GENERATION,
LANGUAGE_DESCRIPTIONS, LANGUAGE_DESCRIPTIONS,
COLORS,
) )
from utils import get_resource_path import re
from voice_profiles import ( import platform
from abogen.utils import get_resource_path
from abogen.voice_profiles import (
load_profiles, load_profiles,
save_profiles, save_profiles,
delete_profile, delete_profile,
@@ -221,6 +225,21 @@ class VoiceMixer(QWidget):
self.slider.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding) self.slider.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)
self.slider.setFixedWidth(SLIDER_WIDTH) self.slider.setFixedWidth(SLIDER_WIDTH)
# Fix slider in Windows
if platform.system() == "Windows":
appstyle = QApplication.instance().style().objectName().lower()
if appstyle != "windowsvista":
# Set custom groove color for disabled state using COLORS["GREY_BACKGROUND"]
self.slider.setStyleSheet(
f"""
QSlider::groove:vertical:disabled {{
background: {COLORS.get("GREY_BACKGROUND")};
width: 4px;
border-radius: 4px;
}}
"""
)
# Connect controls # Connect controls
self.slider.valueChanged.connect(lambda val: self.spin_box.setValue(val / 100)) self.slider.valueChanged.connect(lambda val: self.spin_box.setValue(val / 100))
self.spin_box.valueChanged.connect( self.spin_box.valueChanged.connect(
@@ -271,9 +290,9 @@ class HoverLabel(QLabel):
self.delete_button = QPushButton("×", self) self.delete_button = QPushButton("×", self)
self.delete_button.setFixedSize(16, 16) self.delete_button.setFixedSize(16, 16)
self.delete_button.setStyleSheet( self.delete_button.setStyleSheet(
""" f"""
QPushButton { QPushButton {{
background-color: #ff5555; background-color: {COLORS.get("RED")};
color: white; color: white;
border-radius: 7px; border-radius: 7px;
font-weight: bold; font-weight: bold;
@@ -281,11 +300,11 @@ class HoverLabel(QLabel):
border: none; border: none;
padding: 0px; padding: 0px;
margin: 0px; margin: 0px;
} }}
QPushButton:hover { QPushButton:hover {{
background-color: red; background-color: red;
} }}
""" """
) )
# Make sure the entire button is clickable, not just the text # Make sure the entire button is clickable, not just the text
self.delete_button.setFocusPolicy(Qt.NoFocus) self.delete_button.setFocusPolicy(Qt.NoFocus)
@@ -313,7 +332,9 @@ class VoiceFormulaDialog(QDialog):
self._original_mixed_voice_state = None self._original_mixed_voice_state = None
if parent is not None: if parent is not None:
self._original_profile_name = getattr(parent, "selected_profile_name", None) self._original_profile_name = getattr(parent, "selected_profile_name", None)
self._original_mixed_voice_state = getattr(parent, "mixed_voice_state", None) self._original_mixed_voice_state = getattr(
parent, "mixed_voice_state", None
)
profiles = load_profiles() profiles = load_profiles()
self._virtual_new_profile = False self._virtual_new_profile = False
if not profiles: if not profiles:
@@ -350,6 +371,11 @@ class VoiceFormulaDialog(QDialog):
profile_layout.addLayout(header_layout) profile_layout.addLayout(header_layout)
# Profile list # Profile list
self.profile_list = QListWidget() self.profile_list = QListWidget()
self.profile_list.setSelectionMode(QListWidget.SingleSelection)
self.profile_list.setSelectionBehavior(QListWidget.SelectRows)
self.profile_list.setStyleSheet(
"QListWidget::item:selected { background: palette(highlight); color: palette(highlighted-text); }"
)
icon = QIcon(get_resource_path("abogen.assets", "profile.png")) icon = QIcon(get_resource_path("abogen.assets", "profile.png"))
if self._virtual_new_profile: if self._virtual_new_profile:
item = QListWidgetItem(icon, "New profile") item = QListWidgetItem(icon, "New profile")
@@ -738,7 +764,7 @@ class VoiceFormulaDialog(QDialog):
percentage = weight / total * 100 percentage = weight / total * 100
# Make the voice name bold and include percentage # Make the voice name bold and include percentage
voice_label = HoverLabel( voice_label = HoverLabel(
f'<b><span style="color:#007dff">{name}: {percentage:.1f}%</span></b>', f'<b><span style="color:{COLORS.get("BLUE")}">{name}: {percentage:.1f}%</span></b>',
name, name,
) )
voice_label.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred) voice_label.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
@@ -953,6 +979,19 @@ class VoiceFormulaDialog(QDialog):
return return
super().closeEvent(event) super().closeEvent(event)
def _parse_rgba_to_qcolor(self, rgba_str):
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor
"""Helper to convert 'rgba(R,G,B,A_float)' string to QColor."""
match = re.match(r"rgba\((\d+),\s*(\d+),\s*(\d+),\s*([\d.]+)\)", rgba_str)
if match:
r, g, b = int(match.group(1)), int(match.group(2)), int(match.group(3))
a_float = float(match.group(4))
a_int = int(a_float * 255)
return QColor(r, g, b, a_int)
return Qt.GlobalColor.transparent
def mark_profile_modified(self): def mark_profile_modified(self):
item = self.profile_list.currentItem() item = self.profile_list.currentItem()
if item and not item.text().startswith("*"): if item and not item.text().startswith("*"):
@@ -1052,7 +1091,7 @@ class VoiceFormulaDialog(QDialog):
self, "Import Profiles", "", "JSON Files (*.json)" self, "Import Profiles", "", "JSON Files (*.json)"
) )
if path: if path:
from voice_profiles import load_profiles, save_profiles from abogen.voice_profiles import load_profiles, save_profiles
# Try to read the file and count profiles # Try to read the file and count profiles
try: try:
@@ -1339,23 +1378,23 @@ class VoiceFormulaDialog(QDialog):
self.profile_list.setItemWidget(item, widget) self.profile_list.setItemWidget(item, widget)
def update_profile_list_colors(self): def update_profile_list_colors(self):
from PyQt5.QtCore import Qt
profiles = load_profiles() 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("*")
if self._virtual_new_profile and name == "New profile": if self._virtual_new_profile and name == "New profile":
color = QColor("#ffff00") # yellow color = self._parse_rgba_to_qcolor(COLORS.get("YELLOW_BACKGROUND"))
color.setAlpha(64) # Set transparency item.setData(Qt.BackgroundRole, color)
item.setBackground(color)
elif item.text().startswith("*"): elif item.text().startswith("*"):
color = QColor("#ffff00") # yellow color = self._parse_rgba_to_qcolor(COLORS.get("YELLOW_BACKGROUND"))
color.setAlpha(64) # Set transparency item.setData(Qt.BackgroundRole, color)
item.setBackground(color)
else: else:
item.setBackground(self.profile_list.palette().base().color()) # Use default list item background item.setData(
Qt.BackgroundRole, self.profile_list.palette().base().color()
)
weights = profiles.get(name, {}).get("voices", []) weights = profiles.get(name, {}).get("voices", [])
# Defensive: only sum if weights is a list of (voice, weight) pairs
total = 0 total = 0
if isinstance(weights, list): if isinstance(weights, list):
for entry in weights: for entry in weights:
@@ -1366,9 +1405,8 @@ class VoiceFormulaDialog(QDialog):
): ):
total += entry[1] total += entry[1]
if total == 0: if total == 0:
color = QColor("#ff0000") # red color = self._parse_rgba_to_qcolor(COLORS.get("RED_BACKGROUND"))
color.setAlpha(64) # Set transparency item.setData(Qt.BackgroundRole, color)
item.setBackground(color)
self.update_profile_save_buttons() self.update_profile_save_buttons()
def preview_current_mix(self): def preview_current_mix(self):
+1 -1
View File
@@ -1,5 +1,5 @@
import re import re
from constants import VOICES_INTERNAL from abogen.constants import VOICES_INTERNAL
# Calls parsing and loads the voice to gpu or cpu # Calls parsing and loads the voice to gpu or cpu
+1 -1
View File
@@ -1,6 +1,6 @@
import os import os
import json import json
from utils import get_user_config_path from abogen.utils import get_user_config_path
def _get_profiles_path(): def _get_profiles_path():
+1 -1
View File
@@ -23,7 +23,7 @@ brew install ffmpeg
sudo apt install ffmpeg sudo apt install ffmpeg
``` ```
### Create the Video (.webm) ## Create the Video (.webm)
Run this FFmpeg command to create the tiny video: Run this FFmpeg command to create the tiny video:
BIN
View File
Binary file not shown.

Before

Width:  |  Height:  |  Size: 23 KiB

After

Width:  |  Height:  |  Size: 62 KiB

BIN
View File
Binary file not shown.

Before

Width:  |  Height:  |  Size: 41 KiB

After

Width:  |  Height:  |  Size: 150 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 11 KiB

After

Width:  |  Height:  |  Size: 31 KiB

BIN
View File
Binary file not shown.

After

Width:  |  Height:  |  Size: 125 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 28 KiB

After

Width:  |  Height:  |  Size: 69 KiB