feat: image env var

e3d253b0 · Timothy J. Baek · 2d7d6cff · e3d253b0 · e3d253b0
Commit e3d253b0 authored Apr 23, 2024 by Timothy J. Baek
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 4 deletions

backend/apps/images/main.py backend/apps/images/main.py +4 -4

backend/config.py backend/config.py +4 -0

No files found.
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -35,8 +35,8 @@ from config import (
    ENABLE_IMAGE_GENERATION,
    AUTOMATIC1111_BASE_URL,
    COMFYUI_BASE_URL,
-    OPENAI_API_BASE_URL,
-    OPENAI_API_KEY,
+    IMAGE_OPENAI_API_BASE_URL,
+    IMAGE_OPENAI_API_KEY,
 )


@@ -58,8 +58,8 @@ app.add_middleware(
 app.state.ENGINE = ""
 app.state.ENABLED = ENABLE_IMAGE_GENERATION

-app.state.OPENAI_API_BASE_URL = OPENAI_API_BASE_URL
-app.state.OPENAI_API_KEY = OPENAI_API_KEY
+app.state.OPENAI_API_BASE_URL = IMAGE_OPENAI_API_BASE_URL
+app.state.OPENAI_API_KEY = IMAGE_OPENAI_API_KEY

 app.state.MODEL = ""


--- a/backend/config.py
+++ b/backend/config.py
@@ -487,6 +487,10 @@ AUTOMATIC1111_BASE_URL = os.getenv("AUTOMATIC1111_BASE_URL", "")
 COMFYUI_BASE_URL = os.getenv("COMFYUI_BASE_URL", "")


+IMAGE_OPENAI_API_BASE_URL = os.getenv("IMAGE_OPENAI_API_BASE_URL", OPENAI_API_BASE_URL)
+IMAGE_OPENAI_API_KEY = os.getenv("IMAGE_OPENAI_API_KEY", OPENAI_API_KEY)
+
+
 ####################################
 # Audio
 ####################################