FROM python:3.10

WORKDIR /app

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

COPY . .

# Create writable cache directory
RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache

# Set huggingface cache directory
ENV HF_HOME=/app/.cache/huggingface

CMD ["python", "train.py"]