feat: Docker support

Dockerfile

@@ -0,0 +1 @@
+nvidia.Dockerfile

README.md

@@ -127,6 +127,42 @@ For most users, the easiest way to install Selfie is to follow the [Quick Start]
 
 > **Note**: You can host selfie at a publicly-accessible URL with [ngrok](https://ngrok.com). Add your ngrok token (and optionally, ngrok domain) in `selfie/.env` and run `poetry run python -m selfie --share`.
 
+## Using Docker
+
+You can also run Selfie using Docker. To do so, follow these steps:
+
+1. Ensure that [Docker](https://www.docker.com) is installed.
+2. Clone or [download](https://github.com/vana-com/selfie) selfie repository.
+3. In a terminal, navigate to the project directory.
+
+We provide Dockerfiles for different GPU configurations and CPU-only environments. 
+Choose the appropriate Dockerfile based on your setup:
+
+- NVIDIA GPUs - default image, same as `Dockerfile`. Use `nvidia.Dockerfile` for systems with NVIDIA GPUs. This Dockerfile installs the necessary CUDA toolkit and the accelerated `llama-cpp-python` package.
+- No GPU (CPU-only): Use `cpu.Dockerfile` for systems without a GPU. This Dockerfile uses the standard Python image and runs the application on the CPU.
+
+To build and run the Docker image, navigate to the project directory and run the following commands:
+
+```bash
+# Build the Docker image
+docker build -t selfie .
+
+# Or, for CPU-only image:
+docker build -t selfie -f cpu.Dockerfile .
+
+# Run the Docker container
+docker run -p 8181:8181 \
+  --name selfie \
+  -v $(pwd)/data:/selfie/data \
+  -v $(pwd)/selfie:/selfie/selfie \
+  -v $(pwd)/selfie-ui:/selfie/selfie-ui \
+  -v $HOME/.cache/huggingface:/root/.cache/huggingface \
+  selfie:latest
+```
+This will start the server and the UI in your browser at http://0.0.0.0:8181/. 
+Your personal data will be stored in the `data` directory.
+This mounts your Hugging Face cache into the container, so you don't have to download the models again if you already
+have them.
 
 ## Setting Up Selfie
 

cpu.Dockerfile

@@ -0,0 +1,48 @@
+# Build the UI
+FROM node:18.19-alpine3.18 AS selfie-ui
+
+# Set the working directory in the container
+WORKDIR /selfie
+
+# Copy the package.json and yarn.lock files
+COPY selfie-ui/package.json selfie-ui/yarn.lock ./
+
+# Install dependencies
+RUN yarn install --frozen-lockfile --non-interactive
+
+# Copy the rest of the code
+COPY selfie-ui/ .
+
+# Build the project
+RUN yarn run build
+
+# Use the official Python image with CUDA support
+FROM python:3.11 as selfie
+
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
+
+# Set the working directory
+WORKDIR /selfie
+
+# Copy code and dependencies into the docker image
+COPY . .
+
+# Copy the built UI from the previous stage
+COPY --from=selfie-ui /selfie/out/ ./selfie-ui/out
+
+# Install poetry
+RUN pip install poetry --no-cache-dir
+
+# Install dependencies
+RUN poetry config virtualenvs.create false
+RUN poetry install --no-interaction --no-ansi
+
+# Run the installation script
+RUN bash /selfie/scripts/llama-cpp-python-cublas.sh
+
+EXPOSE 8181
+
+# Run the application
+CMD ["python", "-m", "selfie"]

nvidia.Dockerfile

@@ -0,0 +1,50 @@
+# Build the UI
+FROM node:18.19-alpine3.18 AS selfie-ui
+
+# Set the working directory in the container
+WORKDIR /selfie
+
+# Copy the package.json and yarn.lock files
+COPY selfie-ui/package.json selfie-ui/yarn.lock ./
+
+# Install dependencies
+RUN yarn install --frozen-lockfile --non-interactive
+
+# Copy the rest of the code
+COPY selfie-ui/ .
+
+# Build the project
+RUN yarn run build
+
+# Use pytorch with CUDA support
+FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime as selfie
+
+# Install build tools and compilers
+RUN apt-get update && \
+    apt-get install -y build-essential
+
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
+ENV PIP_NO_CACHE_DIR=1
+
+# Set the working directory
+WORKDIR /selfie
+
+# Copy code and dependencies into the docker image
+COPY . .
+
+# Copy the built UI from the previous stage
+COPY --from=selfie-ui /selfie/out/ ./selfie-ui/out
+
+# Install poetry
+RUN pip install poetry
+
+# Install dependencies
+RUN poetry config virtualenvs.create false
+RUN poetry install --no-interaction --no-ansi
+
+EXPOSE 8181
+
+# Run the application with GPU support
+CMD ["python", "-m", "selfie", "--gpu"]