Update app.py

app.py

@@ -81,6 +81,40 @@ def initialize_backgrounds():
     
     logger.info("모든 배경 옵션 초기화 완료")
 
+# 배경 드롭다운 초기화를 위한 함수 추가
+def initialize_dropdowns():
+    """드롭다운 메뉴 초기화 함수"""
+    # 각 배경 유형별 드롭다운 선택 목록 생성
+    simple_choices = list(SIMPLE_BACKGROUNDS.keys())
+    studio_choices = list(STUDIO_BACKGROUNDS.keys())
+    nature_choices = list(NATURE_BACKGROUNDS.keys())
+    indoor_choices = list(INDOOR_BACKGROUNDS.keys())
+    tech_choices = list(TECHNOLOGY_BACKGROUNDS.keys())
+    colorful_choices = list(COLORFUL_PATTERN_BACKGROUNDS.keys())
+    abstract_choices = list(ABSTRACT_BACKGROUNDS.keys())
+    
+    # 빈 배열 체크 및 로깅
+    if not tech_choices:
+        logger.warning("테크놀로지 배경 옵션이 비어 있습니다. JSON 파일을 확인하세요.")
+        tech_choices = ["기본 테크놀로지 배경"]
+    if not colorful_choices:
+        logger.warning("컬러풀 패턴 배경 옵션이 비어 있습니다. JSON 파일을 확인하세요.")
+        colorful_choices = ["기본 컬러풀 배경"]
+    
+    logger.info(f"테크놀로지 배경 옵션 수: {len(tech_choices)}")
+    logger.info(f"컬러풀 패턴 배경 옵션 수: {len(colorful_choices)}")
+    
+    return {
+        "simple": simple_choices,
+        "studio": studio_choices,
+        "nature": nature_choices,
+        "indoor": indoor_choices,
+        "tech": tech_choices,
+        "colorful": colorful_choices,
+        "abstract": abstract_choices
+    }
+
+
 # ------------------- 기본 유틸리티 함수 -------------------
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
@@ -445,6 +479,9 @@ def generate_product_images(image, bg_type, simple, studio, nature, indoor, tech
 
 # ------------------- Gradio 인터페이스 구성 -------------------
 def create_app():
+    # 드롭다운 옵션 초기화
+    dropdown_options = initialize_dropdowns()
+    
     with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 및 이미지 생성") as demo:
         gr.Markdown("# 고급 상품 이미지 배경 프롬프트 및 이미지 생성")
         gr.Markdown(
@@ -460,50 +497,50 @@ def create_app():
                     value="심플 배경"
                 )
                 simple_dropdown = gr.Dropdown(
-                    choices=list(SIMPLE_BACKGROUNDS.keys()),
-                    value=list(SIMPLE_BACKGROUNDS.keys())[0] if SIMPLE_BACKGROUNDS else None,
+                    choices=dropdown_options["simple"],
+                    value=dropdown_options["simple"][0] if dropdown_options["simple"] else None,
                     label="심플 배경 선택",
                     visible=True,
                     interactive=True
                 )
                 studio_dropdown = gr.Dropdown(
-                    choices=list(STUDIO_BACKGROUNDS.keys()),
-                    value=list(STUDIO_BACKGROUNDS.keys())[0] if STUDIO_BACKGROUNDS else None,
+                    choices=dropdown_options["studio"],
+                    value=dropdown_options["studio"][0] if dropdown_options["studio"] else None,
                     label="스튜디오 배경 선택",
                     visible=False,
                     interactive=True
                 )
                 nature_dropdown = gr.Dropdown(
-                    choices=list(NATURE_BACKGROUNDS.keys()),
-                    value=list(NATURE_BACKGROUNDS.keys())[0] if NATURE_BACKGROUNDS else None,
+                    choices=dropdown_options["nature"],
+                    value=dropdown_options["nature"][0] if dropdown_options["nature"] else None,
                     label="자연 환경 선택",
                     visible=False,
                     interactive=True
                 )
                 indoor_dropdown = gr.Dropdown(
-                    choices=list(INDOOR_BACKGROUNDS.keys()),
-                    value=list(INDOOR_BACKGROUNDS.keys())[0] if INDOOR_BACKGROUNDS else None,
+                    choices=dropdown_options["indoor"],
+                    value=dropdown_options["indoor"][0] if dropdown_options["indoor"] else None,
                     label="실내 환경 선택",
                     visible=False,
                     interactive=True
                 )
                 tech_dropdown = gr.Dropdown(
-                    choices=list(TECHNOLOGY_BACKGROUNDS.keys()),
-                    value=list(TECHNOLOGY_BACKGROUNDS.keys())[0] if TECHNOLOGY_BACKGROUNDS else None,
+                    choices=dropdown_options["tech"],
+                    value=dropdown_options["tech"][0] if dropdown_options["tech"] else None,
                     label="테크놀로지 배경 선택",
                     visible=False,
                     interactive=True
                 )
                 colorful_dropdown = gr.Dropdown(
-                    choices=list(COLORFUL_PATTERN_BACKGROUNDS.keys()),
-                    value=list(COLORFUL_PATTERN_BACKGROUNDS.keys())[0] if COLORFUL_PATTERN_BACKGROUNDS else None,
+                    choices=dropdown_options["colorful"],
+                    value=dropdown_options["colorful"][0] if dropdown_options["colorful"] else None,
                     label="컬러풀 패턴 배경 선택",
                     visible=False,
                     interactive=True
                 )
                 abstract_dropdown = gr.Dropdown(
-                    choices=list(ABSTRACT_BACKGROUNDS.keys()),
-                    value=list(ABSTRACT_BACKGROUNDS.keys())[0] if ABSTRACT_BACKGROUNDS else None,
+                    choices=dropdown_options["abstract"],
+                    value=dropdown_options["abstract"][0] if dropdown_options["abstract"] else None,
                     label="추상/특수 배경 선택",
                     visible=False,
                     interactive=True