From 849a0b973db35282f8886c924ffd7a9b8ff17569 Mon Sep 17 00:00:00 2001
From: Yanyutin753 <132346501+Yanyutin753@users.noreply.github.com>
Date: Sun, 28 Apr 2024 07:21:01 +0800
Subject: [PATCH] recompose the name of environment variables

---
 backend/apps/images/main.py | 8 ++++----
 backend/config.py           | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py
index 3b80a41b..fd72b203 100644
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -32,13 +32,13 @@ import logging
 from config import (
     SRC_LOG_LEVELS,
     CACHE_DIR,
-    IMAGES_GENERATION_ENGINE,
+    IMAGE_GENERATION_ENGINE,
     ENABLE_IMAGE_GENERATION,
     AUTOMATIC1111_BASE_URL,
     COMFYUI_BASE_URL,
     IMAGES_OPENAI_API_BASE_URL,
     IMAGES_OPENAI_API_KEY,
-    IMAGES_MODEL,
+    IMAGE_GENERATION_MODEL,
     IMAGE_SIZE,
     IMAGE_STEPS,
 )
@@ -59,13 +59,13 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-app.state.ENGINE = IMAGES_GENERATION_ENGINE
+app.state.ENGINE = IMAGE_GENERATION_ENGINE
 app.state.ENABLED = ENABLE_IMAGE_GENERATION
 
 app.state.OPENAI_API_BASE_URL = IMAGES_OPENAI_API_BASE_URL
 app.state.OPENAI_API_KEY = IMAGES_OPENAI_API_KEY
 
-app.state.MODEL = IMAGES_MODEL
+app.state.MODEL = IMAGE_GENERATION_MODEL
 
 
 app.state.AUTOMATIC1111_BASE_URL = AUTOMATIC1111_BASE_URL
diff --git a/backend/config.py b/backend/config.py
index b05f5cee..f864062d 100644
--- a/backend/config.py
+++ b/backend/config.py
@@ -531,7 +531,7 @@ WHISPER_MODEL_AUTO_UPDATE = (
 # Images
 ####################################
 
-IMAGES_GENERATION_ENGINE = os.getenv("IMAGES_GENERATION_ENGINE", "")
+IMAGE_GENERATION_ENGINE = os.getenv("IMAGE_GENERATION_ENGINE", "")
 
 ENABLE_IMAGE_GENERATION = (
     os.environ.get("ENABLE_IMAGE_GENERATION", "").lower() == "true"
@@ -549,7 +549,7 @@ IMAGE_SIZE = os.getenv("IMAGE_SIZE", "512x512")
 
 IMAGE_STEPS = int(os.getenv("IMAGE_STEPS", 50))
 
-IMAGES_MODEL = os.getenv("IMAGES_MODEL", "")
+IMAGE_GENERATION_MODEL = os.getenv("IMAGE_GENERATION_MODEL", "")
 
 ####################################
 # Audio