FROM nvidia/cuda:12.3.1-base-ubuntu22.04

ENV DEBIAN_FRONTEND noninteractive
ENV CMDARGS --listen

# Instalar dependências do sistema
RUN apt-get update -y && \
	apt-get install -y git python3 python3-pip && \
	apt-get clean && \
	rm -rf /var/lib/apt/lists/*

# Criar diretório de trabalho
WORKDIR /code

# Instalar pygit2
RUN python3 -m pip install pygit2==1.12.2

# Copiar código do aplicativo
COPY . /code/

# Instalar dependências Python
RUN python3 -m pip install pygit2

# Expor porta 7860
EXPOSE 7860

# Executar como usuário não root
USER user

# Comando de inicialização
CMD ["python3", "entry_with_update.py", "--preset", "realistic", "--share", "--always-high-vram"]