Update app.py

app.py

@@ -144,7 +144,7 @@ def preprocess_prompt(prompt, image1, image2, image3):
     elif "6. 이미지 합성(상품포함)" in prompt:
         prompt = "첫 번째 이미지와 두 번째 이미지(또는 세 번째 이미지)를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합해주세요."
 
-    prompt += " 이미지를 생성해주세요."
+    prompt += " 이미지를 생성해주세요. 이미지에 텍스트나 글자를 포함하지 마세요."
     return prompt
 
 def generate_with_images(prompt, images, variation_index=0):
@@ -162,14 +162,16 @@ def generate_with_images(prompt, images, variation_index=0):
 
         # 변형 인덱스에 따라 프롬프트에 약간의 변화 추가
         variation_suffixes = [
-            " 첫 번째 변형으로 만들어주세요.",
-            " 두 번째 변형으로 더 생생하게 만들어주세요.",
-            " 세 번째 변형으로 더 창의적으로 만들어주세요.",
-            " 네 번째 변형으로 색감을 더 강조해서 만들어주세요."
+            " Create this as the first variation. Do not add any text, watermarks, or labels to the image.",
+            " Create this as the second variation with more vivid colors. Do not add any text, watermarks, or labels to the image.",
+            " Create this as the third variation with a more creative style. Do not add any text, watermarks, or labels to the image.",
+            " Create this as the fourth variation with enhanced details. Do not add any text, watermarks, or labels to the image."
         ]
         
         if variation_index < len(variation_suffixes):
             prompt = prompt + variation_suffixes[variation_index]
+        else:
+            prompt = prompt + " Do not add any text, watermarks, or labels to the image."
 
         contents = [prompt]
         for idx, img in enumerate(images, 1):
@@ -230,13 +232,13 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
                 final_prompt = processed_prompt
         else:
             if len(valid_images) == 1:
-                final_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. Do not include any text or watermarks in the generated image."
                 logger.info("Default prompt generated for single image")
             elif len(valid_images) == 2:
-                final_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. Do not include any text or watermarks in the generated image."
                 logger.info("Default prompt generated for two images")
             else:
-                final_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. Do not include any text or watermarks in the generated image."
                 logger.info("Default prompt generated for three images")
 
         result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
@@ -302,9 +304,9 @@ with gr.Blocks() as demo:
         # 왼쪽 입력 영역
         with gr.Column(scale=1):
             with gr.Row():
-                image1_input = gr.Image(type="pil", label="#1", image_mode="RGB", height=300, width=200)
-                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB", height=300, width=200)
-                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB", height=300, width=200)
+                image1_input = gr.Image(type="pil", label="#1", image_mode="RGB", height=200, width=200)
+                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB", height=200, width=200)
+                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB", height=200, width=200)
             prompt_input = gr.Textbox(
                 lines=3,
                 placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
@@ -322,11 +324,11 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             with gr.Row():
                 with gr.Column():
-                    output_image1 = gr.Image(label="생성된 이미지 #1", height=450, width=450)
-                    output_image3 = gr.Image(label="생성된 이미지 #3", height=450, width=450)
+                    output_image1 = gr.Image(label="생성된 이미지 #1", height=220, width=220)
+                    output_image3 = gr.Image(label="생성된 이미지 #3", height=220, width=220)
                 with gr.Column():
-                    output_image2 = gr.Image(label="생성된 이미지 #2", height=450, width=450)
-                    output_image4 = gr.Image(label="생성된 이미지 #4", height=450, width=450)
+                    output_image2 = gr.Image(label="생성된 이미지 #2", height=220, width=220)
+                    output_image4 = gr.Image(label="생성된 이미지 #4", height=220, width=220)
             
             output_text = gr.Textbox(label="상태 메시지", lines=4)
             prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True, lines=4)