Update app.py

app.py

@@ -144,7 +144,7 @@ def process_images_with_prompt(image1, prompt, variation_index=0, max_retries=3)
             valid_images = [img for img in images if img is not None]
             if not valid_images:
                 return None, "이미지를 업로드해주세요.", ""
-            # 이미 Gemini API로 생성된 프롬프트는 영어이므로 추가 번역은 생략
+            # 프롬프트는 이미 Gemini API를 통해 생성된 영어 텍스트라 가정
             final_prompt = prompt.strip()
             result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
             if result_img is not None:
@@ -187,7 +187,6 @@ NATURE_BACKGROUNDS = load_background_json("nature_backgrounds.json")
 INDOOR_BACKGROUNDS = load_background_json("indoor_backgrounds.json")
 ABSTRACT_BACKGROUNDS = load_background_json("abstract_backgrounds.json")
 
-# 배경이 로드되지 않은 경우 기본값 설정
 if not SIMPLE_BACKGROUNDS:
     SIMPLE_BACKGROUNDS = {"화이트 배경": "white background"}
 if not STUDIO_BACKGROUNDS:
@@ -314,6 +313,13 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstra
             "english": "white background"
         }
 
+# --- 프롬프트에서 원하는 부분(프롬프트 텍스트 블록)만 추출하는 함수 ---
+def filter_prompt_only(prompt):
+    # 예시로, 설명 부분("Explanation of Key Elements:")이 있다면 그 앞부분만 사용
+    if "Explanation of Key Elements:" in prompt:
+        return prompt.split("Explanation of Key Elements:")[0].strip()
+    return prompt.strip()
+
 # ------------------- 통합 기능: 프롬프트 생성 후 이미지 생성 -------------------
 def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_info):
     if image is None:
@@ -331,15 +337,17 @@ def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstr
             "키 발급: https://makersuite.google.com/"
         )
         return None, warning_msg, warning_msg
-    # 생성된 프롬프트를 프롬프트 입력창에 반영하고, 그 프롬프트로 이미지 생성
-    result_image, status, final_prompt = process_images_with_prompt(image, generated_prompt, 0)
+    # 생성된 프롬프트에서 오직 프롬프트 텍스트 블록만 추출
+    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
 
 # ------------------- Gradio 인터페이스 구성 -------------------
 def create_app():
-    with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 생성 및 이미지 생성") as demo:
-        gr.Markdown("# 고급 상품 이미지 배경 프롬프트 생성 및 이미지 생성")
-        gr.Markdown("상품 이미지를 업로드하고, 제품명 및 배경 옵션과 추가 요청사항을 입력하면 Gemini API를 이용하여 영어 프롬프트를 생성하고, 그 프롬프트로 이미지를 생성합니다.")
+    with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 및 이미지 생성") as demo:
+        gr.Markdown("# 고급 상품 이미지 배경 프롬프트 및 이미지 생성")
+        gr.Markdown("상품 이미지를 업로드하고, 제품명, 배경 옵션, 추가 요청사항을 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 그 프롬프트로 이미지를 생성합니다. 생성된 프롬프트는 오직 아래의 Midjourney 프롬프트 텍스트만 표시됩니다.")
         with gr.Row():
             with gr.Column(scale=1):
                 product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 텀블러 등", interactive=True)
@@ -389,7 +397,6 @@ def create_app():
                     lines=3,
                     interactive=True
                 )
-                # 배경 유형 변경 시 해당 드롭다운만 보이도록 업데이트
                 def update_dropdowns(bg_type):
                     return {
                         simple_dropdown: gr.update(visible=(bg_type == "심플 배경")),