FROM python:3.10-slim

ENV PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1 \
    HOST=0.0.0.0 \
    PORT=7860 \
    MODEL_DIR=/app/models/supertonic-2 \
    SUPERTONIC_CACHE_DIR=/app/models/supertonic-2 \
    KITTEN_MODEL_NAME=KittenML/kitten-tts-micro-0.8 \
    KITTEN_CACHE_DIR=/app/models/hf-cache \
    KOKORO_MODEL_DIR=/app/models/kokoro \
    HF_HOME=/app/models/hf-cache \
    HUGGINGFACE_HUB_CACHE=/app/models/hf-cache \
    WORKSPACE_DIR=/tmp/tts-server \
    JOB_DB_PATH=/tmp/tts-server/jobs.db \
    PERSISTENT_OUTPUT_DIR=/store \
    ENABLE_MELO_TTS=true \
    REQUIRE_AUTH=false

RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential \
    ca-certificates \
    curl \
    ffmpeg \
    git \
    libsndfile1 \
    && rm -rf /var/lib/apt/lists/*

RUN useradd -m -u 1000 user

WORKDIR /app

COPY --chown=user:user requirements.txt requirements-melo.txt ./
RUN pip install -r requirements.txt

RUN git clone https://github.com/myshell-ai/MeloTTS.git /tmp/MeloTTS && \
    pip install -e /tmp/MeloTTS

COPY --chown=user:user . /app

RUN mkdir -p /app/models/supertonic-2 /app/models/kokoro /app/models/hf-cache /tmp/tts-server /store && \
    chown -R user:user /tmp/tts-server /store && \
    python -c "from supertonic import TTS; tts = TTS(model_dir='/app/models/supertonic-2', auto_download=True); print(f'Preloaded model at {tts.model_dir}')"

RUN python - <<'PY'
from huggingface_hub import snapshot_download
snapshot_download(
    repo_id='onnx-community/Kokoro-82M-v1.0-ONNX',
    local_dir='/app/models/kokoro',
    allow_patterns=['config.json', 'tokenizer.json', 'tokenizer_config.json', 'onnx/model_q8f16.onnx', 'voices/*.bin'],
)
print('Preloaded Kokoro assets into /app/models/kokoro')
PY

RUN python - <<'PY'
from kittentts import KittenTTS
KittenTTS(model_name='KittenML/kitten-tts-micro-0.8', cache_dir='/app/models/hf-cache')
print('Preloaded Kitten TTS assets into /app/models/hf-cache')
PY

RUN python -m unidic download

RUN python /tmp/MeloTTS/melo/init_downloads.py

RUN python - <<'PY'
from melo.api import TTS
TTS(language='EN', device='cpu', use_hf=True)
print('Verified MeloTTS dependency in the image')
PY

USER user

EXPOSE 7860

CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]