FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04

# Set up environment
ENV DEBIAN_FRONTEND=noninteractive \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1 \
    TZ=Etc/UTC

RUN apt-get update && apt-get install -y \
    git python3 python3-pip python-is-python3 \
    libgl1 libglib2.0-0 wget curl \
 && apt-get clean

WORKDIR /app

# Clone the repo (already in your space, so skip cloning)
COPY . .

# Install requirements
RUN pip install -r requirements.txt

# Launch the web UI
CMD ["python3", "launch.py", "--listen", "--port", "7860", "--enable-insecure-extension-access"]