FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime

SHELL ["/bin/bash", "-c"]

# Environment variables for Hugging Face cache
ENV HF_HOME=/app/hf_cache
ENV TRANSFORMERS_CACHE=/app/hf_cache
ENV HF_TOKEN=${HF_TOKEN}
ENV PATH=/opt/conda/bin:$PATH
ENV MPLCONFIGDIR=/tmp/matplotlib

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git wget curl unzip ffmpeg libgl1-mesa-glx libglib2.0-0 && \
    apt-get clean

# Set up working directory as /app
WORKDIR /app

# Copy project into /app
COPY . /app

# Create conda environment and install dependencies
COPY requirements.txt /app/requirements.txt
RUN conda create -n epic python=3.10 -y && \
    conda run -n epic pip install --upgrade pip && \
    conda run -n epic pip install -r /app/requirements.txt

RUN chmod -R 777 /app /workspace

# Expose Gradio default port
EXPOSE 7860

# Start the Gradio app
CMD ["conda", "run", "--no-capture-output", "-n", "epic", "python", "track-pixels_gradio.py"]