# Sử dụng image python slim
FROM python:3.10-slim

# Cài đặt các thư viện cần thiết
RUN pip install --no-cache-dir transformers flask tensorflow tf-keras

# Đặt biến môi trường cho cache (sử dụng HF_HOME thay vì TRANSFORMERS_CACHE)
ENV HF_HOME=/tmp/.cache

# Tải sẵn mô hình PhoBERT từ TensorFlow
RUN python -c "from transformers import TFAutoModel, AutoTokenizer; TFAutoModel.from_pretrained('vinai/phobert-base'); AutoTokenizer.from_pretrained('vinai/phobert-base')"

# Tạo thư mục làm việc trong container
WORKDIR /app

# Copy mã nguồn vào container
COPY . /app

# Expose cổng 7860 để chạy ứng dụng
EXPOSE 7860

# Chạy ứng dụng
CMD ["python", "app.py"]