Merge pull request #35 from Blazzycrafter/main

Add Edge TTS extras module option to Launcher.bat and launcher.sh
This commit is contained in:
deffcolony
2024-02-26 19:05:21 +01:00
committed by GitHub
2 changed files with 19 additions and 0 deletions

View File

@@ -59,6 +59,7 @@ set "caption_trigger=false"
set "summarize_trigger=false"
set "listen_trigger=false"
set "whisper_trigger=false"
set "edge_tts_trigger=false"
REM Define variables to track module status (XTTS)
set "modules_path=%~dp0modules-xtts.txt"
@@ -903,6 +904,7 @@ call :printModule "4. caption (--enable-modules=caption)" %caption_trigger%
call :printModule "5. summarize (--enable-modules=summarize)" %summarize_trigger%
call :printModule "6. listen (--listen)" %listen_trigger%
call :printModule "7. whisper (--enable-modules=whisper-stt)" %whisper_trigger%
call :printModule "8. Edge-tts (--enable-modules=edge-tts)" %edge_tts_trigger%
echo 0. Back to Toolbox
set "python_command="
@@ -959,11 +961,19 @@ for %%i in (%module_choices%) do (
set "whisper_trigger=true"
)
REM set "python_command= --enable-modules=whisper-stt"
) else if "%%i"=="8" (
if "%edge_tts_trigger%"=="true" (
set "edge_tts_trigger=false"
) else (
set "edge_tts_trigger=true"
)
REM set "python_command= --enable-modules=edge-tts"
) else if "%%i"=="0" (
goto :toolbox
)
)
REM Save the module flags to modules.txt
echo cuda_trigger=%cuda_trigger%>"%~dp0modules.txt"
echo rvc_trigger=%rvc_trigger%>>"%~dp0modules.txt"
@@ -972,6 +982,7 @@ echo caption_trigger=%caption_trigger%>>"%~dp0modules.txt"
echo summarize_trigger=%summarize_trigger%>>"%~dp0modules.txt"
echo listen_trigger=%listen_trigger%>>"%~dp0modules.txt"
echo whisper_trigger=%whisper_trigger%>>"%~dp0modules.txt"
echo edge_tts_trigger=%edge_tts_trigger%>>"%~dp0modules.txt"
REM remove modules_enable
set "modules_enable="
@@ -1001,6 +1012,9 @@ if "%summarize_trigger%"=="true" (
if "%whisper_trigger%"=="true" (
set "modules_enable=%modules_enable%whisper-stt,"
)
if "%edge_tts_trigger%"=="true" (
set "modules_enable=%modules_enable%edge-tts,"
)
REM is modules_enable empty?
if defined modules_enable (

View File

@@ -63,6 +63,7 @@ rvc_trigger="false"
talkinghead_trigger="false"
caption_trigger="false"
summarize_trigger="false"
edge_tts_trigger="false"
# Function to log messages with timestamps and colors
@@ -549,6 +550,7 @@ edit_extras_modules() {
printModule "4. caption (--enable-modules=caption)" "$caption_trigger"
printModule "5. summarize (--enable-modules=summarize)" "$summarize_trigger"
printModule "6. listen (--listen)" "$listen_trigger"
printModule "7. Edge TTS (--enable-modules=edge-tts)" "$edge_tts_trigger"
echo "0. Back to Toolbox"
set "python_command="
@@ -564,6 +566,7 @@ edit_extras_modules() {
4) [ "$caption_trigger" == "true" ] && caption_trigger=false || caption_trigger=true ;;
5) [ "$summarize_trigger" == "true" ] && summarize_trigger=false || summarize_trigger=true ;;
6) [ "$listen_trigger" == "true" ] && listen_trigger=false || listen_trigger=true ;;
7) [ "$edge_tts_trigger" == "true" ] && edge_tts_trigger=false || edge_tts_trigger=true ;;
0) toolbox ;;
esac
done
@@ -576,6 +579,7 @@ edit_extras_modules() {
echo "caption_trigger=$caption_trigger" >> "$modules_file"
echo "summarize_trigger=$summarize_trigger" >> "$modules_file"
echo "listen_trigger=$listen_trigger" >> "$modules_file"
echo "edge_tts_trigger=$edge_tts_trigger" >> "$modules_file"
# Compile the Python command
python_command="python server.py"
@@ -586,6 +590,7 @@ edit_extras_modules() {
[ "$talkinghead_trigger" == "true" ] && modules_enable+="talkinghead,"
[ "$caption_trigger" == "true" ] && modules_enable+="caption,"
[ "$summarize_trigger" == "true" ] && modules_enable+="summarize,"
[ "$edge_tts_trigger" == "true" ] && modules_enable+="edge-tts,"
# Remove the last comma from modules_enable
modules_enable="${modules_enable%,}"