Update app.py

app.py

@@ -293,7 +293,12 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
     return results[0], results[1], results[2], results[3], combined_status, combined_prompts
 
 # Gradio 인터페이스
-with gr.Blocks() as demo:
+with gr.Blocks(css="""
+    .gradio-container {max-width: 650px; margin: 0 auto;}
+    .output-image {height: 300px; width: 300px;}
+    .input-image {height: 200px; width: 200px;}
+    .image-row {display: flex; justify-content: center;}
+""") as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
@@ -306,33 +311,39 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
-            with gr.Row():
-                image1_input = gr.Image(type="pil", label="#1", image_mode="RGB", height=256, width=256)
-                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB", height=256, width=256)
-                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB", height=256, width=256)
+    # 입력 이미지 UI를 균형있게 정렬
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=1, min_width=256):
+            image1_input = gr.Image(type="pil", label="#1", image_mode="RGB", height=256, width=256)
+        with gr.Column(scale=1, min_width=256):
+            image2_input = gr.Image(type="pil", label="#2", image_mode="RGB", height=256, width=256)
+        with gr.Column(scale=1, min_width=256):
+            image3_input = gr.Image(type="pil", label="#3", image_mode="RGB", height=256, width=256)
             prompt_input = gr.Textbox(
                 lines=3,
                 placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
                 label="프롬프트 (선택 사항)"
             )
-            with gr.Row():
-                image_change_btn = gr.Button("이미지변경")
-                text_remove_btn = gr.Button("글자지우기")
-                clothes_change_btn = gr.Button("가상 상품착용")
-                background_change_btn = gr.Button("배경바꾸기")
-                composite_product_btn = gr.Button("부분지우기")
+    # 버튼 균등 배치
+    with gr.Row(equal_height=True, variant="panel"):
+        image_change_btn = gr.Button("이미지변경", min_width=100, scale=1)
+        text_remove_btn = gr.Button("글자지우기", min_width=100, scale=1)
+        clothes_change_btn = gr.Button("가상 상품착용", min_width=100, scale=1)
+        background_change_btn = gr.Button("배경바꾸기", min_width=100, scale=1)
+        composite_product_btn = gr.Button("부분지우기", min_width=100, scale=1)
             submit_btn = gr.Button("이미지 생성 (4장)", variant="primary")
     
-    with gr.Row():
-        with gr.Column():
+        # 출력 이미지 UI를 2x2 균형 그리드로 재구성
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=1, min_width=300):
             output_image1 = gr.Image(label="생성된 이미지 #1", height=300, width=300)
-        with gr.Column():
+        with gr.Column(scale=1, min_width=300):
             output_image2 = gr.Image(label="생성된 이미지 #2", height=300, width=300)
     
-    with gr.Row():
-        with gr.Column():
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=1, min_width=300):
             output_image3 = gr.Image(label="생성된 이미지 #3", height=300, width=300)
-        with gr.Column():
+        with gr.Column(scale=1, min_width=300):
             output_image4 = gr.Image(label="생성된 이미지 #4", height=300, width=300)
     
     output_text = gr.Textbox(label="상태 메시지")
@@ -340,17 +351,17 @@ with gr.Blocks() as demo:
     
     # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
     image_change_btn.click(
-        fn=lambda: "#1 이미지의 [다른 모습]으로 변경하라.",
+        fn=lambda: "(#1의 여성)의 이미지를 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라",
         inputs=[],
         outputs=prompt_input
     )
     text_remove_btn.click(
-        fn=lambda: "#1 이미지의 [중국어를 모두]를 제거하라.",
+        fn=lambda: "(#1 이미지)에 있는 중국어를 모두 제거하라.",
         inputs=[],
         outputs=prompt_input
     )
     clothes_change_btn.click(
-        fn=lambda: "#1이미지에서 [신체 비율과 포즈는 유지한 체] 의상[스타일과 색상을]#2, #3으로 변경해주세요.",
+        fn=lambda: "(#1의 여성모겔)이 신체 비육과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
         inputs=[],
         outputs=prompt_input
     )