Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
ba1328311c | ||
|
|
9e6b9396bc | ||
|
|
2e40db5fe5 | ||
|
|
720bf8697f | ||
|
|
1893d1a130 | ||
|
|
8472baf031 | ||
|
|
6cab618706 | ||
|
|
ebf3b362f7 | ||
|
|
2dd303f53c | ||
|
|
df69fb0b07 | ||
|
|
a0d9771fa6 | ||
|
|
a71c46bb98 | ||
|
|
995f317b4f | ||
|
|
6ef2de2923 | ||
|
|
37c4e86566 | ||
|
|
11071b279b | ||
|
|
5ab7463f35 | ||
|
|
74c9238e3a | ||
|
|
e874e46304 | ||
|
|
b32df6ac15 | ||
|
|
b76dd3561f | ||
|
|
60bfbdcabd | ||
|
|
c128d0851b | ||
|
|
2f3b7762e4 | ||
|
|
0e594f5098 | ||
|
|
14151d4dda | ||
|
|
94ab21fb15 | ||
|
|
1acac3d6c6 | ||
|
|
95ec2dc9d2 | ||
|
|
6b93eb6c1e | ||
|
|
838578a042 | ||
|
|
2f689f0438 | ||
|
|
022025ded6 | ||
|
|
320a7a6aba | ||
|
|
fa82870883 | ||
|
|
a094d4f9da | ||
|
|
bb028af694 | ||
|
|
aeb225ec7c | ||
|
|
06a632459e | ||
|
|
221efbddb4 | ||
|
|
e81a88938a | ||
|
|
65e257077c | ||
|
|
95659d5b7a | ||
|
|
ff57270286 | ||
|
|
4fc61ba99a | ||
|
|
3cd22081f4 | ||
|
|
d20ee95e67 | ||
|
|
cd47f6b123 | ||
|
|
53a2d0a65e | ||
|
|
13318243e4 | ||
|
|
acc457436a | ||
|
|
8d80151889 | ||
|
|
8b0ade5b9f | ||
|
|
9f033e4654 | ||
|
|
1a2b8571d0 | ||
|
|
9ab20be294 | ||
|
|
a34cbfdd55 | ||
|
|
94f5b781cc | ||
|
|
f83e7c3b4e | ||
|
|
607dc89be5 | ||
|
|
47758c4aeb | ||
|
|
26414d4925 | ||
|
|
7ad92e7f5d | ||
|
|
e775cef5bd | ||
|
|
ac9ce144a0 | ||
|
|
f181c06f89 | ||
|
|
9fae4692e3 | ||
|
|
97f0d6441a | ||
|
|
2ea6cb6304 | ||
|
|
5bed68a963 | ||
|
|
be9862d0b0 | ||
|
|
f5a811f0f8 | ||
|
|
dc9d343115 | ||
|
|
c92aa435e5 | ||
|
|
1f425e6bac | ||
|
|
0854544a47 | ||
|
|
52a7922e98 | ||
|
|
558583b16f | ||
|
|
993d065368 | ||
|
|
d7376fa8ef | ||
|
|
2b175b13c4 | ||
|
|
da0445489e | ||
|
|
e4eaa760b9 | ||
|
|
de4017f385 | ||
|
|
f2119aa0f7 | ||
|
|
bb81f3f874 | ||
|
|
fae15eecb1 | ||
|
|
6edcc2125b | ||
|
|
d26909a1d6 | ||
|
|
3be6b842c4 | ||
|
|
e10dc700c3 | ||
|
|
e8e21f949c | ||
|
|
e68a7d72f2 | ||
|
|
fd26db8e8a | ||
|
|
396f1290e7 | ||
|
|
5bc7dd8692 | ||
|
|
b5b68ad51f | ||
|
|
e49ceed40c | ||
|
|
598aec7f52 | ||
|
|
f25d2627cd | ||
|
|
3e6f1bace3 | ||
|
|
fb851ff3f0 | ||
|
|
92bf8f5de8 | ||
|
|
8f318892c2 | ||
|
|
28a62ed39a | ||
|
|
4c08927ed9 | ||
|
|
a93da39711 | ||
|
|
557f4b80b6 |
No files matched your search
@@ -1,6 +1,7 @@
|
|||||||
voices/*.*
|
voices/*.*
|
||||||
models/*.*
|
models/*.*
|
||||||
outputs/*.*
|
outputs/*.*
|
||||||
|
finetune/*.*
|
||||||
config.json
|
config.json
|
||||||
confignew.json
|
confignew.json
|
||||||
models.json
|
models.json
|
||||||
|
|||||||
Vendored
+9
@@ -0,0 +1,9 @@
|
|||||||
|
{
|
||||||
|
"files.associations": {
|
||||||
|
"*.html": "jinja-html"
|
||||||
|
},
|
||||||
|
"emmet.includeLanguages": {
|
||||||
|
"jinja-html": "html",
|
||||||
|
"jinja2": "html"
|
||||||
|
},
|
||||||
|
}
|
||||||
@@ -2,14 +2,16 @@
|
|||||||
AllTalk is an updated version of the Coqui_tts extension for Text Generation web UI. Features include:
|
AllTalk is an updated version of the Coqui_tts extension for Text Generation web UI. Features include:
|
||||||
|
|
||||||
- **Custom Start-up Settings:** Adjust your default start-up settings. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
- **Custom Start-up Settings:** Adjust your default start-up settings. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
||||||
- **Narrarator:** Use different voices for main character and narration. [Example Narration](https://vocaroo.com/18fYWVxiQpk1)
|
- **Narrarator:** Use different voices for main character and narration. [Example Narration](https://vocaroo.com/18nrv7FR6wuA)
|
||||||
- **Low VRAM mode:** Great for people with small GPU memory or if your VRAM is filled by your LLM. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
- **Low VRAM mode:** Great for people with small GPU memory or if your VRAM is filled by your LLM. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
||||||
- **DeepSpeed:** A 3-4x performance boost generating TTS. [DeepSpeed Windows/Linux Instructions](https://github.com/erew123/alltalk_tts?tab=readme-ov-file#deepspeed-installation-options) [Screenshot](https://github.com/erew123/alltalk_tts/assets/35898566/548619c8-5f1b-47d0-a73d-54d2fee3f3db)
|
- **DeepSpeed:** A 3-4x performance boost generating TTS. [DeepSpeed Windows/Linux Instructions](https://github.com/erew123/alltalk_tts?tab=readme-ov-file#deepspeed-installation-options) [Screenshot](https://github.com/erew123/alltalk_tts/assets/35898566/548619c8-5f1b-47d0-a73d-54d2fee3f3db)
|
||||||
- **Local/Custom models:** Use any of the XTTSv2 models (API Local and XTTSv2 Local).
|
- **Local/Custom models:** Use any of the XTTSv2 models (API Local and XTTSv2 Local).
|
||||||
- **Optional wav file maintenance:** Configurable deletion of old output wav files. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
- **Optional wav file maintenance:** Configurable deletion of old output wav files. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
||||||
|
- **Finetuning** Train the model specifically on a voice of your choosing for better reproduction.
|
||||||
- **Documentation:** Fully documented with a built in webpage. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
- **Documentation:** Fully documented with a built in webpage. [Screenshot](https://github.com/erew123/alltalk_tts#screenshots)
|
||||||
- **Console output** Clear command line output for any warnings or issues.
|
- **Console output** Clear command line output for any warnings or issues.
|
||||||
- **Standalone/3rd Party support via JSON calls** Can be used with 3rd party applications via JSON calls.
|
- **API Suite and 3rd Party support via JSON calls** Can be used with 3rd party applications via JSON calls.
|
||||||
|
- **Can be run as a standalone app** Not just inside of text-generation-webui.
|
||||||
|
|
||||||
## Index
|
## Index
|
||||||
|
|
||||||
@@ -17,22 +19,26 @@ AllTalk is an updated version of the Coqui_tts extension for Text Generation web
|
|||||||
- 🟪 [Updating & problems with updating](https://github.com/erew123/alltalk_tts?#-updating)
|
- 🟪 [Updating & problems with updating](https://github.com/erew123/alltalk_tts?#-updating)
|
||||||
- 🟫 [Screenshots](https://github.com/erew123/alltalk_tts#-screenshots)
|
- 🟫 [Screenshots](https://github.com/erew123/alltalk_tts#-screenshots)
|
||||||
- 🟨 [Help with problems](https://github.com/erew123/alltalk_tts?#-help-with-problems)
|
- 🟨 [Help with problems](https://github.com/erew123/alltalk_tts?#-help-with-problems)
|
||||||
|
- ⚫ [Finetuning a model](https://github.com/erew123/alltalk_tts?#-finetuning-a-model)
|
||||||
- 🔵🟢🟡 [DeepSpeed Installation (Windows & Linux)](https://github.com/erew123/alltalk_tts?#-deepspeed-installation-options)
|
- 🔵🟢🟡 [DeepSpeed Installation (Windows & Linux)](https://github.com/erew123/alltalk_tts?#-deepspeed-installation-options)
|
||||||
|
- 🟦 [Running AllTalk as a standalone app](https://github.com/erew123/alltalk_tts?#-running-alltalk-as-a-standalone-app)
|
||||||
|
- 🟠 [API Suite and JSON-CURL](https://github.com/erew123/alltalk_tts?#-api-suite-and-json-curl)
|
||||||
- 🔴 [Future to-do list & Upcoming updates](https://github.com/erew123/alltalk_tts?#-future-to-do-list)
|
- 🔴 [Future to-do list & Upcoming updates](https://github.com/erew123/alltalk_tts?#-future-to-do-list)
|
||||||
|
|
||||||
#### Updates
|
## 🔄 Updates
|
||||||
The latest build (13 Dec 2023) has had the entire text filtering engine and narration engine rebuilt from scratch. It's highly complicated how its actually working, but the end result it a much clearer TTS output and much better control over the narrator option and how to handle text that isnt within quotes or asterisks. It does however mean you need to ensure your character card is set up correctly if using the narrator function. Details are below in the installation notes.
|
Please check the below link to find a list of all recent updates and changes.
|
||||||
|
#### 🔄 **Updates list & bug fixes list** can be found [here](https://github.com/erew123/alltalk_tts/issues/25)
|
||||||
DeepSpeed **v11.2** can be installed within the **default text-generation-webui Python 3.11 environment**. Installs in custom Python environments are possible, but can be more complicated. Instructions [here](https://github.com/erew123/alltalk_tts##deepspeed-installation-options) (or scroll down).
|
|
||||||
|
|
||||||
## 🟩 Installation on Text generation web UI
|
## 🟩 Installation on Text generation web UI
|
||||||
This has been tested on the current Dec 2023 release of Text generation webUI. If you have not updated it for a while, you may wish to update Text generation webUI, [instructions here](https://github.com/oobabooga/text-generation-webui?tab=readme-ov-file#how-to-install)
|
This has been tested on the current Dec 2023 release of Text generation webUI. If you have not updated it for a while, you may wish to update Text generation webUI, [instructions here](https://github.com/oobabooga/text-generation-webui?tab=readme-ov-file#how-to-install)
|
||||||
|
|
||||||
|
If you want to watch a video of how to do the below [link here](https://youtu.be/9BPKuwaav5w)
|
||||||
|
|
||||||
1) In a command prompt/terminal window you need to move into your Text generation webUI folder:<br><br>
|
1) In a command prompt/terminal window you need to move into your Text generation webUI folder:<br><br>
|
||||||
`cd text-generation-webui`
|
`cd text-generation-webui`
|
||||||
|
|
||||||
2) Start the Text generation webUI Python environment for your OS:<br><br>
|
2) Start the Text generation webUI Python environment for your OS with whichever **one** of the below is correct for your OS:<br><br>
|
||||||
`cmd_windows.bat`, `./cmd_linux.sh`, `cmd_macos.sh` or `cmd_wsl.bat`
|
`cmd_windows.bat`, `./cmd_linux.sh`, `cmd_macos.sh` or `cmd_wsl.bat`<br><br> Loading Text-generation-webui's Python Environment is $\textcolor{red}{\textsf{VERY IMPORTANT}}$. If you are uncertain what a loaded Python environment looks like, image [here](https://github.com/erew123/alltalk_tts/issues/25#issuecomment-1869344442) and video [here](https://www.youtube.com/watch?v=9BPKuwaav5w)
|
||||||
|
|
||||||
3) Move into your extensions folder:<br><br>
|
3) Move into your extensions folder:<br><br>
|
||||||
`cd extensions`
|
`cd extensions`
|
||||||
@@ -43,17 +49,17 @@ This has been tested on the current Dec 2023 release of Text generation webUI. I
|
|||||||
5) Move into the **alltalk_tts** folder:<br><br>
|
5) Move into the **alltalk_tts** folder:<br><br>
|
||||||
`cd alltalk_tts`
|
`cd alltalk_tts`
|
||||||
|
|
||||||
6) Install the requirements:<br><br>
|
6) Install one of the two requirements files. Whichever one of the two is correct for your machine type:<br><br>
|
||||||
*Nvidia graphics card machines* - `pip install -r requirements_nvidia.txt`<br><br>
|
**Nvidia graphics card machines** - `pip install -r requirements_nvidia.txt`<br><br>
|
||||||
*Other machines (mac, amd etc)* - `pip install -r requirements_other.txt`
|
**Other machines (mac, amd etc)** - `pip install -r requirements_other.txt`
|
||||||
|
|
||||||
7) **(Optional DeepSpeed)** If you have an Nvidia Graphics card on a system running Linux or Windows and wish to use **DeepSpeed** please follow these instructions [here](https://github.com/erew123/alltalk_tts?#-deepspeed-installation-options). **However**, I would highly reccommend before you install DeepSpeed, you start text-generation-webui up, confirm AllTalk starts correctly and everything is working, as DeepSpeed can add another layer of complications troubleshooting any potential start-up issues. If necessary you can `pip uninstall deepspeed`.
|
7) **(Optional DeepSpeed)** If you have an Nvidia Graphics card on a system running Linux or Windows and wish to use **DeepSpeed** please follow these instructions [here](https://github.com/erew123/alltalk_tts?#-deepspeed-installation-options). **However**, I would highly reccommend before you install DeepSpeed, you start text-generation-webui up, confirm AllTalk starts correctly and everything is working, as DeepSpeed can add another layer of complications troubleshooting any potential start-up issues. If necessary you can `pip uninstall deepspeed`.
|
||||||
|
|
||||||
8) You can now start move back to the main Text generation webUI folder `cd ..` (a few times), start Text generation webUI (`start_windows.bat`,`./start_linux.sh`, `start_macos.sh` or `start_wsl.bat`) and load the AllTalk extension in the Text generation webUI **session** tab.
|
8) You can now start move back to the main Text generation webUI folder `cd ..` (a few times), start Text generation webUI with whichever **one** of the startup scripts is correct for your OS (`start_windows.bat`,`./start_linux.sh`, `start_macos.sh` or `start_wsl.bat`) and load the AllTalk extension in the Text generation webUI **session** tab.<br><br> Starting Text-generation-webui with its correct start-up script is $\textcolor{red}{\textsf{VERY IMPORTANT}}$.
|
||||||
|
|
||||||
9) Please read the note below about start-up times and also the note about ensuring your character cards are set up [correctly](https://github.com/erew123/alltalk_tts#the-one-thing-i-cant-easily-work-around)
|
10) Please read the note below about start-up times and also the note about ensuring your character cards are set up [correctly](https://github.com/erew123/alltalk_tts#the-one-thing-i-cant-easily-work-around)
|
||||||
|
|
||||||
10) Some extra voices downloadable [here](https://drive.google.com/file/d/1bYdZdr3L69kmzUN3vSiqZmLRD7-A3M47/view?usp=drive_link)
|
11) Some extra voices downloadable [here](https://drive.google.com/file/d/1bYdZdr3L69kmzUN3vSiqZmLRD7-A3M47/view?usp=drive_link)
|
||||||
|
|
||||||
#### 🟩 Other installation notes
|
#### 🟩 Other installation notes
|
||||||
On first startup, AllTalk will download the Coqui XTTSv2 2.0.2 model to its **models** folder (1.8GB space required). Check the command prompt/terminal window if you want to know what its doing. After it says "Model Loaded" the Text generation webUI is usually available on its IP address a few seconds later, for you to connect to in your browser.
|
On first startup, AllTalk will download the Coqui XTTSv2 2.0.2 model to its **models** folder (1.8GB space required). Check the command prompt/terminal window if you want to know what its doing. After it says "Model Loaded" the Text generation webUI is usually available on its IP address a few seconds later, for you to connect to in your browser.
|
||||||
@@ -63,17 +69,20 @@ Once the extension is loaded, please find all documentation and settings on the
|
|||||||
**Where to find voices** https://aiartes.com/voiceai or https://commons.wikimedia.org/ or interviews on youtube etc. Instructions on how to cut down and prepare a voice sample are within the built in documentation.
|
**Where to find voices** https://aiartes.com/voiceai or https://commons.wikimedia.org/ or interviews on youtube etc. Instructions on how to cut down and prepare a voice sample are within the built in documentation.
|
||||||
|
|
||||||
#### 🟩 The one thing I cant easily work around
|
#### 🟩 The one thing I cant easily work around
|
||||||
Narrator function specific - With a RP chat with your AI, **on your character card** `parameters menu` > `character tab` > `greeting` make sure that anything in there that is the **narrator is in asterisks** and anything **spoken is in double quotes**, then hit the `save` (💾) button. Greeting paragraphs/sentences are handled differently from how the AI sends text and so its difficut to account for them both.
|
Messages intended for the Narrator should be enclosed in asterisks `*` and those for the character inside quotation marks `"`. However, AI systems often deviate from these rules, resulting in text that is neither in quotes nor asterisks. Sometimes, text may appear with only a single asterisk, and AI models may vary their formatting mid-conversation. For example, they might use asterisks initially and then switch to unmarked text. A properly formatted line should look like this:
|
||||||
|
|
||||||
I could force a delimeter in at this stage, but I know it would/may affect things further down the line in the chat and I need a good think about that before just making a change. This issue **only** affects the greeting card/start of conversation and the "example" card that comes with text-generation-webui suffers this issue (if you want to try it for yourself). So you would put double quotes around like this (from the example card):
|
`"`Hey! I'm so excited to finally meet you. I've heard so many great things about you and I'm eager to pick your brain about computers.`"` `*`She walked across the room and picked up her cup of coffee`*`
|
||||||
|
|
||||||
`"`*Hey! I'm so excited to finally meet you. I've heard so many great things about you and I'm eager to pick your brain about computers. I'm sure you have a wealth of knowledge that I can learn from.*`"`
|
Most narrator/character systems switch voices upon encountering an asterisk or quotation marks, which is somewhat effective. AllTalk has undergone several revisions in its sentence splitting and identification methods. While some irregularities and AI deviations in message formatting are inevitable, any line beginning or ending with an asterisk should now be recognized as Narrator dialogue. Lines enclosed in double quotes are identified as Character dialogue. For any other text, you can choose how AllTalk handles it: whether it should be interpreted as Character or Narrator dialogue (most AI systems tend to lean more towards one format when generating text not enclosed in quotes or asterisks).
|
||||||
|
|
||||||
|
With improvements to the splitter/processor, I'm confident it's functioning well. You can monitor what AllTalk identifies as Narrator lines on the command line and adjust its behavior if needed (Text Not Inside - Function).
|
||||||
|
|
||||||
## 🟪 Updating
|
## 🟪 Updating
|
||||||
This is pretty much a repeat of the installation process.
|
This is pretty much a repeat of the installation process.
|
||||||
|
|
||||||
1) In a command prompt/terminal window you need to move into your Text generation webUI folder:<br><br>
|
1) In a command prompt/terminal window you need to move into your Text generation webUI folder:<br><br>
|
||||||
`cd text-generation-webui`
|
`cd text-generation-webui` and start the Python environment for your OS with whichever **one** of the below is correct for your OS:<br><br>
|
||||||
|
`cmd_windows.bat`, `./cmd_linux.sh`, `cmd_macos.sh` or `cmd_wsl.bat`<br>
|
||||||
|
|
||||||
2) Move into your extensions and alltalk_tts folder:<br><br>
|
2) Move into your extensions and alltalk_tts folder:<br><br>
|
||||||
`cd extensions` then `cd alltalk_tts`
|
`cd extensions` then `cd alltalk_tts`
|
||||||
@@ -81,9 +90,9 @@ This is pretty much a repeat of the installation process.
|
|||||||
3) At the command prompt/terminal, type:<br><br>
|
3) At the command prompt/terminal, type:<br><br>
|
||||||
`git pull`
|
`git pull`
|
||||||
|
|
||||||
4) Install the requirements:<br><br>
|
4) Install the correct requirements for your machine:<br><br>
|
||||||
*Nvidia graphics card machines* - `pip install -r requirements_nvidia.txt`<br><br>
|
**Nvidia graphics card machines** - `pip install -r requirements_nvidia.txt`<br><br>
|
||||||
*Other machines (mac, amd etc)* - `pip install -r requirements_other.txt`
|
**Other machines (mac, amd etc)** - `pip install -r requirements_other.txt`
|
||||||
|
|
||||||
#### 🟪 Updating "git pull" error
|
#### 🟪 Updating "git pull" error
|
||||||
|
|
||||||
@@ -95,6 +104,7 @@ I did leave a mistake in the `/extensions/alltalk_tts/.gitignore` file at one po
|
|||||||
voices/*.*
|
voices/*.*
|
||||||
models/*.*
|
models/*.*
|
||||||
outputs/*.*
|
outputs/*.*
|
||||||
|
finetune/*.*
|
||||||
config.json
|
config.json
|
||||||
confignew.json
|
confignew.json
|
||||||
models.json
|
models.json
|
||||||
@@ -123,9 +133,9 @@ This will download a fresh installation.
|
|||||||
3) Move into the **alltalk_tts** folder:<br><br>
|
3) Move into the **alltalk_tts** folder:<br><br>
|
||||||
`cd alltalk_tts`
|
`cd alltalk_tts`
|
||||||
|
|
||||||
4) Install the requirements:<br><br>
|
4) Install the correct requirements for your machine:<br><br>
|
||||||
*Nvidia graphics card machines* - `pip install -r requirements_nvidia.txt`<br><br>
|
**Nvidia graphics card machines** - `pip install -r requirements_nvidia.txt`<br><br>
|
||||||
*Other machines (mac, amd etc)* - `pip install -r requirements_other.txt`
|
**Other machines (mac, amd etc)** - `pip install -r requirements_other.txt`
|
||||||
|
|
||||||
5) Before starting it up, copy/merge the `models`, `voices` and `outputs` folders over from the `alltalk_tts.old` folder to the newly created `alltalk_tts` folder. This will keep your voices history and also stop it re-downloading the model again.
|
5) Before starting it up, copy/merge the `models`, `voices` and `outputs` folders over from the `alltalk_tts.old` folder to the newly created `alltalk_tts` folder. This will keep your voices history and also stop it re-downloading the model again.
|
||||||
|
|
||||||
@@ -135,11 +145,13 @@ Assuming its all working fine and you are happy, you can delete the old alltalk_
|
|||||||
</details>
|
</details>
|
||||||
|
|
||||||
## 🟫 Screenshots
|
## 🟫 Screenshots
|
||||||
| |  |
|
| |  |
|
||||||
|:---:|:---:|
|
|:---:|:---:|
|
||||||
| |  |
|
| |  |
|
||||||
|
| |  |
|
||||||
|
|
||||||
## 🟨 Help with problems
|
## 🟨 Help with problems
|
||||||
|
#### 🔄 **Minor updates/bug fixes list** can be found [here](https://github.com/erew123/alltalk_tts/issues/25)
|
||||||
|
|
||||||
#### 🟨 How to make a diagnostics report file
|
#### 🟨 How to make a diagnostics report file
|
||||||
<details>
|
<details>
|
||||||
@@ -160,7 +172,7 @@ Assuming its all working fine and you are happy, you can delete the old alltalk_
|
|||||||
<br><br></details>
|
<br><br></details>
|
||||||
|
|
||||||
|
|
||||||
#### 🟨 [AllTalk Startup] Warning TTS Subprocess has NOT started up yet, Will keep trying for 60 seconds maximum. Please wait. It times out after 60 seconds.
|
#### 🟨 [AllTalk Startup] Warning TTS Subprocess has NOT started up yet, Will keep trying for 120 seconds maximum. Please wait. It times out after 120 seconds.
|
||||||
|
|
||||||
<details>
|
<details>
|
||||||
<summary>Click to expand</summary><br>
|
<summary>Click to expand</summary><br>
|
||||||
@@ -169,9 +181,11 @@ Assuming its all working fine and you are happy, you can delete the old alltalk_
|
|||||||
**A)** Its trying to load the voice model into your graphics card VRAM (assuming you have a Nvidia Graphics card, otherwise its your system RAM)<br>
|
**A)** Its trying to load the voice model into your graphics card VRAM (assuming you have a Nvidia Graphics card, otherwise its your system RAM)<br>
|
||||||
**B)** Its trying to start up the mini-webserver and send the "ready" signal back to the main process.
|
**B)** Its trying to start up the mini-webserver and send the "ready" signal back to the main process.
|
||||||
|
|
||||||
|
Before giving other possibilities a go, some people with **old machines** are finding their startup times are **very** slow 2-3 minutes. Ive extended the allowed time within the script from 1 minute to 2 minutes. **If you have an older machine** and wish to try extending this further, you can do so by editing `script.py` and changing line 251 `timeout = 120` changing the timeout to a larger value e.g `timeout = 240` (4 minutes).
|
||||||
|
|
||||||
**Note:** If you need to create a support ticket, please create a `diagnostics.log` report file to submit with a support request. Details on doing this are above.
|
**Note:** If you need to create a support ticket, please create a `diagnostics.log` report file to submit with a support request. Details on doing this are above.
|
||||||
|
|
||||||
Possibilities for this issue are:
|
Other possibilities for this issue are:
|
||||||
|
|
||||||
1) You are starting AllTalk in both your `CMD FLAG.txt` and `settings.yaml` file. The `CMD FLAG.txt` you would have manually edited and the `settings.yaml` is the one you change and save in the `session` tab of text-generation-webui and you can `Save UI defaults to settings.yaml`. Please only have one of those two starting up AllTalk.
|
1) You are starting AllTalk in both your `CMD FLAG.txt` and `settings.yaml` file. The `CMD FLAG.txt` you would have manually edited and the `settings.yaml` is the one you change and save in the `session` tab of text-generation-webui and you can `Save UI defaults to settings.yaml`. Please only have one of those two starting up AllTalk.
|
||||||
|
|
||||||
@@ -196,6 +210,19 @@ Possibilities for this issue are:
|
|||||||
11) You have built yourself a custom Python environment and something is funky with it. This is very hard to diagnose as its not a standard environment. You may want to updating text-generation-webui and re installing its requirements file (whichever one you use that comes down with text-generation-webui).
|
11) You have built yourself a custom Python environment and something is funky with it. This is very hard to diagnose as its not a standard environment. You may want to updating text-generation-webui and re installing its requirements file (whichever one you use that comes down with text-generation-webui).
|
||||||
</details>
|
</details>
|
||||||
|
|
||||||
|
#### 🟨 I think AllTalks requirements file has installed something another extension doesn't like
|
||||||
|
<details>
|
||||||
|
<summary>Click to expand</summary><br>
|
||||||
|
|
||||||
|
Ive paid very close attention to **not** impact what Text-generation-webui is requesting on a factory install. This is one of the requirements of submitting an extension to Text-generation-webui. If you want to look at a comparison of a factory fresh text-generation-webui installed packages (with cuda 12.1, though AllTalk's requirements were set on cuda 11.8) you can find that comparison [here](https://github.com/erew123/alltalk_tts/issues/23). This comparison shows that AllTalk is requesting the same package version numbers as Text-generation-webui or even lower version numbers (meaning AllTalk will not update them to a later version). What other extensions do, I cant really account for that.
|
||||||
|
|
||||||
|
I will note that the TTS engine downgrades Pandas data validator to 1.5.3 though its unlikely to cause any issues. You can upgrade it back to text-generation-webui default (december 2023) with `pip install pandas==2.1.4` when inside of the python environment. I have noticed no ill effects from it being a lower or higher version, as far as AllTalk goes. This is also the same behaviour as the Coqui_tts extension that comes with Text-generation-webui.
|
||||||
|
|
||||||
|
Other people are reporting issues with extensions not starting with errors about Pydantic e.g. ```pydantic.errors.PydanticImportError: BaseSettings` has been moved to the pydantic-settings package. See https://docs.pydantic.dev/2.5/migration/#basesettings-has-moved-to-pydantic-settings for more details.```
|
||||||
|
|
||||||
|
Im not sure if the Pydantic version has been recently updated by the Text-generation-webui installer, but this is nothing to do with AllTalk. The other extension you are having an issue with, need to be updated to work with Pydantic 2.5.x. AllTalk was updated in mid december to work with 2.5.x. I am not specifically condoning doing this, as it may have other knock on effects, but within the text-gen Python environment, you can use `pip install pydantic==2.5.0` or `pip install pydantic==1.10.13` to change the version of Pydantic installed.
|
||||||
|
</details>
|
||||||
|
|
||||||
#### 🟨 I activated DeepSpeed in the settings page, but I didnt install DeepSpeed yet and now I have issues starting up
|
#### 🟨 I activated DeepSpeed in the settings page, but I didnt install DeepSpeed yet and now I have issues starting up
|
||||||
<details>
|
<details>
|
||||||
<summary>Click to expand</summary><br>
|
<summary>Click to expand</summary><br>
|
||||||
@@ -230,7 +257,101 @@ I would suggest following [Problems Updating](https://github.com/erew123/alltalk
|
|||||||
As far as I am aware, these are to do with the chrome browser the gradio text-generation-webui in some way. I raised an issue about this on the text-generation-webui [here](https://github.com/oobabooga/text-generation-webui/issues/4788) where you can see that AllTalk is not loaded and the messages persist. Either way, this is more a warning than an actual issue, so shouldnt affect any functionality of either AllTalk or text-generation-webui, they are more just an annoyance.
|
As far as I am aware, these are to do with the chrome browser the gradio text-generation-webui in some way. I raised an issue about this on the text-generation-webui [here](https://github.com/oobabooga/text-generation-webui/issues/4788) where you can see that AllTalk is not loaded and the messages persist. Either way, this is more a warning than an actual issue, so shouldnt affect any functionality of either AllTalk or text-generation-webui, they are more just an annoyance.
|
||||||
</details>
|
</details>
|
||||||
|
|
||||||
|
#### 🟨 I have multiple GPU's and I have problems running Finetuning
|
||||||
|
|
||||||
|
<details>
|
||||||
|
<summary>Click to expand</summary><br>
|
||||||
|
|
||||||
|
Finetuning pulls in various other scripts and some of those scripts can have issues with multiple Nvidia GPU's being present. Until the people that created those other scripts fix up their code, there is a workaround to temporarily tell your system to only use the 1x of your Nvidia GPU's. To do this:
|
||||||
|
|
||||||
|
- **Windows** - You will start the script with `set CUDA_VISIBLE_DEVICES=0 && python finetune.py`<br>
|
||||||
|
After you have completed training, you can reset back with `set CUDA_VISIBLE_DEVICES=`<br>
|
||||||
|
|
||||||
|
- **Linux** - You will start the script with `CUDA_VISIBLE_DEVICES=0 python finetune.py`<br>
|
||||||
|
After you have completed training, you can reset back with `unset CUDA_VISIBLE_DEVICES`<br>
|
||||||
|
|
||||||
|
Rebooting your system will also unset this. The setting is only applied temporarily.
|
||||||
|
|
||||||
|
Depending on which of your Nvidia GPU's is the more powerful one, you can change the `0` to `1` or whichever of your GPU's is the most powerful.
|
||||||
|
|
||||||
|
</details>
|
||||||
|
|
||||||
|
## ⚫ Finetuning a model
|
||||||
|
If you have a voice that the model doesnt quite reproduce correctly, or indeed you just want to improve the reproduced voice, then finetuning is a way to train your "XTTSv2 local" model **(stored in `/alltalk_tts/models/xxxxx/`)** on a specific voice. For this you will need:
|
||||||
|
|
||||||
|
- An Nvidia graphics card. (Please see this [note](https://github.com/erew123/alltalk_tts#-i-have-multiple-gpus-and-i-have-problems-running-finetuning) if you have multiple Nvidia GPU's).
|
||||||
|
- To install a few portions of the Nvidia CUDA 11.8 Toolkit (this will not impact text-generation-webui's cuda setup.
|
||||||
|
- 18GB of disk space free (most of this is used temporarily)
|
||||||
|
- At least 2 minutes of good quality speech from your chosen speaker in mp3, wav or flacc format, in one or more files (have tested as far as 20 minutes worth of audio).
|
||||||
|
|
||||||
|
#### ⚫ How will this work/How complicated is it?
|
||||||
|
Everything has been done to make this as simple as possible. At its simplest, you can literally just download a large chunk of audio from an interview, and tell the finetuning to strip through it, find spoken parts and build your dataset. You can literally click 4 buttons, then copy a few files and you are done. At it's more complicated end you will clean up the audio a little beforehand, but its still only 4x buttons and copying a few files.
|
||||||
|
|
||||||
|
#### ⚫ The audio you will use
|
||||||
|
I would suggest that if its in an interview format, you cut out the interviewer speaking in audacity or your chosen audio editing package. You dont have to worry about being perfect with your cuts, the finetuning Step 1 will go and find spoken audio and cut it out for you. Is there is music over the spoken parts, for best quality you would cut out those parts, though its not 100% necessary. As always, try to avoid bad quality audio with noises in it (humming sounds, hiss etc). You can try something like [Audioenhancer](https://audioenhancer.ai/) to try clean up noisier audio. There is no need to down-sample any of the audio, all of that is handled for you. Just give the finetuning some good quality audio to work with.
|
||||||
|
|
||||||
|
#### ⚫ Important requirements CUDA 11.8
|
||||||
|
As mentioned you must have a small portion of the Nvidia CUDA Toolkit **11.8** installed. Not higher or lower versions. Specifically **11.8**. You do not have to uninstall any other versions, change any graphics drivers, reinstall torch or anything like that. To keep the download+install as small as possible, you will need to:
|
||||||
|
- Download the **xxx (network)** install of the Nvidia Cuda Toolkit 11.8 from [here](https://developer.nvidia.com/cuda-11-8-0-download-archive)
|
||||||
|
- Run the installer. At minimum, you need to [minimally] install the `nvcc` compiler and the `CUBLAS` development and runtime libraries:
|
||||||
|
- Select **Custom Advanced** as your installation type.
|
||||||
|
- Uncheck all the checkboxes in the list.
|
||||||
|
- Now check the following elements:
|
||||||
|
- `CUDA` > `Development` > `Compiler` > `nvcc`
|
||||||
|
- `CUDA` > `Development` > `Compiler` > `Libraries` > `CUBLAS`
|
||||||
|
- `CUDA` > `Runtime` > `Libraries` > `CUBLAS`
|
||||||
|
- You can now proceed through the install.
|
||||||
|
- When that has installed, open a terminal/command prompt and type `nvcc --version`. If it reports back `Cuda compilation tools, release 11.8.` you are good to go. **Specifically, 11.8**. If not continue to the next step.
|
||||||
|
- For both Windows and Linux, you need to ensure that `nvcc` and the 11.8 cuda library files are in your environments search path. You can undo the changes below after finetuning if you prefer:
|
||||||
|
- **Windows**: Edit the Windows `Path` environment variable and add `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin`
|
||||||
|
- **Linux**: The path may be different depending on what flavour of Linux you are running, so you may need to seek out specific instructions on the internet. Generic paths **may** be:
|
||||||
|
- `export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}` and
|
||||||
|
- `export LD_LIBRARY_PATH=/usr/local/cuda-11.8/bin`
|
||||||
|
- Add these to your '~/.bashrc' if you want this to be permanent and not something you have to set each time you open a new terminal.
|
||||||
|
- When you have made the changes, open a **new** terminal/command prompt (in order to load the new search paths) and `nvcc --version`. It should report back `Cuda compilation tools, release 11.8.` at which point, you are good to go.
|
||||||
|
- If it doesn't report that, check you have correctly set the search environment paths, dont have overlapping other versions of cuda paths etc.
|
||||||
|
|
||||||
|
#### ⚫ Starting Finetuning
|
||||||
|
**NOTE:** Please make sure you have started AllTalk at least once after updating, so that it downloads the additional files needed for finetuning.
|
||||||
|
|
||||||
|
1) Close all other applications that are using your GPU/VRAM and copy your audio samples into:<br><br>
|
||||||
|
`/alltalk_tts/finetune/put-voice-samples-in-here/`
|
||||||
|
3) In a command prompt/terminal window you need to move into your Text generation webUI folder:<br><br>
|
||||||
|
`cd text-generation-webui`
|
||||||
|
|
||||||
|
4) Start the Text generation webUI Python environment for your OS:<br><br>
|
||||||
|
`cmd_windows.bat`, `./cmd_linux.sh`, `cmd_macos.sh` or `cmd_wsl.bat`
|
||||||
|
|
||||||
|
5) You can double check your search path environment still works correctly with `nvcc --version`. It should report back 11.8:<br><br>
|
||||||
|
`Cuda compilation tools, release 11.8.`
|
||||||
|
|
||||||
|
7) Move into your extensions folder:<br><br>
|
||||||
|
`cd extensions`
|
||||||
|
|
||||||
|
8) Move into the **alltalk_tts** folder:<br><br>
|
||||||
|
`cd alltalk_tts`
|
||||||
|
|
||||||
|
9) Install the finetune requirements file: `pip install -r requirements_finetune.txt`
|
||||||
|
|
||||||
|
10) Type `python finetune.py` and it should start up.
|
||||||
|
11) Follow the on-screen instructions when the web interface starts up.
|
||||||
|
12) When you have finished finetuning, the final tab will tell you what to do with your files and how to move your newly trained model to the correct location on disk.
|
||||||
|
|
||||||
|
#### ⚫ Using a Finetuned model in Text-generation-webui
|
||||||
|
|
||||||
|
At the end of the finetune process, you will have an option to `Compact and move model to /trainedmodel/` this will compact the raw training file and move it to `/model/trainedmodel/`. When AllTalk starts up within Text-generation-webui, if it finds a model in this location a new loader will appear in the interface for `XTTSv2 FT` and you can use this to load your finetuned model. <br><br>**Be careful** not to train a new model from the base model, then overwrite your current `/model/trainedmodel/` **if** you want a seperately trained model. This is why there is an `OPTION B` to move your just trained model to `/models/lastfinetuned/`.
|
||||||
|
|
||||||
|
#### ⚫ Training one model with multiple voices
|
||||||
|
|
||||||
|
At the end of the finetune process, you will have an option to `Compact and move model to /trainedmodel/` this will compact the raw training file and move it to `/model/trainedmodel/`. This model will become available when you start up finetuning. You will have a choice to train the Base Model or the `Existing finetuned model` (which is the one in `/model/trainedmodel/`). So you can use this to keep further training this model with additional voices, then copying it back to `/model/trainedmodel/` at the end of training.
|
||||||
|
|
||||||
|
#### ⚫ Do I need to keep the raw training data/model?
|
||||||
|
|
||||||
|
If you've compacted and moved your model, its highly unlikely you would want to keep that data, however the choice is there to keep it if you wish. It will be between 5-10GB in size, so most people will want to delete it.
|
||||||
|
|
||||||
## 🔵🟢🟡 DeepSpeed Installation Options
|
## 🔵🟢🟡 DeepSpeed Installation Options
|
||||||
|
**NOTE**: You **DO NOT** need to set Text-generation-webUI's **--deepspeed** setting for AllTalk to be able to use DeepSpeed. These are two completely separate things and incorrectly setting that on Text-generation-webUI may cause other complications.
|
||||||
|
|
||||||
#### 🔵 Linux Installation
|
#### 🔵 Linux Installation
|
||||||
<details>
|
<details>
|
||||||
<summary>Click to expand: Linux DeepSpeed installation</summary>
|
<summary>Click to expand: Linux DeepSpeed installation</summary>
|
||||||
@@ -381,10 +502,207 @@ def get_shm_size():
|
|||||||
**Removal** - If it became necessary to uninstall DeepSpeed, you can do so with `cmd_windows.bat` and then `pip uninstall deepspeed`<br><br>
|
**Removal** - If it became necessary to uninstall DeepSpeed, you can do so with `cmd_windows.bat` and then `pip uninstall deepspeed`<br><br>
|
||||||
</details>
|
</details>
|
||||||
|
|
||||||
|
## 🟦 Running AllTalk as a standalone app
|
||||||
|
AllTalk will run as a standalone app, as long as you install its requirements files into whatever Python environment you are using. You can follow the steps to install the AllTalk's requirements into whatever Python environment you wish. Because I dont know what Python environment you are wanting to use, I can only give you a loose set of installation instructions.
|
||||||
|
|
||||||
|
Please note, at time of writing, the TTS engine requires Python **3.9.x** to **3.11.x** [TTS Engine details here](https://pypi.org/project/TTS/). AllTalk and its requirements are tested on Python 3.11.x.
|
||||||
|
|
||||||
|
#### 🟦 (Option 1) I already have AllTalk installed as an extension of Text-generation-webui
|
||||||
|
If you already have AllTalk as a extension of Text-generation-webui, and wish to run it as standalone, load Text-generation-webui's Python environment `cmd_windows.bat`, `./cmd_linux.sh`, `cmd_macos.sh` or `cmd_wsl.bat`, move into the AllTalk folder `cd extensions` > `cd alltalk_tts` and start AllTalk with `python script.py`. There is nothing beyond this you would need to do.
|
||||||
|
|
||||||
|
#### 🟦 (Option 2) I wish to do a custom install of AllTalk
|
||||||
|
1) In your chosen disk location git clone this repository:<br><br>
|
||||||
|
`git clone https://github.com/erew123/alltalk_tts`
|
||||||
|
|
||||||
|
2) Move into the **alltalk_tts** folder:<br><br>
|
||||||
|
`cd alltalk_tts`
|
||||||
|
|
||||||
|
3) If you have a custom Python environment, now is the time to load it!
|
||||||
|
|
||||||
|
4) Install one of the two requirements files. Whichever one of the two is correct for your machine type:<br><br>
|
||||||
|
**Nvidia graphics card machines** - `pip install -r requirements_nvidia.txt`<br><br>
|
||||||
|
**Other machines (mac, amd etc)** - `pip install -r requirements_other.txt`
|
||||||
|
|
||||||
|
5) Start AllTalk with `python script.py`
|
||||||
|
|
||||||
|
Deepspeed and other such things can be installed. Please read the relevant instructions for those items, however, make the relevant changes to load your correct Python environment when installing any requirements files and starting AllTalk.
|
||||||
|
|
||||||
|
## 🟠 API Suite and JSON-CURL
|
||||||
|
### 🟠Overview
|
||||||
|
The Text-to-Speech (TTS) Generation API allows you to generate speech from text input using various configuration options. This API supports both character and narrator voices, providing flexibility for creating dynamic and engaging audio content.
|
||||||
|
|
||||||
|
#### 🟠 Ready Endpoint<br>
|
||||||
|
Check if the Text-to-Speech (TTS) service is ready to accept requests.
|
||||||
|
|
||||||
|
- URL: `http://127.0.0.1:7851/api/ready`<br> - Method: `GET`<br>
|
||||||
|
|
||||||
|
`curl -X GET "http://127.0.0.1:7851/api/ready"`
|
||||||
|
|
||||||
|
Response: `Ready`
|
||||||
|
|
||||||
|
#### 🟠 Voices List Endpoint<br>
|
||||||
|
Retrieve a list of available voices for generating speech.
|
||||||
|
|
||||||
|
- URL: `http://127.0.0.1:7851/api/voices`<br> - Method: `GET`<br>
|
||||||
|
|
||||||
|
`curl -X GET "http://127.0.0.1:7851/api/voices"`
|
||||||
|
|
||||||
|
JSON return: `{"voices": ["voice1.wav", "voice2.wav", "voice3.wav"]}`
|
||||||
|
|
||||||
|
#### 🟠 Preview Voice Endpoint
|
||||||
|
Generate a preview of a specified voice with hardcoded settings.
|
||||||
|
|
||||||
|
- URL: `http://127.0.0.1:7851/api/previewvoice/`<br> - Method: `POST`<br> - Content-Type: `application/x-www-form-urlencoded`<br>
|
||||||
|
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/previewvoice/" -F "voice=female_01.wav"`
|
||||||
|
|
||||||
|
Replace `female_01.wav` with the name of the voice sample you want to hear.
|
||||||
|
|
||||||
|
JSON return: `{"status": "generate-success", "output_file_path": "/path/to/outputs/api_preview_voice.wav", "output_file_url": "http://127.0.0.1:7851/audio/api_preview_voice.wav"}`
|
||||||
|
|
||||||
|
#### 🟠 Switching Model Endpoint<br>
|
||||||
|
|
||||||
|
- URL: `http://127.0.0.1:7851/api/reload`<br> - Method: `POST`<br><br>
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/reload?tts_method=API%20Local"`<br>
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/reload?tts_method=API%20TTS"`<br>
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/reload?tts_method=XTTSv2%20Local"`<br>
|
||||||
|
|
||||||
|
Switch between the 3 models respectively.
|
||||||
|
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/reload?tts_method=XTTSv2%20FT"`<br>
|
||||||
|
|
||||||
|
If you have a finetuned model in `/models/trainedmodel/` (will error otherwise)
|
||||||
|
|
||||||
|
JSON return `{"status": "model-success"}`
|
||||||
|
|
||||||
|
#### 🟠 Switch DeepSpeed Endpoint<br>
|
||||||
|
|
||||||
|
- URL: `http://127.0.0.1:7851/api/deepspeed`<br> - Method: `POST`<br><br>
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/deepspeed?new_deepspeed_value=True"`
|
||||||
|
|
||||||
|
Replace True with False to disable DeepSpeed mode.
|
||||||
|
|
||||||
|
JSON return `{"status": "deepspeed-success"}`
|
||||||
|
|
||||||
|
#### 🟠 Switching Low VRAM Endpoint<br>
|
||||||
|
|
||||||
|
- URL: `http://127.0.0.1:7851/api/lowvramsetting`<br> - Method: `POST`<br><br>
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/lowvramsetting?new_low_vram_value=True"`
|
||||||
|
|
||||||
|
Replace True with False to disable Low VRAM mode.
|
||||||
|
|
||||||
|
JSON return `{"status": "lowvram-success"}`
|
||||||
|
|
||||||
|
### 🟠 TTS Generation Endpoint
|
||||||
|
|
||||||
|
- URL: `http://127.0.0.1:7851/api/tts-generate`<br> - Method: `POST`<br> - Content-Type: `application/x-www-form-urlencoded`<br>
|
||||||
|
|
||||||
|
### 🟠 Example command lines
|
||||||
|
Standard TTS speech Example (standard text) generating a time-stamped file<br>
|
||||||
|
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/tts-generate" -d "text_input=All of this is text spoken by the character. This is text not inside quotes, though that doesnt matter in the slightest" -d "text_filtering=standard" -d "character_voice_gen=female_01.wav" -d "narrator_enabled=false" -d "narrator_voice_gen=male_01.wav" -d "text_not_inside=character" -d "language=en" -d "output_file_name=myoutputfile" -d "output_file_timestamp=true" -d "autoplay=true" -d "autoplay_volume=0.8"`<br>
|
||||||
|
|
||||||
|
Narrator Example (standard text) generating a time-stamped file
|
||||||
|
|
||||||
|
`curl -X POST "http://127.0.0.1:7851/api/tts-generate" -d "text_input=*This is text spoken by the narrator* \"This is text spoken by the character\". This is text not inside quotes." -d "text_filtering=standard" -d "character_voice_gen=female_01.wav" -d "narrator_enabled=true" -d "narrator_voice_gen=male_01.wav" -d "text_not_inside=character" -d "language=en" -d "output_file_name=myoutputfile" -d "output_file_timestamp=true" -d "autoplay=true" -d "autoplay_volume=0.8"`<br>
|
||||||
|
|
||||||
|
Note that if your text that needs to be generated contains double quotes you will need to escape them with `\"` (Please see the narrator example).
|
||||||
|
|
||||||
|
### 🟠 Request Parameters
|
||||||
|
🟠 **text_input**: The text you want the TTS engine to produce. Use escaped double quotes for character speech and asterisks for narrator speech if using the narrator function. Example:
|
||||||
|
|
||||||
|
`-d "text_input=*This is text spoken by the narrator* \"This is text spoken by the character\". This is text not inside quotes."`
|
||||||
|
|
||||||
|
🟠 **text_filtering**: Filter for text. Options:
|
||||||
|
|
||||||
|
- **none** No filtering. Whatever is sent will go over to the TTS engine as raw text, which may result in some odd sounds with some special characters.<br>
|
||||||
|
- **standard** Human-readable text and a basic level of filtering, just to clean up some special characters.<br>
|
||||||
|
- **html** HTML content. Where you are using HTML entity's like "<br>
|
||||||
|
|
||||||
|
`-d "text_filtering=none"`<br>
|
||||||
|
`-d "text_filtering=standard"`<br>
|
||||||
|
`-d "text_filtering=html"`<br>
|
||||||
|
|
||||||
|
Example:
|
||||||
|
|
||||||
|
- **Standard Example**: `*This is text spoken by the narrator* "This is text spoken by the character" This is text not inside quotes.`<br>
|
||||||
|
- **HTML Example**: `*This is text spoken by the narrator* "This is text spoken by the character" This is text not inside quotes.`<br>
|
||||||
|
- **None**: `Will just pass whatever characters/text you send at it.`<br>
|
||||||
|
|
||||||
|
🟠 **character_voice_gen**: The WAV file name for the character's voice.<br>
|
||||||
|
|
||||||
|
`-d "character_voice_gen=female_01.wav"`
|
||||||
|
|
||||||
|
🟠 **narrator_enabled**: Enable or disable the narrator function. If true, minimum text filtering is set to standard. Anything between double quotes is considered the character's speech, and anything between asterisks is considered the narrator's speech.
|
||||||
|
|
||||||
|
`-d "narrator_enabled=true"`<br>
|
||||||
|
`-d "narrator_enabled=false"`
|
||||||
|
|
||||||
|
🟠 **narrator_voice_gen**: The WAV file name for the narrator's voice.
|
||||||
|
|
||||||
|
`-d "narrator_voice_gen=male_01.wav"`
|
||||||
|
|
||||||
|
🟠 **text_not_inside**: Specify the handling of lines not inside double quotes or asterisks, for the narrator feature. Options:
|
||||||
|
|
||||||
|
- **character**: Treat as character speech.<br>
|
||||||
|
- **narrator**: Treat as narrator speech.<br>
|
||||||
|
|
||||||
|
`-d "text_not_inside=character"`<br>
|
||||||
|
`-d "text_not_inside=narrator"`
|
||||||
|
|
||||||
|
🟠 **language**: Choose the language for TTS. Options:
|
||||||
|
|
||||||
|
`ar` Arabic<br>
|
||||||
|
`zh-cn` Chinese (Simplified)<br>
|
||||||
|
`cs` Czech<br>
|
||||||
|
`nl` Dutch<br>
|
||||||
|
`en` English<br>
|
||||||
|
`fr` French<br>
|
||||||
|
`de` German<br>
|
||||||
|
`hu` Hungarian<br>
|
||||||
|
`it` Italian<br>
|
||||||
|
`ja` Japanese<br>
|
||||||
|
`ko` Korean<br>
|
||||||
|
`pl` Polish<br>
|
||||||
|
`pt` Portuguese<br>
|
||||||
|
`ru` Russian<br>
|
||||||
|
`es` Spanish<br>
|
||||||
|
`tr` Turkish<br>
|
||||||
|
|
||||||
|
`-d "language=en"`<br>
|
||||||
|
|
||||||
|
🟠 **output_file_name**: The name of the output file (excluding the .wav extension).
|
||||||
|
|
||||||
|
`-d "output_file_name=myoutputfile"`<br>
|
||||||
|
|
||||||
|
🟠 **output_file_timestamp**: Add a timestamp to the output file name. If true, each file will have a unique timestamp; otherwise, the same file name will be overwritten each time you generate TTS.
|
||||||
|
|
||||||
|
`-d "output_file_timestamp=true"`<br>
|
||||||
|
`-d "output_file_timestamp=false"`
|
||||||
|
|
||||||
|
🟠 **autoplay**: Enable or disable playing the generated TTS to your standard sound output device at time of TTS generation.
|
||||||
|
|
||||||
|
`-d "autoplay=true"`<br>
|
||||||
|
`-d "autoplay=false"`
|
||||||
|
|
||||||
|
🟠 **autoplay_volume**: Set the autoplay volume. Should be between 0.1 and 1.0. Needs to be specified in the JSON request even if autoplay is false.
|
||||||
|
|
||||||
|
`-d "autoplay_volume=0.8"`
|
||||||
|
|
||||||
|
### 🟠 TTS Generation Response
|
||||||
|
The API returns a JSON object with the following properties:
|
||||||
|
|
||||||
|
- **status** Indicates whether the generation was successful (generate-success) or failed (generate-failure).<br>
|
||||||
|
- **output_file_path** The on-disk location of the generated WAV file.<br>
|
||||||
|
- **output_file_url** The HTTP location for accessing the generated WAV file for browser playback.<br>
|
||||||
|
- **output_cache_url** The HTTP location for accessing the generated WAV file as a pushed download.<br>
|
||||||
|
|
||||||
|
Example JSON TTS Generation Response:
|
||||||
|
|
||||||
|
`{"status":"generate-success","output_file_path":"C:\\text-generation-webui\\extensions\\alltalk_tts\\outputs\\myoutputfile_1704141936.wav","output_file_url":"http://127.0.0.1:7851/audio/myoutputfile_1704141936.wav","output_cache_url":"http://127.0.0.1:7851/audiocache/myoutputfile_1704141936.wav"}`
|
||||||
|
|
||||||
### 🔴 Future to-do list
|
### 🔴 Future to-do list
|
||||||
- Complete & document the new/full standalone mode API.
|
|
||||||
- Voice output within the command prompt/terminal (TBD).
|
- Voice output within the command prompt/terminal (TBD).
|
||||||
- Correct an issue on incorrect output folder path when running as a standalone app.
|
- Correct a few spelling mistakes in the documentation.
|
||||||
- Correct a few spelling mistakes in the documnentation.
|
|
||||||
- Possibly add some additional TTS engines (TBD).
|
- Possibly add some additional TTS engines (TBD).
|
||||||
- Have a break!
|
- Have a break!
|
||||||
+39
-6
@@ -6,6 +6,7 @@ try:
|
|||||||
import torch
|
import torch
|
||||||
import os # Import the os module
|
import os # Import the os module
|
||||||
import re
|
import re
|
||||||
|
import sys
|
||||||
import glob
|
import glob
|
||||||
import textwrap
|
import textwrap
|
||||||
import packaging.version
|
import packaging.version
|
||||||
@@ -17,7 +18,7 @@ except ImportError as e:
|
|||||||
print(f"\033[91mError importing module: {e}\033[0m\n")
|
print(f"\033[91mError importing module: {e}\033[0m\n")
|
||||||
print("\033[94mPlease ensure you started the Text-generation-webUI Python environment with either\033[0m")
|
print("\033[94mPlease ensure you started the Text-generation-webUI Python environment with either\033[0m")
|
||||||
print("\033[92mcmd_linux.sh\033[0m, \033[92mcmd_windows.bat\033[0m, \033[92mcmd_macos.sh\033[0m, or \033[92mcmd_wsl.bat\033[0m")
|
print("\033[92mcmd_linux.sh\033[0m, \033[92mcmd_windows.bat\033[0m, \033[92mcmd_macos.sh\033[0m, or \033[92mcmd_wsl.bat\033[0m")
|
||||||
print("\033[94mand then try running the diagnostics again.\033[0m")
|
print("\033[94mfrom the text-generation-webui directory, and then try running the diagnostics again.\033[0m")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
@@ -110,6 +111,18 @@ def log_system_info():
|
|||||||
# Package versions using importlib_metadata
|
# Package versions using importlib_metadata
|
||||||
package_versions = {d.metadata['Name']: d.version for d in distributions()}
|
package_versions = {d.metadata['Name']: d.version for d in distributions()}
|
||||||
|
|
||||||
|
# Python environment information
|
||||||
|
python_executable = sys.executable
|
||||||
|
python_version_info = sys.version_info
|
||||||
|
python_virtual_env = os.environ.get('VIRTUAL_ENV', 'N/A')
|
||||||
|
|
||||||
|
# Conda environment information
|
||||||
|
conda_env = os.environ.get('CONDA_DEFAULT_ENV', 'N/A')
|
||||||
|
|
||||||
|
# Get Path environment information
|
||||||
|
search_path = sys.path
|
||||||
|
path_env = os.environ.get('PATH', 'N/A')
|
||||||
|
|
||||||
# Compare with requirements file
|
# Compare with requirements file
|
||||||
requirements_file = get_requirements_file()
|
requirements_file = get_requirements_file()
|
||||||
if requirements_file:
|
if requirements_file:
|
||||||
@@ -134,14 +147,24 @@ def log_system_info():
|
|||||||
|
|
||||||
# Log and print information
|
# Log and print information
|
||||||
logging.info(f"OS Version: {os_version}")
|
logging.info(f"OS Version: {os_version}")
|
||||||
logging.info(f"Note: Windows 11 build is 10.x.22xxx")
|
logging.info(f"Note: Windows 11 will list as build is 10.x.22xxx")
|
||||||
logging.info(f"Python Version: {python_version}")
|
|
||||||
logging.info(f"Torch Version: {torch_version}")
|
logging.info(f"Torch Version: {torch_version}")
|
||||||
logging.info(f"System RAM: {system_ram}")
|
logging.info(f"System RAM: {system_ram}")
|
||||||
logging.info(f"CUDA_HOME: {cuda_home}")
|
logging.info(f"CUDA_HOME: {cuda_home}")
|
||||||
logging.info(f"Port Status: {port_status}")
|
logging.info(f"Port Status: {port_status}")
|
||||||
|
logging.info(f"Python Version: {platform.python_version()}")
|
||||||
|
logging.info(f"Python Version Info: {python_version_info}")
|
||||||
|
logging.info(f"Python Executable: {python_executable}")
|
||||||
|
logging.info(f"Python Virtual Environment: {python_virtual_env} (Should be N/A when in Text-generation-webui Conda Python environment)")
|
||||||
|
logging.info(f"Conda Environment: {conda_env}")
|
||||||
|
logging.info("\nPython Search Path:")
|
||||||
|
for path in search_path:
|
||||||
|
logging.info(f" {path}")
|
||||||
|
logging.info("\nOS PATH Environment Variable:")
|
||||||
|
for path in path_env.split(';'):
|
||||||
|
logging.info(f" {path}")
|
||||||
if required_packages: # Check if the dictionary is not empty
|
if required_packages: # Check if the dictionary is not empty
|
||||||
logging.info("Package Versions:")
|
logging.info("\nPackage Versions:")
|
||||||
max_package_length = max(len(package) for package in required_packages.keys())
|
max_package_length = max(len(package) for package in required_packages.keys())
|
||||||
for package_name, (operator, required_version) in required_packages.items():
|
for package_name, (operator, required_version) in required_packages.items():
|
||||||
installed_version = installed_packages.get(package_name, 'Not installed')
|
installed_version = installed_packages.get(package_name, 'Not installed')
|
||||||
@@ -153,12 +176,22 @@ def log_system_info():
|
|||||||
|
|
||||||
# Print to screen
|
# Print to screen
|
||||||
print(f"\n\033[94mOS Version:\033[0m \033[92m{os_version}\033[0m")
|
print(f"\n\033[94mOS Version:\033[0m \033[92m{os_version}\033[0m")
|
||||||
print(f"\033[94mOS Ver note:\033[0m \033[92mWindows 11 build is 10.x.22xxx\033[0m")
|
print(f"\033[94mOS Ver note:\033[0m \033[92m(Windows 11 will say build is 10.x.22xxx)\033[0m")
|
||||||
print(f"\033[94mCUDA_HOME:\033[0m \033[92m{cuda_home}\033[0m")
|
print(f"\033[94mCUDA_HOME:\033[0m \033[92m{cuda_home}\033[0m")
|
||||||
print(f"\033[94mSystem RAM:\033[0m \033[92m{system_ram}\033[0m")
|
print(f"\033[94mSystem RAM:\033[0m \033[92m{system_ram}\033[0m")
|
||||||
print(f"\033[94mPort Status:\033[0m \033[92m{port_status}\033[0m")
|
print(f"\033[94mPort Status:\033[0m \033[92m{port_status}\033[0m")
|
||||||
print(f"\033[94mTorch Version:\033[0m \033[92m{torch_version}\033[0m")
|
print(f"\033[94mTorch Version:\033[0m \033[92m{torch_version}\033[0m")
|
||||||
print(f"\033[94mPython Version:\033[0m \033[92m{python_version}\033[0m")
|
print(f"\033[94mPython Version:\033[0m \033[92m{platform.python_version()}\033[0m")
|
||||||
|
print(f"\033[94mPython Version Info:\033[0m \033[92m{python_version_info}\033[0m")
|
||||||
|
print(f"\033[94mPython Executable:\033[0m \033[92m{python_executable}\033[0m")
|
||||||
|
print(f"\033[94mPython Virtual Environment:\033[0m \033[92m{python_virtual_env}\033[0m (Should be N/A when in Text-generation-webui Conda Python environment)")
|
||||||
|
print(f"\033[94mConda Environment:\033[0m \033[92m{conda_env}\033[0m")
|
||||||
|
print(f"\n\033[94mPython Search Path:\033[0m")
|
||||||
|
for path in search_path:
|
||||||
|
print(f" {path}")
|
||||||
|
print(f"\n\033[94mOS Search PATH Environment Variable:\033[0m")
|
||||||
|
for path in path_env.split(';'):
|
||||||
|
print(f" {path}")
|
||||||
if required_packages: # Check if the dictionary is not empty
|
if required_packages: # Check if the dictionary is not empty
|
||||||
print("\033[94m\nRequirements file package comparison:\033[0m")
|
print("\033[94m\nRequirements file package comparison:\033[0m")
|
||||||
max_package_length = max(len(package) for package in required_packages.keys())
|
max_package_length = max(len(package) for package in required_packages.keys())
|
||||||
|
|||||||
+1349
File diff suppressed because it is too large.
Load diff
@@ -0,0 +1,26 @@
|
|||||||
|
import torch
|
||||||
|
import os
|
||||||
|
MODEL_PATH = "best_model.pth"
|
||||||
|
# Check if best_model.pth exists
|
||||||
|
if not os.path.exists(MODEL_PATH):
|
||||||
|
print(f"\nCannot find \033[94m{MODEL_PATH}\033[0m in the current folder.")
|
||||||
|
|
||||||
|
# Check if model.pth exists as an alternative
|
||||||
|
ALTERNATIVE_PATH = "model.pth"
|
||||||
|
if os.path.exists(ALTERNATIVE_PATH):
|
||||||
|
print(f"\nIf your model is currently called \033[93m{ALTERNATIVE_PATH}\033[0m please rename it to \033[94mbest_model.pth\033[0m.")
|
||||||
|
else:
|
||||||
|
print(f"Please put \033[91mbest_model.pth\033[0m in the current folder or rename your model file to \033[94mbest_model.pth\033[0m.")
|
||||||
|
exit(1)
|
||||||
|
# Attempt to load the model
|
||||||
|
try:
|
||||||
|
checkpoint = torch.load(MODEL_PATH, map_location=torch.device("cpu"))
|
||||||
|
except Exception as e:
|
||||||
|
print("Error loading checkpoint:", e)
|
||||||
|
raise
|
||||||
|
del checkpoint["optimizer"]
|
||||||
|
for key in list(checkpoint["model"].keys()):
|
||||||
|
if "dvae" in key:
|
||||||
|
del checkpoint["model"][key]
|
||||||
|
torch.save(checkpoint, "model.pth")
|
||||||
|
print(f"\nYour compacted model is created \033[93mmodel.pth\033[0m.")
|
||||||
Whitespace-only changes.
+4
-1
@@ -6,6 +6,9 @@
|
|||||||
"README.md": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/README.md?download=true",
|
"README.md": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/README.md?download=true",
|
||||||
"config.json": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/config.json?download=true",
|
"config.json": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/config.json?download=true",
|
||||||
"model.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/model.pth?download=true",
|
"model.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/model.pth?download=true",
|
||||||
|
"dvae.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/dvae.pth?download=true",
|
||||||
|
"mel_stats.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/mel_stats.pth?download=true",
|
||||||
|
"speakers_xtts.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/speakers_xtts.pth?download=true",
|
||||||
"vocab.json": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/vocab.json?download=true"
|
"vocab.json": "https://huggingface.co/coqui/XTTS-v2/resolve/v2.0.2/vocab.json?download=true"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
gradio>=3.50.2
|
||||||
|
torch>=2.1.0+cu118
|
||||||
|
torchaudio>=2.1.0+cu118
|
||||||
|
TTS==0.21.3
|
||||||
|
faster-whisper>=0.10.0
|
||||||
|
tqdm>=4.66.0
|
||||||
|
pandas>=1.5.0
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
numpy>=1.22.0
|
numpy>=1.22.0
|
||||||
soundfile>=0.12.1
|
soundfile>=0.12.1
|
||||||
uvicorn>=0.24.0.post1
|
uvicorn>=0.24.0.post1
|
||||||
TTS>=0.22.0
|
TTS>=0.21.3
|
||||||
torch>=2.1.0+cu118
|
torch>=2.1.0+cu118
|
||||||
torchaudio>=2.1.0+cu118
|
torchaudio>=2.1.0+cu118
|
||||||
fastapi>=0.104.1
|
fastapi>=0.104.1
|
||||||
@@ -10,3 +10,6 @@ requests>=2.31.0
|
|||||||
tqdm>=4.66.1
|
tqdm>=4.66.1
|
||||||
importlib-metadata>=4.8.1
|
importlib-metadata>=4.8.1
|
||||||
packaging>=23.2
|
packaging>=23.2
|
||||||
|
pydantic>=1.10.13
|
||||||
|
sounddevice>=0.4.6
|
||||||
|
python-multipart>=0.0.6
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
numpy>=1.22.0
|
numpy>=1.22.0
|
||||||
soundfile>=0.12.1
|
soundfile>=0.12.1
|
||||||
uvicorn>=0.24.0.post1
|
uvicorn>=0.24.0.post1
|
||||||
TTS>=0.22.0
|
TTS>=0.21.3
|
||||||
torch>=2.1.0
|
torch>=2.1.0
|
||||||
torchaudio>=2.1.0
|
torchaudio>=2.1.0
|
||||||
fastapi>=0.104.1
|
fastapi>=0.104.1
|
||||||
@@ -10,3 +10,6 @@ requests>=2.31.0
|
|||||||
tqdm>=4.66.1
|
tqdm>=4.66.1
|
||||||
importlib-metadata>=4.8.1
|
importlib-metadata>=4.8.1
|
||||||
packaging>=23.2
|
packaging>=23.2
|
||||||
|
pydantic>=1.10.13
|
||||||
|
sounddevice>=0.4.6
|
||||||
|
python-multipart>=0.0.6
|
||||||
@@ -39,8 +39,8 @@ try:
|
|||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
logger.error(
|
logger.error(
|
||||||
f"[{params['branding']}]\033[91mWarning\033[0m Could not find the TTS module. Make sure to install the requirements for the {params['branding']} extension."
|
f"[{params['branding']}]\033[91mWarning\033[0m Could not find the TTS module. Make sure to install the requirements for the {params['branding']} extension."
|
||||||
f"[{params['branding']}]\033[91mWarning\033[0m Linux / Mac:\npip install -r extensions/alltalk_tts/requirements.txt\n"
|
f"[{params['branding']}]\033[91mWarning\033[0m Linux / Mac:\npip install -r /alltalk_tts/requirements.txt\n"
|
||||||
f"[{params['branding']}]\033[91mWarning\033[0m Windows:\npip install -r extensions\\alltalk_tts\\requirements.txt\n"
|
f"[{params['branding']}]\033[91mWarning\033[0m Windows:\npip install -r \\alltalk_tts\\requirements.txt\n"
|
||||||
f"[{params['branding']}]\033[91mWarning\033[0m If you used the one-click installer, paste the command above in the terminal window launched after running the cmd_ script. On Windows, that's cmd_windows.bat."
|
f"[{params['branding']}]\033[91mWarning\033[0m If you used the one-click installer, paste the command above in the terminal window launched after running the cmd_ script. On Windows, that's cmd_windows.bat."
|
||||||
)
|
)
|
||||||
raise
|
raise
|
||||||
@@ -62,7 +62,9 @@ try:
|
|||||||
from modules.utils import gradio
|
from modules.utils import gradio
|
||||||
# This is set to check if the script is being run within text generation webui or as a standalone script. False is running as part of text gen web ui or a gradio interface
|
# This is set to check if the script is being run within text generation webui or as a standalone script. False is running as part of text gen web ui or a gradio interface
|
||||||
running_in_standalone = False
|
running_in_standalone = False
|
||||||
|
output_folder_wav = params["output_folder_wav"]
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
|
output_folder_wav = params["output_folder_wav_standalone"]
|
||||||
print(f"[{params['branding']}Startup] Running script.py in standalone mode")
|
print(f"[{params['branding']}Startup] Running script.py in standalone mode")
|
||||||
# This is set to check if the script is being run within text generation webui or as a standalone script. true means standalone
|
# This is set to check if the script is being run within text generation webui or as a standalone script. true means standalone
|
||||||
running_in_standalone = True
|
running_in_standalone = True
|
||||||
@@ -75,6 +77,8 @@ with open(this_dir / "languages.json", encoding="utf8") as f:
|
|||||||
languages = json.load(f)
|
languages = json.load(f)
|
||||||
# Create a global lock
|
# Create a global lock
|
||||||
process_lock = threading.Lock()
|
process_lock = threading.Lock()
|
||||||
|
# Base setting for a possible FineTuned model existing and the loader being available
|
||||||
|
tts_method_xtts_ft = False
|
||||||
|
|
||||||
# Gather the voice files
|
# Gather the voice files
|
||||||
def get_available_voices():
|
def get_available_voices():
|
||||||
@@ -108,7 +112,6 @@ def delete_old_files(folder_path, days_to_keep):
|
|||||||
|
|
||||||
# Extract settings using params dictionary
|
# Extract settings using params dictionary
|
||||||
delete_output_wavs_setting = params["delete_output_wavs"]
|
delete_output_wavs_setting = params["delete_output_wavs"]
|
||||||
output_folder_wav = params["output_folder_wav"]
|
|
||||||
output_folder_wav = os.path.normpath(output_folder_wav)
|
output_folder_wav = os.path.normpath(output_folder_wav)
|
||||||
|
|
||||||
# Check and perform file deletion
|
# Check and perform file deletion
|
||||||
@@ -138,10 +141,23 @@ def check_required_files():
|
|||||||
subprocess.run(["python", str(download_script_path)])
|
subprocess.run(["python", str(download_script_path)])
|
||||||
print(f"[{params['branding']}Startup] All required files are present.")
|
print(f"[{params['branding']}Startup] All required files are present.")
|
||||||
|
|
||||||
|
|
||||||
# STARTUP Call Check routine
|
# STARTUP Call Check routine
|
||||||
check_required_files()
|
check_required_files()
|
||||||
|
|
||||||
|
##################################################
|
||||||
|
#### Check to see if a finetuned model exists ####
|
||||||
|
##################################################
|
||||||
|
# Set the path to the directory
|
||||||
|
trained_model_directory = this_dir / "models" / "trainedmodel"
|
||||||
|
# Check if the directory "trainedmodel" exists
|
||||||
|
finetuned_model = trained_model_directory.exists()
|
||||||
|
# If the directory exists, check for the existence of the required files
|
||||||
|
# If true, this will add a extra option in the Gradio interface for loading Xttsv2 FT
|
||||||
|
if finetuned_model:
|
||||||
|
required_files = ["model.pth", "config.json", "vocab.json"]
|
||||||
|
finetuned_model = all((trained_model_directory / file).exists() for file in required_files)
|
||||||
|
if finetuned_model:
|
||||||
|
print(f"[{params['branding']}Startup] Finetuned model \033[93mDetected\033[0m")
|
||||||
|
|
||||||
####################################################
|
####################################################
|
||||||
#### SET GRADIO BUTTONS BASED ON confignew.json ####
|
#### SET GRADIO BUTTONS BASED ON confignew.json ####
|
||||||
@@ -248,10 +264,10 @@ else:
|
|||||||
# Cleanly kill off this script, but allow text-generation-webui to keep running, albeit without this alltalk_tts
|
# Cleanly kill off this script, but allow text-generation-webui to keep running, albeit without this alltalk_tts
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
timeout = 60 # Adjust the timeout as needed
|
timeout = 120 # Adjust the timeout as needed
|
||||||
|
|
||||||
# Introduce a delay before starting the check loop
|
# Introduce a delay before starting the check loop
|
||||||
time.sleep(25) # Wait 25 secs before checking if the tts_server.py has started up.
|
time.sleep(26) # Wait 26 secs before checking if the tts_server.py has started up.
|
||||||
start_time = time.time()
|
start_time = time.time()
|
||||||
while time.time() - start_time < timeout:
|
while time.time() - start_time < timeout:
|
||||||
try:
|
try:
|
||||||
@@ -261,9 +277,9 @@ while time.time() - start_time < timeout:
|
|||||||
except requests.RequestException as e:
|
except requests.RequestException as e:
|
||||||
# Print the exception for debugging purposes
|
# Print the exception for debugging purposes
|
||||||
print(
|
print(
|
||||||
f"[{params['branding']}Startup] \033[91mWarning\033[0m TTS Subprocess has NOT started up yet, Will keep trying for 60 seconds maximum. Please wait."
|
f"[{params['branding']}Startup] \033[91mWarning\033[0m TTS Subprocess has NOT started up yet, Will keep trying for 120 seconds maximum. Please wait."
|
||||||
)
|
)
|
||||||
time.sleep(1)
|
time.sleep(4)
|
||||||
else:
|
else:
|
||||||
print(
|
print(
|
||||||
f"[{params['branding']}Startup] Startup timed out. Check the server logs for more information."
|
f"[{params['branding']}Startup] Startup timed out. Check the server logs for more information."
|
||||||
@@ -271,11 +287,13 @@ else:
|
|||||||
# Cleanly kill off this script, but allow text-generation-webui to keep running, albeit without this alltalk_tts
|
# Cleanly kill off this script, but allow text-generation-webui to keep running, albeit without this alltalk_tts
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
|
|
||||||
#####################################
|
#####################################
|
||||||
#### MODEL LOADING AND UNLOADING ####
|
#### MODEL LOADING AND UNLOADING ####
|
||||||
#####################################
|
#####################################
|
||||||
# MODEL - Swap model based on Gradio selection API TTS, API Local, XTTSv2 Local
|
# MODEL - Swap model based on Gradio selection API TTS, API Local, XTTSv2 Local
|
||||||
def send_reload_request(tts_method):
|
def send_reload_request(tts_method):
|
||||||
|
global tts_method_xtts_ft
|
||||||
try:
|
try:
|
||||||
params["tts_model_loaded"] = False
|
params["tts_model_loaded"] = False
|
||||||
url = f"{base_url}/api/reload"
|
url = f"{base_url}/api/reload"
|
||||||
@@ -294,17 +312,26 @@ def send_reload_request(tts_method):
|
|||||||
params["tts_method_api_tts"] = True
|
params["tts_method_api_tts"] = True
|
||||||
params["deepspeed_activate"] = False
|
params["deepspeed_activate"] = False
|
||||||
audio_path = this_dir / "templates" / "apitts.wav"
|
audio_path = this_dir / "templates" / "apitts.wav"
|
||||||
|
tts_method_xtts_ft = False
|
||||||
elif tts_method == "API Local":
|
elif tts_method == "API Local":
|
||||||
params["tts_method_api_tts"] = False
|
params["tts_method_api_tts"] = False
|
||||||
params["tts_method_xtts_local"] = False
|
params["tts_method_xtts_local"] = False
|
||||||
params["tts_method_api_local"] = True
|
params["tts_method_api_local"] = True
|
||||||
params["deepspeed_activate"] = False
|
params["deepspeed_activate"] = False
|
||||||
audio_path = this_dir / "templates" / "apilocal.wav"
|
audio_path = this_dir / "templates" / "apilocal.wav"
|
||||||
|
tts_method_xtts_ft = False
|
||||||
elif tts_method == "XTTSv2 Local":
|
elif tts_method == "XTTSv2 Local":
|
||||||
params["tts_method_api_tts"] = False
|
params["tts_method_api_tts"] = False
|
||||||
params["tts_method_api_local"] = False
|
params["tts_method_api_local"] = False
|
||||||
params["tts_method_xtts_local"] = True
|
params["tts_method_xtts_local"] = True
|
||||||
audio_path = this_dir / "templates" / "xttslocal.wav"
|
audio_path = this_dir / "templates" / "xttslocal.wav"
|
||||||
|
tts_method_xtts_ft = False
|
||||||
|
elif tts_method == "XTTSv2 FT":
|
||||||
|
params["tts_method_api_tts"] = False
|
||||||
|
params["tts_method_api_local"] = False
|
||||||
|
params["tts_method_xtts_local"] = False
|
||||||
|
audio_path = this_dir / "templates" / "xttsfinetuned.wav"
|
||||||
|
tts_method_xtts_ft = True
|
||||||
return f'<audio src="file/{audio_path}" controls autoplay></audio>'
|
return f'<audio src="file/{audio_path}" controls autoplay></audio>'
|
||||||
except requests.exceptions.RequestException as e:
|
except requests.exceptions.RequestException as e:
|
||||||
# Handle the HTTP request error
|
# Handle the HTTP request error
|
||||||
@@ -482,6 +509,54 @@ def voice_preview(string):
|
|||||||
return f"[{params['branding']}Server] Audio generation failed. Status: {generate_response.get('status')}"
|
return f"[{params['branding']}Server] Audio generation failed. Status: {generate_response.get('status')}"
|
||||||
|
|
||||||
|
|
||||||
|
def process_text(text):
|
||||||
|
# Normalize HTML encoded quotes
|
||||||
|
text = html.unescape(text)
|
||||||
|
|
||||||
|
# Replace ellipsis with a single dot
|
||||||
|
text = re.sub(r'\.{3,}', '.', text)
|
||||||
|
|
||||||
|
# Pattern to identify combined narrator and character speech
|
||||||
|
combined_pattern = r'(\*[^*"]+\*|"[^"*]+")'
|
||||||
|
|
||||||
|
# List to hold parts of speech along with their type
|
||||||
|
ordered_parts = []
|
||||||
|
|
||||||
|
# Track the start of the next segment
|
||||||
|
start = 0
|
||||||
|
|
||||||
|
# Find all matches
|
||||||
|
for match in re.finditer(combined_pattern, text):
|
||||||
|
# Add the text before the match, if any, as ambiguous
|
||||||
|
if start < match.start():
|
||||||
|
ambiguous_text = text[start:match.start()].strip()
|
||||||
|
if ambiguous_text:
|
||||||
|
ordered_parts.append(('ambiguous', ambiguous_text))
|
||||||
|
|
||||||
|
# Add the matched part as either narrator or character
|
||||||
|
matched_text = match.group(0)
|
||||||
|
if matched_text.startswith('*') and matched_text.endswith('*'):
|
||||||
|
ordered_parts.append(('narrator', matched_text.strip('*').strip()))
|
||||||
|
elif matched_text.startswith('"') and matched_text.endswith('"'):
|
||||||
|
ordered_parts.append(('character', matched_text.strip('"').strip()))
|
||||||
|
else:
|
||||||
|
# In case of mixed or improperly formatted parts
|
||||||
|
if '*' in matched_text:
|
||||||
|
ordered_parts.append(('narrator', matched_text.strip('*').strip('"')))
|
||||||
|
else:
|
||||||
|
ordered_parts.append(('character', matched_text.strip('"').strip('*')))
|
||||||
|
|
||||||
|
# Update the start of the next segment
|
||||||
|
start = match.end()
|
||||||
|
|
||||||
|
# Add any remaining text after the last match as ambiguous
|
||||||
|
if start < len(text):
|
||||||
|
ambiguous_text = text[start:].strip()
|
||||||
|
if ambiguous_text:
|
||||||
|
ordered_parts.append(('ambiguous', ambiguous_text))
|
||||||
|
|
||||||
|
return ordered_parts
|
||||||
|
|
||||||
#################################
|
#################################
|
||||||
#### TTS STANDARD GENERATION ####
|
#### TTS STANDARD GENERATION ####
|
||||||
#################################
|
#################################
|
||||||
@@ -490,6 +565,7 @@ def output_modifier(string, state):
|
|||||||
if not params["activate"]:
|
if not params["activate"]:
|
||||||
return string
|
return string
|
||||||
original_string = string
|
original_string = string
|
||||||
|
#print("ORIGINAL STRING IS:", original_string)
|
||||||
cleaned_string = before_audio_generation(string, params)
|
cleaned_string = before_audio_generation(string, params)
|
||||||
if cleaned_string is None:
|
if cleaned_string is None:
|
||||||
return
|
return
|
||||||
@@ -500,85 +576,55 @@ def output_modifier(string, state):
|
|||||||
if process_lock.acquire(blocking=False):
|
if process_lock.acquire(blocking=False):
|
||||||
try:
|
try:
|
||||||
if params["narrator_enabled"]:
|
if params["narrator_enabled"]:
|
||||||
# print(original_string)
|
processed_parts = process_text(original_string)
|
||||||
# Do Some basic stripping and remove any double CR's
|
|
||||||
processed_string = (
|
|
||||||
original_string
|
|
||||||
.replace("***", "*")
|
|
||||||
.replace("**", "*")
|
|
||||||
.replace("\n\n", "\n")
|
|
||||||
)
|
|
||||||
# Clean up two asterisks (narrators) being made between a newline, making it one sentence.
|
|
||||||
processed_string = re.sub(r'\.\*\n\*', '. ', processed_string)
|
|
||||||
# Clean up a few other bits.
|
|
||||||
processed_string = (
|
|
||||||
processed_string
|
|
||||||
.replace("'", "'")
|
|
||||||
.replace("\n", " ")
|
|
||||||
# Add special characters to the quote so that we can use it to identify things later after its been split
|
|
||||||
.replace('"', '"<')
|
|
||||||
# Capture new conversations which wont have things like "e in them
|
|
||||||
.replace('"', '"<')
|
|
||||||
)
|
|
||||||
#capturing another outlier in inital character paragraph
|
|
||||||
# print("processed string 1 is:", processed_string)
|
|
||||||
processed_string = processed_string.replace('"<. *', '"< *"')
|
|
||||||
processed_string = processed_string.replace('< *"', '< *')
|
|
||||||
processed_string = processed_string.replace('. *', '< *')
|
|
||||||
# print("processed string 2 is:", processed_string)
|
|
||||||
# Set up a tracking of the individual wav files.
|
|
||||||
audio_files_all_paragraphs = []
|
audio_files_all_paragraphs = []
|
||||||
# Split the line using " and ".* " (so end of sentences, leaving special characters added to the start of all OTHER sentences, bar possibly the first one if its starting with a *
|
for part_type, part in processed_parts:
|
||||||
parts = re.split(r'"|\.\*', processed_string)
|
|
||||||
audio_files_paragraph = []
|
|
||||||
for i, part in enumerate(parts):
|
|
||||||
# Skip parts that are too short
|
# Skip parts that are too short
|
||||||
if len(part.strip()) <= 1:
|
if len(part.strip()) <= 3:
|
||||||
continue
|
continue
|
||||||
# Figure out which type of line it is, then replace characters as necessary to avoid TTS trying to pronunce them, htmlunescape after.
|
|
||||||
# Character will always be a < with a letter immediately after it
|
# Determine the voice to use based on the part type
|
||||||
if '<' in part and '< *' not in part and '<*' not in part and '< *' not in part and '< ' not in part and '< ' not in part:
|
if part_type == 'narrator':
|
||||||
cleaned_part = html.unescape(part.replace('<', ''))
|
|
||||||
voice_to_use = params["voice"]
|
|
||||||
#Narrator will always be an * or < with an * a position or two after it.
|
|
||||||
elif '<*' in part or '< *' in part or '< *' in part or '*' in part:
|
|
||||||
cleaned_part = html.unescape(part.replace('<*', '').replace('< *', '').replace('< *', '').replace('*', ''))
|
|
||||||
voice_to_use = params["narrator_voice"]
|
voice_to_use = params["narrator_voice"]
|
||||||
#If the other two dont capture it, aka, the AI gave no * or " on the line, use non_quoted_text_is aka user interface, user can choose Char or Narrator
|
print(f"[{params['branding']}TTSGen] \033[92mNarrator\033[0m") # Green
|
||||||
elif non_quoted_text_is:
|
elif part_type == 'character':
|
||||||
cleaned_part = html.unescape(part.replace('< ', '').replace('< ', '').replace('< ', ''))
|
|
||||||
voice_to_use = params["voice"]
|
voice_to_use = params["voice"]
|
||||||
|
print(f"[{params['branding']}TTSGen] \033[36mCharacter\033[0m") # Yellow
|
||||||
else:
|
else:
|
||||||
cleaned_part = html.unescape(part.replace('< ', '').replace('< ', '').replace('< ', ''))
|
# Handle ambiguous parts based on user preference
|
||||||
voice_to_use = params["narrator_voice"]
|
voice_to_use = params["voice"] if non_quoted_text_is else params["narrator_voice"]
|
||||||
# Check if character name exists and if not, just call it TTSOUT_
|
voice_description = "\033[36mCharacter (Text-not-inside)\033[0m" if non_quoted_text_is else "\033[92mNarrator (Text-not-inside)\033[0m"
|
||||||
if "character_menu" in state:
|
print(f"[{params['branding']}TTSGen] {voice_description}")
|
||||||
output_file = Path(f'{params["output_folder_wav"]}/{state["character_menu"]}_{int(time.time())}_{i}.wav')
|
|
||||||
else:
|
# Replace multiple exclamation marks, question marks, or other punctuation with a single instance
|
||||||
output_file = Path(f'{params["output_folder_wav"]}/TTSOUT_{int(time.time())}_{i}.wav')
|
cleaned_part = re.sub(r'([!?.])\1+', r'\1', part)
|
||||||
# Generate that TTS and output to a file
|
# Further clean to remove any other unwanted characters
|
||||||
output_file_str = output_file.as_posix()
|
cleaned_part = re.sub(r'[^a-zA-Z0-9\s\.,;:!?\-\'"]', '', cleaned_part)
|
||||||
|
# Remove all newline characters (single or multiple)
|
||||||
|
cleaned_part = re.sub(r'\n+', ' ', cleaned_part)
|
||||||
|
|
||||||
|
# Generate TTS and output to a file
|
||||||
|
output_filename = get_output_filename(state)
|
||||||
generate_response = send_generate_request(
|
generate_response = send_generate_request(
|
||||||
cleaned_part, voice_to_use, language_code, output_file_str
|
cleaned_part, voice_to_use, language_code, output_filename
|
||||||
)
|
)
|
||||||
audio_path = generate_response.get("data", {}).get("audio_path")
|
audio_path = generate_response.get("data", {}).get("audio_path")
|
||||||
audio_files_paragraph.append(audio_path)
|
audio_files_all_paragraphs.append(audio_path)
|
||||||
# Accumulate audio files within the paragraph
|
|
||||||
audio_files_all_paragraphs.extend(audio_files_paragraph)
|
|
||||||
# Combine audio files across paragraphs
|
# Combine audio files across paragraphs
|
||||||
final_output_file = combine(
|
final_output_file = combine(
|
||||||
audio_files_all_paragraphs, params["output_folder_wav"], state
|
audio_files_all_paragraphs, params["output_folder_wav"], state
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
processed_string = (
|
# Decode HTML entities first
|
||||||
original_string
|
cleaned_part = html.unescape(original_string)
|
||||||
.replace("***", "")
|
# Replace multiple instances of certain punctuation marks with a single instance
|
||||||
.replace("**", "")
|
cleaned_part = re.sub(r'([!?.])\1+', r'\1', cleaned_part)
|
||||||
.replace("*", "")
|
# Further clean to remove any other unwanted characters
|
||||||
.replace("\n\n", "\n")
|
cleaned_part = re.sub(r'[^a-zA-Z0-9\s\.,;:!?\-\'"]', '', cleaned_part)
|
||||||
.replace("'", "'")
|
# Remove all newline characters (single or multiple)
|
||||||
)
|
cleaned_part = re.sub(r'\n+', ' ', cleaned_part)
|
||||||
cleaned_part = html.unescape(processed_string)
|
|
||||||
# Process the part and give it a non-character name if being used vai API or standalone.
|
# Process the part and give it a non-character name if being used vai API or standalone.
|
||||||
if "character_menu" in state:
|
if "character_menu" in state:
|
||||||
output_file = Path(f'{params["output_folder_wav"]}/{state["character_menu"]}_{int(time.time())}.wav')
|
output_file = Path(f'{params["output_folder_wav"]}/{state["character_menu"]}_{int(time.time())}.wav')
|
||||||
@@ -627,9 +673,10 @@ def output_modifier(string, state):
|
|||||||
|
|
||||||
|
|
||||||
def get_output_filename(state):
|
def get_output_filename(state):
|
||||||
return Path(
|
if "character_menu" in state:
|
||||||
f'{params["output_folder_wav"]}/{state["character_menu"]}_{str(uuid.uuid4())[:8]}.wav'
|
return Path(f'{params["output_folder_wav"]}/{state["character_menu"]}_{str(uuid.uuid4())[:8]}.wav').as_posix()
|
||||||
).as_posix()
|
else:
|
||||||
|
return Path(f'{params["output_folder_wav"]}/TTSOUT_{str(uuid.uuid4())[:8]}.wav').as_posix()
|
||||||
|
|
||||||
|
|
||||||
###############################################
|
###############################################
|
||||||
@@ -685,10 +732,6 @@ def input_modifier(string, state):
|
|||||||
shared.processing_message = "*Is recording a voice message...*"
|
shared.processing_message = "*Is recording a voice message...*"
|
||||||
return string
|
return string
|
||||||
|
|
||||||
def custom_css():
|
|
||||||
path_to_css = Path(f"{this_dir}/style.css")
|
|
||||||
return open(path_to_css, "r").read()
|
|
||||||
|
|
||||||
|
|
||||||
def ui():
|
def ui():
|
||||||
with gr.Accordion(params["branding"] + " TTS (XTTSv2)"):
|
with gr.Accordion(params["branding"] + " TTS (XTTSv2)"):
|
||||||
@@ -782,14 +825,17 @@ def ui():
|
|||||||
deepspeed_checkbox_play = gr.HTML(visible=False)
|
deepspeed_checkbox_play = gr.HTML(visible=False)
|
||||||
|
|
||||||
with gr.Row():
|
with gr.Row():
|
||||||
|
model_loader_choices=["API TTS", "API Local", "XTTSv2 Local"]
|
||||||
|
if finetuned_model:
|
||||||
|
model_loader_choices.append("XTTSv2 FT")
|
||||||
tts_radio_buttons = gr.Radio(
|
tts_radio_buttons = gr.Radio(
|
||||||
choices=["API TTS", "API Local", "XTTSv2 Local"],
|
choices=model_loader_choices,
|
||||||
label="Select TTS Generation Method (Read NOTE)",
|
label="Select TTS Generation Method (Read NOTE)",
|
||||||
value=gr_modelchoice, # Set the default value
|
value=gr_modelchoice, # Set the default value
|
||||||
)
|
)
|
||||||
tts_radio_buttons_play = gr.HTML(visible=False)
|
tts_radio_buttons_play = gr.HTML(visible=False)
|
||||||
explanation_text = gr.HTML(
|
explanation_text = gr.HTML(
|
||||||
f"<p>NOTE: Switching Model Type, Low VRAM & DeepSpeed, each takes 15 seconds. Each TTS generation method has a slightly different sound. DeepSpeed checkbox is only visible if DeepSpeed is present. Readme & Settings: <a href='http://{params['ip_address']}:{params['port_number']}'>http://{params['ip_address']}:{params['port_number']}</a>"
|
f"<p>⚙️ <a href='http://{params['ip_address']}:{params['port_number']}'>Click here for the Settings page, help & instructions</a><a href='http://{params['ip_address']}:{params['port_number']}'></a>⚙️<br>NOTE: Switching Model Type, Low VRAM & DeepSpeed, each takes 15 seconds. Each TTS generation method has a slightly different sound. DeepSpeed checkbox is only visible if DeepSpeed is installed on your system."
|
||||||
)
|
)
|
||||||
|
|
||||||
with gr.Row():
|
with gr.Row():
|
||||||
@@ -888,4 +934,4 @@ if running_in_standalone:
|
|||||||
try:
|
try:
|
||||||
time.sleep(1) # Add a small delay to avoid high CPU usage
|
time.sleep(1) # Add a small delay to avoid high CPU usage
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
break # Allow graceful exit on Ctrl+C
|
break # Allow graceful exit on Ctrl+C
|
||||||
@@ -1,8 +0,0 @@
|
|||||||
.SDAP .hires_opts input[type="number"] {
|
|
||||||
width: 6em !important;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* silero_tts preview */
|
|
||||||
.form:has(> #silero_preview_text) {
|
|
||||||
min-width: 75%
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,584 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>AllTalk TTS Generator</title>
|
||||||
|
|
||||||
|
<style>
|
||||||
|
body,
|
||||||
|
textarea,
|
||||||
|
select,
|
||||||
|
input[type="range"],
|
||||||
|
input[type="number"],
|
||||||
|
button {
|
||||||
|
font-family: Arial, sans-serif;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
margin: 20px;
|
||||||
|
background-color: #f7f7f7;
|
||||||
|
color: #333;
|
||||||
|
}
|
||||||
|
|
||||||
|
h1 {
|
||||||
|
color: #333;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-container {
|
||||||
|
max-width: 80%;
|
||||||
|
margin: 50px auto;
|
||||||
|
padding: 20px;
|
||||||
|
background: #fff;
|
||||||
|
border-radius: 8px;
|
||||||
|
box-shadow: 0 2px 5px rgba(0, 0, 0, .2);
|
||||||
|
}
|
||||||
|
|
||||||
|
.settings-section {
|
||||||
|
display: flex;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
justify-content: flex-start;
|
||||||
|
/* Align items to the start */
|
||||||
|
align-items: center;
|
||||||
|
/* Align items vertically */
|
||||||
|
gap: 20px;
|
||||||
|
/* This creates a consistent gap between all child elements */
|
||||||
|
}
|
||||||
|
|
||||||
|
.setting {
|
||||||
|
margin-bottom: 10px;
|
||||||
|
/* Space from bottom if wrapped */
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
/* Removed flex property to stop forcing them to grow */
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
label {
|
||||||
|
font-size: 0.9em;
|
||||||
|
color: #333;
|
||||||
|
margin-bottom: 5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
textarea {
|
||||||
|
resize: vertical;
|
||||||
|
width: 100%;
|
||||||
|
padding: 10px;
|
||||||
|
margin-bottom: 16px;
|
||||||
|
border: 1px solid #ccc;
|
||||||
|
border-radius: 4px;
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Ensure select and input elements do not exceed the width of their parent */
|
||||||
|
select,
|
||||||
|
input[type="range"],
|
||||||
|
input[type="number"] {
|
||||||
|
width: 100%;
|
||||||
|
/* They will fill the width of their .setting container */
|
||||||
|
padding: 10px;
|
||||||
|
margin-bottom: 16px;
|
||||||
|
border: 1px solid #ccc;
|
||||||
|
border-radius: 4px;
|
||||||
|
}
|
||||||
|
|
||||||
|
@media (min-width: 768px) {
|
||||||
|
|
||||||
|
select,
|
||||||
|
input[type="range"],
|
||||||
|
input[type="number"] {
|
||||||
|
width: auto;
|
||||||
|
/* Only auto width for larger screens */
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
button {
|
||||||
|
background-color: #4caf50;
|
||||||
|
color: #fff;
|
||||||
|
padding: 10px 20px;
|
||||||
|
border: none;
|
||||||
|
border-radius: 4px;
|
||||||
|
cursor: pointer;
|
||||||
|
display: inline-block;
|
||||||
|
margin: 5px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
button:hover {
|
||||||
|
background-color: #45a049;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dark-mode {
|
||||||
|
background-color: #121212;
|
||||||
|
color: #ffffff;
|
||||||
|
/* Text color for dark mode background */
|
||||||
|
}
|
||||||
|
|
||||||
|
.dark-mode .form-container {
|
||||||
|
background-color: #1e1e1e;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Modify the text color for elements within dark mode */
|
||||||
|
.dark-mode label,
|
||||||
|
.dark-mode textarea,
|
||||||
|
.dark-mode select,
|
||||||
|
.dark-mode input[type="range"],
|
||||||
|
.dark-mode input[type="number"],
|
||||||
|
.dark-mode button,
|
||||||
|
.dark-mode .error {
|
||||||
|
color: #ffffff;
|
||||||
|
/* Text color for dark mode elements */
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Modify text color for input elements, textarea, and dropdowns in dark mode */
|
||||||
|
.dark-mode input,
|
||||||
|
.dark-mode textarea,
|
||||||
|
.dark-mode select {
|
||||||
|
color: #000000;
|
||||||
|
/* Text color for input elements, textarea, and dropdowns in dark mode */
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Set background color for textarea in dark mode */
|
||||||
|
.dark-mode textarea {
|
||||||
|
background-color: #C0C0C0;
|
||||||
|
/* Light grey background color for textarea in dark mode */
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Set background color for option elements within select in dark mode */
|
||||||
|
.dark-mode select {
|
||||||
|
background-color: #C0C0C0;
|
||||||
|
/* Light grey background color for options in dark mode */
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Additional styles if needed for fixed-width dropdowns */
|
||||||
|
.fixed-size-dropdown {
|
||||||
|
width: 130px;
|
||||||
|
/* Adjust the width as needed */
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-overlay {
|
||||||
|
position: fixed;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
background: rgba(0, 0, 0, 0.5);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
z-index: 1000;
|
||||||
|
}
|
||||||
|
|
||||||
|
.spinner {
|
||||||
|
border: 5px solid #f3f3f3;
|
||||||
|
border-top: 5px solid #3498db;
|
||||||
|
border-radius: 50%;
|
||||||
|
width: 50px;
|
||||||
|
height: 50px;
|
||||||
|
animation: spin 2s linear infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes spin {
|
||||||
|
0% {
|
||||||
|
transform: rotate(0deg);
|
||||||
|
}
|
||||||
|
|
||||||
|
100% {
|
||||||
|
transform: rotate(360deg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.error {
|
||||||
|
color: #ff0000;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* Add space between sliders */
|
||||||
|
.slider-setting {
|
||||||
|
margin-right: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* Adjust the thumb appearance for a more standard line slider look */
|
||||||
|
input[type="range"]::-webkit-slider-thumb,
|
||||||
|
input[type="range"]::-moz-range-thumb,
|
||||||
|
input[type="range"]::-ms-thumb {
|
||||||
|
-webkit-appearance: none;
|
||||||
|
appearance: none;
|
||||||
|
width: 16px;
|
||||||
|
/* Smaller width for a more standard thumb size */
|
||||||
|
height: 16px;
|
||||||
|
/* Smaller height for a more standard thumb size */
|
||||||
|
border-radius: 50%;
|
||||||
|
/* Keep it rounded, or set to 0 for square */
|
||||||
|
background: #3071a9;
|
||||||
|
cursor: pointer;
|
||||||
|
margin-top: -7px;
|
||||||
|
/* Adjust if needed to align thumb with the track */
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
|
||||||
|
</head>
|
||||||
|
<h1>AllTalk TTS Generator</h1>
|
||||||
|
|
||||||
|
<div class="form-container">
|
||||||
|
<div class="input-section">
|
||||||
|
<label for="text_input">Text Input:</label>
|
||||||
|
<textarea id="text_input" rows="8"></textarea> <!-- Removed cols attribute -->
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="settings-section">
|
||||||
|
|
||||||
|
<div class="setting voice-setting">
|
||||||
|
<label for="character_voice_gen">Character Voice:</label>
|
||||||
|
<select id="character_voice_gen"></select>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="setting language-setting">
|
||||||
|
<label for="language">Language:</label>
|
||||||
|
<select id="language" name="language" class="fixed-size-dropdown">
|
||||||
|
<option value="ar">Arabic</option>
|
||||||
|
<option value="zh-cn">Chinese (Simplified)</option>
|
||||||
|
<option value="cs">Czech</option>
|
||||||
|
<option value="nl">Dutch</option>
|
||||||
|
<option value="en" selected>English</option>
|
||||||
|
<option value="fr">French</option>
|
||||||
|
<option value="de">German</option>
|
||||||
|
<option value="hu">Hungarian</option>
|
||||||
|
<option value="it">Italian</option>
|
||||||
|
<option value="ja">Japanese</option>
|
||||||
|
<option value="ko">Korean</option>
|
||||||
|
<option value="pl">Polish</option>
|
||||||
|
<option value="pt">Portuguese</option>
|
||||||
|
<option value="ru">Russian</option>
|
||||||
|
<option value="es">Spanish</option>
|
||||||
|
<option value="tr">Turkish</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
|
||||||
|
<div class="setting play-mode-setting">
|
||||||
|
<label for="playMode">Playback:</label>
|
||||||
|
<select id="playMode" name="playMode" class="fixed-size-dropdown">
|
||||||
|
<option value="local" selected>In Browser</option>
|
||||||
|
<option value="remote">On Server</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="setting theme-mode-setting">
|
||||||
|
<label for="themeMode">Theme:</label>
|
||||||
|
<select id="themeMode" name="themeMode" class="fixed-size-dropdown">
|
||||||
|
<option value="light" selected>Light</option>
|
||||||
|
<option value="dark">Dark</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="setting chunk-size-setting">
|
||||||
|
<label for="chunkSize">Sentence Chunk Size:</label>
|
||||||
|
<input type="range" id="chunkSize" name="chunkSize" class="slider-setting" min="1" max="10" value="4"
|
||||||
|
oninput="updateChunkSizeValue(this.value)">
|
||||||
|
<span id="chunkSizeValue">3</span> <!-- Display the value of the slider -->
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="setting volume-setting">
|
||||||
|
<label for="volume">Volume:</label>
|
||||||
|
<input type="range" id="volume" name="volume" class="slider-setting" min="0.1" max="1.0" step="0.1"
|
||||||
|
value="0.8" oninput="updateVolumeValue(this.value)">
|
||||||
|
<span id="volumeValue">0.8</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
|
||||||
|
<div class="action-buttons">
|
||||||
|
<button onclick="generateTTS()">Generate TTS</button>
|
||||||
|
<button onclick="stopPlayback()">Stop Playback</button>
|
||||||
|
<button onclick="previewVoice()">Preview Voice</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div id="progressIndicator" class="progress-overlay" style="display: none;">
|
||||||
|
<div class="spinner"></div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
async function fetchVoices() {
|
||||||
|
try {
|
||||||
|
const response = await fetch("http://127.0.0.1:7851/api/voices");
|
||||||
|
const result = await response.json();
|
||||||
|
const characterVoiceSelect = document.getElementById("character_voice_gen");
|
||||||
|
|
||||||
|
result.voices.forEach(voice => {
|
||||||
|
const option = document.createElement("option");
|
||||||
|
option.value = voice;
|
||||||
|
option.text = voice;
|
||||||
|
characterVoiceSelect.add(option);
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
console.error(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function updateVolumeValue(value) {
|
||||||
|
document.getElementById("volumeValue").innerText = value;
|
||||||
|
|
||||||
|
// Set the volume for browser playback
|
||||||
|
audioElement.volume = parseFloat(value);
|
||||||
|
}
|
||||||
|
|
||||||
|
let stopTTSGeneration = false;
|
||||||
|
let audioQueue = [];
|
||||||
|
let currentlyPlaying = false;
|
||||||
|
const audioElement = new Audio();
|
||||||
|
|
||||||
|
async function generateTTS() {
|
||||||
|
try {
|
||||||
|
// Reset the flag at the start
|
||||||
|
stopTTSGeneration = false;
|
||||||
|
|
||||||
|
const textInput = document.getElementById("text_input").value;
|
||||||
|
const language = document.getElementById("language").value;
|
||||||
|
const characterVoice = document.getElementById("character_voice_gen").value;
|
||||||
|
const volume = parseFloat(document.getElementById("volume").value);
|
||||||
|
const chunkSize = parseInt(document.getElementById("chunkSize").value);
|
||||||
|
// Retrieve the value of the selected option for playMode from the dropdown
|
||||||
|
const playModeValue = document.getElementById("playMode").value;
|
||||||
|
|
||||||
|
// Clean up the textInput
|
||||||
|
let cleanedText = textInput.replace(/ \- | \– /g, ' '); // Replaces " - " and " – " with " "
|
||||||
|
cleanedText = cleanedText.replace(/\.\s'/g, ".'"); // Correctly remove space between a period and a single quote
|
||||||
|
cleanedText = cleanedText.replace(/'/g, ""); // Remove all single quotes
|
||||||
|
|
||||||
|
// Split text into chunks based on the specified number of sentences
|
||||||
|
const sentences = cleanedText.match(/("[^"]+"|[^.!?]+[.!?]+)/g);
|
||||||
|
const chunks = [];
|
||||||
|
for (let i = 0; i < sentences.length; i += chunkSize) {
|
||||||
|
let chunk = sentences.slice(i, i + chunkSize).map(s => s.trim()).join(' ');
|
||||||
|
chunks.push(chunk);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use playModeValue for the conditional check
|
||||||
|
if (playModeValue === "remote") {
|
||||||
|
for (const chunk of chunks) {
|
||||||
|
if (stopTTSGeneration) break; // Stop if flag is set
|
||||||
|
await generateAndPlayRemote(chunk, language, characterVoice, volume);
|
||||||
|
}
|
||||||
|
} else if (playModeValue === "local") {
|
||||||
|
generateAndPlayLocal(chunks, 0);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Error in generateTTS:", error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
async function generateAndPlayLocal(chunks, index) {
|
||||||
|
if (index >= chunks.length || stopTTSGeneration) {
|
||||||
|
return; // Stop if no more chunks or flag is set
|
||||||
|
}
|
||||||
|
if (index >= chunks.length) {
|
||||||
|
// No more chunks to process
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const chunk = chunks[index];
|
||||||
|
const data = new URLSearchParams({
|
||||||
|
text_input: chunk.trim(),
|
||||||
|
text_filtering: "standard",
|
||||||
|
character_voice_gen: document.getElementById("character_voice_gen").value,
|
||||||
|
narrator_enabled: "false",
|
||||||
|
narrator_voice_gen: document.getElementById("character_voice_gen").value,
|
||||||
|
text_not_inside: "character",
|
||||||
|
language: document.getElementById("language").value,
|
||||||
|
output_file_name: "myoutputfile",
|
||||||
|
output_file_timestamp: "true",
|
||||||
|
autoplay: "false",
|
||||||
|
autoplay_volume: document.getElementById("volume").value
|
||||||
|
});
|
||||||
|
|
||||||
|
const response = await fetch("http://127.0.0.1:7851/api/tts-generate", {
|
||||||
|
method: "POST",
|
||||||
|
headers: {
|
||||||
|
"Content-Type": "application/x-www-form-urlencoded",
|
||||||
|
},
|
||||||
|
body: data,
|
||||||
|
});
|
||||||
|
|
||||||
|
const result = await response.json();
|
||||||
|
if (result.output_file_url) {
|
||||||
|
audioQueue.push(result.output_file_url);
|
||||||
|
if (!currentlyPlaying) {
|
||||||
|
playNextLocalChunk();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Recursively call the function with the next index
|
||||||
|
generateAndPlayLocal(chunks, index + 1);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Error in generateAndPlayLocal with chunk index", index, ":", error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function generateAndPlayRemote(chunk, language, characterVoice, volume) {
|
||||||
|
try {
|
||||||
|
if (stopTTSGeneration) {
|
||||||
|
console.log("TTS generation stopped.");
|
||||||
|
return; // Early return if stop flag is set
|
||||||
|
}
|
||||||
|
|
||||||
|
const data = new URLSearchParams({
|
||||||
|
text_input: chunk.trim(),
|
||||||
|
text_filtering: "standard",
|
||||||
|
character_voice_gen: characterVoice,
|
||||||
|
narrator_enabled: "false",
|
||||||
|
narrator_voice_gen: characterVoice,
|
||||||
|
text_not_inside: "character",
|
||||||
|
language: language,
|
||||||
|
output_file_name: "myoutputfile",
|
||||||
|
output_file_timestamp: "true",
|
||||||
|
autoplay: "true",
|
||||||
|
autoplay_volume: volume.toString(),
|
||||||
|
});
|
||||||
|
|
||||||
|
const response = await fetch("http://127.0.0.1:7851/api/tts-generate", {
|
||||||
|
method: "POST",
|
||||||
|
headers: {
|
||||||
|
"Content-Type": "application/x-www-form-urlencoded",
|
||||||
|
},
|
||||||
|
body: data,
|
||||||
|
});
|
||||||
|
|
||||||
|
const result = await response.json();
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new Error(`API error: ${result.detail || 'Unknown error'}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log("TTS generation complete for chunk:", chunk);
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Error in generateAndPlayRemote:", error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function previewVoice() {
|
||||||
|
const characterVoice = document.getElementById("character_voice_gen").value;
|
||||||
|
try {
|
||||||
|
const formData = new FormData();
|
||||||
|
formData.append("voice", characterVoice);
|
||||||
|
|
||||||
|
const response = await fetch("http://127.0.0.1:7851/api/previewvoice/", {
|
||||||
|
method: "POST",
|
||||||
|
body: formData,
|
||||||
|
});
|
||||||
|
|
||||||
|
const result = await response.json();
|
||||||
|
if (result.status === "generate-success") {
|
||||||
|
// Play the preview audio
|
||||||
|
const audioPreview = new Audio(result.output_file_url);
|
||||||
|
audioPreview.play();
|
||||||
|
} else {
|
||||||
|
throw new Error('Preview generation failed');
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error in previewVoice:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function playNextLocalChunk() {
|
||||||
|
if (audioQueue.length > 0) {
|
||||||
|
currentlyPlaying = true;
|
||||||
|
const audioUrl = audioQueue.shift();
|
||||||
|
audioElement.src = audioUrl;
|
||||||
|
audioElement.play();
|
||||||
|
|
||||||
|
audioElement.onended = () => {
|
||||||
|
currentlyPlaying = false;
|
||||||
|
playNextLocalChunk();
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function toggleThemeMode() {
|
||||||
|
// Retrieve the value of the selected option for themeMode from the dropdown
|
||||||
|
const themeModeValue = document.getElementById('themeMode').value;
|
||||||
|
|
||||||
|
if (themeModeValue === "dark") {
|
||||||
|
document.body.classList.add('dark-mode');
|
||||||
|
localStorage.setItem('darkMode', 'enabled'); // Assuming saveDarkModePreference sets this
|
||||||
|
} else {
|
||||||
|
document.body.classList.remove('dark-mode');
|
||||||
|
localStorage.setItem('darkMode', 'disabled'); // Assuming saveDarkModePreference sets this
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function updateChunkSizeValue(value) {
|
||||||
|
document.getElementById("chunkSizeValue").innerText = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Optionally, save the dark mode preference in localStorage
|
||||||
|
function saveDarkModePreference(isDarkMode) {
|
||||||
|
localStorage.setItem('darkMode', isDarkMode ? 'enabled' : 'disabled');
|
||||||
|
}
|
||||||
|
|
||||||
|
function stopPlayback() {
|
||||||
|
// Stop current audio playback
|
||||||
|
audioElement.pause();
|
||||||
|
audioElement.currentTime = 0;
|
||||||
|
|
||||||
|
// Clear the audio queue, reset playback flag, and stop further TTS generation
|
||||||
|
audioQueue = [];
|
||||||
|
currentlyPlaying = false;
|
||||||
|
stopTTSGeneration = true; // Set flag to stop TTS generation
|
||||||
|
}
|
||||||
|
|
||||||
|
function showProgress() {
|
||||||
|
document.getElementById('progressIndicator').style.display = 'flex';
|
||||||
|
}
|
||||||
|
|
||||||
|
function hideProgress() {
|
||||||
|
document.getElementById('progressIndicator').style.display = 'none';
|
||||||
|
}
|
||||||
|
|
||||||
|
function displayOutput(result) {
|
||||||
|
const outputContainer = document.getElementById("output");
|
||||||
|
outputContainer.innerHTML = `<p>Status: ${result.status}</p>`;
|
||||||
|
if (result.output_file_path) {
|
||||||
|
outputContainer.innerHTML += `<p>File Path: ${result.output_file_path}</p>`;
|
||||||
|
}
|
||||||
|
if (result.output_file_url) {
|
||||||
|
outputContainer.innerHTML += `<p>File URL: ${result.output_file_url}</p>`;
|
||||||
|
}
|
||||||
|
if (result.error) {
|
||||||
|
outputContainer.innerHTML += `<p class="error">Error: ${result.error}</p>`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
window.onload = function() {
|
||||||
|
fetchVoices();
|
||||||
|
|
||||||
|
const isDarkModeEnabled = localStorage.getItem('darkMode') === 'enabled';
|
||||||
|
const themeModeSelect = document.getElementById('themeMode');
|
||||||
|
if (isDarkModeEnabled) {
|
||||||
|
document.body.classList.add('dark-mode');
|
||||||
|
themeModeSelect.value = 'dark';
|
||||||
|
} else {
|
||||||
|
themeModeSelect.value = 'light';
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add change event listener to the themeMode dropdown
|
||||||
|
themeModeSelect.addEventListener('change', toggleThemeMode);
|
||||||
|
|
||||||
|
// Initialize other elements if necessary
|
||||||
|
};
|
||||||
|
|
||||||
|
</script>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
||||||
File diff suppressed because it is too large.
Load diff
Binary file not shown.
+470
-721
File diff suppressed because it is too large.
Load diff
Reference in new issue
Block a user