FROM python:3.10-slim

ENV PYTHONUNBUFFERED=1
ENV HF_HOME=/tmp/huggingface
ENV HOME=/tmp

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

WORKDIR /app

RUN pip install --no-cache-dir torch transformers peft trl bitsandbytes accelerate wandb datasets huggingface_hub matplotlib Pillow

RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git

RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/ && \
    mkdir -p /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/ && \
    mkdir -p /app/data/

COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
COPY data/ /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/
COPY data/ /app/data/
COPY train.py .
COPY evaluate.py .

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