Update app.py

app.py

@@ -17,6 +17,67 @@ 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}")
+
+# ------------------- 전역 변수 설정 -------------------
+SIMPLE_BACKGROUNDS = {}
+STUDIO_BACKGROUNDS = {}
+NATURE_BACKGROUNDS = {}
+INDOOR_BACKGROUNDS = {}
+TECHNOLOGY_BACKGROUNDS = {}
+COLORFUL_PATTERN_BACKGROUNDS = {}
+ABSTRACT_BACKGROUNDS = {}
+
+# ------------------- 배경 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:
+            data = json.load(f)
+            logger.info(f"{filename} 파일을 성공적으로 로드했습니다. {len(data)} 항목 포함.")
+            return data
+    except FileNotFoundError:
+        logger.info(f"{filename} 파일이 없습니다.")
+        return {}
+    except Exception as e:
+        logger.warning(f"{filename} 파일 로드 중 오류 발생: {str(e)}.")
+        return {}
+
+# ------------------- 배경 초기화 함수 -------------------
+def initialize_backgrounds():
+    global SIMPLE_BACKGROUNDS, STUDIO_BACKGROUNDS, NATURE_BACKGROUNDS, INDOOR_BACKGROUNDS
+    global TECHNOLOGY_BACKGROUNDS, COLORFUL_PATTERN_BACKGROUNDS, ABSTRACT_BACKGROUNDS
+    
+    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")
+    TECHNOLOGY_BACKGROUNDS = load_background_json("technology_backgrounds.json")
+    COLORFUL_PATTERN_BACKGROUNDS = load_background_json("colorful_pattern_backgrounds.json")
+    ABSTRACT_BACKGROUNDS = load_background_json("abstract_backgrounds.json")
+    
+    # 기본값 설정 (JSON 파일이 없거나 비어있는 경우)
+    if not SIMPLE_BACKGROUNDS:
+        SIMPLE_BACKGROUNDS = {"화이트 배경": "white background"}
+    if not STUDIO_BACKGROUNDS:
+        STUDIO_BACKGROUNDS = {"미니멀 플랫레이": "minimalist flat lay with clean white background"}
+    if not NATURE_BACKGROUNDS:
+        NATURE_BACKGROUNDS = {"열대 해변": "tropical beach with crystal clear water"}
+    if not INDOOR_BACKGROUNDS:
+        INDOOR_BACKGROUNDS = {"미니멀 스칸디나비안 거실": "minimalist Scandinavian living room"}
+    if not TECHNOLOGY_BACKGROUNDS:
+        TECHNOLOGY_BACKGROUNDS = {"다이나믹 스플래시": "dynamic water splash interaction with product"}
+    if not COLORFUL_PATTERN_BACKGROUNDS:
+        COLORFUL_PATTERN_BACKGROUNDS = {"화려한 꽃 패턴": "vibrant floral pattern backdrop"}
+    if not ABSTRACT_BACKGROUNDS:
+        ABSTRACT_BACKGROUNDS = {"네온 조명": "neon lights abstract background"}
+    
+    logger.info("모든 배경 옵션 로드 완료")
+
 # ------------------- 기본 이미지 생성 관련 함수 -------------------
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
@@ -86,32 +147,39 @@ def generate_with_images(prompt, images, variation_index=0):
             return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
         client = genai.Client(api_key=api_key)
         logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}, 변형 인덱스: {variation_index}")
+        
+        # 향상된 변형 접미사
         variation_suffixes = [
-            " Create this as the first variation. Do not add any text, watermarks, or labels to the image.",
-            " Create this as the second variation with more vivid colors. Do not add any text, watermarks, or labels to the image.",
-            " Create this as the third variation with a more creative style. Do not add any text, watermarks, or labels to the image.",
-            " Create this as the fourth variation with enhanced details. Do not add any text, watermarks, or labels to the image."
+            " Create this as a professional studio product shot with precise focus on the product details. Do not add any text, watermarks, or labels to the image.",
+            " Create this as a high-contrast artistic studio shot with dramatic lighting and shadows. Do not add any text, watermarks, or labels to the image.",
+            " Create this as a soft-lit elegantly styled product shot with complementary elements. Do not add any text, watermarks, or labels to the image.",
+            " Create this as a high-definition product photography with perfect color accuracy and detail preservation. Do not add any text, watermarks, or labels to the image."
         ]
+        
         if variation_index < len(variation_suffixes):
             prompt = prompt + variation_suffixes[variation_index]
         else:
-            prompt = prompt + " Do not add any text, watermarks, or labels to the image."
+            prompt = prompt + " Create as high-end commercial product photography. Do not add any text, watermarks, or labels to the image."
+        
         contents = [prompt]
         for idx, img in enumerate(images, 1):
             if img is not None:
                 contents.append(img)
                 logger.info(f"이미지 #{idx} 추가됨")
+        
+        # 향상된 생성 설정 - 더 높은 temperature로 창의성 증가, 더 높은 max_output_tokens
         response = client.models.generate_content(
             model="gemini-2.0-flash-exp-image-generation",
             contents=contents,
             config=types.GenerateContentConfig(
                 response_modalities=['Text', 'Image'],
-                temperature=1,
-                top_p=0.95,
-                top_k=40,
-                max_output_tokens=8192
+                temperature=1.05,  # 약간 높은 온도로 더 다양한 결과
+                top_p=0.97,       # 약간 높임
+                top_k=50,         # 다양성 증가
+                max_output_tokens=10240  # 더 높은 토큰 한도
             )
         )
+        
         with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
             temp_path = tmp.name
             result_text = ""
@@ -160,117 +228,21 @@ def process_images_with_prompt(image1, prompt, variation_index=0, max_retries=3)
             time.sleep(1)
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
 
-# ------------------- 상품 프롬프트 생성 관련 함수 -------------------
-# 배경 JSON 파일 경로 설정
-BACKGROUNDS_DIR = "./background"
-if not os.path.exists(BACKGROUNDS_DIR):
-    os.makedirs(BACKGROUNDS_DIR)
-    logger.info(f"배경 디렉토리를 생성했습니다: {BACKGROUNDS_DIR}")
-else:
-    logger.info(f"배경 디렉토리가 이미 존재합니다: {BACKGROUNDS_DIR}")
-
-def load_background_json(filename):
-    file_path = os.path.join(BACKGROUNDS_DIR, filename)
-    try:
-        with open(file_path, 'r', encoding='utf-8') as f:
-            data = json.load(f)
-            logger.info(f"{filename} 파일을 성공적으로 로드했습니다. {len(data)} 항목 포함.")
-            return data
-    except Exception as e:
-        logger.warning(f"{filename} 파일 로드 중 오류 발생: {str(e)}. 기본값을 사용합니다.")
-        return {}
-
-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")
-
-if not SIMPLE_BACKGROUNDS:
-    SIMPLE_BACKGROUNDS = {"화이트 배경": "white background"}
-if not STUDIO_BACKGROUNDS:
-    STUDIO_BACKGROUNDS = {"제품 사진 스튜디오": "product photography studio"}
-if not NATURE_BACKGROUNDS:
-    NATURE_BACKGROUNDS = {"열대 해변": "tropical beach"}
-if not INDOOR_BACKGROUNDS:
-    INDOOR_BACKGROUNDS = {"모던 리빙룸": "modern living room"}
-if not ABSTRACT_BACKGROUNDS:
-    ABSTRACT_BACKGROUNDS = {"네온 조명": "neon lights"}
-
-# ------------------- 시스템 인스트럭션 수정 (상품 포커스, 고화질 렌더링 및 상품 테두리 빛 최소화) -------------------
-def generate_system_instruction():
-    return """당신은 상품 이미지의 배경을 변경하기 위한 고품질 프롬프트를 생성하는 전문가입니다.
-사용자가 제공하는 상품명, 배경 유형, 추가 요청사항을 바탕으로 미드저니(Midjourney)에 사용할 수 있는 상세하고 전문적인 프롬프트를 영어로 생성해주세요.
-다음 가이드라인을 반드시 준수해야 합니다:
-1. 상품을 "#1"로 지정하여 참조합니다. (예: "skincare tube (#1)")
-2. *** 매우 중요: 상품의 원래 특성(디자인, 색상, 형태, 로고, 패키지 등)은 어떤 상황에서도 절대 변경하지 않습니다. ***
-3. *** 상품의 본질적 특성을 유지하되, 상품에 포커스를 맞춰 모든 세부 사항이 선명하게 드러나도록 하며, 고화질(ultra high resolution)로 렌더링되어 화질 저하 없이 표현되어야 합니다. ***
-4. 이미지 비율은 정확히 1:1(정사각형) 형식으로 지정합니다. 프롬프트에 "square format", "1:1 ratio" 또는 "aspect ratio 1:1"을 명시적으로 포함합니다.
-5. 상품은 반드시 정사각형 구도의 정중앙에 배치되어야 합니다.
-6. 상품을 이미지의 주요 초점으로 부각시키고, 상품의 비율이 전체 이미지에서 크게 차지하도록 합니다.
-7. 상품 이미지 컷아웃(#1)의 기본 형태와 색상은 유지하면서, 선택한 환경에 자연스럽게 통합되도록 합니다.
-8. 고급스러운 상업적 이미지를 위한 다음 환경 요소들을 포��하세요:
-   - 상품과 어울리는 주변 환경/배경 요소
-   - 환경의 조명 효과는 부드럽고 자연스러운 조명만 적용하며, 상품 테두리의 빛 효과는 최소화합니다.
-   - 자연스러운 그림자와 빛 표현
-   - 상품의 용도나 장점을 암시하는 배경 요소
-   - 프로페셔널한 상업 사진 효과 (피사계 심도, 소프트 포커스, 스튜디오 조명 등)
-9. 프롬프트에 다음 요소들을 명시적으로 포함하세요:
-   - "highly detailed commercial photography"
-   - "award-winning product photography"
-   - "professional advertising imagery"
-   - "studio quality"
-   - "magazine advertisement quality"
-10. 배경 환경 요소를 상품 카테고리에 맞게 선택합니다.
-11. 사용자가 제공한 구체적인 배경과 추가 요청사항을 정확히 반영합니다.
-12. 프롬프트는 미드저니 AI에 최적화되어야 합니다.
-13. 프롬프트 끝에 "--ar 1:1 --s 750 --q 2" 파라미터를 추가하여 미드저니에서 고품질 정사각형 비율을 강제합니다.
-"""
-
-def generate_prompt_with_gemini(product_name, background_info, additional_info=""):
-    GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
-    if not GEMINI_API_KEY:
-        return "Gemini API 키가 설정되지 않았습니다. 환경 변수 GEMINI_API_KEY를 설정하거나 코드에 직접 입력하세요."
-    try:
-        genai_generative.configure(api_key=GEMINI_API_KEY)
-        prompt_request = f"""
-상품명: {product_name}
-배경 유형: {background_info.get('english', 'studio')}
-배경 카테고리: {background_info.get('category', '')}
-배경 이름: {background_info.get('name', '')}
-추가 요청사항: {additional_info}
-중요 요구사항:
-1. 상품이 크게 부각되고 이미지에서 중심적인 위치를 차지하도록 프롬프트를 생성해주세요.
-2. 이미지는 정확히 1:1 비율(정사각형)이어야 합니다.
-3. 상품은 정사각형 프레임의 정중앙에 위치해야 합니다.
-4. 상품의 디자인, 색상, 형태, 로고 등 본질적 특성은 절대 수정하지 마세요.
-5. 환경과의 자연스러운 통합을 위한 조명 효과와 그림자는 적용하되, 상품 테두리의 빛 효과는 최소화합니다.
-6. 상품을 더 돋보이게 하는 배경 환경을 설명해주세요.
-7. 고급스러운 상업 광고 품질의 이미지가 되도록 환경 설명을 해주세요.
-8. 프롬프트 끝에 미드저니 파라미터 "--ar 1:1 --s 750 --q 2"를 추가해주세요.
-한국어 입력 내용을 영어로 적절히 번역하여 반영해주세요.
-"""
-        model = genai_generative.GenerativeModel(
-            'gemini-2.0-flash',
-            system_instruction=generate_system_instruction()
-        )
-        response = model.generate_content(
-            prompt_request,
-            generation_config=genai_generative.types.GenerationConfig(
-                temperature=0.7,
-                top_p=0.95,
-                top_k=64,
-                max_output_tokens=1024,
-            )
-        )
-        response_text = response.text.strip()
-        if "--ar 1:1" not in response_text:
-            response_text = response_text.rstrip(".") + ". --ar 1:1 --s 750 --q 2"
-        return response_text
-    except Exception as e:
-        return f"프롬프트 생성 중 오류가 발생했습니다: {str(e)}"
+# ------------------- 프롬프트에서 오직 Midjourney 프롬프트 텍스트만 추출하는 함수 -------------------
+def filter_prompt_only(prompt):
+    idx = prompt.find("Highly detailed commercial photography")
+    if idx != -1:
+        prompt_text = prompt[idx:].strip()
+        end_idx = prompt_text.rfind("--ar 1:1 --s 750 --q 2")
+        if end_idx != -1:
+            end_idx += len("--ar 1:1 --s 750 --q 2")
+            prompt_text = prompt_text[:end_idx].strip()
+        return prompt_text
+    return prompt.strip()
 
-def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstract):
+# ------------------- 선택된 배경 정보 가져오기 함수 -------------------
+def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract):
+    """선택된 배경 정보를 가져오는 함수"""
     if bg_type == "심플 배경":
         return {
             "category": "심플 배경",
@@ -295,6 +267,18 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstra
             "name": indoor,
             "english": INDOOR_BACKGROUNDS.get(indoor, "indoor environment")
         }
+    elif bg_type == "테크놀로지 배경":
+        return {
+            "category": "테크놀로지 배경",
+            "name": tech,
+            "english": TECHNOLOGY_BACKGROUNDS.get(tech, "technology environment")
+        }
+    elif bg_type == "컬러풀 패턴 배경":
+        return {
+            "category": "컬러풀 패턴 배경",
+            "name": colorful,
+            "english": COLORFUL_PATTERN_BACKGROUNDS.get(colorful, "colorful pattern background")
+        }
     elif bg_type == "추상/특수 배경":
         return {
             "category": "추상/특수 배경",
@@ -308,24 +292,114 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstra
             "english": "white background"
         }
 
-# --- 프롬프트에서 오직 Midjourney 프롬프트 텍스트만 추출하는 함수 ---
-def filter_prompt_only(prompt):
-    idx = prompt.find("Highly detailed commercial photography")
-    if idx != -1:
-        prompt_text = prompt[idx:].strip()
-        end_idx = prompt_text.rfind("--ar 1:1 --s 750 --q 2")
-        if end_idx != -1:
-            end_idx += len("--ar 1:1 --s 750 --q 2")
-            prompt_text = prompt_text[:end_idx].strip()
-        return prompt_text
-    return prompt.strip()
+# ------------------- 향상된 시스템 인스트럭션 생성 함수 -------------------
+def generate_enhanced_system_instruction():
+    """향상된 시스템 인스트럭션 생성 함수"""
+    return """당신은 상품 이미지의 배경을 변경하기 위한 최고 품질의 프롬프트를 생성하는 전문가입니다.
+사용자가 제공하는 상품명, 배경 유형, 추가 요청사항을 바탕으로 미드저니(Midjourney)에 사용할 수 있는 상세하고 전문적인 프롬프트를 영어로 생성해주세요.
+
+다음 가이드라인을 반드시 준수해야 합니다:
+
+1. 상품을 "#1"로 지정하여 참조합니다. (예: "skincare tube (#1)")
+
+2. *** 매우 중요: 상품의 원래 특성(디자인, 색상, 형태, 로고, 패키지 등)은 어떤 상황에서도 절대 변경하지 않습니다. ***
+
+3. *** 상품의 본질적 특성을 유지하되, 상품에 포커스를 맞춰 모든 세부 사항이 선명하게 드러나도록 하며, 
+   8K 해상도(8K resolution), 오버샤프닝 없는 초고화질(ultra high definition without oversharpening)로 렌더링되어야 합니다. ***
+
+4. 이미지 비율은 정확히 1:1(정사각형) 형식으로 지정합니다. 프롬프트에 "square format", "1:1 ratio" 또는 "aspect ratio 1:1"을 명시적으로 포함합니다.
+
+5. 상품은 반드시 정사각형 구도의 정중앙에 배치되어야 하며, 적절한 크기로 표현하여 디테일이 완벽하게 보이도록 합니다.
+
+6. 상품을 이미지의 주요 초점으로 부각시키고, 상품의 비율이 전체 이미지에서 60-70% 이상 차지하도록 합니다.
+
+7. 조명 설명을 매우 구체적으로 해주세요. 예: "soft directional lighting from left side", "dramatic rim lighting", "diffused natural light through windows"
+
+8. 배경의 재질과 질감을 상세히 설명해주세요. 예: "polished marble surface", "rustic wooden table with visible grain", "matte concrete wall with subtle texture"
+
+9. 프롬프트에 다음 요소들을 명시적으로 포함하되, 사용 맥락에 적절하게 변형하세요:
+   - "award-winning product photography" 
+   - "magazine-worthy commercial product shot"
+   - "professional advertising imagery with perfect exposure"
+   - "studio lighting with color-accurate rendering"
+   - "8K ultra high definition product showcase"
+   - "commercial product photography with precise detail rendering"
+   - "ultra high definition"
+   - "crystal clear details"
+
+10. 사용자가 제공한 구체적인 배경과 추가 요청사항을 프롬프트에 정확히 반영하고 확장합니다.
+
+11. 프롬프트 끝에 "--ar 1:1 --s 750 --q 2 --v 5.2" 파라미터를 추가하여 미드저니에서 고품질 정사각형 비율을 강제합니다.
+"""
+
+# ------------------- 프롬프트 생성 함수 -------------------
+def generate_prompt_with_gemini(product_name, background_info, additional_info=""):
+    """향상된 프롬프트 생성 함수"""
+    GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
+    if not GEMINI_API_KEY:
+        return "Gemini API 키가 설정되지 않았습니다. 환경 변수 GEMINI_API_KEY를 설정하거나 코드에 직접 입력하세요."
+    
+    try:
+        genai_generative.configure(api_key=GEMINI_API_KEY)
+        
+        # 더 상세한 프롬프트 요청 템플릿
+        prompt_request = f"""
+상품명: {product_name}
+배경 유형: {background_info.get('english', 'studio')}
+배경 카테고리: {background_info.get('category', '')}
+배경 이름: {background_info.get('name', '')}
+추가 요청사항: {additional_info}
+
+중요 요구사항:
+1. 상품(#1)이 이미지 구도에서 중심적인 위치를 차지하며 적절한 크기(이미지의 60-70%)로 표현되도록 프롬프트를 생성해주세요.
+2. 이미지는 정확히 1:1 비율(정사각형)이어야 합니다.
+3. 상품의 디자인, 색상, 형태, 로고 등 본질적 특성은 절대 수정하지 마세요.
+4. 구체적인 조명 기법을 상세히 명시해주세요:
+   - 정확한 조명 위치 (예: "45-degree key light from upper left")
+   - 조명 품질 (예: "soft diffused light", "hard directional light")
+   - 조명 강도와 색온도 (예: "warm tungsten key light with cool blue fill")
+   - 반사와 그림자 처리 방식 (예: "controlled specular highlights with soft shadow transitions")
+5. 상품을 더 돋보이게 하는 보조 요소(props)를 자연스럽게 활용하되, 상품이 항상 주인공이어야 합니다.
+6. 배경 재질과 표면 질감을 구체적으로 설명하고, 상품과의 상호작용 방식을 명시해주세요.
+7. 색상 구성(color palette, color harmonies)을 명확히 해주세요.
+8. 고급스러운 상업 광고 품질의 이미지가 되도록 프롬프트를 작성해주세요.
+9. 프롬프트 끝에 미드저니 파라미터 "--ar 1:1 --s 750 --q 2 --v 5.2"를 추가해주세요.
+
+한국어 입력 내용을 전문적인 영어로 번역하여 반영해주세요.
+"""
+        # 향상된 시스템 인스트럭션 사용
+        model = genai_generative.GenerativeModel(
+            'gemini-2.0-flash',
+            system_instruction=generate_enhanced_system_instruction()
+        )
+        
+        # 더 창의적인 결과를 위한 생성 설정 조정
+        response = model.generate_content(
+            prompt_request,
+            generation_config=genai_generative.types.GenerationConfig(
+                temperature=0.8,       # 더 창의적인 결과를 위해 온도 상향
+                top_p=0.97,
+                top_k=64,
+                max_output_tokens=1600,  # 더 상세한 프롬프트 허용
+            )
+        )
+        
+        response_text = response.text.strip()
+        
+        # 미드저니 파라미터가 없을 경우 추가
+        if "--ar 1:1" not in response_text:
+            response_text = response_text.rstrip(".") + ". --ar 1:1 --s 750 --q 2 --v 5.2"
+            
+        return response_text
+    except Exception as e:
+        return f"프롬프트 생성 중 오류가 발생했습니다: {str(e)}"
 
 # ------------------- 단일 이미지 생성 함수 -------------------
-def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_info):
+def generate_product_image(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, product_name, additional_info):
     if image is None:
         return None, "이미지를 업로드해주세요.", "이미지를 업로드 후 프롬프트를 생성해주세요."
     product_name = product_name.strip() or "제품"
-    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, abstract)
+    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract)
     generated_prompt = generate_prompt_with_gemini(product_name, background_info, additional_info)
     if "Gemini API 키가 설정되지 않았습니다" in generated_prompt:
         warning_msg = (
@@ -341,11 +415,11 @@ def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstr
     return result_image, status, final_prompt
 
 # ------------------- 4장 이미지 생성 함수 -------------------
-def generate_product_images(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_info):
+def generate_product_images(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, product_name, additional_info):
     if image is None:
         return None, None, None, None, "이미지를 업로드해주세요.", "이미지를 업로드 후 프롬프트를 생성해주세요."
     product_name = product_name.strip() or "제품"
-    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, abstract)
+    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract)
     generated_prompt = generate_prompt_with_gemini(product_name, background_info, additional_info)
     if "Gemini API 키가 설정되지 않았습니다" in generated_prompt:
         warning_msg = (
@@ -377,9 +451,9 @@ def create_app():
         )
         with gr.Row():
             with gr.Column(scale=1):
-                product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 텀블러 등", interactive=True)
+                product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 스마트워치, 향수, 운동화 등", interactive=True)
                 background_type = gr.Radio(
-                    choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "추상/특수 배경"],
+                    choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "테크놀로지 배경", "컬러풀 패턴 배경", "추상/특수 배경"],
                     label="배경 유형",
                     value="심플 배경"
                 )
@@ -411,6 +485,20 @@ def create_app():
                     visible=False,
                     interactive=True
                 )
+                tech_dropdown = gr.Dropdown(
+                    choices=list(TECHNOLOGY_BACKGROUNDS.keys()),
+                    value=list(TECHNOLOGY_BACKGROUNDS.keys())[0] if TECHNOLOGY_BACKGROUNDS 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,
+                    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,
@@ -430,12 +518,14 @@ def create_app():
                         studio_dropdown: gr.update(visible=(bg_type == "스튜디오 배경")),
                         nature_dropdown: gr.update(visible=(bg_type == "자연 환경")),
                         indoor_dropdown: gr.update(visible=(bg_type == "실내 환경")),
+                        tech_dropdown: gr.update(visible=(bg_type == "테크놀로지 배경")),
+                        colorful_dropdown: gr.update(visible=(bg_type == "컬러풀 패턴 배경")),
                         abstract_dropdown: gr.update(visible=(bg_type == "추상/특수 배경"))
                     }
                 background_type.change(
                     fn=update_dropdowns,
                     inputs=[background_type],
-                    outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown]
+                    outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown]
                 )
                 image_input = gr.Image(label="상품 이미지 업로드", type="pil")
                 with gr.Row():
@@ -455,17 +545,21 @@ def create_app():
                 prompt_output_multi = gr.Textbox(label="생성된 프롬프트 (영어)", lines=6)
         single_btn.click(
             fn=generate_product_image,
-            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown, product_name, additional_info],
+            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, product_name, additional_info],
             outputs=[image_output, status_output, prompt_output]
         )
         multi_btn.click(
             fn=generate_product_images,
-            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown, product_name, additional_info],
+            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, product_name, additional_info],
             outputs=[image_output1, image_output2, image_output3, image_output4, status_output_multi, prompt_output_multi]
         )
     return demo
 
+# ------------------- 메인 실행 함수 -------------------
 if __name__ == "__main__":
+    # 배경 옵션 초기화
+    initialize_backgrounds()
+    # 앱 생성 및 실행
     app = create_app()
     app.queue()
-    app.launch()
+    app.launch()