From 6bb45db97e904c4fcfcd1315eaf93b73e1ecb031 Mon Sep 17 00:00:00 2001
From: Lan Tian <xuyh0120@outlook.com>
Date: Tue, 26 Nov 2024 00:00:54 -0800
Subject: [PATCH] stable-diffusion-webui: integrate with open webui

---
 nixos/optional-apps/open-webui.nix             |  6 ++++++
 nixos/optional-apps/stable-diffusion-webui.nix | 11 ++++++++---
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/nixos/optional-apps/open-webui.nix b/nixos/optional-apps/open-webui.nix
index 5a41f421..13343b20 100644
--- a/nixos/optional-apps/open-webui.nix
+++ b/nixos/optional-apps/open-webui.nix
@@ -31,6 +31,12 @@
       OAUTH_ROLES_CLAIM = "groups";
       OPENID_PROVIDER_URL = "https://login.lantian.pub/.well-known/openid-configuration";
 
+      ENABLE_IMAGE_GENERATION = "true";
+      IMAGE_GENERATION_ENGINE = "automatic1111";
+      AUTOMATIC1111_BASE_URL = "https://stable-diffusion.xuyh0120.win";
+      IMAGE_SIZE = "512x512";
+      IMAGE_STEPS = "20";
+
       ENABLE_RAG_WEB_SEARCH = "true";
       ENABLE_SEARCH_QUERY = "true";
       RAG_WEB_SEARCH_ENGINE = "brave";
diff --git a/nixos/optional-apps/stable-diffusion-webui.nix b/nixos/optional-apps/stable-diffusion-webui.nix
index dcbff72a..a2a16f18 100644
--- a/nixos/optional-apps/stable-diffusion-webui.nix
+++ b/nixos/optional-apps/stable-diffusion-webui.nix
@@ -1,5 +1,6 @@
 {
   LT,
+  pkgs,
   ...
 }:
 let
@@ -16,6 +17,10 @@ let
     "venv"
     "repositories"
   ];
+
+  webui-user-sh = pkgs.writeShellScript "webui-user.sh" ''
+    export COMMANDLINE_ARGS="--api --disable-console-progressbars --xformers --no-half-vae"
+  '';
 in
 {
   virtualisation.oci-containers.containers.stable-diffusion = {
@@ -27,9 +32,9 @@ in
     ];
     image = "universonic/stable-diffusion-webui";
     ports = [ "127.0.0.1:${LT.portStr.StableDiffusionWebUI}:8080" ];
-    volumes = builtins.map (
-      f: "/var/lib/stable-diffusion/${f}:/app/stable-diffusion-webui/${f}"
-    ) subfolders;
+    volumes = [
+      "${webui-user-sh}:/app/stable-diffusion-webui/webui-user.sh:ro"
+    ] ++ builtins.map (f: "/var/lib/stable-diffusion/${f}:/app/stable-diffusion-webui/${f}") subfolders;
   };
 
   # Container uses UID/GID 1000