Update app.py

app.py

@@ -101,9 +101,6 @@ def preprocess_prompt(prompt, image1, image2, image3):
         else:
             prompt = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
     
-    elif "3. 얼굴바꾸기" in prompt:
-        prompt = "첫 번째 이미지의 인물 얼굴을 두 번째 이미지의 얼굴로 자연스럽게 교체해주세요. 얼굴의 표정과 특징은 두 번째 이미지를 따르되, 나머지 부분은 첫 번째 이미지를 유지해주세요."
-    
     elif "4. 옷바꾸기" in prompt:
         prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 변경해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
     
@@ -113,12 +110,6 @@ def preprocess_prompt(prompt, image1, image2, image3):
     elif "6. 이미지 합성(상품포함)" in prompt:
         prompt = "첫 번째 이미지와 두 번째 이미지(또는 세 번째 이미지)를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합해주세요."
     
-    elif "7. 이미지 합성(스타일적용)" in prompt:
-        prompt = "첫 번째 이미지의 내용을 두 번째 이미지의 스타일로 변경해주세요. 첫 번째 이미지의 주요 피사체와 구도는 유지하되, 두 번째 이미지의 예술적 스타일, 색상, 질감을 적용해주세요."
-    
-    elif "을 붉은색으로 바꿔라" in prompt or "를 붉은색으로 바꿔라" in prompt:
-        prompt = "첫 번째 이미지를 붉은색 톤으로 변경해주세요. 전체적인 색상을 붉은 계열로 조정하고 자연스러운 느낌을 유지해주세요."
-    
     prompt += " 이미지를 생성해주세요."
     
     return prompt
@@ -179,49 +170,47 @@ def generate_with_images(prompt, images):
 
 def process_images_with_prompt(image1, image2, image3, prompt):
     """
-    3개의 이미지와 프롬프트를 처리하는 함수
+    3개의 이미지와 프롬프트를 처리하는 함수.
+    최종적으로 영어로 번역된 프롬프트(final_prompt)를 생성하여 API에 적용하고, 그 결과와 함께 반환합니다.
     """
     try:
         images = [image1, image2, image3]
         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():
+            return None, "적어도 하나의 이미지를 업로드해주세요.", ""
+        
+        # 프롬프트가 입력된 경우: 전처리 후 최종 영어 프롬프트 생성
+        if prompt and prompt.strip():
+            processed_prompt = preprocess_prompt(prompt, image1, image2, image3)
+            if re.search("[가-힣]", processed_prompt):
+                final_prompt = translate_prompt_to_english(processed_prompt)
+            else:
+                final_prompt = processed_prompt
+        else:
+            # 프롬프트 미입력 시, 이미지 개수에 따른 기본 프롬프트 (영어)
             if len(valid_images) == 1:
-                prompt = "Please creatively transform this image into a more vivid and artistic version."
+                final_prompt = "Please creatively transform this image into a more vivid and artistic version."
                 logger.info("Default prompt generated for single image")
             elif len(valid_images) == 2:
-                prompt = "Please seamlessly composite these two images, integrating their key elements harmoniously into a single image."
+                final_prompt = "Please seamlessly composite these two images, integrating their key elements harmoniously into a single image."
                 logger.info("Default prompt generated for two images")
             else:
-                prompt = "Please creatively composite these three images, combining their main elements into a cohesive and natural scene."
+                final_prompt = "Please creatively composite these three images, combining their main elements into a cohesive and natural scene."
                 logger.info("Default prompt generated for three images")
-        else:
-            prompt = preprocess_prompt(prompt, image1, image2, image3)
-        return generate_with_images(prompt, valid_images)
+        
+        result_img, status = generate_with_images(final_prompt, valid_images)
+        return result_img, status, final_prompt
     except Exception as e:
         logger.exception("이미지 처리 중 오류 발생:")
-        return None, f"오류 발생: {str(e)}"
+        return None, f"오류 발생: {str(e)}", prompt
 
 def process_and_show_prompt(image1, image2, image3, prompt):
-    images = [image1, image2, image3]
-    valid_images = [img for img in images if img is not None]
+    """
+    최종적으로 이미지 생성 및 영어 최종 프롬프트를 반환하는 함수.
+    """
     try:
-        auto_prompt = prompt
-        if not prompt or not prompt.strip():
-            if len(valid_images) == 1:
-                auto_prompt = "Please creatively transform this image into a more vivid and artistic version."
-            elif len(valid_images) == 2:
-                auto_prompt = "Please seamlessly composite these two images, integrating their key elements harmoniously into a single image."
-            else:
-                auto_prompt = "Please creatively composite these three images, combining their main elements into a cohesive and natural scene."
-        else:
-            auto_prompt = preprocess_prompt(prompt, image1, image2, image3)
-        result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
-        return result_img, status, auto_prompt
+        result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt)
+        return result_img, status, final_prompt
     except Exception as e:
         logger.exception("처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}", prompt
@@ -248,35 +237,31 @@ with gr.Blocks() as demo:
                 placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
                 label="프롬프트 (선택 사항)"
             )
-            # 마크다운으로 선택 옵션 리스트 표시 (여러 항목)
+            # 마크다운으로 선택 옵션 리스트 표시 (얼굴바꾸기, 이미지 합성(스타일적용) 제외)
             gr.Markdown(
                 """
                 **선택 옵션:**
 
                 1. **이미지 변경:** `#1 이미지의 [다른 모습]으로 변경하라.`
                 2. **글자지우기:** `#1 이미지의 [중국어를 모두]를 제거하라.`
-                3. **얼굴바꾸기:** `#1 이미지의 인물 얼굴을 #2 이미지의 얼굴로 교체하라.`
-                4. **옷바꾸기:** `#1 이미지의 인물 의상을 #2 이미지의 의상으로 변경하라.`
-                5. **배경바꾸기:** `#1 이미지의 배경을 #2 이미지의 배경으로 변경하라.`
-                6. **이미지 합성(상품포함):** `#1 이미지와 #2 이미지(또는 #3 이미지)를 자연스럽게 합성하라.`
-                7. **이미지 합성(스타일적용):** `#1 이미지의 내용을 #2 이미지의 스타일로 변경하라.`
+                3. **옷바꾸기:** `#1 이미지의 인물 의상을 #2 이미지의 의상으로 변경하라.`
+                4. **배경바꾸기:** `#1 이미지의 배경을 #2 이미지의 배경으로 변경하라.`
+                5. **이미지 합성(상품포함):** `#1 이미지와 #2 이미지(또는 #3 이미지)를 자연스럽게 합성하라.`
                 """
             )
             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)
+            prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
     
-    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (입력값은 한국어로 지정됨)
+    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구 입력)
     image_change_btn.click(
         fn=lambda: "#1 이미지의 [다른 모습]으로 변경하라.",
         inputs=[],
@@ -287,11 +272,6 @@ with gr.Blocks() as demo:
         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=[],
@@ -307,11 +287,6 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=prompt_input
     )
-    composite_style_btn.click(
-        fn=lambda: "#1 이미지의 내용을 #2 이미지의 스타일로 변경하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
     
     submit_btn.click(
         fn=process_and_show_prompt,
@@ -327,7 +302,7 @@ with gr.Blocks() as demo:
         2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
         3. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다.
         4. **선택 옵션**: 위의 마크다운 형식의 옵션에서 원하는 항목을 클릭하면 프롬프트 입력란에 한국어 문구가 채워지고,
-           이후 Gemini‑2.0‑flash 모델을 통해 영어로 번역되어 적용됩니다.
+           이후 Gemini‑2.0‑flash 모델을 통해 영어로 번역되어 실제 API 호출 프롬프트로 적용됩니다.
         
         > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
         """