FROM python:3.10-slim

WORKDIR /app

# Enable unbuffered logging
ENV PYTHONUNBUFFERED=1

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

COPY requirements.txt .

RUN pip install --no-cache-dir -r requirements.txt

COPY . .

# Download the model from Hugging Face
RUN mkdir -p checkpoints && \
    curl -fSL -o checkpoints/best_model.pt "https://huggingface.co/TVQuyet05/En-Vi-Translator/resolve/main/best_model.pt"

EXPOSE 7860

HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health

CMD ["streamlit", "run", "web_translation/app.py", "--server.port=7860", "--server.address=0.0.0.0"]