FROM python:3.9

# Set environment variables
ENV VIRTUAL_ENV=/opt/venv
ENV PATH="$VIRTUAL_ENV/bin:$PATH"

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt \
    && mkdir cache \
    && chmod 777 cache
RUN pip install --upgrade pip \
    && pip install numpy \
    && pip install git+https://github.com/facebookresearch/segment-anything.git \
    && pip install git+https://github.com/huggingface/transformers.git

COPY . .

# Set permissions for cache directory if it exists
RUN if [ ! -d "cache" ]; then mkdir cache; fi && chmod 777 cache

EXPOSE 7860

CMD ["shiny", "run", "app.py", "--host", "0.0.0.0", "--port", "7860"]