Update app.py

app.py

@@ -151,6 +151,7 @@ def generate_with_images(prompt, images, variation_index=0):
     """
     API 호출을 통해 이미지를 생성하고 결과 이미지를 반환합니다.
     variation_index로 다양한 변화를 줍니다.
+    생성된 이미지는 일관된 크기로 조정됩니다.
     """
     try:
         api_key = os.environ.get("GEMINI_API_KEY")
@@ -206,6 +207,11 @@ def generate_with_images(prompt, images, variation_index=0):
             result_img = Image.open(temp_path)
             if result_img.mode == "RGBA":
                 result_img = result_img.convert("RGB")
+                
+            # 이미지 크기를 512x512로 조정하여 일관된 출력 크기 유지
+            target_size = (512, 512)
+            result_img = result_img.resize(target_size, Image.Resampling.LANCZOS)
+            
             return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
     except Exception as e:
         logger.exception("이미지 생성 중 오류 발생:")
@@ -293,6 +299,7 @@ with gr.Blocks() as demo:
         <div style="text-align: center; margin-bottom: 1rem;">
             <h1>4장 이미지 생성기 (예제 포함)</h1>
             <p>이미지를 업로드하고 "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.</p>
+            <p>모든 이미지는 고정된 크기로 표시됩니다.</p>
         </div>
         """
     )
@@ -300,9 +307,9 @@ 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")
-                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB")
-                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB")
+                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)
             prompt_input = gr.Textbox(
                 lines=3,
                 placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
@@ -318,15 +325,15 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
-            output_image1 = gr.Image(label="생성된 이미지 #1")
+            output_image1 = gr.Image(label="생성된 이미지 #1", height=300, width=300)
         with gr.Column():
-            output_image2 = gr.Image(label="생성된 이미지 #2")
+            output_image2 = gr.Image(label="생성된 이미지 #2", height=300, width=300)
     
     with gr.Row():
         with gr.Column():
-            output_image3 = gr.Image(label="생성된 이미지 #3")
+            output_image3 = gr.Image(label="생성된 이미지 #3", height=300, width=300)
         with gr.Column():
-            output_image4 = gr.Image(label="생성된 이미지 #4")
+            output_image4 = gr.Image(label="생성된 이미지 #4", height=300, width=300)
     
     output_text = gr.Textbox(label="상태 메시지")
     prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
@@ -343,7 +350,7 @@ with gr.Blocks() as demo:
         outputs=prompt_input
     )
     clothes_change_btn.click(
-        fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유지한 체 (#2의 선글라스)와 (#3의 청바지)를 착용하여 이미지가 자연스럽게 어우러지도록 생성해주세요.",
+        fn=lambda: "#1이미지에서 [신체 비율과 포즈는 유지한 체] 의상[스타일과 색상을]#2, #3으로 변경해주세요.",
         inputs=[],
         outputs=prompt_input
     )