Update app.py

app.py

@@ -35,6 +35,9 @@ def initialize_api_keys():
     key1 = os.environ.get("GEMINI_API_KEY_1", "")
     key2 = os.environ.get("GEMINI_API_KEY_2", "")
     key3 = os.environ.get("GEMINI_API_KEY_3", "")
+    key4 = os.environ.get("GEMINI_API_KEY_4", "")
+    key5 = os.environ.get("GEMINI_API_KEY_5", "")
+
     
     # 빈 문자열이 아닌 키만 추가
     if key1:
@@ -43,6 +46,10 @@ def initialize_api_keys():
         API_KEYS.append(key2)
     if key3:
         API_KEYS.append(key3)
+    if key4:
+        API_KEYS.append(key4)
+    if key5:
+        API_KEYS.append(key5)
     
     # 기존 GEMINI_API_KEY가 있으면 추가
     default_key = os.environ.get("GEMINI_API_KEY", "")
@@ -729,6 +736,8 @@ def generate_product_image(image, bg_type, simple, studio, nature, indoor, speci
             "1. 환경 변수: export GEMINI_API_KEY_1=\"your-api-key-1\"\n"
             "2. 환경 변수: export GEMINI_API_KEY_2=\"your-api-key-2\"\n"
             "3. 환경 변수: export GEMINI_API_KEY_3=\"your-api-key-3\"\n"
+            "4. 환경 변수: export GEMINI_API_KEY_4=\"your-api-key-4\"\n"
+            "5. 환경 변수: export GEMINI_API_KEY_5=\"your-api-key-5\"\n"
             "키 발급: https://aistudio.google.com/apikey"
         )
         return None, warning_msg, warning_msg
@@ -750,6 +759,8 @@ def generate_product_images(image, bg_type, simple, studio, nature, indoor, spec
             "1. 환경 변수: export GEMINI_API_KEY_1=\"your-api-key-1\"\n"
             "2. 환경 변수: export GEMINI_API_KEY_2=\"your-api-key-2\"\n"
             "3. 환경 변수: export GEMINI_API_KEY_3=\"your-api-key-3\"\n"
+            "4. 환경 변수: export GEMINI_API_KEY_4=\"your-api-key-4\"\n"
+            "5. 환경 변수: export GEMINI_API_KEY_5=\"your-api-key-5\"\n"
             "키 발급: https://aistudio.google.com/apikey"
         )
         return None, None, None, None, warning_msg, warning_msg
@@ -1074,7 +1085,7 @@ def create_app():
             if actual_api_key:
                 os.environ["GEMINI_API_KEY"] = actual_api_key  # 환경 변수 업데이트
             else:
-                return None, None, None, None, "API 키가 설정되지 않았습니다. 환경 변수에 GEMINI_API_KEY_1, GEMINI_API_KEY_2, GEMINI_API_KEY_3 중 하나 이상을 설정해주세요."
+                return None, None, None, None, "API 키가 설정되지 않았습니다. 환경 변수에 GEMINI_API_KEY_1, GEMINI_API_KEY_2, GEMINI_API_KEY_3, GEMINI_API_KEY_4, GEMINI_API_KEY_5 중 하나 이상을 설정해주세요."
             result_img, status, _ = generate_product_image(image, bg_type, simple, studio, nature, indoor, special, product_name, additional_info)
             return result_img, None, None, None, status
 
@@ -1091,7 +1102,7 @@ def create_app():
             if actual_api_key:
                 os.environ["GEMINI_API_KEY"] = actual_api_key  # 환경 변수 업데이트
             else:
-                return None, None, None, None, "API 키가 설정되지 않았습니다. 환경 변수에 GEMINI_API_KEY_1, GEMINI_API_KEY_2, GEMINI_API_KEY_3 중 하나 이상을 설정해주세요."
+                return None, None, None, None, "API 키가 설정되지 않았습니다. 환경 변수에 GEMINI_API_KEY_1, GEMINI_API_KEY_2, GEMINI_API_KEY_3, GEMINI_API_KEY_4, GEMINI_API_KEY_5 중 하나 이상을 설정해주세요."
             
             img1, img2, img3, img4, combined_status, _ = generate_product_images(image, bg_type, simple, studio, nature, indoor, special, product_name, additional_info)