feat: 1차 구현 오나료

Signed-off-by: longshiine <[email protected]>

app.py

@@ -1,6 +1,8 @@
 import requests
 import gradio as gr
 import zipfile
+import os
+import tempfile
 from PIL import Image
 from io import BytesIO
 from rembg import remove
@@ -149,30 +151,34 @@ def paste_to_background_type_c(background: Image.Image, product: Image.Image, ma
         print(f"Failed to paste product images to background: {e}")
         return None
     
-def create_zip_file(images: list, zip_filename: str = "images.zip") -> bytes:
-    # 메모리 내에서 ZIP 파일 생성
-    zip_buffer = BytesIO()
-    
-    with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zipf:
-        for i, image in enumerate(images):
-            # 이미지를 메모리 버퍼에 저장
-            image_buffer = BytesIO()
-            image.save(image_buffer, format="PNG")
-            image_buffer.seek(0)
-            
-            # 버퍼 내용을 ZIP 파일에 추가
-            zipf.writestr(f"image_{i + 1}.png", image_buffer.getvalue())
+def create_zip_file(images: list) -> str:
+    # 임시 파일 생성
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".zip") as temp_zip:
+        with zipfile.ZipFile(temp_zip, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            for i, image in enumerate(images):
+                # 이미지를 메모리 버퍼에 저장
+                image_buffer = BytesIO()
+                image.save(image_buffer, format="PNG")
+                image_buffer.seek(0)
+                
+                # 버퍼 내용을 ZIP 파일에 추가
+                zipf.writestr(f"image_{i + 1}.png", image_buffer.getvalue())
+        
+        # 임시 파일의 경로를 반환
+        temp_zip_path = temp_zip.name
     
-    # ZIP 파일 데이터를 반환
-    zip_buffer.seek(0)
-    return zip_buffer.getvalue()
+    return temp_zip_path
 
 def image_processing_single(background_image: Image.Image, product_image_url: str):
     # 각종 설정 값
+    background_image_size = 1024 # 배경 이미지의 크기
     product_image_size = 650 # 배경이 제거된 제품 이미지의 크기
     type_b_margin = 15
     type_c_margin_1, type_c_margin_2 = 15, 45 # (1=가운데 두개의 마진, 2= 중앙 한개의 top 마진)
     
+    # 배경 이미지 크기 조정
+    background_image = background_image.resize((background_image_size, background_image_size))
+    
     # 이미지 다운로드
     original_image = download_image(product_image_url)
 
@@ -192,16 +198,30 @@ def image_processing_single(background_image: Image.Image, product_image_url: st
     
     # 결과 이미지 반환
     image_list = [type_a_image, type_b_image, type_c_image]
-    image_zip_file = create_zip_file(image_list)
+    zip_file_path = create_zip_file(image_list)
     
-    return image_list #, image_zip_file
+    return image_list, zip_file_path
 
-def image_processing_multi(background_image: Image.Image, product_image_url_list: list):
+def image_processing_batch(background_image: Image.Image, product_image_url_file_path: str):
     # 각종 설정 값
     product_image_size = 650 # 배경이 제거된 제품 이미지의 크기
     type_b_margin = 15
     type_c_margin_1, type_c_margin_2 = 15, 45 # (1=가운데 두개의 마진, 2= 중앙 한개의 top 마진)
     
+    # file to url
+    def file_to_list(file_path: str) -> list:
+        # 파일을 열고 읽기
+        with open(file_path, "r") as f:
+            # 파일 내용을 읽어서 줄바꿈(엔터)로 구분된 각 URL을 리스트로 변환
+            content = f.read()
+            # 줄바꿈으로 URL 분리
+            url_list = [url.strip() for url in content.splitlines() if url.strip()]
+
+        return url_list
+    
+    product_image_url_list = file_to_list(product_image_url_file_path)
+    
+    preview_image_list, image_list = [], []
     for idx, product_image_url in enumerate(product_image_url_list):
         # 이미지 다운로드
         original_image = download_image(product_image_url)
@@ -220,22 +240,67 @@ def image_processing_multi(background_image: Image.Image, product_image_url_list
         type_b_image = paste_to_background_type_b(background_image.copy(), resized_image, type_b_margin)
         type_c_image = paste_to_background_type_c(background_image.copy(), resized_image, type_c_margin_1, type_c_margin_2)
         
+        image_list.append(type_a_image)
+        image_list.append(type_b_image)
+        image_list.append(type_c_image)
+        if idx == 0:
+            preview_image_list = [type_a_image, type_b_image, type_c_image]
+            
+    zip_file_path = create_zip_file(image_list)
+    
+    return preview_image_list, zip_file_path
+        
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-    with gr.Row():
-        background_image = gr.Image(label="Background Image", type="pil")
-        with gr.Tab("Single"):
-            link = gr.Textbox(label="Image URL")
-            btn_single = gr.Button("Generate Images")
-        with gr.Tab("Multi"):
-            links = gr.File(label="Image URLs (.txt file)")
-            btn_multi = gr.Button("Generate Images")
+    gr.HTML("""
+        <h1>Product Image Generator</h1>
+        <p>1. 배경 이미지를 업로드해주세요.<p>
+        <p>2-1. Single 모드의 경우, 이미지의 링크를 입력해주세요.<p>
+        <p>2-2. Batch 모드의 경우, 이미지 링크들이 포함된 .txt 파일(형식 = 링크1(엔터) 링크2(엔터) ..)을 업로드해주세요.<p>
+        <p>3. Generate Images 버튼을 눌러 제품 이미지를 생성해주세요.<p>
+    """)
     
+    with gr.Row(elem_classes="input_box"):
+        with gr.Column():
+            background_image = gr.Image(label="Background Image", type="pil", height=300)
+            gr.Examples(
+                examples=[["examples/background_1.png"], ["examples/background_2.png"], ["examples/background_3.png"], ["examples/background_4.png"]], # 예제 이미지 경로 리스트
+                inputs=background_image,  # 예제를 클릭했을 때 해당 이미지가 이 input에 적용됨
+                label="예제 배경 이미지")
+        with gr.Tab("Single Mode"):
+            with gr.Column():
+                link = gr.Textbox(label="Image URL")
+                gr.Examples(
+                    examples=["https://url.kr/yx85be", "https://url.kr/if4wxo"],
+                    inputs=link,
+                    label="예제 이미지 링크"
+                )
+            btn_single = gr.Button("Generate Images")
+        with gr.Tab("Batch Mode"):
+            gr.HTML("<div>링크1(엔터) 링크2(엔터) ... 으로 구성된 .txt 파일을 업로드</div>")
+            with gr.Column():
+                links = gr.File(label="Image URLs (.txt file)", height=100)
+                gr.Examples(
+                    examples=[["examples/links.txt"]],
+                    inputs=links,
+                    label="예제 .txt 파일"
+                )
+            btn_batch = gr.Button("Generate Images")
+                
+    gr.HTML("""
+        <br><br>
+        <h1>생성 결과 확인</h1>
+        <p>생성은 1개 이미지 당 15~30초 소요되며(배경제거 AI 모듈 처리시간), 결과는 다음과 같이 제공됩니다.</p>
+        <p>1. 첫번째 이미지에 대한 프리뷰(3개 타입)</p> 
+        <p>2. 결과 이미지 전체가 담긴 ZIP 파일(results.zip) 다운로드 링크가 제공됩니다</p>
+    """)
+
     with gr.Row():
         preview = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery", columns=[3], rows=[1], object_fit="contain", height="auto")
         output_zip = gr.File(label="Download Result Zip File")
         
     # evnets
-    btn_single.click(fn=image_processing_single, inputs=[background_image, link], outputs=[preview], api_name="image_processing")
+    btn_single.click(fn=image_processing_single, inputs=[background_image, link], outputs=[preview, output_zip], api_name="image_processing_single")
+    btn_batch.click(fn=image_processing_batch, inputs=[background_image, links], outputs=[preview, output_zip], api_name="image_processing_batch")
 
 demo.launch()

links.txt → examples/links.txt

@@ -1,10 +1,10 @@
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/drb/drb00370/v/67.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/cgn/cgn01206/l/73.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/sre/sre01256/l/96.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/lex/lex19936/v/115.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/cey/cey66577/v/6.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/nor/nor01844/v/39.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/swv/swv06072/v/8.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/age/age00855/v/40.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/nhi/nhi03515/v/41.jpg,
-https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/nhi/nhi03512/v/49.jpg,
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/drb/drb00370/v/67.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/cgn/cgn01206/l/73.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/sre/sre01256/l/96.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/lex/lex19936/v/115.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/cey/cey66577/v/6.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/nor/nor01844/v/39.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/swv/swv06072/v/8.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/age/age00855/v/40.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/nhi/nhi03515/v/41.jpg
+https://cloudinary.images-iherb.com/image/upload/f_auto,q_auto:eco/images/nhi/nhi03512/v/49.jpg