tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

Add IMAGE_GENERATION_ENABLED env var

Browse source 
With default of `False`.
This commit is contained in:
Self Denial 2024-04-10 23:21:12 -06:00
parent 9e726a32e6
commit 3b8ea55bc8
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
backend/apps/images/main.py
View file

@ -29,7 +29,7 @@ import base64
import json
import logging
from config import SRC_LOG_LEVELS, CACHE_DIR, AUTOMATIC1111_BASE_URL, COMFYUI_BASE_URL
from config import SRC_LOG_LEVELS, CACHE_DIR, IMAGE_GENERATION_ENABLED, AUTOMATIC1111_BASE_URL, COMFYUI_BASE_URL
log = logging.getLogger(__name__)
@ -48,7 +48,7 @@ app.add_middleware(
)
app.state.ENGINE = ""
app.state.ENABLED = False
app.state.ENABLED = IMAGE_GENERATION_ENABLED
app.state.OPENAI_API_KEY = ""
app.state.MODEL = ""

3
backend/config.py
View file

@ -452,5 +452,8 @@ WHISPER_MODEL_DIR = os.getenv("WHISPER_MODEL_DIR", f"{CACHE_DIR}/whisper/models"
# Images
####################################
IMAGE_GENERATION_ENABLED = (
os.environ.get("IMAGE_GENERATION_ENABLED", "").lower() == "true"
)
AUTOMATIC1111_BASE_URL = os.getenv("AUTOMATIC1111_BASE_URL", "")
COMFYUI_BASE_URL = os.getenv("COMFYUI_BASE_URL", "")