# 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 torch transformers flask

# Đặt biến môi trường cho cache
ENV TRANSFORMERS_CACHE=/tmp/.cache

# Tải sẵn mô hình PhoBERT để tránh lỗi khi chạy
RUN python -c "from transformers import AutoTokenizer; 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"]