Create app.py

app.py

@@ -0,0 +1,313 @@
+import gradio as gr
+import replicate
+import os
+import requests
+from PIL import Image, ImageDraw, ImageFilter, ImageEnhance
+import io
+import base64
+import tempfile
+import numpy as np
+
+# Replicate API 토큰 설정 (환경변수에서 가져오기)
+REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
+if REPLICATE_API_TOKEN:
+    os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
+
+def upload_image_to_temp_url(image):
+    """이미지를 임시 URL로 업로드하는 함수 (실제로는 base64 인코딩)"""
+    if image is None:
+        return None
+    
+    # PIL Image를 base64로 인코딩
+    buffer = io.BytesIO()
+    image.save(buffer, format='PNG')
+    buffer.seek(0)
+    
+    # 임시 파일로 저장
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmp_file:
+        image.save(tmp_file.name, format='PNG')
+        return tmp_file.name
+
+def create_white_background_shadow(image, shadow_intensity=0.3, shadow_offset=(10, 10)):
+    """흰 배경에 그림자 효과 추가"""
+    if image.mode != 'RGBA':
+        image = image.convert('RGBA')
+    
+    # 원본 이미지 크기
+    width, height = image.size
+    
+    # 그림자를 위한 여백 추가
+    shadow_margin = max(shadow_offset) + 20
+    new_width = width + shadow_margin * 2
+    new_height = height + shadow_margin * 2
+    
+    # 흰색 배경 생성
+    result = Image.new('RGBA', (new_width, new_height), (255, 255, 255, 255))
+    
+    # 그림자 생성
+    shadow = Image.new('RGBA', (new_width, new_height), (0, 0, 0, 0))
+    shadow_draw = ImageDraw.Draw(shadow)
+    
+    # 원본 이미지의 알파 채널을 이용해 그림자 마스크 생성
+    alpha = image.split()[-1]
+    shadow_pos = (shadow_margin + shadow_offset[0], shadow_margin + shadow_offset[1])
+    shadow.paste(Image.new('RGBA', image.size, (0, 0, 0, int(255 * shadow_intensity))), shadow_pos, alpha)
+    
+    # 그림자 블러 효과
+    shadow = shadow.filter(ImageFilter.GaussianBlur(radius=8))
+    
+    # 그림자를 배경에 합성
+    result = Image.alpha_composite(result, shadow)
+    
+    # 원본 이미지를 중앙에 배치
+    image_pos = (shadow_margin, shadow_margin)
+    result.paste(image, image_pos, image)
+    
+    return result.convert('RGB')
+
+def enhance_image_quality(image):
+    """이미지 화질 향상"""
+    # 선명도 향상
+    enhancer = ImageEnhance.Sharpness(image)
+    image = enhancer.enhance(1.2)
+    
+    # 대비 향상
+    enhancer = ImageEnhance.Contrast(image)
+    image = enhancer.enhance(1.1)
+    
+    # 색상 채도 향상
+    enhancer = ImageEnhance.Color(image)
+    image = enhancer.enhance(1.05)
+    
+    return image
+
+def rotate_image_slightly(image, angle=5):
+    """이미지를 약간 회전시켜 아이템 위너 회피"""
+    return image.rotate(angle, expand=True, fillcolor=(255, 255, 255))
+
+def check_white_background_ratio(image, threshold=0.85):
+    """흰색 배경 비율 확인 (85% 규칙)"""
+    if image.mode != 'RGB':
+        image = image.convert('RGB')
+    
+    pixels = np.array(image)
+    white_pixels = np.sum(np.all(pixels >= [240, 240, 240], axis=2))
+    total_pixels = pixels.shape[0] * pixels.shape[1]
+    white_ratio = white_pixels / total_pixels
+    
+    return white_ratio >= threshold, white_ratio
+
+def process_with_ai_enhancement(image_path, prompt_option):
+    """AI를 이용한 이미지 품질 향상"""
+    if not REPLICATE_API_TOKEN:
+        return None, "Replicate API 토큰이 설정되지 않았습니다."
+    
+    try:
+        # 프롬프트 옵션별 설정
+        prompts = {
+            "기본 품질향상": "Enhance image quality, make it more professional and crisp, white background",
+            "제품 사진 최적화": "Transform into professional product photography, clean white background, high quality, studio lighting",
+            "쿠팡 스타일": "Make it look like a professional e-commerce product photo for Korean online shopping, clean white background, perfect lighting",
+            "프리미엄 느낌": "Make it look premium and luxury, professional product shot, perfect white background, studio quality"
+        }
+        
+        selected_prompt = prompts.get(prompt_option, prompts["기본 품질향상"])
+        
+        # Replicate API 호출
+        output = replicate.run(
+            "black-forest-labs/flux-kontext-pro",
+            input={
+                "prompt": selected_prompt,
+                "input_image": open(image_path, "rb"),
+                "output_format": "jpg"
+            }
+        )
+        
+        # 결과 이미지 다운로드
+        response = requests.get(output)
+        if response.status_code == 200:
+            enhanced_image = Image.open(io.BytesIO(response.content))
+            return enhanced_image, "AI 향상 완료"
+        else:
+            return None, "AI 처리 실패"
+            
+    except Exception as e:
+        return None, f"AI 처리 중 오류: {str(e)}"
+
+def create_coupang_thumbnail(image, apply_shadow, apply_rotation, rotation_angle, ai_enhancement, prompt_option):
+    """쿠팡 썸네일 생성 메인 함수"""
+    if image is None:
+        return None, "이미지를 업로드해주세요."
+    
+    try:
+        result_image = image.copy()
+        status_messages = []
+        
+        # 1단계: AI 향상 (선택사항)
+        if ai_enhancement and REPLICATE_API_TOKEN:
+            temp_path = upload_image_to_temp_url(result_image)
+            if temp_path:
+                enhanced_img, ai_msg = process_with_ai_enhancement(temp_path, prompt_option)
+                if enhanced_img:
+                    result_image = enhanced_img
+                    status_messages.append(f"✅ AI 향상: {ai_msg}")
+                else:
+                    status_messages.append(f"❌ AI 향상: {ai_msg}")
+                os.unlink(temp_path)  # 임시 파일 삭제
+        
+        # 2단계: 화질 향상
+        result_image = enhance_image_quality(result_image)
+        status_messages.append("✅ 화질 향상 완료")
+        
+        # 3단계: 각도 변경 (아이템 위너 회피)
+        if apply_rotation:
+            result_image = rotate_image_slightly(result_image, rotation_angle)
+            status_messages.append(f"✅ 각도 변경: {rotation_angle}도 회전")
+        
+        # 4단계: 흰 배경 + 그림자 효과
+        if apply_shadow:
+            result_image = create_white_background_shadow(result_image)
+            status_messages.append("✅ 그림자 효과 적용")
+        else:
+            # 그림자 없이 흰 배경만 적용
+            if result_image.mode == 'RGBA':
+                white_bg = Image.new('RGB', result_image.size, (255, 255, 255))
+                white_bg.paste(result_image, mask=result_image.split()[-1] if result_image.mode == 'RGBA' else None)
+                result_image = white_bg
+        
+        # 85% 흰색 배경 규칙 확인
+        is_compliant, white_ratio = check_white_background_ratio(result_image)
+        if is_compliant:
+            status_messages.append(f"✅ 85% 흰색 배경 규칙 준수 ({white_ratio:.1%})")
+        else:
+            status_messages.append(f"⚠️ 흰색 배경 비율 부족 ({white_ratio:.1%}) - 85% 미만")
+        
+        # 최종 이미지 크기 조정 (쿠팡 권장 사이즈)
+        result_image = result_image.resize((1000, 1000), Image.Resampling.LANCZOS)
+        status_messages.append("✅ 1000x1000 크기로 조정 완료")
+        
+        return result_image, "\n".join(status_messages)
+        
+    except Exception as e:
+        return None, f"처리 중 오류가 발생했습니다: {str(e)}"
+
+# Gradio 인터페이스 생성
+def create_interface():
+    with gr.Blocks(title="🛒 쿠팡 썸네일 생성기", theme=gr.themes.Soft()) as iface:
+        gr.Markdown("""
+        # 🛒 쿠팡 썸네일 생성기
+        
+        배경이 제거된 상품 이미지를 업로드하면 쿠팡 규정에 맞는 썸네일로 자동 변환해드립니다!
+        
+        ## 💡 적용되는 쿠팡 규정 대응 전략:
+        - ✅ **흰 배경 + 85% 규칙** 완벽 준수
+        - ✅ **AI 마법 프롬프트**로 화질 향상 
+        - ✅ **각도 변경**으로 아이템 위너 회피
+        - ✅ **그림자 효과**로 자연스러운 입체감 연출
+        """)
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### 📤 이미지 업로드")
+                input_image = gr.Image(
+                    label="배경 제거된 상품 이미지", 
+                    type="pil",
+                    height=300
+                )
+                
+                gr.Markdown("### ⚙️ 옵션 설정")
+                
+                with gr.Group():
+                    gr.Markdown("**🎨 효과 옵션**")
+                    apply_shadow = gr.Checkbox(
+                        label="그림자 효과 적용", 
+                        value=True,
+                        info="자연스러운 입체감 연출"
+                    )
+                    apply_rotation = gr.Checkbox(
+                        label="각도 변경 적용", 
+                        value=True,
+                        info="아이템 위너 회피"
+                    )
+                    rotation_angle = gr.Slider(
+                        minimum=-15, 
+                        maximum=15, 
+                        value=5, 
+                        step=1,
+                        label="회전 각도",
+                        visible=True
+                    )
+                
+                with gr.Group():
+                    gr.Markdown("**🤖 AI 향상 (Replicate API 필요)**")
+                    ai_enhancement = gr.Checkbox(
+                        label="AI 품질 향상 사용", 
+                        value=False,
+                        info="Replicate API 토큰이 필요합니다"
+                    )
+                    prompt_option = gr.Dropdown(
+                        choices=["기본 품질향상", "제품 사진 최적화", "쿠팡 스타일", "프리미엄 느낌"],
+                        value="쿠팡 스타일",
+                        label="AI 향상 스타일"
+                    )
+                
+                process_btn = gr.Button("🚀 쿠팡 썸네일 생성", variant="primary", size="lg")
+            
+            with gr.Column(scale=1):
+                gr.Markdown("### 📥 결과")
+                output_image = gr.Image(
+                    label="쿠팡 규정 썸네일", 
+                    height=400
+                )
+                status_output = gr.Textbox(
+                    label="처리 상태", 
+                    lines=8,
+                    max_lines=15
+                )
+        
+        # 사용 예시
+        gr.Markdown("""
+        ---
+        ## 📋 사용 방법
+        1. **배경이 제거된 상품 이미지**를 업로드하세요
+        2. 원하는 **효과 옵션**을 선택하세요
+        3. **AI 향상**을 사용하려면 Replicate API 토큰을 환경변수로 설정하세요
+        4. **'쿠팡 썸네일 생성'** 버튼을 클릭하세요
+        
+        ## 🔑 API 설정 (선택사항)
+        AI 향상 기능을 사용하려면 환경변수를 설정하세요:
+        ```bash
+        export REPLICATE_API_TOKEN=your_token_here
+        ```
+        """)
+        
+        # 이벤트 처리
+        def update_rotation_visibility(apply_rot):
+            return gr.update(visible=apply_rot)
+        
+        apply_rotation.change(update_rotation_visibility, apply_rotation, rotation_angle)
+        
+        process_btn.click(
+            create_coupang_thumbnail,
+            inputs=[
+                input_image, 
+                apply_shadow, 
+                apply_rotation, 
+                rotation_angle, 
+                ai_enhancement, 
+                prompt_option
+            ],
+            outputs=[output_image, status_output]
+        )
+    
+    return iface
+
+# 앱 실행
+if __name__ == "__main__":
+    app = create_interface()
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True,
+        show_error=True
+    )