Update app.py

app.py

@@ -20,13 +20,60 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
+# ------------------- API 키 순환 시스템 -------------------
+API_KEYS = []  # API 키 목록
+current_key_index = 0  # 현재 사용 중인 키 인덱스
+
+def initialize_api_keys():
+    """API 키 목록을 초기화하는 함수"""
+    global API_KEYS
+    # 환경 변수에서 API 키 가져오기
+    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:
+        API_KEYS.append(key1)
+    if key2:
+        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", "")
+    if default_key and default_key not in API_KEYS:
+        API_KEYS.append(default_key)
+    
+    logger.info(f"API 키 {len(API_KEYS)}개가 로드되었습니다.")
+
+def get_next_api_key():
+    """다음 API 키를 가져오는 함수"""
+    global current_key_index
+    
+    if not API_KEYS:
+        return None
+    
+    # 현재 키 가져오기
+    api_key = API_KEYS[current_key_index]
+    
+    # 다음 키 인덱스로 업데이트
+    current_key_index = (current_key_index + 1) % len(API_KEYS)
+    
+    return api_key
+
 # ========== 이미지 생성기 관련 함수 ==========
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
 
-def translate_prompt_to_english(prompt, api_key=None):
-    # 기존 함수 유지하되 API 키 파라미터 추가
+def translate_prompt_to_english(prompt):
     if not re.search("[가-힣]", prompt):
         return prompt
     
@@ -35,16 +82,15 @@ def translate_prompt_to_english(prompt, api_key=None):
     prompt = prompt.replace("#3", "IMAGE_TAG_THREE")
     
     try:
-        # 사용자 입력 API 키 또는 환경변수 사용
-        used_api_key = api_key if api_key else os.environ.get("GEMINI_API_KEY")
-        if not used_api_key:
+        api_key = get_next_api_key()
+        if not api_key:
             logger.error("Gemini API 키가 설정되지 않았습니다.")
             prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
             prompt = prompt.replace("IMAGE_TAG_TWO", "#2")
             prompt = prompt.replace("IMAGE_TAG_THREE", "#3")
             return prompt
             
-        client = genai.Client(api_key=used_api_key)
+        client = genai.Client(api_key=api_key)
         translation_prompt = f"""
         Translate the following Korean text to English:
         
@@ -140,15 +186,13 @@ def preprocess_prompt(prompt, image1, image2, image3):
     prompt += " 이미지를 생성해주세요. 이미지에 텍스트나 글자를 포함하지 마세요."
     return prompt
 
-def generate_with_images(prompt, images, variation_index=0, api_key=None):
-    # API 키 파라미터 추가
+def generate_with_images(prompt, images, variation_index=0):
     try:
-        # 사용자 입력 API 키 또는 환경변수 사용
-        used_api_key = api_key if api_key else os.environ.get("GEMINI_API_KEY")
-        if not used_api_key:
-            return None, "API 키가 설정되지 않았습니다. API 키를 입력하거나 환경변수를 확인해주세요."
+        api_key = get_next_api_key()
+        if not api_key:
+            return None, "API 키가 설정되지 않았습니다. 환경 변수에 GEMINI_API_KEY_1, GEMINI_API_KEY_2, GEMINI_API_KEY_3, GEMINI_API_KEY_4, GEMINI_API_KEY_5 중 하나 이상을 설정해주세요."
 
-        client = genai.Client(api_key=used_api_key)
+        client = genai.Client(api_key=api_key)
         logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}, 변형 인덱스: {variation_index}")
 
         variation_suffixes = [
@@ -203,8 +247,7 @@ def generate_with_images(prompt, images, variation_index=0, api_key=None):
         logger.exception("이미지 생성 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
-def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0, max_retries=3, api_key=None):
-    # API 키 파라미터 추가
+def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0, max_retries=3):
     retry_count = 0
     last_error = None
     
@@ -218,7 +261,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
             if prompt and prompt.strip():
                 processed_prompt = preprocess_prompt(prompt, image1, image2, image3)
                 if re.search("[가-힣]", processed_prompt):
-                    final_prompt = translate_prompt_to_english(processed_prompt, api_key)
+                    final_prompt = translate_prompt_to_english(processed_prompt)
                 else:
                     final_prompt = processed_prompt
             else:
@@ -232,7 +275,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
                     final_prompt = "Please creatively composite these three images, combining their main elements into a cohesive and natural scene. Do not include any text or watermarks in the generated image."
                     logger.info("Default prompt generated for three images")
 
-            result_img, status = generate_with_images(final_prompt, valid_images, variation_index, api_key)
+            result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
             if result_img is not None:
                 return result_img, status, final_prompt
             else:
@@ -248,8 +291,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
     
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
 
-def generate_multiple_images(image1, image2, image3, prompt, api_key=None, progress=gr.Progress()):
-    # API 키 파라미터 추가
+def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
     results = []
     statuses = []
     prompts = []
@@ -261,7 +303,7 @@ def generate_multiple_images(image1, image2, image3, prompt, api_key=None, progr
     
     for i in range(num_images):
         progress((i / num_images), desc=f"{i+1}/{num_images} 이미지 생성 중...")
-        result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt, i, max_retries, api_key)
+        result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt, i, max_retries)
         
         if result_img is not None:
             results.append(result_img)
@@ -451,30 +493,6 @@ body {
     max-width: 100% !important;
 }
 
-/* 상단 헤더 */
-.app-header {
-    background: linear-gradient(135deg, var(--primary-color), var(--secondary-color));
-    color: white;
-    padding: 2rem;
-    border-radius: var(--border-radius);
-    margin-bottom: 1.5rem;
-    box-shadow: var(--shadow);
-    text-align: center;
-}
-
-.app-header h1 {
-    margin: 0;
-    font-size: 2.5rem;
-    font-weight: 700;
-    letter-spacing: -0.5px;
-}
-
-.app-header p {
-    margin: 0.75rem 0 0;
-    font-size: 1.1rem;
-    opacity: 0.9;
-}
-
 /* 패널 스타일링 */
 .gr-group {
     background-color: var(--card-bg);
@@ -605,85 +623,6 @@ body {
     margin-bottom: 1.2rem;
 }
 
-/* 사용자 매뉴얼 */
-.user-manual {
-    background-color: white;
-    padding: 2rem;
-    border-radius: var(--border-radius);
-    box-shadow: var(--shadow);
-    margin-top: 2rem;
-}
-
-.manual-title {
-    font-size: 1.8rem;
-    font-weight: 700;
-    color: var(--primary-color);
-    margin-bottom: 1.5rem;
-    text-align: center;
-    display: flex;
-    align-items: center;
-    justify-content: center;
-}
-
-.manual-title i {
-    margin-right: 0.5rem;
-    font-size: 1.8rem;
-}
-
-.manual-section {
-    margin-bottom: 1.5rem;
-    padding: 1.2rem;
-    background-color: #f8faff;
-    border-radius: calc(var(--border-radius) - 5px);
-}
-
-.manual-section-title {
-    font-size: 1.3rem;
-    font-weight: 700;
-    margin-bottom: 1rem;
-    color: var(--primary-color);
-    display: flex;
-    align-items: center;
-}
-
-.manual-section-title i {
-    margin-right: 0.5rem;
-    font-size: 1.2rem;
-}
-
-.manual-text {
-    font-size: 1rem;
-    line-height: 1.7;
-}
-
-.manual-text strong {
-    color: var(--accent-color);
-}
-
-.tip-box {
-    background-color: rgba(255, 107, 107, 0.1);
-    border-left: 3px solid var(--accent-color);
-    padding: 1rem 1.2rem;
-    margin: 1rem 0;
-    border-radius: 8px;
-}
-
-/* 로딩 애니메이션 */
-.progress-container {
-    background-color: rgba(255, 255, 255, 0.9);
-    border-radius: var(--border-radius);
-    padding: 2rem;
-    box-shadow: var(--shadow);
-    text-align: center;
-}
-
-.progress-bar {
-    height: 8px;
-    border-radius: 4px;
-    background: linear-gradient(90deg, var(--primary-color), var(--secondary-color));
-    margin: 1rem 0;
-}
-
 /* 메인 컨텐츠 스크롤바 */
 ::-webkit-scrollbar {
     width: 8px;
@@ -760,74 +699,6 @@ body {
     animation: fadeIn 0.5s ease-out;
 }
 
-/* 사용 가이드 스타일 */
-.guide-container {
-    background-color: var(--card-bg);
-    border-radius: var(--border-radius);
-    box-shadow: var(--shadow);
-    padding: 1.5rem;
-    margin-bottom: 1.5rem;
-    border: 1px solid rgba(0, 0, 0, 0.04);
-}
-
-.guide-title {
-    font-size: 1.5rem;
-    font-weight: 700;
-    color: var(--primary-color);
-    margin-bottom: 1.5rem;
-    padding-bottom: 0.5rem;
-    border-bottom: 2px solid var(--primary-color);
-    display: flex;
-    align-items: center;
-}
-
-.guide-title i {
-    margin-right: 0.8rem;
-    font-size: 1.5rem;
-}
-
-.guide-item {
-    display: flex;
-    margin-bottom: 1rem;
-    align-items: flex-start;
-}
-
-.guide-number {
-    background-color: var(--primary-color);
-    color: white;
-    width: 25px;
-    height: 25px;
-    border-radius: 50%;
-    display: flex;
-    align-items: center;
-    justify-content: center;
-    font-weight: bold;
-    margin-right: 10px;
-    flex-shrink: 0;
-}
-
-.guide-text {
-    flex: 1;
-    line-height: 1.6;
-}
-
-.guide-text a {
-    color: var(--primary-color);
-    text-decoration: underline;
-    font-weight: 600;
-}
-
-.guide-text a:hover {
-    color: var(--accent-color);
-}
-
-.guide-highlight {
-    background-color: rgba(251, 127, 13, 0.1);
-    padding: 2px 5px;
-    border-radius: 4px;
-    font-weight: 500;
-}
-
 /* Examples 섹션 스타일 */
 .examples-section {
     display: grid;
@@ -865,64 +736,10 @@ fontawesome_link = """
 <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css" crossorigin="anonymous" referrerpolicy="no-referrer" />
 """
 
-# 앱 헤더 HTML
-header_html = """
-<div class="app-header">
-    <h1>✨ 끝장AI - 이커머스전문 이미지생성기(Ver 1.3) ✨</h1>
-    <p style="font-size:18px; margin-top:10px;">
-        이커머스에 최적화된 이미지생성 서비스입니다. 다양한 메뉴를 제공하오니 업무에 잘 활용하시기 바랍니다.
-    </p>
-</div>
-"""
-
-# 섹션 간 여백용 HTML
-section_spacing_html = """
-<div style="margin-bottom: 2rem;"></div>
-"""
-
-# 이미지 생성기 사용 가이드 HTML
-image_generator_guide_html = """
-<div class="guide-container">
-    <div class="guide-title"><i class="fas fa-book"></i> 이미지 생성기 사용 가이드</div>
-    <div class="guide-item">
-        <div class="guide-number">1</div>
-        <div class="guide-text">
-            <strong class="guide-highlight">Gemini API 키가 필요합니다.</strong> 무료이니 <a href="https://aistudio.google.com/apikey" target="_blank">Google AI Studio</a>를 통해 발급받아 사용해주세요.
-        </div>
-    </div>
-    <div class="guide-item">
-        <div class="guide-number">2</div>
-        <div class="guide-text">
-            구글 이미지 생성 모델의 특성상 원본 유지가 잘되지 않습니다. 예시와 프롬프트 템플릿을 참고하여 여러번 이미지를 생성해주세요.
-        </div>
-    </div>
-    <div class="guide-item">
-        <div class="guide-number">3</div>
-        <div class="guide-text">
-            자세한 사용법은 <a href="http://xn--ai-7i4i628e.com/링크" target="_blank">활용법 보러가기</a>에서 확인해주세요.
-        </div>
-    </div>
-</div>
-"""
-
-# 이미지 편집기 사용 가이드 HTML
-image_editor_guide_html = """
-<div class="guide-container">
-    <div class="guide-title"><i class="fas fa-book"></i> 이미지 편집기 사용 가이드</div>
-    <div class="guide-item">
-        <div class="guide-number">1</div>
-        <div class="guide-text">
-            생성된 이미지를 추가적으로 다양한 필터를 적용하여 실시간으로 편집이 가능합니다.
-        </div>
-    </div>
-    <div class="guide-item">
-        <div class="guide-number">2</div>
-        <div class="guide-text">
-            자세한 사용법은 <a href="http://xn--ai-7i4i628e.com/링크" target="_blank">활용법 보러가기</a>에서 확인해주세요.
-        </div>
-    </div>
-</div>
-"""
+# 제목과 사용 가이드 제거
+header_html = ""
+image_generator_guide_html = ""
+image_editor_guide_html = ""
 
 # UI 구성
 with gr.Blocks(css=custom_css, theme=gr.themes.Default(
@@ -931,24 +748,17 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
     font=[gr.themes.GoogleFont("Noto Sans KR"), "ui-sans-serif", "system-ui"]
 )) as demo:
     gr.HTML(fontawesome_link)
-    gr.HTML(header_html)
+    # 제목 제거
+    # gr.HTML(header_html)
     
     with gr.Tabs(elem_classes="tabs") as tabs:
         # 이미지 생성기 탭
         with gr.Tab("✨ 이미지 생성기", elem_classes="tab-content"):
-            # 사용 가이드 섹션 추가
-            gr.HTML(image_generator_guide_html)
+            # 사용 가이드 섹션 제거
+            # gr.HTML(image_generator_guide_html)
             with gr.Row(equal_height=True):
                 with gr.Column(scale=1):
-                    # ======== API 키 설정 섹션 ========
-                    with gr.Group():                        
-                        gr.HTML('<div class="section-title"><i class="fas fa-key"></i> <span>API 키 설정</span></div>')
-                        api_key_input = gr.Textbox(
-                            type="password",
-                            label="Gemini API 키 (선택사항)",
-                            placeholder="API 키를 입력하세요",
-                            elem_classes="gr-text-input"
-                        )
+                    # API 키 입력 섹션 제거
                     
                     # ======== 이미지 업로드 및 설정 섹션 ========
                     with gr.Group():
@@ -1039,8 +849,8 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
         
         # 이미지 편집기 탭 추가
         with gr.Tab("🎨 이미지 편집기", elem_classes="tab-content"):
-            # 사용 가이드 섹션 추가
-            gr.HTML(image_editor_guide_html)
+            # 사용 가이드 섹션 제거
+            # gr.HTML(image_editor_guide_html)
             
             with gr.Row():
                 # 왼쪽 열: 비율 1
@@ -1154,21 +964,20 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
         outputs=prompt_input
     )    
     
-    
-    # 단일 이미지 생성 버튼 이벤트 연결 - API 키 입력값 활용
-    def generate_single_image_with_key(image1, image2, image3, prompt, api_key):
-        return process_images_with_prompt(image1, image2, image3, prompt, 0, 3, api_key)
+    # 단일 이미지 생성 버튼 이벤트 연결 - API 키 입력값 제거
+    def generate_single_image(image1, image2, image3, prompt):
+        return process_images_with_prompt(image1, image2, image3, prompt, 0, 3)
     
     submit_single_btn.click(
-        fn=lambda image1, image2, image3, prompt, api_key: generate_single_image_with_key(image1, image2, image3, prompt, api_key),
-        inputs=[image1_input, image2_input, image3_input, prompt_input, api_key_input],
+        fn=generate_single_image,
+        inputs=[image1_input, image2_input, image3_input, prompt_input],
         outputs=[output_image1, output_text, prompt_display],
     )
     
-    # 4장 이미지 생성 버튼 이벤트 연결 - API 키 입력값 활용
+    # 4장 이미지 생성 버튼 이벤트 연결 - API 키 입력값 제거
     submit_btn.click(
-        fn=lambda image1, image2, image3, prompt, api_key: generate_multiple_images(image1, image2, image3, prompt, api_key),
-        inputs=[image1_input, image2_input, image3_input, prompt_input, api_key_input],
+        fn=generate_multiple_images,
+        inputs=[image1_input, image2_input, image3_input, prompt_input],
         outputs=[output_image1, output_image2, output_image3, output_image4, output_text, prompt_display],
     )
     
@@ -1220,5 +1029,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
         outputs=download_output
     )
 
+# API 키 초기화 및 애플리케이션 실행
+initialize_api_keys()  # API 키 초기화 함수 호출
 demo.queue()
 demo.launch(share=False, inbrowser=True, width="100%")  # width 파라미터 추가