Update app.py

app.py

@@ -491,6 +491,9 @@ def create_app():
         with gr.Row():
             with gr.Column(scale=1):
                 product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 스마트워치, 향수, 운동화 등", interactive=True)
+                # 상품 이미지 업로드를 상품명 아래로 이동
+                image_input = gr.Image(label="상품 이미지 업로드", type="pil")
+                
                 background_type = gr.Radio(
                     choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "테크놀로지 배경", "컬러풀 패턴 배경", "추상/특수 배경"],
                     label="배경 유형",
@@ -566,32 +569,45 @@ def create_app():
                     inputs=[background_type],
                     outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown]
                 )
-                image_input = gr.Image(label="상품 이미지 업로드", type="pil")
+                
                 with gr.Row():
                     single_btn = gr.Button("프롬프트 및 단일 이미지 생성", variant="primary")
                     multi_btn = gr.Button("프롬프트 및 4장 이미지 생성", variant="primary")
+            
             with gr.Column(scale=1):
-                # 단일 이미지 출력 영역
+                # 공통 프롬프트 출력 영역
                 prompt_output = gr.Textbox(label="생성된 프롬프트 (영어)", lines=6)
-                image_output = gr.Image(label="생성된 이미지", type="filepath")
+                
+                # 이미지 레이아웃 변경: 2x2 그리드로 배치
+                with gr.Row():
+                    image_output1 = gr.Image(label="이미지 #1", type="filepath")
+                    image_output2 = gr.Image(label="이미지 #2", type="filepath")
+                
+                with gr.Row():
+                    image_output3 = gr.Image(label="이미지 #3", type="filepath")
+                    image_output4 = gr.Image(label="이미지 #4", type="filepath")
+                
                 status_output = gr.Textbox(label="결과 정보", lines=3)
-                # 4장 이미지 출력 영역
-                image_output1 = gr.Image(label="이미지 #1", type="filepath")
-                image_output2 = gr.Image(label="이미지 #2", type="filepath")
-                image_output3 = gr.Image(label="이미지 #3", type="filepath")
-                image_output4 = gr.Image(label="이미지 #4", type="filepath")
-                status_output_multi = gr.Textbox(label="4장 생성 결과 정보", lines=3)
-                prompt_output_multi = gr.Textbox(label="생성된 프롬프트 (영어)", lines=6)
+        
+        # 단일 이미지 생성 함수 수정
+        def modified_single_image_gen(*args):
+            # 기존 함수 호출
+            result_img, status, prompt = generate_product_image(*args)
+            # 단일 이미지를 첫 번째 이미지 슬롯에 배치하고 나머지는 None으로 설정
+            return prompt, result_img, None, None, None, status
+        
         single_btn.click(
-            fn=generate_product_image,
+            fn=modified_single_image_gen,
             inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, product_name, additional_info],
-            outputs=[image_output, status_output, prompt_output]
+            outputs=[prompt_output, image_output1, image_output2, image_output3, image_output4, status_output]
         )
+        
         multi_btn.click(
             fn=generate_product_images,
             inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, product_name, additional_info],
-            outputs=[image_output1, image_output2, image_output3, image_output4, status_output_multi, prompt_output_multi]
+            outputs=[image_output1, image_output2, image_output3, image_output4, status_output, prompt_output]
         )
+    
     return demo
 
 # ------------------- 메인 실행 함수 -------------------