Update app.py

app.py

@@ -10,111 +10,29 @@ GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
 # Gemini API 초기화
 genai.configure(api_key=GEMINI_API_KEY)
 
-# 배경 옵션 - 간단히 구분
-# 1. 심플 배경 (미니멀한 배경, 단색 배경 등)
-SIMPLE_BACKGROUNDS = {
-    "화이트 배경": "white background",
-    "블랙 배경": "black background",
-    "그라데이션 배경": "gradient background",
-    "베이지 배경": "beige background",
-    "그레이 배경": "gray background",
-    "파스텔 배경": "pastel background",
-    "그린 배경": "green background",
-    "블루 배경": "blue background",
-    "핑크 배경": "pink background",
-    "옐로우 배경": "yellow background",
-    "퍼플 배경": "purple background",
-    "오렌지 배경": "orange background",
-    "브라운 배경": "brown background",
-    "민트 배경": "mint background",
-    "라벤더 배경": "lavender background",
-    "코랄 배경": "coral background",
-    "터쿼이즈 배경": "turquoise background",
-    "네이비 배경": "navy background",
-    "샌드 배경": "sand background",
-    "페이퍼 배경": "paper background"
-}
+# 배경 JSON 파일 경로 설정
+BACKGROUNDS_DIR = os.path.join(os.path.dirname(__file__), "background")
 
-# 2. 스튜디오 배경 (제품 사진 스튜디오 느낌)
-STUDIO_BACKGROUNDS = {
-    "제품 사진 스튜디오": "product photography studio",
-    "미니멀 스튜디오": "minimal studio",
-    "하이키 스튜디오": "high key studio",
-    "로우키 스튜디오": "low key studio",
-    "소프트박스 조명": "softbox lighting studio",
-    "링라이트 스튜디오": "ring light studio",
-    "리플렉터 스튜디오": "reflector studio",
-    "그림자 스튜디오": "shadow studio",
-    "백라이트 스튜디오": "backlight studio",
-    "사이드라이트 스튜디오": "sidelight studio",
-    "스포트라이트 스튜디오": "spotlight studio",
-    "텍스처 배경 스튜디오": "textured background studio",
-    "대리석 배경 스튜디오": "marble background studio",
-    "콘크리트 배경 스튜디오": "concrete background studio",
-    "우드 배경 스튜디오": "wooden background studio",
-    "패브릭 배경 스튜디오": "fabric background studio",
-    "글래스 표면 스튜디오": "glass surface studio",
-    "메탈 표면 스튜디오": "metal surface studio",
-    "반사 표면 스튜디오": "reflective surface studio",
-    "매트 표면 스튜디오": "matte surface studio"
-}
+# JSON 파일이 존재하지 않을 경우 디렉토리 생성
+if not os.path.exists(BACKGROUNDS_DIR):
+    os.makedirs(BACKGROUNDS_DIR)
 
-# 3. 자연 환경
-NATURE_BACKGROUNDS = {
-    "열대 해변": "tropical beach",
-    "푸른 바다": "blue ocean",
-    "산 풍경": "mountain landscape",
-    "숲 풍경": "forest landscape",
-    "꽃밭": "flower garden",
-    "초원": "meadow",
-    "사막": "desert",
-    "강가": "riverside",
-    "호수가": "lakeside",
-    "폭포": "waterfall",
-    "일몰 풍경": "sunset landscape",
-    "일출 풍경": "sunrise landscape",
-    "안개 낀 풍경": "foggy landscape",
-    "별이 빛나는 하늘": "starry sky",
-    "눈 덮인 풍경": "snowy landscape"
-}
-
-# 4. 실내 환경
-INDOOR_BACKGROUNDS = {
-    "모던 리빙룸": "modern living room",
-    "미니멀 주방": "minimal kitchen",
-    "럭셔리 욕실": "luxury bathroom",
-    "침실": "bedroom",
-    "홈 오피스": "home office",
-    "카페": "cafe",
-    "레스토랑": "restaurant",
-    "호텔 로비": "hotel lobby",
-    "아트 갤러리": "art gallery",
-    "서재": "library",
-    "스파": "spa",
-    "쇼룸": "showroom",
-    "부티크": "boutique",
-    "편집샵": "concept store",
-    "일본식 인테리어": "japanese interior"
-}
+# JSON 파일 로드 함수
+def load_background_json(filename):
+    file_path = os.path.join(BACKGROUNDS_DIR, filename)
+    try:
+        with open(file_path, 'r', encoding='utf-8') as f:
+            return json.load(f)
+    except FileNotFoundError:
+        print(f"경고: {filename} 파일을 찾을 수 없습니다. 기본값을 사용합니다.")
+        return {}
 
-# 5. 추상/특수 배경
-ABSTRACT_BACKGROUNDS = {
-    "네온 조명": "neon lights",
-    "유체 추상화": "fluid abstract",
-    "기하학적 패턴": "geometric patterns",
-    "미니멀 추상": "minimal abstract",
-    "네온 추상": "neon abstract",
-    "가상 환경": "virtual environment",
-    "판타지 세계": "fantasy world",
-    "공상과학 배경": "sci-fi background",
-    "디지털 세계": "digital world",
-    "우주 공간": "outer space",
-    "미래 도시": "future city",
-    "판타지 자연": "fantasy nature",
-    "글리치 효과": "glitch effect",
-    "홀로그램 배경": "hologram background",
-    "입자 효과": "particle effect"
-}
+# 배경 데이터 로드
+SIMPLE_BACKGROUNDS = load_background_json("simple_backgrounds.json")
+STUDIO_BACKGROUNDS = load_background_json("studio_backgrounds.json")
+NATURE_BACKGROUNDS = load_background_json("nature_backgrounds.json")
+INDOOR_BACKGROUNDS = load_background_json("indoor_backgrounds.json")
+ABSTRACT_BACKGROUNDS = load_background_json("abstract_backgrounds.json")
 
 def generate_system_instruction():
     return """당신은 상품 이미지의 배경을 변경하기 위한 고품질 프롬프트를 생성하는 전문가입니다.
@@ -234,7 +152,7 @@ def create_app():
                     with gr.TabItem("심플 배경"):
                         simple_dropdown = gr.Dropdown(
                             choices=list(SIMPLE_BACKGROUNDS.keys()),
-                            value=list(SIMPLE_BACKGROUNDS.keys())[0],
+                            value=list(SIMPLE_BACKGROUNDS.keys())[0] if SIMPLE_BACKGROUNDS else None,
                             label="심플 배경 선택",
                             interactive=True
                         )
@@ -242,7 +160,7 @@ def create_app():
                     with gr.TabItem("스튜디오 배경"):
                         studio_dropdown = gr.Dropdown(
                             choices=list(STUDIO_BACKGROUNDS.keys()),
-                            value=list(STUDIO_BACKGROUNDS.keys())[0],
+                            value=list(STUDIO_BACKGROUNDS.keys())[0] if STUDIO_BACKGROUNDS else None,
                             label="스튜디오 배경 선택",
                             interactive=True
                         )
@@ -250,7 +168,7 @@ def create_app():
                     with gr.TabItem("자연 환경"):
                         nature_dropdown = gr.Dropdown(
                             choices=list(NATURE_BACKGROUNDS.keys()),
-                            value=list(NATURE_BACKGROUNDS.keys())[0],
+                            value=list(NATURE_BACKGROUNDS.keys())[0] if NATURE_BACKGROUNDS else None,
                             label="자연 환경 선택",
                             interactive=True
                         )
@@ -258,7 +176,7 @@ def create_app():
                     with gr.TabItem("실내 환경"):
                         indoor_dropdown = gr.Dropdown(
                             choices=list(INDOOR_BACKGROUNDS.keys()),
-                            value=list(INDOOR_BACKGROUNDS.keys())[0],
+                            value=list(INDOOR_BACKGROUNDS.keys())[0] if INDOOR_BACKGROUNDS else None,
                             label="실내 환경 선택",
                             interactive=True
                         )
@@ -266,7 +184,7 @@ def create_app():
                     with gr.TabItem("추상/특수 배경"):
                         abstract_dropdown = gr.Dropdown(
                             choices=list(ABSTRACT_BACKGROUNDS.keys()),
-                            value=list(ABSTRACT_BACKGROUNDS.keys())[0],
+                            value=list(ABSTRACT_BACKGROUNDS.keys())[0] if ABSTRACT_BACKGROUNDS else None,
                             label="추상/특수 배경 선택",
                             interactive=True
                         )