Update app.py

app.py

@@ -106,8 +106,8 @@ def preprocess_prompt(prompt, image1, image2, image3):
     elif "5. 배경바꾸기" in prompt:
         prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 변경해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
 
-    elif "6. 이미지 합성(상품포함)" in prompt:
-        prompt = "첫 번째 이미지와 두 번째 이미지(또는 세 번째 이미지)를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합해주세요."
+    elif "상품합성" in prompt:
+        prompt = "#1 상품 이미지에 #2 이미지 또는 #3 이미지를 [모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합하여]자연스럽게 합성하라."
 
     prompt += " 이미지를 생성해주세요."
     return prompt
@@ -233,7 +233,8 @@ with gr.Blocks() as demo:
                 text_remove_btn = gr.Button("글자지우기")
                 clothes_change_btn = gr.Button("옷바꾸기")
                 background_change_btn = gr.Button("배경바꾸기")
-                composite_product_btn = gr.Button("이미지 합성(상품포함)")
+                # 버튼 라벨과 클릭 시 반환되는 프롬프트 수정
+                composite_product_btn = gr.Button("상품합성")
             submit_btn = gr.Button("이미지 생성", variant="primary")
         with gr.Column():
             output_image = gr.Image(label="생성된 이미지")
@@ -262,7 +263,7 @@ with gr.Blocks() as demo:
         outputs=prompt_input
     )
     composite_product_btn.click(
-        fn=lambda: "#1 이미지와 #2 이미지(또는 #3 이미지)를 자연스럽게 합성하라.",
+        fn=lambda: "#1 상품 이미지에 #2 이미지 또는 #3 이미지를 [모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합하여]자연스럽게 합성하라.",
         inputs=[],
         outputs=prompt_input
     )
@@ -286,4 +287,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)