Update app.py

app.py

@@ -1,23 +1,69 @@
-import subprocess
-import os
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
-# Install necessary packages
-subprocess.call(['pip', 'install', 'torch', 'torchvision'])
-subprocess.call(['pip', 'install', 'pygit2==1.12.2'])
+ENV DEBIAN_FRONTEND=noninteractive \
+    TZ=America/Los_Angeles
 
-# Rimuovi CUDA completamente dall'ambiente
-os.environ.pop('CUDA', None)
-os.environ.pop('CUDA_VISIBLE_DEVICES', None)
+ARG USE_PERSISTENT_DATA
 
-# Clone the repository
-git_clone_cmd = ['git', 'clone', 'https://github.com/lllyasviel/Fooocus.git']
-subprocess.call(git_clone_cmd)
+RUN apt-get update && apt-get install -y \
+    git \
+    make build-essential libssl-dev zlib1g-dev \
+    libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
+    libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git-lfs  \
+    ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
+    && rm -rf /var/lib/apt/lists/* \
+    && git lfs install
 
-# Navigate to the cloned directory
-os.chdir('Fooocus')
+WORKDIR /code
 
-# Run the Python script with specified arguments
-python_cmd = ['python', 'entry_with_update.py', '--share', '--always-high-vram']
+COPY ./requirements.txt /code/requirements.txt
 
-# Run the Python script
-subprocess.call(python_cmd)
+# User
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+# Pyenv
+RUN curl https://pyenv.run | bash
+ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
+
+ARG PYTHON_VERSION=3.10.12
+# Python
+RUN pyenv install $PYTHON_VERSION && \
+    pyenv global $PYTHON_VERSION && \
+    pyenv rehash && \
+    pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir \
+    datasets \
+    huggingface-hub "protobuf<4" "click<8.1"
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set the working directory to /data if USE_PERSISTENT_DATA is set, otherwise set to $HOME/app
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+
+RUN git clone https://github.com/lllyasviel/Fooocus . && \
+    # pin to a specific commit
+    git checkout bd4d40203c7e727fd06fc58ec122c25befcc5644 && \
+    pip install --no-cache-dir xformers==0.0.20 triton==2.0.0 torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 && \
+    pip install --no-cache-dir -r requirements_versions.txt
+
+# remove the next line if you're running on your own GPU, it set max images to 3 and disables the API
+RUN sed -i "s|image_number = gr.Slider(label='Image Number', minimum=1, maximum=32|image_number = gr.Slider(label='Image Number', minimum=1, maximum=3|" webui.py && \
+    sed -i "s|shared.gradio_root = gr.Blocks(title='Fooocus ' + fooocus_version.version, css=modules.html.css).queue()|shared.gradio_root = gr.Blocks(title='Fooocus ' + fooocus_version.version, css=modules.html.css).queue(concurrency_count=1,api_open=False)|" webui.py && \
+    sed -i "s|shared.gradio_root.launch(inbrowser=True, server_name=args.listen, server_port=args.port, share=args.share)|shared.gradio_root.launch(inbrowser=True, server_name=args.listen, server_port=args.port, share=args.share, show_api=False)|" webui.py
+
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    PYTHONPATH=$HOME/app \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
+
+CMD ["python", "entry_with_update.py", "--listen"]