Update app.py

app.py

@@ -224,8 +224,167 @@ def process_images_with_prompt(image1, image2, image3, prompt, bg_options=None):
         logger.exception("이미지 처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}", prompt
 
+def update_preview(product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood):
+    preview = f"상품명: {product_name}\n"
+    preview += f"제품군: {product_category}\n"
+    preview += f"배경 유형: {background_type}\n"
+    preview += f"세부 스타일: {background_style}\n"
+    
+    if lighting:
+        preview += f"조명: {lighting}\n"
+    if texture and texture != "선택 안함":
+        preview += f"배경 질감: {texture}\n"
+    if color_tone and color_tone != "선택 안함":
+        preview += f"색감: {color_tone}\n"
+    if mood and mood != "선택 안함":
+        preview += f"분위기: {mood}\n"
+    
+    # 선택 사항을 기반으로 LLM 스타일 프롬프트 생성
+    llm_prompt = generate_llm_prompt(product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood)
+    
+    preview += f"\n생성된 LLM 프롬프트:\n{llm_prompt}"
+    
+    return preview, llm_prompt
+
+# LLM 스타일 프롬프트 생성 함수
+def generate_llm_prompt(product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood):
+    # 기본 프롬프트 템플릿
+    prompt = f"{product_name}을(를) "
+    
+    # 배경 유형 및 스타일 추가
+    if background_type == "자연 배경":
+        if background_style == "숲":
+            prompt += "울창한 숲속의 자연 환경에 "
+        elif background_style == "해변":
+            prompt += "청량한 해변가의 모래 위에 "
+        elif background_style == "산":
+            prompt += "웅장한 산의 정상에서 맑은 하늘을 배경으로 "
+        elif background_style == "들판":
+            prompt += "넓게 펼쳐진 초원 위에 "
+        elif background_style == "계곡":
+            prompt += "시원한 계곡물이 흐르는 바위 위에 "
+        elif background_style == "열대 정원":
+            prompt += "생동감 넘치는 열대 식물로 둘러싸인 공간에 "
+        elif background_style == "사막":
+            prompt += "끝없이 펼쳐진 황금빛 사막 위에 "
+        elif background_style == "눈 풍경":
+            prompt += "하얀 눈으로 덮인 순백의 풍경 위에 "
+        else:
+            prompt += f"{background_style} 자연 환경 속에 "
+    elif background_type == "실내 배경":
+        if background_style == "거실":
+            prompt += "모던하고 세련된 거실 공간의 테이블 위에 "
+        elif background_style == "주방":
+            prompt += "깔끔한 주방 카운터 위에 "
+        elif background_style == "서재":
+            prompt += "지적인 분위기의 서재 책상 위에 "
+        elif background_style == "화장대":
+            prompt += "세련된 화장대 위에 "
+        elif background_style == "호텔 로비":
+            prompt += "고급스러운 호텔 로비의 대리석 테이블 위에 "
+        elif background_style == "카페":
+            prompt += "감성적인 카페 테이블 위에 "
+        elif background_style == "미니멀 공간":
+            prompt += "미니멀한 디자인의 깔끔한 공간에 "
+        else:
+            prompt += f"{background_style} 실내 공간에 "
+    elif background_type == "특수 배경":
+        if background_style == "도시 풍경":
+            prompt += "현대적인 도시 스카이라인을 배경으로 "
+        elif background_style == "루프탑":
+            prompt += "도시 전경이 보이는 루프탑 테이블 위에 "
+        elif background_style == "해안가":
+            prompt += "아름다운 해안가의 절벽 위에 "
+        elif background_style == "모던 스튜디오":
+            prompt += "프로페셔널한 모던 스튜디오 세트에 "
+        elif background_style == "빈티지 공간":
+            prompt += "클래식한 분위기의 빈티지 공간에 "
+        elif background_style == "미니어처 공간":
+            prompt += "정교하게 꾸며진 미니어처 세트 위에 "
+        else:
+            prompt += f"{background_style} 특수 환경에 "
+    
+    # 조명, 질감, 색감, 분위기 추가
+    lighting_text = ""
+    if lighting == "자연광":
+        lighting_text = "부드러운 자연광이 비치는 가운데"
+    elif lighting == "따뜻한 조명":
+        lighting_text = "따뜻한 황금빛 조명이 감싸는 가운데"
+    elif lighting == "그림자 있는 조명":
+        lighting_text = "드라마틱한 그림자를 만드는 조명 아래"
+    elif lighting == "미니멀 조명":
+        lighting_text = "깔끔하고 미니멀한 조명 아래"
+    
+    if lighting_text:
+        prompt += lighting_text + " "
+    
+    # 질감 추가
+    if texture and texture != "선택 안함":
+        if texture == "흰 벽":
+            prompt += "깔끔한 흰 벽을 배경으로 "
+        elif texture == "대리석":
+            prompt += "고급스러운 대리석 질감 위에 "
+        elif texture == "나무":
+            prompt += "따뜻한 느낌의 원목 질감 위에 "
+        elif texture == "벽돌":
+            prompt += "클래식한 벽돌 질감을 배경으로 "
+        elif texture == "흰 천":
+            prompt += "부드러운 흰 천 위에 "
+        elif texture == "시멘트":
+            prompt += "모던한 시멘트 질감 위에 "
+    
+    # 색감 추가
+    if color_tone and color_tone != "선택 안함":
+        if color_tone == "화이트 톤":
+            prompt += "순백의 깔끔한 색감으로 "
+        elif color_tone == "블루 톤":
+            prompt += "시원한 블루 톤의 색감으로 "
+        elif color_tone == "따뜻한 색상":
+            prompt += "따뜻한 파스텔 색감으로 "
+        elif color_tone == "모노톤":
+            prompt += "세련된 모노톤 색감으로 "
+        elif color_tone == "파스텔":
+            prompt += "부드러운 파스텔 색감으로 "
+    
+    # 분위기 추가
+    if mood and mood != "선택 안함":
+        if mood == "포근한":
+            prompt += "포근하고 아늑한 분위기 속에서 "
+        elif mood == "심플한":
+            prompt += "심플하고 깔끔한 분위기 속에서 "
+        elif mood == "모던한":
+            prompt += "모던하고 세련된 분위기 속에서 "
+        elif mood == "내추럴":
+            prompt += "자연스럽고 편안한 분위기 속에서 "
+        elif mood == "미니멀":
+            prompt += "미니멀리즘의 정제된 분위기 속에서 "
+    
+    # 마무리
+    if product_category == "화장품":
+        prompt += "완벽히 디테일이 살아있는 화장품 제품을 고급스럽게 연출한 상업용 제품 사진"
+    elif product_category == "가방":
+        prompt += "질감과 디자인이 돋보이는 가방 제품을 고급스럽게 연출한 상업용 제품 사진"
+    elif product_category == "액세서리":
+        prompt += "섬세한 디테일이 빛나는 액세서리를 고급스럽게 연출한 상업용 제품 사진"
+    elif product_category == "보석류":
+        prompt += "화려한 광채가 돋보이는 보석 제품을 고급스럽게 연출한 상업용 제품 사진"
+    elif product_category == "홈 데코":
+        prompt += "공간에 어울리는 홈 데코 제품을 세련되게 연출한 상업용 제품 사진"
+    elif product_category == "신발":
+        prompt += "디자인과 질감이 돋보이는 신발을 세련되게 연출한 상업용 제품 사진"
+    elif product_category == "의류":
+        prompt += "소재와 실루엣이 돋보이는 의류를 세련되게 연출한 상업용 제품 사진"
+    elif product_category == "주방용품":
+        prompt += "실용성과 디자인이 돋보이는 주방용품을 세련되게 연출한 상업용 제품 사진"
+    elif product_category == "전자기기":
+        prompt += "첨단 기술과 디자인이 돋보이는 전자기기를 세련되게 연출한 상업용 제품 사진"
+    else:
+        prompt += f"디테일과 품질이 돋보이는 {product_category} 제품을 세련되게 연출한 상업용 제품 사진"
+    
+    return prompt
+
 def process_and_show_prompt(image1, image2, image3, prompt, product_name, product_category, 
-                            background_type, background_style, lighting, texture, color_tone, mood):
+                            background_type, background_style, lighting, texture, color_tone, mood, llm_prompt=None):
     try:
         # 배경 옵션을 딕셔너리로 구성
         bg_options = {
@@ -238,7 +397,13 @@ def process_and_show_prompt(image1, image2, image3, prompt, product_name, produc
             "color_tone": color_tone if color_tone != "선택 안함" else "",
             "mood": mood if mood != "선택 안함" else ""
         }
+        
+        # LLM 프롬프트가 있으면 사용, 없으면 기본 프롬프트 생성
+        if llm_prompt and llm_prompt.strip():
+            # LLM 프롬프트를 그대로 사용
+            prompt = llm_prompt
             
+        # 이미지 처리 및 프롬프트 생성
         result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt, bg_options)
         
         # 선택된 옵션을 문자열로 구성
@@ -288,23 +453,6 @@ def update_background_style_options(background_type):
     else:
         return gr.update(choices=["선택하세요"], value="선택하세요")
 
-# 선택 내용 미리보기 업데이트 함수
-def update_preview(product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood):
-    preview = f"상품명: {product_name}\n"
-    preview += f"제품군: {product_category}\n"
-    preview += f"배경 유형: {background_type}\n"
-    preview += f"세부 스타일: {background_style}\n"
-    
-    if lighting:
-        preview += f"조명: {lighting}\n"
-    if texture and texture != "선택 안함":
-        preview += f"배경 질감: {texture}\n"
-    if color_tone and color_tone != "선택 안함":
-        preview += f"색감: {color_tone}\n"
-    if mood and mood != "선택 안함":
-        preview += f"분위기: {mood}\n"
-    
-    return preview
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
@@ -417,6 +565,13 @@ with gr.Blocks() as demo:
                 visible=False
             )
             
+            # LLM 프롬프트 입력창
+            llm_prompt_input = gr.Textbox(
+                label="프롬프트 (선택 사항)",
+                lines=3,
+                visible=False
+            )
+            
             submit_btn = gr.Button("이미지 생성", variant="primary")
         
         with gr.Column(scale=1):
@@ -426,134 +581,113 @@ with gr.Blocks() as demo:
             kr_prompt_display = gr.Textbox(label="사용된 프롬프트 (한국어)", visible=True)
             options_summary = gr.Textbox(label="선택된 옵션 요약", lines=8, visible=True)
     
-    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
-    image_change_btn.click(
-        fn=lambda: "#1 이미지의 [다른 모습]으로 변경하라.",
-        inputs=[],
-        outputs=prompt_input
-    ).then(
-        fn=lambda: "이미지 변경",
-        inputs=[],
-        outputs=current_selection
-    ).then(
-        fn=toggle_background_options,
-        inputs=[current_selection],
-        outputs=[background_header, product_name, product_category, background_type, 
-                 background_style, additional_header, lighting, texture, color_tone, 
-                 mood, selection_preview]
-    )
-    
-    text_remove_btn.click(
-        fn=lambda: "#1 이미지의 [중국어를 모두]를 제거하라.",
-        inputs=[],
-        outputs=prompt_input
-    ).then(
-        fn=lambda: "글자지우기",
-        inputs=[],
-        outputs=current_selection
-    ).then(
-        fn=toggle_background_options,
-        inputs=[current_selection],
-        outputs=[background_header, product_name, product_category, background_type, 
-                 background_style, additional_header, lighting, texture, color_tone, 
-                 mood, selection_preview]
-    )
-    
-    clothes_change_btn.click(
-        fn=lambda: "#1이미지에서 [신체 비율과 포즈는 유지한 체] 의상[스타일과 색상을]#2, #3으로 변경해주세요.",
-        inputs=[],
-        outputs=prompt_input
-    ).then(
-        fn=lambda: "옷바꾸기",
-        inputs=[],
-        outputs=current_selection
-    ).then(
-        fn=toggle_background_options,
-        inputs=[current_selection],
-        outputs=[background_header, product_name, product_category, background_type, 
-                 background_style, additional_header, lighting, texture, color_tone, 
-                 mood, selection_preview]
-    )
-    
     # 배경 변경 버튼 클릭 이벤트를 더 간단하게 수정
     def show_background_options():
         return ("배경 변경", "배경바꾸기", 
                 gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), 
                 gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), 
                 gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),
-                gr.update(visible=True), "")  # 마지막에 selection_preview 내용 추가
+                gr.update(visible=True), "", "")  # 마지막에 selection_preview와 llm_prompt 내용 추가
     
     background_change_btn.click(
         fn=show_background_options,
         inputs=[],
         outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
                  background_type, background_style, additional_header, lighting, texture, 
-                 color_tone, mood, selection_preview]
+                 color_tone, mood, selection_preview, llm_prompt_input]
     )
     
-    composite_product_btn.click(
-        fn=lambda: "이미지 #1의 모델이 필요에 따라 자세나 표정을 유동적으로 변형한 후, 이미지 #2(또는 #3)의 상품을 최대한 원래 형태와 디테일을 유지하면서 자연스럽게 착용한 모습을 합성하라.",
+    # 다른 기능 버튼들도 간단한 방식으로 수정
+    def hide_background_options(prompt_value):
+        return (prompt_value, "기타",
+                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), 
+                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), 
+                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
+                gr.update(visible=False), "", "")  # 마지막에 selection_preview와 llm_prompt 내용 추가
+    
+    image_change_btn.click(
+        fn=lambda: hide_background_options("#1 이미지의 [다른 모습]으로 변경하라."),
         inputs=[],
-        outputs=prompt_input
-    ).then(
-        fn=lambda: "상품합성",
+        outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
+                 background_type, background_style, additional_header, lighting, texture, 
+                 color_tone, mood, selection_preview, llm_prompt_input]
+    )
+    
+    text_remove_btn.click(
+        fn=lambda: hide_background_options("#1 이미지의 [중국어를 모두]를 제거하라."),
         inputs=[],
-        outputs=current_selection
-    ).then(
-        fn=toggle_background_options,
-        inputs=[current_selection],
-        outputs=[background_header, product_name, product_category, background_type, 
-                 background_style, additional_header, lighting, texture, color_tone, 
-                 mood, selection_preview]
+        outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
+                 background_type, background_style, additional_header, lighting, texture, 
+                 color_tone, mood, selection_preview, llm_prompt_input]
+    )
+    
+    clothes_change_btn.click(
+        fn=lambda: hide_background_options("#1이미지에서 [신체 비율과 포즈는 유지한 체] 의상[스타일과 색상을]#2, #3으로 변경해주세요."),
+        inputs=[],
+        outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
+                 background_type, background_style, additional_header, lighting, texture, 
+                 color_tone, mood, selection_preview, llm_prompt_input]
+    )
+    
+    composite_product_btn.click(
+        fn=lambda: hide_background_options("이미지 #1의 모델이 필요에 따라 자세나 표정을 유동적으로 변형한 후, 이미지 #2(또는 #3)의 상품을 최대한 원래 형태와 디테일을 유지하면서 자연스럽게 착용한 모습을 합성하라."),
+        inputs=[],
+        outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
+                 background_type, background_style, additional_header, lighting, texture, 
+                 color_tone, mood, selection_preview, llm_prompt_input]
     )
     
-    # 배경 유형에 따라 세부 스타일 옵션 업데이트
+    # 선택 옵션 변경 시 미리보기와 LLM 프롬프트 업데이트
+    def update_all_previews(product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood):
+        preview, llm_prompt = update_preview(product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood)
+        return preview, llm_prompt
+    
+    # 배경 유형에 따라 세부 스타일 옵션 업데이트 (먼저 수행)
     background_type.change(
         fn=update_background_style_options,
         inputs=[background_type],
         outputs=[background_style]
+    ).then(
+        fn=update_all_previews,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview, llm_prompt_input]
     )
     
-    # 선택 내용 미리보기 업데이트 (모든 선택 컴포넌트의 변경 이벤트에 연결)
+    # 모든 선택 컴포넌트의 변경 이벤트에 연결
     product_name.change(
-        fn=update_preview,
+        fn=update_all_previews,
         inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
+        outputs=[selection_preview, llm_prompt_input]
     )
     product_category.change(
-        fn=update_preview,
-        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
-    )
-    background_type.change(
-        fn=update_preview,
+        fn=update_all_previews,
         inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
+        outputs=[selection_preview, llm_prompt_input]
     )
     background_style.change(
-        fn=update_preview,
+        fn=update_all_previews,
         inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
+        outputs=[selection_preview, llm_prompt_input]
     )
     lighting.change(
-        fn=update_preview,
+        fn=update_all_previews,
         inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
+        outputs=[selection_preview, llm_prompt_input]
     )
     texture.change(
-        fn=update_preview,
+        fn=update_all_previews,
         inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
+        outputs=[selection_preview, llm_prompt_input]
     )
     color_tone.change(
-        fn=update_preview,
+        fn=update_all_previews,
         inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
+        outputs=[selection_preview, llm_prompt_input]
     )
     mood.change(
-        fn=update_preview,
+        fn=update_all_previews,
         inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
-        outputs=[selection_preview]
+        outputs=[selection_preview, llm_prompt_input]
     )
     
     # 이미지 생성 버튼 클릭 이벤트
@@ -561,7 +695,7 @@ with gr.Blocks() as demo:
         fn=process_and_show_prompt,
         inputs=[image1_input, image2_input, image3_input, prompt_input, 
                 product_name, product_category, background_type, background_style, 
-                lighting, texture, color_tone, mood],
+                lighting, texture, color_tone, mood, llm_prompt_input],
         outputs=[output_image, output_text, prompt_display, options_summary, kr_prompt_display],
     )
     
@@ -569,12 +703,14 @@ with gr.Blocks() as demo:
         """
         ### 사용 방법:
         
-        1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
-        2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
-        3. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구로 입력하시면 됩니다.
-        4. **배경 변경**: 배경바꾸기 버튼을 클릭하면 추가 옵션이 나타납니다.
+        1. 상품 이미지를 업로드하세요 (#1 이미지).
+        2. "배경바꾸기" 버튼을 클릭하면 배경 변경 옵션이 나타납니다.
+        3. 상품명을 입력하고 제품군, 배경 유형, 배경 세부 스타일을 선택하세요.
+        4. 추가 옵션(조명, 질감, 색감, 분위기)을 선택하면 더 세밀한 배경을 생성할 수 있습니다.
+        5. "이미지 생성" 버튼을 클릭하면 선택한 옵션에 따라 배경이 변경된 이미지가 생성됩니다.
+        6. 프롬프트 입력란에 직접 프롬프트를 작성할 수도 있습니다.
         
-        > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
+        > **팁**: 선택 내용 미리보기에서 생성된 LLM 프롬프트를 확인할 수 있습니다.
         """
     )