app.py

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

# 환경 변수 로드
load_dotenv()

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

# 캐시된 이미지 저장소
IMAGE_CACHE = {}

# CSS 및 아이콘 링크 (예시 탭 전용)
custom_css = """
:root {
    --background-color: #FFFFFF;
    --text-color: #334155;
    --border-radius: 18px;
    --shadow: 0 8px 30px rgba(0,0,0,0.08);
}

.example-gallery {
    display: grid;
    grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
    gap: 20px;
    padding: 20px;
}
.example-item img {
    width: 100%;
    height: 120px;
    object-fit: cover;
    border-radius: var(--border-radius);
}

.image-container {
    border-radius: var(--border-radius);
    overflow: hidden;
    background-color: white;
    transition: box-shadow 0.3s;
}
.image-container:hover {
    box-shadow: var(--shadow);
}

.footer { visibility: hidden; }
"""
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(path):
    if path not in IMAGE_CACHE:
        try:
            img = Image.open(path)
            IMAGE_CACHE[path] = img
        except Exception as e:
            logger.error(f"이미지 로드 실패: {path} - {e}")
            return None
    return IMAGE_CACHE[path]


def preload_example_images():
    for ex in product_background_examples:
        load_image_cached(ex[0])
        load_image_cached(ex[5])

# 예시 선택 핸들러

def load_example(evt: gr.SelectData):
    ex = product_background_examples[evt.index]
    return ex[0], ex[1], ex[2], ex[3], ex[4] or "(없음)", ex[5]

# 첫 번째 예시 자동 로드

def load_first_example():
    if product_background_examples:
        ex = product_background_examples[0]
        return ex[0], ex[1], ex[2], ex[3], ex[4] or "(없음)", ex[5]
    return None, "", "", "", "", None

# Gradio 앱 생성

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.Tabs():
            with gr.TabItem("예시 결과 보기"):
                # 메인 뷰 컴포넌트 먼저 정의
                with gr.Row():
                    example_input_image = gr.Image(label="입력 이미지", elem_classes="image-container")
                    with gr.Column():
                        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="결과 이미지", elem_classes="image-container")

                # 예시 갤러리
                with gr.Column():
                    gr.HTML('<div class="section-title"><img src="https://cdn-icons-png.flaticon.com/512/681/681443.png"> 상품 배경 이미지 예시 갤러리</div>')
                    with gr.Row(elem_classes="example-gallery"):
                        for idx, ex in enumerate(product_background_examples):
                            item = gr.Image(value=ex[5], show_label=False, elem_classes="example-item")
                            item.select(fn=load_example, inputs=None,
                                        outputs=[example_input_image,
                                                 example_bg_type,
                                                 example_bg_option,
                                                 example_product_name,
                                                 example_additional_info,
                                                 example_output_image])

                # 첫 번째 예시 자동 로드
                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.launch(share=False, inbrowser=True)