Update app.py

app.py

@@ -313,11 +313,12 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstra
             "english": "white background"
         }
 
-# --- 프롬프트에서 원하는 부분(프롬프트 텍스트 블록)만 추출하는 함수 ---
+# --- 프롬프트에서 오직 Midjourney 프롬프트 텍스트만 추출하는 함수 ---
 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()
+    # "Highly detailed commercial photography" 문자열부터 시작하는 부분을 반환
+    idx = prompt.find("Highly detailed commercial photography")
+    if idx != -1:
+        return prompt[idx:].strip()
     return prompt.strip()
 
 # ------------------- 통합 기능: 프롬프트 생성 후 이미지 생성 -------------------
@@ -337,7 +338,7 @@ def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstr
             "키 발급: https://makersuite.google.com/"
         )
         return None, warning_msg, warning_msg
-    # 생성된 프롬프트에서 오직 프롬프트 텍스트 블록만 추출
+    # 생성된 프롬프트에서 오직 Midjourney 프롬프트 텍스트 블록만 추출
     final_prompt = filter_prompt_only(generated_prompt)
     # 추출된 프롬프트를 사용해 이미지 생성
     result_image, status = process_images_with_prompt(image, final_prompt, 0)[:2]
@@ -347,7 +348,7 @@ def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstr
 def create_app():
     with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 및 이미지 생성") as demo:
         gr.Markdown("# 고급 상품 이미지 배경 프롬프트 및 이미지 생성")
-        gr.Markdown("상품 이미지를 업로드하고, 제품명, 배경 옵션, 추가 요청사항을 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 그 프롬프트로 이미지를 생성합니다. 생성된 프롬프트는 오직 아래의 Midjourney 프롬프트 텍스트만 표시됩니다.")
+        gr.Markdown("상품 이미지를 업로드하고, 제품명, 배경 옵션, 추가 요청사항을 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 그 프롬프트로 이미지를 생성합니다. 생성된 프롬프트는 오직 Midjourney 프롬프트 텍스트만 표시됩니다.")
         with gr.Row():
             with gr.Column(scale=1):
                 product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 텀블러 등", interactive=True)