Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import logging
 import re
 from io import BytesIO
+import time
 
 from google import genai
 from google.genai import types
@@ -146,9 +147,10 @@ def preprocess_prompt(prompt, image1, image2, image3):
     prompt += " 이미지를 생성해주세요."
     return prompt
 
-def generate_with_images(prompt, images):
+def generate_with_images(prompt, images, variation_index=0):
     """
     API 호출을 통해 이미지를 생성하고 결과 이미지를 반환합니다.
+    variation_index로 다양한 변화를 줍니다.
     """
     try:
         api_key = os.environ.get("GEMINI_API_KEY")
@@ -156,7 +158,18 @@ def generate_with_images(prompt, images):
             return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
 
         client = genai.Client(api_key=api_key)
-        logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}")
+        logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}, 변형 인덱스: {variation_index}")
+
+        # 변형 인덱스에 따라 프롬프트에 약간의 변화 추가
+        variation_suffixes = [
+            " 첫 번째 변형으로 만들어주세요.",
+            " 두 번째 변형으로 더 생생하게 만들어주세요.",
+            " 세 번째 변형으로 더 창의적으로 만들어주세요.",
+            " 네 번째 변형으로 색감을 더 강조해서 만들어주세요."
+        ]
+        
+        if variation_index < len(variation_suffixes):
+            prompt = prompt + variation_suffixes[variation_index]
 
         contents = [prompt]
         for idx, img in enumerate(images, 1):
@@ -198,7 +211,7 @@ def generate_with_images(prompt, images):
         logger.exception("이미지 생성 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
-def process_images_with_prompt(image1, image2, image3, prompt):
+def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0):
     """
     3개의 이미지와 프롬프트를 처리하여 최종 영어 프롬프트(final_prompt)를 생성한 후,
     API를 호출하여 결과 이미지를 반환합니다.
@@ -226,27 +239,60 @@ def process_images_with_prompt(image1, image2, image3, prompt):
                 final_prompt = "Please creatively composite these three images, combining their main elements into a cohesive and natural scene."
                 logger.info("Default prompt generated for three images")
 
-        result_img, status = generate_with_images(final_prompt, valid_images)
+        result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
         return result_img, status, final_prompt
     except Exception as e:
         logger.exception("이미지 처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}", prompt
 
-def process_and_show_prompt(image1, image2, image3, prompt):
-    try:
-        result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt)
-        return result_img, status, final_prompt
-    except Exception as e:
-        logger.exception("처리 중 오류 발생:")
-        return None, f"오류 발생: {str(e)}", prompt
+def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
+    """
+    여러 개의 이미지를 차례대로 생성합니다.
+    """
+    results = []
+    statuses = []
+    prompts = []
+    
+    num_images = 4  # 생성할 이미지 수
+    
+    progress(0, desc="이미지 생성 준비 중...")
+    
+    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)
+        
+        if result_img is not None:
+            results.append(result_img)
+            statuses.append(f"이미지 #{i+1}: {status}")
+            prompts.append(f"이미지 #{i+1}: {final_prompt}")
+        else:
+            # 에러가 발생한 경우에도 결과 목록에 None을 추가
+            results.append(None)
+            statuses.append(f"이미지 #{i+1} 생성 실패: {status}")
+            prompts.append(f"이미지 #{i+1}: {final_prompt}")
+        
+        # API 호출 사이에 약간의 간격을 두어 속도 제한 방지
+        time.sleep(1)
+    
+    progress(1.0, desc="이미지 생성 완료!")
+    
+    # 결과가 없는 경우 빈 슬롯 채우기
+    while len(results) < 4:
+        results.append(None)
+    
+    # 상태 메시지와 프롬프트 정보 결합
+    combined_status = "\n".join(statuses)
+    combined_prompts = "\n".join(prompts)
+    
+    return results[0], results[1], results[2], results[3], combined_status, combined_prompts
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>간단한 이미지 생성기 (예제 포함)</h1>
-            <p>이미지를 업로드하거나 아래 예제 버튼을 클릭��면 처리된 결과가 down 폴더에 저장됩니다.</p>
+            <h1>4장 이미지 생성기 (예제 포함)</h1>
+            <p>이미지를 업로드하고 "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.</p>
         </div>
         """
     )
@@ -268,11 +314,22 @@ with gr.Blocks() as demo:
                 clothes_change_btn = gr.Button("가상 상품착용")
                 background_change_btn = gr.Button("배경바꾸기")
                 composite_product_btn = gr.Button("부분지우기")
-            submit_btn = gr.Button("이미지 생성", variant="primary")
+            submit_btn = gr.Button("이미지 생성 (4장)", variant="primary")
+    
+    with gr.Row():
         with gr.Column():
-            output_image = gr.Image(label="생성된 이미지")
-            output_text = gr.Textbox(label="상태 메시지")
-            prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
+            output_image1 = gr.Image(label="생성된 이미지 #1")
+        with gr.Column():
+            output_image2 = gr.Image(label="생성된 이미지 #2")
+    
+    with gr.Row():
+        with gr.Column():
+            output_image3 = gr.Image(label="생성된 이미지 #3")
+        with gr.Column():
+            output_image4 = gr.Image(label="생성된 이미지 #4")
+    
+    output_text = gr.Textbox(label="상태 메시지")
+    prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
     
     # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
     image_change_btn.click(
@@ -302,9 +359,9 @@ with gr.Blocks() as demo:
     )
     
     submit_btn.click(
-        fn=process_and_show_prompt,
+        fn=generate_multiple_images,
         inputs=[image1_input, image2_input, image3_input, prompt_input],
-        outputs=[output_image, output_text, prompt_display],
+        outputs=[output_image1, output_image2, output_image3, output_image4, output_text, prompt_display],
     )
     
     gr.Markdown(
@@ -314,6 +371,7 @@ with gr.Blocks() as demo:
         1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
         2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
         3. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구로 입력하시면 됩니다.
+        4. **다양한 이미지**: "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.
         
         > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
         """