FROM nvcr.io/nvidia/modulus/modulus:24.12

# Install additional dependencies
COPY requirements.txt /workspace/requirements.txt
RUN pip install -r /workspace/requirements.txt

# Copy application files
COPY app.py /workspace/
COPY conf /workspace/conf/
COPY outputs/main/flow_network.0.pth /workspace/outputs/main/

# Set working directory
WORKDIR /workspace

# Expose Streamlit port
EXPOSE 8501

# Set environment variable to disable CUDA
ENV CUDA_VISIBLE_DEVICES=""

# Command to run the Streamlit app
CMD ["streamlit", "run", "app.py", "--server.address", "0.0.0.0"]