stephan/audio-engine-hub
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Add XTTS v2 support, refactor Docker/GPU infra, and improve Piper engine

Browse Source 
- Add XTTS v2 configuration to .env.example
- Refactor Dockerfile to multi-stage build with CUDA 12.1 support
- Update Makefile with Kokoro and XTTS test environment targets
- Refactor Piper engine (app/engines/piper.py) to use python module execution
- Add comprehensive documentation for Kokoro and XTTS plans
- Add helper scripts and patches for build process
This commit is contained in:
Stephan Wilkens
2025-12-13 11:37:58 +01:00
parent d6d1fe9d23
commit 91887ae296
19 changed files with 1625 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
Dockerfile
View File

@ -1,42 +1,56 @@
# Stage 1: Builder
FROM python:3.11 as builder
# Stage 1: Builder (for heavy Python dependencies including CUDA-enabled PyTorch)
FROM nvidia/cuda:12.1.0-devel-ubuntu22.04 as builder
WORKDIR /opt/venv
# Create a virtual environment and install dependencies
COPY requirements.txt .
RUN python -m venv . && . /opt/venv/bin/activate && pip install --no-cache-dir -r requirements.txt
# Install Python 3.11 and venv in the builder stage
RUN apt-get update && apt-get install -y python3.11 python3.11-venv
# Stage 2: Runner (The final image)
# Create a virtual environment and install Python dependencies
COPY requirements.txt .
COPY patches/ ./patches/
RUN python3.11 -m venv . && . /opt/venv/bin/activate && pip install --no-cache-dir -r requirements.txt && \
chmod +x ./patches/fix-misaki-espeak.sh && ./patches/fix-misaki-espeak.sh
# Stage 2: Runner (The final slim image)
FROM python:3.11-slim
# Install system dependencies needed at runtime
# ffmpeg is required for audio conversion
# espeak-ng is required for Kokoro TTS phonemization
RUN apt-get update && apt-get install -y --no-install-recommends \
ffmpeg \
espeak-ng \
&& rm -rf /var/lib/apt/lists/*
# Create an unprivileged user
RUN useradd --create-home --shell /bin/bash appuser
# Set working directory for the application code
WORKDIR /home/appuser
# Copy the virtual environment from the builder stage
# Copy the entire virtual environment from the builder stage
COPY --from=builder /opt/venv /opt/venv
# Add venv to PATH and PYTHONPATH so packages and scripts work correctly
ENV PATH="/opt/venv/bin:$PATH"
ENV PYTHONPATH="/opt/venv/lib/python3.11/site-packages:$PYTHONPATH"
# Create espeak-ng-data symlink for misaki compatibility
RUN mkdir -p /home/runner/work/espeakng-loader/espeakng-loader/espeak-ng/_dynamic/share/ && \
ln -sf /usr/lib/x86_64-linux-gnu/espeak-ng-data /home/runner/work/espeakng-loader/espeakng-loader/espeak-ng/_dynamic/share/espeak-ng-data
# Copy the application code into the container
COPY app/ ./app
# Set the PATH to include the virtual environment's bin directory
ENV PATH="/opt/venv/bin:$PATH"
# Expose the application port
EXPOSE 8000
# Run as the unprivileged user
USER appuser
# Command to run the application using gunicorn for production
CMD ["gunicorn", "-k", "uvicorn.workers.UvicornWorker", "-w", "2", "-b", "0.0.0.0:8000", "app.main:app"]
# Command to run the application using uvicorn for development with auto-reloading
CMD ["python3.11", "-m", "uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--log-level", "debug"]