From e3d253b0403cd55950a37a95a24e1dee1a96bc94 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Tue, 23 Apr 2024 06:53:04 -0400
Subject: [PATCH] feat: image env var

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

diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py
index a3939d20..5f7ce388 100644
--- 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 = ""
 
diff --git a/backend/config.py b/backend/config.py
index 199037e4..4d05a001 100644
--- 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
 ####################################