feat: 제품 이미지 사이즈 조절 추가

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

.gitignore

@@ -1 +1,2 @@
-__pycache__/
+__pycache__/
+.DS_Store

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from infra import image_processing_single, image_processing_batch
 
-with gr.Blocks(theme='NoCrypt/miku', title='Image Generator') as demo:
+with gr.Blocks(theme='Zarkel/IBM_Carbon_Theme', title='Image Generator') as demo:
     gr.HTML("""
         <h1>Product Image Generator</h1>
         <p>1. 배경 이미지를 업로드해주세요.<p>
@@ -19,6 +19,7 @@ with gr.Blocks(theme='NoCrypt/miku', title='Image Generator') as demo:
                 label="예제 배경 이미지")
         with gr.Tab("Single Mode"):
             with gr.Column():
+                product_image_size = gr.Slider(512, 1024, value=800, label="제품 이미지 사이즈")
                 link = gr.Textbox(label="Image URL")
                 gr.Examples(
                     examples=["https://url.kr/yx85be", "https://url.kr/if4wxo"],
@@ -30,6 +31,7 @@ with gr.Blocks(theme='NoCrypt/miku', title='Image Generator') as demo:
             gr.HTML("<div>링크1(엔터) 링크2(엔터) ... 으로 구성된 .txt 파일을 업로드</div>")
             with gr.Column():
                 links = gr.File(label="Image URLs (.txt file)", height=100)
+                product_image_size_batch = gr.Slider(512, 1024, value=800, label="제품 이미지 사이즈")
                 gr.Examples(
                     examples=[["examples/links.txt"]],
                     inputs=links,
@@ -50,7 +52,7 @@ with gr.Blocks(theme='NoCrypt/miku', title='Image Generator') as demo:
         output_zip = gr.File(label="Download Result Zip File")
         
     # evnets
-    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")
+    btn_single.click(fn=image_processing_single, inputs=[background_image, link, product_image_size], outputs=[preview, output_zip], api_name="image_processing_single")
+    btn_batch.click(fn=image_processing_batch, inputs=[background_image, links, product_image_size_batch], outputs=[preview, output_zip], api_name="image_processing_batch")
 
 demo.launch()

infra.py

@@ -78,84 +78,68 @@ def resize_image(image: Image.Image, max_size: int) -> Image.Image:
         return None
     
 def paste_to_background_type_a(background: Image.Image, product: Image.Image) -> Image.Image:
-    # 배경에 제품 이미지 합성
     try:
-        # 배경 이미지 크기 가져오기
         bg_width, bg_height = background.size
-        
-        # 제품 이미지 크기 가져오기
         product_width, product_height = product.size
-        
-        # 제품 이미지를 배경 이미지 중앙에 위치시키기
+
+        # 제품 이미지를 배경 이미지 중앙에 위치시키기 (음수 오프셋 허용)
         offset = ((bg_width - product_width) // 2, (bg_height - product_height) // 2)
-        
-        # 알파 채널을 고려하여 합성
+
+        # 알파 채널을 고려하여 합성 (음수 오프셋 지원)
         background.paste(product, offset, mask=product)
         return background
-    
+
     except Exception as e:
         print(f"Failed to paste product image to background: {e}")
         return None
-    
+
 def paste_to_background_type_b(background: Image.Image, product: Image.Image, margin: int = 10) -> Image.Image:
     try:
-        # 배경 이미지 크기 가져오기
         bg_width, bg_height = background.size
-        
-        # 제품 이미지 크기 가져오기
         product_width, product_height = product.size
-        
-        # 두 제품 이미지를 위한 전체 너비 계산 (제품 이미지 두 개 + 마진)
+
+        # 두 제품 이미지를 위한 전체 너비 계산 (음수 값 허용)
         total_width = (product_width * 2) + margin
-        
-        if total_width > bg_width:
-            raise ValueError("Background is too narrow to fit two product images with the specified margin.")
-        
-        # 첫 번째 제품 이미지의 왼쪽 상단 좌표 계산
+
+        # 첫 번째 제품 이미지의 왼쪽 상단 좌표 계산 (음수 오프셋 허용)
         left_offset = (bg_width - total_width) // 2
         top_offset = (bg_height - product_height) // 2
-        
+
+        # 첫 번째 제품 이미지 합성
+        background.paste(product, (left_offset, top_offset), mask=product)
+
         # 두 번째 제품 이미지의 왼쪽 상단 좌표 계산
         right_offset = left_offset + product_width + margin
-        
-        # 첫 번째 제품 이미지 배경에 합성
-        background.paste(product, (left_offset, top_offset), mask=product)
-        
-        # 두 번째 제품 이미지 배경에 합성
+
+        # 두 번째 제품 이미지 합성
         background.paste(product, (right_offset, top_offset), mask=product)
         return background
-    
+
     except Exception as e:
         print(f"Failed to paste product images to background: {e}")
         return None
 
 def paste_to_background_type_c(background: Image.Image, product: Image.Image, margin: int = 10, top_margin: int = 15) -> Image.Image:
     try:
-        # 배경 이미지 크기 가져오기
         bg_width, bg_height = background.size
-        
-        # 제품 이미지 크기 가져오기
         product_width, product_height = product.size
-        
-        # 먼저 type_b 형태로 아래 두 제품 이미지를 배치
+
+        # 아래 두 제품 이미지를 배치
         background_with_two_products = paste_to_background_type_b(background, product, margin)
 
-        # 중앙 상단에 위치할 제품 이미지의 오프셋 계산
+        # 중앙 상단에 위치할 제품 이미지의 오프셋 계산 (음수 오프셋 허용)
         center_offset_x = (bg_width - product_width) // 2
         center_offset_y = ((bg_height - product_height) // 2) + top_margin
-        
-        if center_offset_y < 0:
-            raise ValueError("Background is too small to fit three product images with the specified top margin.")
-        
-        # 세 번째 제품 이미지 배경에 합성 (중앙 상단)
+
+        # 세 번째 제품 이미지 합성 (중앙 상단)
         background_with_two_products.paste(product, (center_offset_x, center_offset_y), mask=product)
-        
-        # 결과 이미지 저장 및 반환
+
         return background_with_two_products
-    
+
     except Exception as e:
         print(f"Failed to paste product images to background: {e}")
         return None
+
     
 def create_zip_file(images: list) -> str:
     # 임시 파일 생성
@@ -174,11 +158,8 @@ def create_zip_file(images: list) -> str:
         temp_zip_path = temp_zip.name
     
     return temp_zip_path
-
-def image_processing_single(background_image: Image.Image, product_image_url: str):    
-    # 배경 이미지 크기 조정
-    background_image = background_image.resize((BACKGROUND_IMAGE_MAX_SIZE, BACKGROUND_IMAGE_MAX_SIZE))
     
+def image_processing(background_image: Image.Image, product_image_url: str, product_image_size: int = 650):    
     # 이미지 다운로드
     original_image = download_image(product_image_url)
 
@@ -189,20 +170,31 @@ def image_processing_single(background_image: Image.Image, product_image_url: st
     cropped_image = crop_image(removebg_image)
     
     # 크롭된 이미지 원하는 사이즈로 resize
-    resized_image = resize_image(cropped_image, PRODUCT_IMAGE_MAX_SIZE)
+    resized_image = resize_image(cropped_image, product_image_size)
     
     # type_a 합성
     type_a_image = paste_to_background_type_a(background_image.copy(), resized_image)
     type_b_image = paste_to_background_type_b(background_image.copy(), resized_image, TYPD_B_MARGIN)
     type_c_image = paste_to_background_type_c(background_image.copy(), resized_image, TYPE_C_MARGIN_1, TYPE_C_MARGIN_2)
     
+    # 결과 이미지 반환
+    return type_a_image, type_b_image, type_c_image
+
+
+def image_processing_single(background_image: Image.Image, product_image_url: str, product_image_size: int = 650): 
+    # 배경 이미지 크기 조정
+    background_image = background_image.resize((BACKGROUND_IMAGE_MAX_SIZE, BACKGROUND_IMAGE_MAX_SIZE))
+    
+    # 이미지 프로세싱
+    type_a_image, type_b_image, type_c_image = image_processing(background_image, product_image_url, product_image_size)
+    
     # 결과 이미지 반환
     image_list = [type_a_image, type_b_image, type_c_image]
     zip_file_path = create_zip_file(image_list)
     
     return image_list, zip_file_path
 
-def image_processing_batch(background_image: Image.Image, product_image_url_file_path: str):    
+def image_processing_batch(background_image: Image.Image, product_image_url_file_path: str, product_image_size: int = 650):    
     # 배경 이미지 크기 조정
     background_image = background_image.resize((BACKGROUND_IMAGE_MAX_SIZE, BACKGROUND_IMAGE_MAX_SIZE))
     
@@ -221,22 +213,7 @@ def image_processing_batch(background_image: Image.Image, product_image_url_file
     
     preview_image_list, image_list = [], []
     for idx, product_image_url in enumerate(product_image_url_list):
-        # 이미지 다운로드
-        original_image = download_image(product_image_url)
-
-        # 이미지 누끼따기
-        removebg_image = remove_background(original_image)
-        
-        # 이미지 크롭
-        cropped_image = crop_image(removebg_image)
-        
-        # 크롭된 이미지 원하는 사이즈로 resize
-        resized_image = resize_image(cropped_image, PRODUCT_IMAGE_MAX_SIZE)
-        
-        # type_a 합성
-        type_a_image = paste_to_background_type_a(background_image.copy(), resized_image)
-        type_b_image = paste_to_background_type_b(background_image.copy(), resized_image, TYPD_B_MARGIN)
-        type_c_image = paste_to_background_type_c(background_image.copy(), resized_image, TYPE_C_MARGIN_1, TYPE_C_MARGIN_2)
+        type_a_image, type_b_image, type_c_image = image_processing(background_image, product_image_url, product_image_size)
         
         image_list.append(type_a_image)
         image_list.append(type_b_image)