Update app.py

app.py

@@ -27,7 +27,6 @@ def translate_prompt_to_english(prompt):
     입력된 프롬프트에 한글이 포함되어 있으면 Gemini‑2.0‑flash 모델을 사용하여 영어로 번역합니다.
     한글이 없으면 원본 프롬프트를 그대로 반환합니다.
     """
-    # 한글이 포함되어 있는지 확인
     if not re.search("[가-힣]", prompt):
         return prompt
     try:
@@ -67,12 +66,10 @@ def preprocess_prompt(prompt, image1, image2, image3):
     """
     프롬프트를 처리하고 기능 명령을 해석
     """
-    # 이미지 없는 참조 확인 및 처리
     has_img1 = image1 is not None
     has_img2 = image2 is not None
     has_img3 = image3 is not None
     
-    # #1, #2, #3 참조를 설명으로 변환 (이미지가 없는 경우 무시)
     if "#1" in prompt and not has_img1:
         prompt = prompt.replace("#1", "첫 번째 이미지(없음)")
     else:
@@ -88,7 +85,6 @@ def preprocess_prompt(prompt, image1, image2, image3):
     else:
         prompt = prompt.replace("#3", "세 번째 이미지")
     
-    # 기능 명령 해석
     if "1. 이미지 변경" in prompt:
         desc_match = re.search(r'#1을 "(.*?)"으로 바꿔라', prompt)
         if desc_match:
@@ -109,22 +105,16 @@ def preprocess_prompt(prompt, image1, image2, image3):
         prompt = "첫 번째 이미지의 인물 얼굴을 두 번째 이미지의 얼굴로 자연스럽게 교체해주세요. 얼굴의 표정과 특징은 두 번째 이미지를 따르되, 나머지 부분은 첫 번째 이미지를 유지해주세요."
     
     elif "4. 옷바꾸기" in prompt:
-        if "#3" in prompt or "또는 #3" in prompt:
-            prompt = "첫 번째 이미지의 인물 의상을 두 번째 또는 세 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 참조 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
-        else:
-            prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+        prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 변경해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
     
     elif "5. 배경바꾸기" in prompt:
-        prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 자연스럽게 교체해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
+        prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 변경해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
     
     elif "6. 이미지 합성(상품포함)" in prompt:
-        if "#3" in prompt or "또는 #3" in prompt:
-            prompt = "첫 번째 이미지와 두 번째, 세 번째 이미지를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 잘 보이도록 조화롭게 통합해주세요."
-        else:
-            prompt = "첫 번째 이미지와 두 번째 이미지를 자연스럽게 합성해주세요. 두 이미지의 주요 요소를 포함하고, 특히 상품이 잘 보이도록 조화롭게 통합해주세요."
+        prompt = "첫 번째 이미지와 두 번째 이미지(또는 세 번째 이미지)를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합해주세요."
     
     elif "7. 이미지 합성(스타일적용)" in prompt:
-        prompt = "첫 번째 이미지의 내용을 두 번째 이미지의 스타일로 변환해주세요. 첫 번째 이미지의 주요 피사체와 구도는 유지하되, 두 번째 이미지의 예술적 스타일, 색상, 질감을 적용해주세요."
+        prompt = "첫 번째 이미지의 내용을 두 번째 이미지의 스타일로 변경해주세요. 첫 번째 이미지의 주요 피사체와 구도는 유지하되, 두 번째 이미지의 예술적 스타일, 색상, 질감을 적용해주세요."
     
     elif "을 붉은색으로 바꿔라" in prompt or "를 붉은색으로 바꿔라" in prompt:
         prompt = "첫 번째 이미지를 붉은색 톤으로 변경해주세요. 전체적인 색상을 붉은 계열로 조정하고 자연스러운 느낌을 유지해주세요."
@@ -196,7 +186,7 @@ def process_images_with_prompt(image1, image2, image3, prompt):
         valid_images = [img for img in images if img is not None]
         if not valid_images:
             return None, "적어도 하나의 이미지를 업로드해주세요."
-        # 프롬프트가 비어있지 않고 한글이 포함된 경우 영어로 ���역
+        # 프롬프트가 한국어라면 번역 후 적용
         if prompt and re.search("[가-힣]", prompt):
             prompt = translate_prompt_to_english(prompt)
         if not prompt or not prompt.strip():
@@ -258,28 +248,67 @@ with gr.Blocks() as demo:
                 placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
                 label="프롬프트 (선택 사항)"
             )
-            with gr.Row():
-                gr.Markdown(
-                    "**선택 옵션:**\n\n"
-                    "- **이미지 변경**: `Change image #1 to [another look]`\n"
-                    "- **글자지우기**: `#1 Remove [all Chinese] from the image.`"
-                )
+            # 마크다운으로 선택 옵션 리스트 표시 (여러 항목)
+            gr.Markdown(
+                """
+                **선택 옵션:**
+
+                1. **이미지 변경:** `#1 이미지의 [다른 모습]으로 변경하라.`
+                2. **글자지우기:** `#1 이미지의 [중국어를 모두]를 제거하라.`
+                3. **얼굴바꾸기:** `#1 이미지의 인물 얼굴을 #2 이미지의 얼굴로 교체하라.`
+                4. **옷바꾸기:** `#1 이미지의 인물 의상을 #2 이미지의 의상으로 변경하라.`
+                5. **배경바꾸기:** `#1 이미지의 배경을 #2 이미지의 배경으로 변경하라.`
+                6. **이미지 합성(상품포함):** `#1 이미지와 #2 이미지(또는 #3 이미지)를 자연스럽게 합성하라.`
+                7. **이미지 합성(스타일적용):** `#1 이미지의 내용을 #2 이미지의 스타일로 변경하라.`
+                """
+            )
             with gr.Row():
                 image_change_btn = gr.Button("이미지 변경")
                 text_remove_btn = gr.Button("글자지우기")
+                face_swap_btn = gr.Button("얼굴바꾸기")
+                clothes_change_btn = gr.Button("옷바꾸기")
+                background_change_btn = gr.Button("배경바꾸기")
+                composite_product_btn = gr.Button("이미지 합성(상품포함)")
+                composite_style_btn = gr.Button("이미지 합성(스타일적용)")
             submit_btn = gr.Button("이미지 생성", variant="primary")
         with gr.Column():
             output_image = gr.Image(label="생성된 이미지")
             output_text = gr.Textbox(label="상태 메시지")
             prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
     
+    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (입력값은 한국어로 지정됨)
     image_change_btn.click(
-        fn=lambda: "Change image #1 to [another look]",
+        fn=lambda: "#1 이미지의 [다른 모습]으로 변경하라.",
         inputs=[],
         outputs=prompt_input
     )
     text_remove_btn.click(
-        fn=lambda: "#1 Remove [all Chinese] from the image.",
+        fn=lambda: "#1 이미지의 [중국어를 모두]를 제거하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    face_swap_btn.click(
+        fn=lambda: "#1 이미지의 인물 얼굴을 #2 이미지의 얼굴로 교체하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    clothes_change_btn.click(
+        fn=lambda: "#1 이미지의 인물 의상을 #2 이미지의 의상으로 변경하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    background_change_btn.click(
+        fn=lambda: "#1 이미지의 배경을 #2 이미지의 배경으로 변경하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    composite_product_btn.click(
+        fn=lambda: "#1 이미지와 #2 이미지(또는 #3 이미지)를 자연스럽게 합성하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    composite_style_btn.click(
+        fn=lambda: "#1 이미지의 내용을 #2 이미지의 스타일로 변경하라.",
         inputs=[],
         outputs=prompt_input
     )
@@ -297,8 +326,8 @@ with gr.Blocks() as demo:
         1. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
         2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
         3. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다.
-        4. **선택 옵션**: 위의 선택 옵션에서 원하는 항목을 클릭하면 프롬프트 입력란에 바로 적용됩니다.
-        5. **한국어 프롬프트 자동 번역**: 한국어로 입력 시 Gemini‑2.0‑flash 모델을 이용해 영어로 번역 후 적용됩니다.
+        4. **선택 옵션**: 위의 마크다운 형식의 옵션에서 원하는 항목을 클릭하면 프롬프트 입력란에 한국어 문구가 채워지고,
+           이후 Gemini‑2.0‑flash 모델을 통해 영어로 번역되어 적용됩니다.
         
         > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
         """