# Use the official Python image
FROM python:3.9

# Install system dependencies including libGL
RUN apt-get update && apt-get install -y libgl1-mesa-glx

RUN apt-get update && apt-get install -y \
    libgl1-mesa-glx \
    libglib2.0-0 \
    curl \
    build-essential

# Install Rust
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
ENV PATH="/root/.cargo/bin:${PATH}"

# Set a working directory
WORKDIR /app

# Copy your Streamlit app files to the container
COPY . .

# Install Python dependencies
COPY requirements.txt .
RUN pip install --upgrade pip
RUN pip install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install -r requirements.txt

# Expose the port your Streamlit app is running on
EXPOSE 8501

# Start your Streamlit app
CMD ["streamlit", "run", "app.py"]