Update app.py

app.py

@@ -197,7 +197,7 @@ if not INDOOR_BACKGROUNDS:
 if not ABSTRACT_BACKGROUNDS:
     ABSTRACT_BACKGROUNDS = {"네온 조명": "neon lights"}
 
-# ------------------- 시스템 인스트럭션 수정 (상품 포커스 및 고화질 렌더링 추가) -------------------
+# ------------------- 시스템 인스트럭션 (상품 포커스 및 고화질 렌더링) -------------------
 def generate_system_instruction():
     return """당신은 상품 이미지의 배경을 변경하기 위한 고품질 프롬프트를 생성하는 전문가입니다.
 사용자가 제공하는 상품명, 배경 유형, 추가 요청사항을 바탕으로 미드저니(Midjourney)에 사용할 수 있는 상세하고 전문적인 프롬프트를 영어로 생성해주세요.
@@ -310,23 +310,20 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstra
 
 # --- 프롬프트에서 오직 Midjourney 프롬프트 텍스트만 추출하는 함수 ---
 def filter_prompt_only(prompt):
-    # "Highly detailed commercial photography" 문자열부터 시작하는 부분을 찾음
     idx = prompt.find("Highly detailed commercial photography")
     if idx != -1:
         prompt_text = prompt[idx:].strip()
-        # "--ar 1:1 --s 750 --q 2" 파라미터까지 포함한 후 그 이후 설명이 있다면 제거
         end_idx = prompt_text.rfind("--ar 1:1 --s 750 --q 2")
         if end_idx != -1:
-            # 파라미터 텍스트 길이만큼 더해서 잘라냄
             end_idx += len("--ar 1:1 --s 750 --q 2")
             prompt_text = prompt_text[:end_idx].strip()
         return prompt_text
     return prompt.strip()
 
-# ------------------- 통합 기능: 프롬프트 생성 후 이미지 생성 -------------------
-def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_info):
+# ------------------- 통합 기능: 4장 이미지 생성 -------------------
+def generate_product_images_multiple(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_info):
     if image is None:
-        return None, "이미지를 업로드해주세요.", "이미지를 업로드 후 프롬프트를 생성해주세요."
+        return None, None, None, None, "이미지를 업로드해주세요.", "이미지를 업로드 후 프롬프트를 생성해주세요."
     product_name = product_name.strip() or "제품"
     background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, abstract)
     # Gemini API로 프롬프트 생성
@@ -339,18 +336,23 @@ def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstr
             "2. 코드 내 직접 입력: GEMINI_API_KEY = \"your-api-key\"\n"
             "키 발급: https://makersuite.google.com/"
         )
-        return None, warning_msg, warning_msg
-    # 생성된 프롬프트에서 오직 Midjourney 프롬프트 텍스트 블록만 추출
+        return None, None, None, None, warning_msg, warning_msg
     final_prompt = filter_prompt_only(generated_prompt)
-    # 추출된 프롬프트를 사용해 이미지 생성
-    result_image, status = process_images_with_prompt(image, final_prompt, 0)[:2]
-    return result_image, status, final_prompt
+    images_list = []
+    statuses = []
+    for i in range(4):
+        result_img, status, _ = process_images_with_prompt(image, final_prompt, i)
+        images_list.append(result_img)
+        statuses.append(f"이미지 #{i+1}: {status}")
+        time.sleep(1)
+    combined_status = "\n".join(statuses)
+    return images_list[0], images_list[1], images_list[2], images_list[3], combined_status, final_prompt
 
 # ------------------- Gradio 인터페이스 구성 -------------------
 def create_app():
-    with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 및 이미지 생성") as demo:
-        gr.Markdown("# 고급 상품 이미지 배경 프롬프트 및 이미지 생성")
-        gr.Markdown("상품 이미지를 업로드하고, 제품명, 배경 옵션, 추가 요청사항을 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 그 프롬프트로 이미지를 생성합니다. 생성된 프롬프트는 불필요한 설명 없이 오직 Midjourney 프롬프트 텍스트만 표시됩니다.")
+    with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 및 4장 이미지 생성") as demo:
+        gr.Markdown("# 고급 상품 이미지 배경 프롬프트 및 4장 이미지 생성")
+        gr.Markdown("상품 이미지를 업로드하고, 제품명, 배경 옵션, 추가 요청사항을 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 그 프롬프트를 기반으로 4장의 이미지가 순차적으로 생성됩니다. 생성된 프롬프트는 불필요한 설명 없이 오직 Midjourney 프롬프트 텍스트만 표시됩니다.")
         with gr.Row():
             with gr.Column(scale=1):
                 product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 텀블러 등", interactive=True)
@@ -414,15 +416,18 @@ def create_app():
                     outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown]
                 )
                 image_input = gr.Image(label="상품 이미지 업로드", type="pil")
-                generate_btn = gr.Button("프롬프트 및 이미지 생성", variant="primary")
+                generate_multiple_btn = gr.Button("4장 이미지 생성", variant="primary")
             with gr.Column(scale=1):
-                prompt_output = gr.Textbox(label="생성된 프롬프트 (영어)", lines=6)
-                image_output = gr.Image(label="생성된 이미지", type="filepath")
+                image_output1 = gr.Image(label="이미지 #1", type="filepath")
+                image_output2 = gr.Image(label="이미지 #2", type="filepath")
+                image_output3 = gr.Image(label="이미지 #3", type="filepath")
+                image_output4 = gr.Image(label="이미지 #4", type="filepath")
                 status_output = gr.Textbox(label="결과 정보", lines=3)
-        generate_btn.click(
-            fn=generate_product_image,
+                prompt_output = gr.Textbox(label="생성된 프롬프트 (영어)", lines=6)
+        generate_multiple_btn.click(
+            fn=generate_product_images_multiple,
             inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown, product_name, additional_info],
-            outputs=[image_output, status_output, prompt_output]
+            outputs=[image_output1, image_output2, image_output3, image_output4, status_output, prompt_output]
         )
     return demo