FROM python:3.9-slim

WORKDIR /app

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

COPY src/ ./src/

# Upgrade pip dan install hanya dependency utama (tanpa requirements.txt dulu)
RUN pip install --upgrade pip

# Install clean versi terbaru yang kompatibel
RUN pip install --no-cache-dir \
    torch \
    torchaudio \
    transformers>=4.40.1 \
    accelerate>=0.20.3 \
    sentencepiece \
    streamlit \
    librosa \
    soundfile \
    numpy \
    yt-dlp

# Optional: debug versi
RUN python -c "import transformers; print('Transformers version:', transformers.__version__)"

EXPOSE 8501

HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1

ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]