Update app.py

app.py

@@ -61,7 +61,7 @@ def translate_prompt_to_english(prompt):
         logger.exception("번역 중 오류 발생:")
         return prompt
 
-def preprocess_prompt(prompt, image1, image2, image3, background_options=None):
+def preprocess_prompt(prompt, image1, image2, image3, bg_options=None):
     """
     프롬프트를 처리하고 기능 명령을 해석
     """
@@ -105,13 +105,30 @@ def preprocess_prompt(prompt, image1, image2, image3, background_options=None):
 
     elif "5. 배경바꾸기" in prompt or "배경 변경" in prompt:
         # 배경 옵션이 있으면 그것을 기반으로 프롬프트 생성
-        if background_options and all(background_options.values()):
-            product_category = background_options.get("product_category", "일반 상품")
-            background_type = background_options.get("background_type", "자연 배경")
-            background_style = background_options.get("background_style", "숲")
+        if bg_options and all([bg_options.get("product_name"), bg_options.get("background_type"), bg_options.get("background_style")]):
+            product_name = bg_options.get("product_name", "")
+            product_category = bg_options.get("product_category", "일반 상품")
+            background_type = bg_options.get("background_type", "자연 배경")
+            background_style = bg_options.get("background_style", "숲")
+            lighting = bg_options.get("lighting", "자연광")
+            texture = bg_options.get("texture", "")
+            color_tone = bg_options.get("color_tone", "")
+            mood = bg_options.get("mood", "")
             
-            # 상품 카테고리별 특화된 프롬프트 생성
-            prompt = f"첫 번째 이미지의 {product_category} 상품은 그대로 유지하면서, 배경을 {background_type} 중 {background_style}으로 자연스럽게 변경해주세요. 상품의 색상, 질감, 형태는 완벽하게 보존하고 고품질 상품 사진 스타일로 만들어주세요."
+            # 상세 프롬프트 생성
+            prompt = f"{product_name} 상품 이미지의 배경을 {background_type} 중 {background_style}으로 자연스럽게 변경해주세요. "
+            
+            # 추가 세부 사항 포함
+            if lighting:
+                prompt += f"{lighting} 조명을 사용하고, "
+            if texture:
+                prompt += f"{texture} 질감을 표현하며, "
+            if color_tone:
+                prompt += f"{color_tone} : 전체 색감으로 처리하고, "
+            if mood:
+                prompt += f"{mood}한 분위기로 "
+            
+            prompt += f"연출해주세요. {product_category} 상품의 색상, 질감, 형태는 완벽하게 보존하고 고품질 상품 사진 스타일로 제작해주세요."
         else:
             prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 변경해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
 
@@ -173,7 +190,7 @@ def generate_with_images(prompt, images):
         logger.exception("이미지 생성 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
-def process_images_with_prompt(image1, image2, image3, prompt, background_options=None):
+def process_images_with_prompt(image1, image2, image3, prompt, bg_options=None):
     """
     3개의 이미지와 프롬프트를 처리하여 최종 영어 프롬프트(final_prompt)를 생성한 후,
     API를 호출하여 결과 이미지를 반환합니다.
@@ -185,7 +202,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, background_option
             return None, "적어도 하나의 이미지를 업로드해주세요.", ""
 
         if prompt and prompt.strip():
-            processed_prompt = preprocess_prompt(prompt, image1, image2, image3, background_options)
+            processed_prompt = preprocess_prompt(prompt, image1, image2, image3, bg_options)
             if re.search("[가-힣]", processed_prompt):
                 final_prompt = translate_prompt_to_english(processed_prompt)
             else:
@@ -207,35 +224,58 @@ def process_images_with_prompt(image1, image2, image3, prompt, background_option
         logger.exception("이미지 처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}", prompt
 
-def process_and_show_prompt(image1, image2, image3, prompt, product_category=None, background_type=None, background_style=None):
+def process_and_show_prompt(image1, image2, image3, prompt, product_name, product_category, 
+                            background_type, background_style, lighting, texture, color_tone, mood):
     try:
-        # 배경 옵션이 있을 경우 딕셔너리로 전달
-        background_options = None
-        if all([product_category, background_type, background_style]):
-            background_options = {
-                "product_category": product_category,
-                "background_type": background_type,
-                "background_style": background_style
-            }
+        # 배경 옵션을 딕셔너리로 구성
+        bg_options = {
+            "product_name": product_name,
+            "product_category": product_category,
+            "background_type": background_type,
+            "background_style": background_style,
+            "lighting": lighting,
+            "texture": texture,
+            "color_tone": color_tone,
+            "mood": mood
+        }
             
-        result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt, background_options)
+        result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt, bg_options)
         
         # 선택된 옵션을 문자열로 구성
-        options_summary = ""
-        if background_options:
-            options_summary = f"제품군: {product_category}\n배경 유형: {background_type}\n세부 스타일: {background_style}"
+        options_summary = f"상품명: {product_name}\n"
+        options_summary += f"제품군: {product_category}\n"
+        options_summary += f"배경 유형: {background_type}\n"
+        options_summary += f"세부 스타일: {background_style}\n"
+        
+        if lighting:
+            options_summary += f"조명: {lighting}\n"
+        if texture:
+            options_summary += f"배경 질감: {texture}\n"
+        if color_tone:
+            options_summary += f"색감: {color_tone}\n"
+        if mood:
+            options_summary += f"분위기: {mood}\n"
         
-        return result_img, status, final_prompt, options_summary
+        # 한국어와 영어 프롬프트 비교를 위해 원본 프롬프트 확인
+        kr_prompt = preprocess_prompt(prompt, image1, image2, image3, bg_options)
+        
+        return result_img, status, final_prompt, options_summary, kr_prompt
     except Exception as e:
         logger.exception("처리 중 오류 발생:")
-        return None, f"오류 발생: {str(e)}", prompt, ""
+        return None, f"오류 발생: {str(e)}", prompt, "", ""
 
 # 선택 옵션 표시/숨김 처리 함수
 def toggle_background_options(selection):
     if selection == "배경바꾸기":
-        return gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+        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))
     else:
-        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return (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))
 
 # 배경 유형에 따른 세부 스타일 옵션 업데이트
 def update_background_style_options(background_type):
@@ -248,13 +288,31 @@ 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:
+        preview += f"배경 질감: {texture}\n"
+    if color_tone:
+        preview += f"색감: {color_tone}\n"
+    if mood:
+        preview += f"분위기: {mood}\n"
+    
+    return preview
+
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>간단한 이미지 생성기 (예제 포함)</h1>
-            <p>이미지를 업로드하거나 아래 예제 버튼을 클릭하면 처리된 결과가 down 폴더에 저장됩니다.</p>
+            <h1>이커머스 상품 이미지 배경 생성기</h1>
+            <p>상품 이미지를 업로드하고 원하는 배경 옵션을 선택하여 새로운 배경 이미지를 생성하세요.</p>
         </div>
         """
     )
@@ -262,11 +320,12 @@ with gr.Blocks() as demo:
     current_selection = gr.State("")
     
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
             with gr.Row():
-                image1_input = gr.Image(type="pil", label="#1", image_mode="RGB")
-                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB")
-                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB")
+                image1_input = gr.Image(type="pil", label="#1 상품 이미지", image_mode="RGB")
+                image2_input = gr.Image(type="pil", label="#2 (선택사항)", image_mode="RGB")
+                image3_input = gr.Image(type="pil", label="#3 (선택사항)", image_mode="RGB")
+            
             prompt_input = gr.Textbox(
                 lines=3,
                 placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
@@ -277,12 +336,19 @@ with gr.Blocks() as demo:
                 image_change_btn = gr.Button("이미지 변경")
                 text_remove_btn = gr.Button("글자지우기")
                 clothes_change_btn = gr.Button("옷바꾸기")
-                background_change_btn = gr.Button("배경바꾸기")
+                background_change_btn = gr.Button("배경바꾸기", variant="primary")
                 composite_product_btn = gr.Button("상품합성")
             
-            # 배경 변경 관련 옵션들 (기본적으로 숨김 상태)
-            with gr.Row(visible=False) as background_options_row:
-                gr.HTML("<h3>배경 변경 옵션</h3>")
+            # 배경 변경 관련 섹션 (기본적으로 숨김 상태)
+            with gr.Row(visible=False) as background_header:
+                gr.HTML("<h3 style='margin-top: 20px; margin-bottom: 10px; width: 100%; text-align: center;'>배경 변경 옵션</h3>")
+            
+            # 상품명 입력
+            product_name = gr.Textbox(
+                label="상품명 입력",
+                placeholder="상품명을 입력해주세요",
+                visible=False
+            )
             
             # 제품군 카테고리 드롭다운
             product_category = gr.Dropdown(
@@ -308,13 +374,58 @@ with gr.Blocks() as demo:
                 visible=False
             )
             
+            # 추가 선택 옵션 헤더
+            with gr.Row(visible=False) as additional_header:
+                gr.HTML("<h4 style='margin-top: 15px; margin-bottom: 10px; width: 100%; text-align: center;'>추가 선택 옵션</h4>")
+            
+            # 조명 드롭다운
+            lighting = gr.Dropdown(
+                choices=["자연광", "따뜻한 조명", "그림자 있는 조명", "미니멀 조명"],
+                label="조명 선택",
+                value="자연광",
+                visible=False
+            )
+            
+            # 배경 질감 드롭다운
+            texture = gr.Dropdown(
+                choices=["흰 벽", "대리석", "나무", "벽돌", "흰 천", "시멘트"],
+                label="배경 질감 선택",
+                value="",
+                visible=False
+            )
+            
+            # 색감 드롭다운
+            color_tone = gr.Dropdown(
+                choices=["화이트 톤", "블루 톤", "따뜻한 색상", "모노톤", "파스텔"],
+                label="색감 선택",
+                value="",
+                visible=False
+            )
+            
+            # 분위기 드롭다운
+            mood = gr.Dropdown(
+                choices=["포근한", "심플한", "모던한", "내추럴", "미니멀"],
+                label="분위기 선택",
+                value="",
+                visible=False
+            )
+            
+            # 선택 내용 미리보기
+            selection_preview = gr.Textbox(
+                label="선택 내용 미리보기",
+                lines=8,
+                readonly=True,
+                visible=False
+            )
+            
             submit_btn = gr.Button("이미지 생성", variant="primary")
         
-        with gr.Column():
+        with gr.Column(scale=1):
             output_image = gr.Image(label="생성된 이미지")
             output_text = gr.Textbox(label="상태 메시지")
             prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
-            options_summary = gr.Textbox(label="선택된 옵션", visible=True)
+            kr_prompt_display = gr.Textbox(label="사용된 프롬프트 (한국어)", visible=True)
+            options_summary = gr.Textbox(label="선택된 옵션 요약", lines=8, visible=True)
     
     # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
     image_change_btn.click(
@@ -328,7 +439,9 @@ with gr.Blocks() as demo:
     ).then(
         fn=toggle_background_options,
         inputs=[current_selection],
-        outputs=[background_options_row, product_category, background_type, background_style]
+        outputs=[background_header, product_name, product_category, background_type, 
+                 background_style, additional_header, lighting, texture, color_tone, 
+                 mood, selection_preview]
     )
     
     text_remove_btn.click(
@@ -342,7 +455,9 @@ with gr.Blocks() as demo:
     ).then(
         fn=toggle_background_options,
         inputs=[current_selection],
-        outputs=[background_options_row, product_category, background_type, background_style]
+        outputs=[background_header, product_name, product_category, background_type, 
+                 background_style, additional_header, lighting, texture, color_tone, 
+                 mood, selection_preview]
     )
     
     clothes_change_btn.click(
@@ -356,7 +471,9 @@ with gr.Blocks() as demo:
     ).then(
         fn=toggle_background_options,
         inputs=[current_selection],
-        outputs=[background_options_row, product_category, background_type, background_style]
+        outputs=[background_header, product_name, product_category, background_type, 
+                 background_style, additional_header, lighting, texture, color_tone, 
+                 mood, selection_preview]
     )
     
     background_change_btn.click(
@@ -370,7 +487,9 @@ with gr.Blocks() as demo:
     ).then(
         fn=toggle_background_options,
         inputs=[current_selection],
-        outputs=[background_options_row, product_category, background_type, background_style]
+        outputs=[background_header, product_name, product_category, background_type, 
+                 background_style, additional_header, lighting, texture, color_tone, 
+                 mood, selection_preview]
     )
     
     composite_product_btn.click(
@@ -384,7 +503,9 @@ with gr.Blocks() as demo:
     ).then(
         fn=toggle_background_options,
         inputs=[current_selection],
-        outputs=[background_options_row, product_category, background_type, background_style]
+        outputs=[background_header, product_name, product_category, background_type, 
+                 background_style, additional_header, lighting, texture, color_tone, 
+                 mood, selection_preview]
     )
     
     # 배경 유형에 따라 세부 스타일 옵션 업데이트
@@ -394,13 +515,58 @@ with gr.Blocks() as demo:
         outputs=[background_style]
     )
     
+    # 선택 내용 미리보기 업데이트 (모든 선택 컴포넌트의 변경 이벤트에 연결)
+    product_name.change(
+        fn=update_preview,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview]
+    )
+    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,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview]
+    )
+    background_style.change(
+        fn=update_preview,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview]
+    )
+    lighting.change(
+        fn=update_preview,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview]
+    )
+    texture.change(
+        fn=update_preview,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview]
+    )
+    color_tone.change(
+        fn=update_preview,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview]
+    )
+    mood.change(
+        fn=update_preview,
+        inputs=[product_name, product_category, background_type, background_style, lighting, texture, color_tone, mood],
+        outputs=[selection_preview]
+    )
+    
     # 이미지 생성 버튼 클릭 이벤트
     submit_btn.click(
         fn=process_and_show_prompt,
-        inputs=[image1_input, image2_input, image3_input, prompt_input, product_category, background_type, background_style],
-        outputs=[output_image, output_text, prompt_display, options_summary],
+        inputs=[image1_input, image2_input, image3_input, prompt_input, 
+                product_name, product_category, background_type, background_style, 
+                lighting, texture, color_tone, mood],
+        outputs=[output_image, output_text, prompt_display, options_summary, kr_prompt_display],
     )
     
+    
     gr.Markdown(
         """
         ### 사용 방법: