app.py

import os
import logging
import json
import time
from PIL import Image
import gradio as gr
from dotenv import load_dotenv

load_dotenv()

# ------------------- 로깅 설정 -------------------
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

# ------------------- 배경 디렉토리 설정 -------------------
BACKGROUNDS_DIR = "./background"
if not os.path.exists(BACKGROUNDS_DIR):
    os.makedirs(BACKGROUNDS_DIR)
    logger.info(f"배경 디렉토리를 생성했습니다: {BACKGROUNDS_DIR}")

# 예시 데이터 로드
from db_examples import product_background_examples

# ------------------- 전역 변수 설정 -------------------
IMAGE_CACHE = {}  # 이미지 캐시 추가

# 커스텀 CSS 스타일
custom_css = """
:root {
    --primary-color: #FB7F0D;
    --secondary-color: #ff9a8b;
    --accent-color: #FF6B6B;
    --background-color: #FFFFFF; /* 베이지색(#FFF3E9)에서 완전한 흰색(#FFFFFF)으로 변경 */
    --card-bg: #ffffff;
    --text-color: #334155;
    --border-radius: 18px;
    --shadow: 0 8px 30px rgba(251, 127, 13, 0.08);
}

/* 예시 갤러리 스타일 */
.example-gallery {
    display: grid;
    grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
    gap: 20px;
    padding: 20px;
}

.example-item {
    cursor: pointer;
    border: 1px solid rgba(0, 0, 0, 0.1);
    border-radius: var(--border-radius);
    overflow: hidden;
    transition: all 0.3s ease;
    background: white;
}

.example-item:hover {
    box-shadow: 0 10px 20px rgba(0, 0, 0, 0.1);
    transform: translateY(-2px);
}

.example-item img {
    width: 100%;
    height: 250px;
    object-fit: cover;
}

.example-label {
    padding: 10px;
    text-align: center;
    font-weight: bold;
    background: rgba(251, 127, 13, 0.1);
}

.example-detail-view {
    margin-bottom: 30px;
}

.example-params {
    background: white;
    padding: 20px;
    border-radius: var(--border-radius);
    box-shadow: var(--shadow);
}

.example-params p {
    margin: 10px 0;
    font-size: 16px;
}

.example-params strong {
    color: var(--primary-color);
}

/* 선택된 예시 하이라이트 */
.example-item.selected {
    border: 3px solid var(--primary-color);
}

/* ── 그룹 래퍼 배경 완전 제거 ── */
.custom-section-group,
.gr-block.gr-group {
    background-color: var(--background-color) !important; /* 변경된 배경색 변수 사용 */
    box-shadow: none !important;
}
.custom-section-group::before,
.custom-section-group::after,
.gr-block.gr-group::before,
.gr-block.gr-group::after {
    display: none !important;
    content: none !important;
}

/* 그룹 컨테이너 배경을 아이보리로, 그림자 제거 */
.custom-section-group {
    background-color: var(--background-color) !important; /* 변경된 배경색 변수 사용 */
    box-shadow: none !important;
}
/* 상단·하단에 그려지는 회색 캡(둥근 모서리) 제거 */
.custom-section-group::before,
.custom-section-group::after {
    display: none !important;
    content: none !important;
}

body {
    font-family: 'Pretendard', 'Noto Sans KR', -apple-system, BlinkMacSystemFont, sans-serif;
    background-color: var(--background-color); /* 변경된 배경색 변수 사용 */
    color: var(--text-color);
    line-height: 1.6;
    margin: 0;
    padding: 0;
}

/* 푸터 숨김 설정 추가 */
footer {
    visibility: hidden;
}

.gradio-container {
    width: 100%;  /* 전체 너비 100% 고정 */
    margin: 0 auto;
    padding: 20px;
    background-color: var(--background-color); /* 변경된 배경색 변수 사용 */
}

/* 콘텐츠 박스 (프레임) 스타일 */
.custom-frame {
    background-color: var(--card-bg);
    border: 1px solid rgba(0, 0, 0, 0.04);
    border-radius: var(--border-radius);
    padding: 20px;
    margin: 10px 0;
    box-shadow: var(--shadow);
}

/* 섹션 그룹 스타일 - 회색 배경 완전 제거 */
.custom-section-group {
    margin-top: 20px;
    padding: 0;
    border: none;
    border-radius: 0;
    background-color: var(--background-color); /* 회색 → 아이보리(전체 배경색) */
    box-shadow: none !important; /* 혹시 남아있는 그림자도 같이 제거 */
}

/* 버튼 스타일 - 글자 크기 18px */
.custom-button {
    border-radius: 30px !important;
    background: linear-gradient(135deg, var(--primary-color), var(--secondary-color)) !important;
    color: white !important;
    font-size: 18px !important;
    padding: 10px 20px !important;
    border: none;
    box-shadow: 0 4px 8px rgba(251, 127, 13, 0.25);
    transition: transform 0.3s ease;
}
.custom-button:hover {
    transform: translateY(-2px);
    box-shadow: 0 6px 12px rgba(251, 127, 13, 0.3);
}

/* 제목 스타일 (모든 항목명이 동일하게 custom-title 클래스로) */
.custom-title {
    font-size: 28px;
    font-weight: bold;
    margin-bottom: 10px;
    color: var(--text-color);
    border-bottom: 2px solid var(--primary-color);
    padding-bottom: 5px;
}

/* 이미지 컨테이너 */
.image-container {
    border-radius: var(--border-radius);
    overflow: hidden;
    border: 1px solid rgba(0, 0, 0, 0.08);
    transition: all 0.3s ease;
    background-color: white;
    aspect-ratio: 1 / 1;  /* 정사각형 비율 강제 */
}

.image-container:hover {
    box-shadow: 0 10px 20px rgba(0, 0, 0, 0.1);
}

.image-container img {
    width: 100%;
    height: 100%;
    object-fit: contain;  /* 이미지 비율 유지하면서 컨테이너에 맞춤 */
}

/* 입력 필드 스타일 */
.gr-input, .gr-text-input, .gr-sample-inputs {
    border-radius: var(--border-radius) !important;
    border: 1px solid #dddddd !important;
    padding: 12px !important;
    box-shadow: inset 0 1px 3px rgba(0, 0, 0, 0.05) !important;
    transition: all 0.3s ease !important;
}

.gr-input:focus, .gr-text-input:focus {
    border-color: var(--primary-color) !important;
    outline: none !important;
    box-shadow: 0 0 0 2px rgba(251, 127, 13, 0.2) !important;
}

/* 메인 컨텐츠 스크롤바 */
::-webkit-scrollbar {
    width: 8px;
    height: 8px;
}

::-webkit-scrollbar-track {
    background: rgba(0, 0, 0, 0.05);
    border-radius: 10px;
}

::-webkit-scrollbar-thumb {
    background: var(--primary-color);
    border-radius: 10px;
}

/* 애니메이션 스타일 */
@keyframes fadeIn {
    from { opacity: 0; transform: translateY(10px); }
    to { opacity: 1; transform: translateY(0); }
}

.fade-in {
    animation: fadeIn 0.5s ease-out;
}

/* 반응형 */
@media (max-width: 768px) {
    .button-grid {
        grid-template-columns: repeat(2, 1fr);
    }
}

/* 섹션 제목 스타일 - 참조코드 스타일과 동일하게 적용 */
.section-title {
    display: flex;
    align-items: center;
    font-size: 20px;
    font-weight: 700;
    color: #333333;
    margin-bottom: 10px;
    padding-bottom: 5px;
    border-bottom: 2px solid #FB7F0D;
    font-family: 'Pretendard', 'Noto Sans KR', -apple-system, BlinkMacSystemFont, sans-serif;
}

.section-title img {
    margin-right: 10px;
    width: 24px;
    height: 24px;
}
"""

# FontAwesome 아이콘 포함
fontawesome_link = """
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css" crossorigin="anonymous" referrerpolicy="no-referrer" />
"""

# ------------------- 예시 탭을 위한 함수 -------------------
def load_image_cached(image_path):
    """이미지를 캐시하여 로드하는 함수"""
    global IMAGE_CACHE
    if image_path not in IMAGE_CACHE:
        try:
            img = Image.open(image_path)
            # 큰 이미지는 미리 리사이즈하여 캐시
            if max(img.size) > 1000:
                ratio = 1000 / max(img.size)
                new_size = (int(img.size[0] * ratio), int(img.size[1] * ratio))
                img = img.resize(new_size, Image.Resampling.LANCZOS)
            IMAGE_CACHE[image_path] = img
        except Exception as e:
            logger.error(f"이미지 로드 실패: {image_path}, 에러: {e}")
            return None
    return IMAGE_CACHE[image_path]

def preload_example_images():
    """예시 이미지들을 미리 로드하는 함수"""
    for example in product_background_examples:
        load_image_cached(example[0])  # 입력 이미지
        load_image_cached(example[5])  # 결과 이미지

def create_app():
    with gr.Blocks(css=custom_css, theme=gr.themes.Default(
        primary_hue="orange",
        secondary_hue="orange",
        font=[gr.themes.GoogleFont("Noto Sans KR"), "ui-sans-serif", "system-ui"]
    )) as demo:
        gr.HTML(fontawesome_link)
        
        # 상품 배경 이미지 예시 갤러리 섹션
        with gr.Column(elem_classes="custom-frame"):
            gr.HTML('<div class="section-title"><img src="https://cdn-icons-png.flaticon.com/512/681/681443.png"> 상품 배경 이미지 예시 갤러리</div>')
            
            # 상단 메인 뷰 영역
            with gr.Row():
                example_input_image = gr.Image(
                    label="입력 이미지", 
                    height=400,
                    width=400,
                    elem_classes="image-container",
                    show_label=True,
                    show_download_button=True,
                    container=True,
                    scale=1
                )
                with gr.Column(elem_classes="example-params"):
                    example_bg_type = gr.Textbox(label="배경 유형", interactive=False)
                    example_bg_option = gr.Textbox(label="배경 선택", interactive=False)
                    example_product_name = gr.Textbox(label="상품명", interactive=False)
                    example_additional_info = gr.Textbox(label="추가 요청사항", interactive=False)
                example_output_image = gr.Image(
                    label="결과 이미지", 
                    height=400,
                    width=400,
                    elem_classes="image-container",
                    show_label=True,
                    show_download_button=True,
                    container=True,
                    scale=1
                )
        
        # 배경 유형별 갤러리 섹션
        with gr.Column(elem_classes="custom-frame"):
            gr.HTML('<div class="section-title"><img src="https://cdn-icons-png.flaticon.com/512/3068/3068327.png"> 배경 유형별 예시</div>')
            
            # 예시들을 배경 유형별로 그룹화
            examples_by_type = {}
            for example in product_background_examples:
                bg_type = example[1]  # 배경 유형
                if bg_type not in examples_by_type:
                    examples_by_type[bg_type] = []
                examples_by_type[bg_type].append(example)
            
            # 배경 유형별 갤러리 섹션 생성
            for bg_type, examples in examples_by_type.items():
                gr.Markdown(f"### {bg_type}")
                # 10개씩 한 줄로 표시
                for i in range(0, len(examples), 10):
                    with gr.Row():
                        for j in range(10):
                            if i + j < len(examples):
                                example = examples[i + j]
                                with gr.Column(scale=1, min_width=100):
                                    # 결과 이미지만 표시
                                    display_img = gr.Image(
                                        value=example[5], 
                                        label=None,  # 레이블 제거
                                        elem_classes="example-item",
                                        height=120,  # 크기 축소
                                        width=120,   # 크기 축소
                                        show_label=False,  # 레이블 숨기기
                                        show_download_button=False,  # 다운로드 버튼 숨기기
                                        show_share_button=False,  # 공유 버튼 숨기기
                                        container=False  # 컨테이너 테두리 제거
                                    )
                                    
                                    def make_example_handler(ex):
                                        def handler():
                                            # 로딩 상태 표시를 위한 중간 업데이트
                                            yield (
                                                gr.update(value=None),  # 임시로 이미지 비우기
                                                gr.update(value="로딩 중..."),
                                                gr.update(value="로딩 중..."),
                                                gr.update(value="로딩 중..."),
                                                gr.update(value="로딩 중..."),
                                                gr.update(value=None)
                                            )
                                            
                                            # 실제 데이터 로드
                                            yield (
                                                ex[0],  # 입력 이미지
                                                ex[1],  # 배경 유형
                                                ex[2],  # 배경 선택
                                                ex[3],  # 상품명
                                                ex[4] if ex[4] else "(없음)",  # 추가 요청사항
                                                ex[5]   # 결과 이미지
                                            )
                                        return handler
                                    
                                    display_img.select(
                                        fn=make_example_handler(example),
                                        outputs=[
                                            example_input_image,
                                            example_bg_type,
                                            example_bg_option,
                                            example_product_name,
                                            example_additional_info,
                                            example_output_image
                                        ],
                                        queue=True  # 요청을 큐에 넣어 순차적으로 처리
                                    )
                            else:
                                # 빈 공간 유지
                                with gr.Column(scale=1, min_width=100):
                                    pass
        
        # 페이지 로드 시 첫 번째 예시 자동 표시
        def load_first_example():
            if product_background_examples:
                first_example = product_background_examples[0]
                return (
                    first_example[0],  # 입력 이미지
                    first_example[1],  # 배경 유형
                    first_example[2],  # 배경 선택
                    first_example[3],  # 상품명
                    first_example[4] if first_example[4] else "(없음)",  # 추가 요청사항
                    first_example[5]   # 결과 이미지
                )
            return None, "", "", "", "", None
        
        demo.load(
            fn=load_first_example,
            outputs=[
                example_input_image,
                example_bg_type,
                example_bg_option,
                example_product_name,
                example_additional_info,
                example_output_image
            ]
        )
    
    return demo

if __name__ == "__main__":
    preload_example_images()  # 예시 이미지 미리 로드
    app = create_app()
    app.queue(max_size=10)  # 요청을 순차적으로 처리하도록 큐 설정
    app.launch(share=False, inbrowser=True, width="100%")