Update app.py

app.py

@@ -34,6 +34,7 @@ ABSTRACT_BACKGROUNDS = {}
 
 # ------------------- 배경 JSON 파일 로드 함수 -------------------
 def load_background_json(filename):
+    """배경 JSON 파일 로드 함수"""
     file_path = os.path.join(BACKGROUNDS_DIR, filename)
     try:
         with open(file_path, 'r', encoding='utf-8') as f:
@@ -47,11 +48,13 @@ def load_background_json(filename):
         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")
@@ -60,9 +63,9 @@ def initialize_backgrounds():
     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"}
+        SIMPLE_BACKGROUNDS = {"클래식 화이트": "clean white background with soft even lighting"}
     if not STUDIO_BACKGROUNDS:
         STUDIO_BACKGROUNDS = {"미니멀 플랫레이": "minimalist flat lay with clean white background"}
     if not NATURE_BACKGROUNDS:
@@ -74,11 +77,11 @@ def initialize_backgrounds():
     if not COLORFUL_PATTERN_BACKGROUNDS:
         COLORFUL_PATTERN_BACKGROUNDS = {"화려한 꽃 패턴": "vibrant floral pattern backdrop"}
     if not ABSTRACT_BACKGROUNDS:
-        ABSTRACT_BACKGROUNDS = {"네온 조명": "neon lights abstract background"}
+        ABSTRACT_BACKGROUNDS = {"네온 라이트": "neon light abstract background with vibrant glowing elements"}
     
-    logger.info("모든 배경 옵션 로드 완료")
+    logger.info("모든 배경 옵션 초기화 완료")
 
-# ------------------- 기본 이미지 생성 관련 함수 -------------------
+# ------------------- 기본 유틸리티 함수 -------------------
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
@@ -140,6 +143,7 @@ def preprocess_prompt(prompt, image1):
     prompt += " 이미지를 생성해주세요. 이미지에 텍스트나 글자를 포함하지 마세요."
     return prompt
 
+# ------------------- 이미지 생성 함수 -------------------
 def generate_with_images(prompt, images, variation_index=0):
     try:
         api_key = os.environ.get("GEMINI_API_KEY")
@@ -228,7 +232,7 @@ def process_images_with_prompt(image1, prompt, variation_index=0, max_retries=3)
             time.sleep(1)
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
 
-# ------------------- 프롬프트에서 오직 Midjourney 프롬프트 텍스트만 추출하는 함수 -------------------
+# ------------------- 프롬프트 관련 함수 -------------------
 def filter_prompt_only(prompt):
     idx = prompt.find("Highly detailed commercial photography")
     if idx != -1:
@@ -240,7 +244,6 @@ def filter_prompt_only(prompt):
         return prompt_text
     return prompt.strip()
 
-# ------------------- 선택된 배경 정보 가져오기 함수 -------------------
 def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract):
     """선택된 배경 정보를 가져오는 함수"""
     if bg_type == "심플 배경":
@@ -292,7 +295,7 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech,
             "english": "white background"
         }
 
-# ------------------- 향상된 시스템 인스트럭션 생성 함수 -------------------
+# ------------------- 프롬프트 생성 함수 -------------------
 def generate_enhanced_system_instruction():
     """향상된 시스템 인스트럭션 생성 함수"""
     return """당신은 상품 이미지의 배경을 변경하기 위한 최고 품질의 프롬프트를 생성하는 전문가입니다.
@@ -332,7 +335,6 @@ def generate_enhanced_system_instruction():
 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", "")
@@ -557,8 +559,9 @@ def create_app():
 
 # ------------------- 메인 실행 함수 -------------------
 if __name__ == "__main__":
-    # 배경 옵션 초기화
+    # 배경 옵션 초기화 - JSON 파일에서 로드
     initialize_backgrounds()
+    
     # 앱 생성 및 실행
     app = create_app()
     app.queue()