edit : api key 파일로 관리

custom_apps/gemini/main.py

@@ -7,14 +7,18 @@ from PIL import Image
 from io import BytesIO
 
 from main_rest.app.utils.date_utils import D
-from const import TEMP_FOLDER
-
+from const import TEMP_FOLDER, API_KEY_MANAGER
 
 def gemini_image(prompt, folder=None):
     from custom_logger.main_log import main_logger as LOG
     
     image_path = ''
-    client = genai.Client(api_key="AIzaSyCSw4pcPDYdAnjzBB7J9ZKXtRJJvunjWtA") # a2tec key
+    api_key = API_KEY_MANAGER.get_api_key()
+    
+    if api_key is None:
+        raise Exception("API 키 세팅 필요! - 서버를 다시 구동하거나, API키 파일을 확인")
+    
+    client = genai.Client(api_key=API_KEY_MANAGER.get_api_key()) # a2tec key
 
     for i in range(3):
         response = client.models.generate_content(