Update app.py

app.py

@@ -144,7 +144,6 @@ 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를 통해 생성된 영어 텍스트라 가정
             final_prompt = prompt.strip()
             result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
             if result_img is not None:
@@ -198,24 +197,24 @@ if not INDOOR_BACKGROUNDS:
 if not ABSTRACT_BACKGROUNDS:
     ABSTRACT_BACKGROUNDS = {"네온 조명": "neon lights"}
 
-# ------------------- 시스템 인스트럭션 수정 (고화질 및 상품 포커스 추가) -------------------
+# ------------------- 시스템 인스트럭션 수정 (상품 포커스 및 고화질 렌더링 추가) -------------------
 def generate_system_instruction():
     return """당신은 상품 이미지의 배경을 변경하기 위한 고품질 프롬프트를 생성하는 전문가입니다.
 사용자가 제공하는 상품명, 배경 유형, 추가 요청사항을 바탕으로 미드저니(Midjourney)에 사용할 수 있는 상세하고 전문적인 프롬프트를 영어로 생성해주세요.
 다음 가이드라인을 반드시 준수해야 합니다:
 1. 상품을 "#1"로 지정하여 참조합니다. (예: "skincare tube (#1)")
 2. *** 매우 중요: 상품의 원래 특성(디자인, 색상, 형태, 로고, 패키지 등)은 어떤 상황에서도 절대 변경하지 않습니다. ***
-3. *** 상품의 본질적 특성을 유지하되, 상품에 포커스를 맞춰 모든 세부 사항이 선명하게 드러나도록 해야 하며, 고화질(ultra high resolution)로 렌더링되어 화질 저하 없이 표현되어야 합니다. ***
+3. *** 상품의 본질적 특성을 유지하되, 상품에 포커스를 맞춰 모든 세부 사항이 선명하게 드러나도록 하며, 고화질(ultra high resolution)로 렌더링되어 화질 저하 없이 표현되어야 합니다. ***
 4. 이미지 비율은 정확히 1:1(정사각형) 형식으로 지정합니다. 프롬프트에 "square format", "1:1 ratio" 또는 "aspect ratio 1:1"을 명시적으로 포함합니다.
 5. 상품은 반드시 정사각형 구도의 정중앙에 배치되어야 합니다.
 6. 상품을 이미지의 주요 초점으로 부각시키고, 상품의 비율이 전체 이미지에서 크게 차지하도록 합니다.
 7. 상품 이미지 컷아웃(#1)의 기본 형태와 색상은 유지하면서, 선택한 환경에 자연스럽게 통합되도록 합니다.
 8. 고급스러운 상업적 이미지를 위한 다음 환경 요소들을 포함하세요:
-   - 상품과 어울리는 주변 환경/배경 요소를 추가합니다.
-   - 환경의 조명 효과(림 라이트, 백라이트, 소프트박스 등)를 설명합니다.
-   - 상품이 환경에 자연스럽게 존재하는 것처럼 보이도록 적절한 그림자와 빛 표현을 포함합니다.
-   - 상품의 용도나 장점을 간접적으로 암시하는 배경 요소를 포함합니다.
-   - 프로페셔널한 상업 사진 효과(선택적 피사계 심도, 소프트 포커스, 스튜디오 조명 등)를 명시합니다.
+   - 상품과 어울리는 주변 환경/배경 요소
+   - 환경의 조명 효과(림 라이트, 백라이트, 소프트박스 등)
+   - 자연스러운 그림자와 빛 표현
+   - 상품의 용도나 장점을 암시하는 배경 요소
+   - 프로페셔널한 상업 사진 효과 (피사계 심도, 소프트 포커스, 스튜디오 조명 등)
 9. 프롬프트에 다음 요소들을 명시적으로 포함하세요:
    - "highly detailed commercial photography"
    - "award-winning product photography"
@@ -311,10 +310,17 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstra
 
 # --- 프롬프트에서 오직 Midjourney 프롬프트 텍스트만 추출하는 함수 ---
 def filter_prompt_only(prompt):
-    # "Highly detailed commercial photography" 문자열부터 시작하는 부분을 반환
+    # "Highly detailed commercial photography" 문자열부터 시작하는 부분을 찾음
     idx = prompt.find("Highly detailed commercial photography")
     if idx != -1:
-        return prompt[idx:].strip()
+        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()
 
 # ------------------- 통합 기능: 프롬프트 생성 후 이미지 생성 -------------------
@@ -344,7 +350,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)