Update app.py

app.py

@@ -249,18 +249,16 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
             if result_img is not None:
                 return result_img, status, final_prompt
             else:
-                # API 호출은 성공했지만 이미지 생성 실패
                 last_error = status
                 retry_count += 1
                 logger.warning(f"이미지 생성 실패, 재시도 {retry_count}/{max_retries}: {status}")
-                time.sleep(1)  # 1초 대기 후 재시도
+                time.sleep(1)
         except Exception as e:
             last_error = str(e)
             retry_count += 1
             logger.exception(f"이미지 처리 중 오류 발생, 재시도 {retry_count}/{max_retries}:")
-            time.sleep(1)  # 1초 대기 후 재시도
+            time.sleep(1)
     
-    # 모든 재시도 실패 후
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
 
 def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
@@ -285,21 +283,17 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
             statuses.append(f"이미지 #{i+1}: {status}")
             prompts.append(f"이미지 #{i+1}: {final_prompt}")
         else:
-            # 에러가 발생한 경우에도 결과 목록에 None을 추가
             results.append(None)
             statuses.append(f"이미지 #{i+1} 생성 실패: {status}")
             prompts.append(f"이미지 #{i+1}: {final_prompt}")
         
-        # API 호출 사이에 약간의 간격을 두어 속도 제한 방지
         time.sleep(1)
     
     progress(1.0, desc="이미지 생성 완료!")
     
-    # 결과가 없는 경우 빈 슬롯 채우기
     while len(results) < 4:
         results.append(None)
     
-    # 상태 메시지와 프롬프트 정보 결합
     combined_status = "\n".join(statuses)
     combined_prompts = "\n".join(prompts)
     
@@ -310,15 +304,14 @@ with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>4장 이미지 생성기 (예제 포함)</h1>
-            <p>이미지를 업로드하고 "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.</p>
-            <p>모든 이미지는 균일한 크기로 표시됩니다.</p>
+            <h1>이커머스용 이미지 생성기</h1>
+            <p>이미지를 업로드하고 예제를 참가하여 프롬프트를 수정하고 "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.</p>
         </div>
         """
     )
     
     with gr.Row():
-        # 왼쪽 입력 영역
+        # 왼쪽 입력 영역 (예제 이미지를 포함하여 사용 방법 아래에 배치)
         with gr.Column(scale=1):
             with gr.Row():
                 image1_input = gr.Image(type="pil", label="#1", image_mode="RGB", height=300, width=200)
@@ -330,15 +323,15 @@ with gr.Blocks() as demo:
                 label="프롬프트 (선택 사항)"
             )
             with gr.Row():
-                image_change_btn1 = gr.Button("이미지변경-1")
-                image_change_btn2 = gr.Button("이미지변경-2")
-                text_remove_btn = gr.Button("글자지우기")
-                text_change_btn = gr.Button("글자변경하기")
+                image_change_btn1 = gr.Button("이미지 변경-1")
+                image_change_btn2 = gr.Button("이미지 변경-2")
+                text_remove_btn = gr.Button("글자 지우기")
+                text_change_btn = gr.Button("글자 변경하기")
                 clothes_change_btn1 = gr.Button("가상 상품착용-1")
-                clothes_change_btn2 = gr.Button("상품착용-2")
+                clothes_change_btn2 = gr.Button("가상 상품착용-2")
                 holding_product_btn = gr.Button("상품들고 있기")
-                background_change_btn = gr.Button("배경바꾸기")
-                composite_product_btn = gr.Button("부분지우기")
+                background_change_btn = gr.Button("배경 바꾸기")
+                composite_product_btn = gr.Button("부분 지우기")
             submit_btn = gr.Button("이미지 생성 (4장)", variant="primary")
             
             gr.Markdown(
@@ -349,11 +342,32 @@ with gr.Blocks() as demo:
                 2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
                 3. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구로 입력하시면 됩니다.
                 4. **다양한 이미지**: "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.
+                5. **예제선택**: 다양한 예제를 통해 미리 테스트해보세요.
                 
                 > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
                 """
             )
             
+            # 예제 이���지: 사용 방법 아래 왼쪽 영역에 배치
+            gr.Markdown("## 예제 이미지")
+            examples = [
+                ["down/모델.jpg", None, None, "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라."],
+                ["down/상어레고모형.jpg", None, None, "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라."],
+                ["down/중국어.png", None, None, "(#1 이미지)에 있는 중국어를 모두 제거하라."],
+                ["down/텍스트.webp", None, None, '(#1의 텍스트)를 스타일을 유지한체 텍스트만 "Hello"로 바꿔라'],
+                ["down/모델.jpg", "down/선글라스.png", "down/청바지.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라."],
+                ["down/모델.jpg", "down/선글라스.png", "down/카페전경.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라."],
+                ["down/모델.jpg", "down/와인잔.png", None, "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 와인잔)을 여성모델이 홍보할 와인잔을 돋보이게 들고 있는 자연스러운 모습으로 이미지를 생성하라."],
+                ["down/모델.jpg", "down/카페전경.png", None, "(#1의 여성모델)이 (#2 이미지의 배경)을 주요 피사체는 그대로 유지하여 이미지의 분위기가 자연스럽게 어우러지도록 생성하라."],
+                ["down/상어레고모형.jpg", None, None, "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지해 해야한다."]
+            ]
+            
+            gr.Examples(
+                examples=examples,
+                inputs=[image1_input, image2_input, image3_input, prompt_input],
+                elem_id="examples-grid"
+            )
+            
         # 오른쪽 출력 영역
         with gr.Column(scale=1):
             with gr.Row():
@@ -369,7 +383,7 @@ with gr.Blocks() as demo:
     
     # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
     image_change_btn1.click(
-        fn=lambda: "(#1의 여성)의 이미지를 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라.",
+        fn=lambda: "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라.",
         inputs=[],
         outputs=prompt_input
     )
@@ -389,17 +403,17 @@ with gr.Blocks() as demo:
         outputs=prompt_input
     )
     clothes_change_btn1.click(
-        fn=lambda: "(#1의 여성모겔)이 신체 비육과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
+        fn=lambda: "(#1의 여성모겔)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
         inputs=[],
         outputs=prompt_input
     )
     clothes_change_btn2.click(
-        fn=lambda: "(#1의 여성모델)이 신체 비육과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라.",
+        fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라.",
         inputs=[],
         outputs=prompt_input
     )
     holding_product_btn.click(
-        fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 크지 않는 와인잔)를 직접 모델이 사용하기 위해 들고 있는 자연스러운 모습으로 이미지를 생성하라.",
+        fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 와인잔)을 여성모델이 홍보할 와인잔을 돋보이게 들고 있는 자연스러운 모습으로 이미지를 생성하라.",
         inputs=[],
         outputs=prompt_input
     )
@@ -419,26 +433,6 @@ with gr.Blocks() as demo:
         inputs=[image1_input, image2_input, image3_input, prompt_input],
         outputs=[output_image1, output_image2, output_image3, output_image4, output_text, prompt_display],
     )
-    
-    gr.Markdown("## 예제 이미지", elem_classes="gr-examples-header")
-    
-    examples = [
-        ["down/모���.jpg", None, None, "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라."],
-        ["down/상어레고모형.png", None, None, "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라."],
-        ["down/중국어.png", None, None, "(#1 이미지)에 있는 중국어를 모두 제거하라."],
-        ["down/텍스트.webp", None, None, '(#1의 텍스트)를 스타일을 유지한체 텍스트만 "Hello"로 바꿔라'],
-        ["down/모델.jpg", "down/선글라스.png", "down/청바지.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라."],
-        ["down/모델.jpg", "down/선글라스.png", "down/카페전경.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라."],
-        ["down/모델.jpg", "down/와인잔.png", None, "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 와인잔)을 여성모델이 홍보할 와인잔을 돋보이게 들고 있는 자연스러운 모습으로 이미지를 생성하라."],
-        ["down/모델.jpg", "down/카페전경.png", None, "(#1의 여성모델)이 (#2 이미지의 배경)을 주요 피사체는 그대로 유지하여 이미지의 분위기가 자연스럽게 어우러지도록 생성하라."],
-        ["down/상어레고모형.png", None, None, "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지해 해야한다."]
-    ]
-    
-    gr.Examples(
-        examples=examples,
-        inputs=[image1_input, image2_input, image3_input, prompt_input],
-        elem_id="examples-grid"
-    )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)