# The base image, which will be the starting point for the Docker image.
# We're using a PyTorch image built from https://github.com/allenai/docker-images
# because PyTorch is really big we want to install it first for caching.
FROM pytorch/pytorch:2.2.2-cuda11.8-cudnn8-runtime

ENV LC_ALL=C.UTF-8
ENV LANG=C.UTF-8

# This is the directory that files will be copied into.
# It's also the directory that you'll start in if you connect to the image.
WORKDIR /stage/

### SPECIFIC TO MY SETTING
COPY magentic-one/dv_comparison/requirements.txt .
RUN pip install -r requirements.txt

# Copy the file `main.py` to `/stage/main.py`
# You might need multiple of these statements to copy all the files you need for your experiment.
COPY app.py .

# Copy the folder `scripts` to `scripts/`
# You might need multiple of these statements to copy all the folders you need for your experiment.
COPY data_collection data_collection/
COPY datavoyager_utils datavoyager_utils/
COPY discoverybench discoverybench/
COPY magentic-one magentic-one/

RUN sed -i '440 c # ' /opt/conda/lib/python3.10/site-packages/autogen_agentchat/teams/_group_chat/_magentic_one/_magentic_one_orchestrator.py

