Update app.py

app.py

@@ -53,6 +53,8 @@ def initialize_backgrounds():
     """모든 배경 옵션 초기화 함수"""
     global SIMPLE_BACKGROUNDS, STUDIO_BACKGROUNDS, NATURE_BACKGROUNDS, INDOOR_BACKGROUNDS
     global TECHNOLOGY_BACKGROUNDS, COLORFUL_PATTERN_BACKGROUNDS, ABSTRACT_BACKGROUNDS
+    global JEWELRY_BACKGROUNDS  # 쥬얼리 배경 추가
+
     
     # 디렉토리 내 모든 파일 로깅
     logger.info(f"Backgrounds 디렉토리 경로: {BACKGROUNDS_DIR}")
@@ -66,6 +68,8 @@ def initialize_backgrounds():
     TECHNOLOGY_BACKGROUNDS = load_background_json("tech-backgrounds-final.json")  # 파일 이름 수정
     COLORFUL_PATTERN_BACKGROUNDS = load_background_json("colorful-pattern-backgrounds.json")  # 파일 이름 수정
     ABSTRACT_BACKGROUNDS = load_background_json("abstract_backgrounds.json")
+    JEWELRY_BACKGROUNDS = load_background_json("jewelry-backgrounds.json")
+
     
     # 기본값 설정 (파일이 없거나 비어있는 경우)
     if not SIMPLE_BACKGROUNDS:
@@ -82,6 +86,9 @@ def initialize_backgrounds():
         COLORFUL_PATTERN_BACKGROUNDS = {"화려한 꽃 패턴": "vibrant floral pattern backdrop"}
     if not ABSTRACT_BACKGROUNDS:
         ABSTRACT_BACKGROUNDS = {"네온 라이트": "neon light abstract background with vibrant glowing elements"}
+    if not jewelry_choices:
+        logger.warning("쥬얼리 배경 옵션이 비어 있습니다. JSON 파일을 확인하세요.")
+        jewelry_choices = ["기본 쥬얼리 배경"]
     
     logger.info("모든 배경 옵션 초기화 완료")
 
@@ -96,17 +103,9 @@ def initialize_dropdowns():
     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)}")
+    jewelry_choices = list(JEWELRY_BACKGROUNDS.keys())
+
+
     
     return {
         "simple": simple_choices,
@@ -116,6 +115,8 @@ def initialize_dropdowns():
         "tech": tech_choices,
         "colorful": colorful_choices,
         "abstract": abstract_choices
+        "jewelry": jewelry_choices  # 새로 추가
+
     }
 
 
@@ -353,6 +354,14 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech,
             "name": abstract,
             "english": ABSTRACT_BACKGROUNDS.get(abstract, "abstract background")
         }
+
+    elif bg_type == "쥬얼리 배경":
+        return {
+            "category": "쥬얼리 배경",
+            "name": jewelry,
+            "english": JEWELRY_BACKGROUNDS.get(jewelry, "jewelry backdrop")
+        }
+
     else:
         return {
             "category": "기본 배경",
@@ -517,7 +526,7 @@ def create_app():
                 image_input = gr.Image(label="상품 이미지 업로드", type="pil")
                 
                 background_type = gr.Radio(
-                    choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "테크놀로지 배경", "컬러풀 패턴 배경", "추상/특수 배경"],
+                    choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "테크놀로지 배경", "컬러풀 패턴 배경", "추상/특수 배경", "쥬얼리 배경"],
                     label="배경 유형",
                     value="심플 배경"
                 )
@@ -569,6 +578,15 @@ def create_app():
                     label="추상/특수 배경 선택",
                     visible=False,
                     interactive=True
+                    
+                jewelry_dropdown = gr.Dropdown(
+                    choices=dropdown_options["jewelry"],
+                    value=dropdown_options["jewelry"][0] if dropdown_options["jewelry"] else None,
+                    label="쥬얼리 배경 선택",
+                    visible=False,
+                    interactive=True
+                )
+                    
                 )
                 additional_info = gr.Textbox(
                     label="추가 요청사항 (선택사항)",
@@ -585,6 +603,8 @@ def create_app():
                         tech_dropdown: gr.update(visible=(bg_type == "테크놀로지 배경")),
                         colorful_dropdown: gr.update(visible=(bg_type == "컬러풀 패턴 배경")),
                         abstract_dropdown: gr.update(visible=(bg_type == "추상/특수 배경"))
+                        jewelry_dropdown: gr.update(visible=(bg_type == "쥬얼리 배경"))  # 쥬얼리 드롭다운 추가
+                        
                     }
                 background_type.change(
                     fn=update_dropdowns,