Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import sys
 import base64
 import io
@@ -5,7 +6,6 @@ import logging
 import tempfile
 import traceback
 import requests
-import json
 from PIL import Image
 import gradio as gr
 from openai import OpenAI
@@ -40,147 +40,26 @@ else:
     logger.warning("GEMINI_API_KEY not found or empty, Gemini client not initialized")
     gemini_client = None
 
-# 설정값들을 환경변수에서 로드하는 함수들
-def get_app_password():
-    """환경변수에서 앱 비밀번호를 가져옵니다."""
-    password = os.environ.get("APP_PASSWORD")
-    if password:
-        logger.info("App password loaded from environment variable")
-        return password
-    else:
-        logger.warning("APP_PASSWORD environment variable not found, using default password")
-        return "1089"  # 기본 비밀번호
-
-# 배경 데이터를 환경변수에서 로드하는 함수
-def load_backgrounds_from_env():
-    """환경변수에서 배경 데이터를 로드합니다."""
-    try:
-        # 환경변수에서 JSON 문자열로 저장된 배경 데이터 로드
-        backgrounds_data = os.environ.get("BACKGROUNDS_DATA")
-        if backgrounds_data:
-            logger.info("Loading backgrounds data from environment variable")
-            return json.loads(backgrounds_data)
-        else:
-            logger.warning("BACKGROUNDS_DATA environment variable not found, using default backgrounds")
-            return get_default_backgrounds()
-    except Exception as e:
-        logger.error(f"Error loading backgrounds from environment: {e}")
-        logger.info("Falling back to default backgrounds")
-        return get_default_backgrounds()
-
-def get_default_backgrounds():
-    """기본 배경 데이터를 반환합니다."""
-    return {
-        "SIMPLE_BACKGROUNDS": {
-            "화이트 기본": "A clean, minimalistic digital background scene featuring a seamless pure white backdrop and a smooth white surface. The lighting is bright and evenly diffused from above, eliminating harsh shadows and creating a crisp, studio-lit environment. There are no products or objects present — only the empty background remains. The composition is ultra-clean and modern, ideal for overlaying lifestyle or hydration-related products. The mood is fresh, light, and professional. Ultra-sharp, studio-quality, high resolution, photo-realistic",
-            "회색 투톤": "A clean and minimal product photography background featuring a seamless, near-white light gray tone across both the backdrop and surface. The entire scene is evenly lit with bright, diffused lighting from above, eliminating harsh shadows and producing a calm, sophisticated atmosphere. The composition remains empty — with no products or objects present — offering a soft and airy visual ideal for modern lifestyle, wellness, or eco-friendly branding. The lighting gently enhances spatial depth without creating reflections. Ultra-clear, studio-quality, photo-realistic"
-        },
-        "STUDIO_BACKGROUNDS": {
-            "연녹색 장미 정원": "A high-resolution, commercial-style digital photograph of a sophisticated background setup designed for a luxury product shoot. The scene features soft green tones with lush fern leaves and creamy pastel roses (white, peach, blush pink) artfully arranged. The flowers are placed on a white polished marble platform with soft veining, surrounded by elegant foliage. The lighting is natural and diffused from the top left, producing soft shadows and a serene mood. There is no product in the scene — only the floral arrangement and surface remain, creating a refined, editorial-quality backdrop suitable for overlaying product photography. No object should be centered; the composition should remain balanced and inviting. ultra-detailed, 8K quality, photo-realistic studio setting."
-        },
-        "NATURE_BACKGROUNDS": {
-            "작은 파도가 있는 해변": "A serene and natural beach-themed product photography scene. The background features warm golden sand and gentle turquoise waves softly rolling onto the shore, with white foam catching the golden light of a late afternoon sun. Around the edges of the frame — but not overlapping the center foreground — small natural elements like seashells, conch shells, and bits of coral are randomly scattered with varied sizes, positions, and quantities to evoke a natural, organic arrangement. The product area remains clean and visually unobstructed to maintain focus. Lighting is soft and ambient with subtle natural shadows to highlight the uploaded product. The overall atmosphere is peaceful, sunlit, and summer-inspired. Ultra photo-realistic, studio-quality"
-        },
-        "INDOOR_BACKGROUNDS": {
-            "기본 책상": "A bright, Scandinavian-inspired home office featuring a matte white desk with crisp edges placed against a soft sage green wall. Natural daylight gently enters from the side, casting diffused shadows across the surface and enhancing the calm, focused atmosphere. On the desk are neatly arranged everyday essentials like a few colorful pens, a potted plant, and an open notebook. A neutral-toned ergonomic chair is tucked under the desk, and the surrounding area is intentionally minimal to emphasize clarity and productivity. The entire setting evokes a modern, clutter-free work environment., photo-realistic, high-resolution."
-        },
-        "SPECIAL_BACKGROUNDS": {
-            "네이비 빈티지 플로럴 벽지": "A richly detailed studio photography background featuring an ornate, vintage-inspired floral wallpaper design. The backdrop showcases vibrant red, pink, yellow, and green botanical motifs intricately woven across a deep navy blue base. The floral pattern is symmetrical and bold, giving the composition an artistic, maximalist aesthetic. The floor is a clean, solid blue surface, providing visual contrast and modern balance. Lighting is bright and evenly diffused, emphasizing both the product and the intricate details of the wallpaper. Ideal for showcasing creative, lifestyle, or design-forward products. Ultra-detailed, high-resolution, ."
-        },
-        "JEWELRY_BACKGROUNDS": {
-            "화이트 미러 스팟 라이트": "A luxury jewelry product photoshoot featuring a soft white backdrop and a polished white mirrored surface. A single focused overhead light beam softly illuminates the center of the scene, fading smoothly toward the edges with a clean gradient. The glossy white surface reflects the jewelry item with crystal clarity — capturing the shape, facets, and symmetry of the design in perfect mirror-like detail. The reflection appears sharp, clean, and vertically aligned beneath the product, enhancing the sense of luxury and balance. The atmosphere is premium, minimal, and luminous — ideal for high-end diamond earrings or bridal jewelry. Ultra-detailed, studio-quality, photo-realistic"
-        },
-        "SPECIAL_EFFECTS_BACKGROUNDS": {
-            "블루블랙 큰 물방울 효과": "A deep black and vivid cobalt blue gradient backdrop with a reflective surface splashed by crystalline water droplets frozen mid-air. Backlighting enhances shimmer and motion, creating a sense of waterproof resilience. High-impact, ultra-detailed."
-        }
-    }
+# 환경변수에서 배경 프롬프트 로드
+BACKGROUNDS_DATA = os.environ.get("BACKGROUNDS_DATA", "")
+# 환경변수에서 비밀번호 로드
+APP_PASSWORD = os.environ.get("APP_PASSWORD", "")
 
-# 배경 데이터 로드
-backgrounds_data = load_backgrounds_from_env()
-SIMPLE_BACKGROUNDS = backgrounds_data.get("SIMPLE_BACKGROUNDS", {})
-STUDIO_BACKGROUNDS = backgrounds_data.get("STUDIO_BACKGROUNDS", {})
-NATURE_BACKGROUNDS = backgrounds_data.get("NATURE_BACKGROUNDS", {})
-INDOOR_BACKGROUNDS = backgrounds_data.get("INDOOR_BACKGROUNDS", {})
-SPECIAL_BACKGROUNDS = backgrounds_data.get("SPECIAL_BACKGROUNDS", {})
-JEWELRY_BACKGROUNDS = backgrounds_data.get("JEWELRY_BACKGROUNDS", {})
-SPECIAL_EFFECTS_BACKGROUNDS = backgrounds_data.get("SPECIAL_EFFECTS_BACKGROUNDS", {})
+if not BACKGROUNDS_DATA:
+    logger.error("BACKGROUNDS_DATA environment variable not found")
+    raise ValueError("BACKGROUNDS_DATA 환경변수가 설정되지 않았습니다.")
 
-# 앱 비밀번호 로드
-APP_PASSWORD = get_app_password()
-
-# 환경변수 검증 함수
-def validate_environment_variables():
-    """필수 환경변수가 설정되었는지 확인합니다."""
-    logger.info("=== 환경변수 검증 시작 ===")
-    
-    # 비밀번호 검증
-    logger.info(f"APP_PASSWORD: {'설정됨' if os.environ.get('APP_PASSWORD') else '기본값 사용 (1089)'}")
-    
-    # API 키 검증
-    openai_key = os.environ.get("OPENAI_API_KEY")
-    replicate_token = os.environ.get("REPLICATE_API_TOKEN") 
-    gemini_key = os.environ.get("GEMINI_API_KEY")
-    
-    logger.info(f"OPENAI_API_KEY: {'설정됨' if openai_key else '설정되지 않음'}")
-    logger.info(f"REPLICATE_API_TOKEN: {'설정됨' if replicate_token else '설정되지 않음'}")
-    logger.info(f"GEMINI_API_KEY: {'설정됨' if gemini_key else '설정되지 않음'}")
-    
-    # 배경 데이터 검증
-    backgrounds_data_env = os.environ.get("BACKGROUNDS_DATA")
-    if backgrounds_data_env:
-        logger.info("BACKGROUNDS_DATA: 환경변수에서 설정됨")
-        try:
-            parsed_data = json.loads(backgrounds_data_env)
-            bg_counts = {
-                category: len(backgrounds) 
-                for category, backgrounds in parsed_data.items()
-            }
-            logger.info(f"배경 데이터 로드 성공: {bg_counts}")
-            
-            # 각 카테고리별 상세 정보
-            for category, backgrounds in parsed_data.items():
-                if backgrounds:
-                    sample_keys = list(backgrounds.keys())[:3]  # 처음 3개만 표시
-                    logger.info(f"  {category}: {len(backgrounds)}개 - 예시: {sample_keys}")
-                else:
-                    logger.warning(f"  {category}: 비어있음")
-                    
-        except Exception as e:
-            logger.error(f"BACKGROUNDS_DATA JSON 파싱 오류: {e}")
-            logger.error("배경 데이터 형식이 올바른지 확인해주세요.")
-    else:
-        logger.info("BACKGROUNDS_DATA: 환경변수 없음 (기본값 사용)")
-        
-    # 실제 로드된 배경 데이터 확인
-    actual_bg_counts = {
-        "SIMPLE_BACKGROUNDS": len(SIMPLE_BACKGROUNDS),
-        "STUDIO_BACKGROUNDS": len(STUDIO_BACKGROUNDS),
-        "NATURE_BACKGROUNDS": len(NATURE_BACKGROUNDS),
-        "INDOOR_BACKGROUNDS": len(INDOOR_BACKGROUNDS),
-        "SPECIAL_BACKGROUNDS": len(SPECIAL_BACKGROUNDS),
-        "JEWELRY_BACKGROUNDS": len(JEWELRY_BACKGROUNDS),
-        "SPECIAL_EFFECTS_BACKGROUNDS": len(SPECIAL_EFFECTS_BACKGROUNDS)
-    }
-    logger.info(f"실제 로드된 배경 데이터: {actual_bg_counts}")
-    
-    # 경고 메시지
-    missing_apis = []
-    if not openai_key:
-        missing_apis.append("OpenAI (GPT 모델 사용 불가)")
-    if not replicate_token:
-        missing_apis.append("Replicate (Flux 모델 및 화질개선 사용 불가)")
-    if not gemini_key:
-        missing_apis.append("Gemini (번역 기능 사용 불가)")
-    
-    if missing_apis:
-        logger.warning(f"누락된 API 키로 인해 다음 기능이 제한됩니다: {', '.join(missing_apis)}")
-    else:
-        logger.info("모든 API 키가 설정되어 전체 기능 사용 가능합니다.")
-    
-    logger.info("=== 환경변수 검증 완료 ===")
+if not APP_PASSWORD:
+    logger.error("APP_PASSWORD environment variable not found")
+    raise ValueError("APP_PASSWORD 환경변수가 설정되지 않았습니다.")
 
-# 환경변수 검증 실행
-validate_environment_variables()
+# 환경변수 데이터를 파이썬 코드로 실행하여 배경 딕셔너리들 생성
+try:
+    exec(BACKGROUNDS_DATA)
+    logger.info("Background data loaded from environment variable")
+except Exception as e:
+    logger.error(f"Failed to load background data: {e}")
+    raise ValueError(f"배경 데이터 로드 실패: {e}")
 
 # 임시 파일 저장 함수
 def save_uploaded_file(uploaded_file, suffix='.png'):
@@ -652,70 +531,58 @@ def create_gradio_interface():
                                 value="심플 배경"
                             )
                             
-                            # 드롭다운 컴포넌트들 - 환경변수에서 로드된 데이터 사용
-                            simple_choices = list(SIMPLE_BACKGROUNDS.keys()) if SIMPLE_BACKGROUNDS else []
-                            studio_choices = list(STUDIO_BACKGROUNDS.keys()) if STUDIO_BACKGROUNDS else []
-                            nature_choices = list(NATURE_BACKGROUNDS.keys()) if NATURE_BACKGROUNDS else []
-                            indoor_choices = list(INDOOR_BACKGROUNDS.keys()) if INDOOR_BACKGROUNDS else []
-                            special_choices = list(SPECIAL_BACKGROUNDS.keys()) if SPECIAL_BACKGROUNDS else []
-                            jewelry_choices = list(JEWELRY_BACKGROUNDS.keys()) if JEWELRY_BACKGROUNDS else []
-                            special_effects_choices = list(SPECIAL_EFFECTS_BACKGROUNDS.keys()) if SPECIAL_EFFECTS_BACKGROUNDS else []
-                            
-                            # 환경변수가 없어서 배경 데이터가 비어있는 경우 경고
-                            if not any([simple_choices, studio_choices, nature_choices, indoor_choices, special_choices, jewelry_choices, special_effects_choices]):
-                                logger.error("모든 배경 카테고리가 비어있습니다. BACKGROUNDS_DATA 환경변수를 설정해주세요.")
-                            
+                            # 드롭다운 컴포넌트들
                             simple_dropdown = gr.Dropdown(
-                                choices=simple_choices,
-                                value=simple_choices[0] if simple_choices else None,
+                                choices=list(SIMPLE_BACKGROUNDS.keys()),
+                                value=list(SIMPLE_BACKGROUNDS.keys())[0] if SIMPLE_BACKGROUNDS else None,
                                 label="심플 배경 선택",
                                 visible=True,
                                 interactive=True
                             )
                             
                             studio_dropdown = gr.Dropdown(
-                                choices=studio_choices,
-                                value=studio_choices[0] if studio_choices else None,
+                                choices=list(STUDIO_BACKGROUNDS.keys()),
+                                value=list(STUDIO_BACKGROUNDS.keys())[0] if STUDIO_BACKGROUNDS else None,
                                 label="스튜디오 배경 선택",
                                 visible=False,
                                 interactive=True
                             )
                             
                             nature_dropdown = gr.Dropdown(
-                                choices=nature_choices,
-                                value=nature_choices[0] if nature_choices else None,
+                                choices=list(NATURE_BACKGROUNDS.keys()),
+                                value=list(NATURE_BACKGROUNDS.keys())[0] if NATURE_BACKGROUNDS else None,
                                 label="자연 환경 선택",
                                 visible=False,
                                 interactive=True
                             )
                             
                             indoor_dropdown = gr.Dropdown(
-                                choices=indoor_choices,
-                                value=indoor_choices[0] if indoor_choices else None,
+                                choices=list(INDOOR_BACKGROUNDS.keys()),
+                                value=list(INDOOR_BACKGROUNDS.keys())[0] if INDOOR_BACKGROUNDS else None,
                                 label="실내 환경 선택",
                                 visible=False,
                                 interactive=True
                             )
                             
                             special_dropdown = gr.Dropdown(
-                                choices=special_choices,
-                                value=special_choices[0] if special_choices else None,
+                                choices=list(SPECIAL_BACKGROUNDS.keys()),
+                                value=list(SPECIAL_BACKGROUNDS.keys())[0] if SPECIAL_BACKGROUNDS else None,
                                 label="특수배경 선택",
                                 visible=False,
                                 interactive=True
                             )
                             
                             jewelry_dropdown = gr.Dropdown(
-                                choices=jewelry_choices,
-                                value=jewelry_choices[0] if jewelry_choices else None,
+                                choices=list(JEWELRY_BACKGROUNDS.keys()),
+                                value=list(JEWELRY_BACKGROUNDS.keys())[0] if JEWELRY_BACKGROUNDS else None,
                                 label="주얼리 배경 선택",
                                 visible=False,
                                 interactive=True
                             )
                             
                             special_effects_dropdown = gr.Dropdown(
-                                choices=special_effects_choices,
-                                value=special_effects_choices[0] if special_effects_choices else None,
+                                choices=list(SPECIAL_EFFECTS_BACKGROUNDS.keys()),
+                                value=list(SPECIAL_EFFECTS_BACKGROUNDS.keys())[0] if SPECIAL_EFFECTS_BACKGROUNDS else None,
                                 label="특수효과 배경 선택",
                                 visible=False,
                                 interactive=True
@@ -723,58 +590,14 @@ def create_gradio_interface():
                             
                             # 드롭다운 변경 함수
                             def update_dropdowns(bg_type):
-                                """배경 유형에 따라 드롭다운을 업데이트합니다."""
-                                logger.info(f"Updating dropdowns for background type: {bg_type}")
-                                
-                                # 각 배경 카테고리의 키 목록 가져오기
-                                simple_choices = list(SIMPLE_BACKGROUNDS.keys()) if SIMPLE_BACKGROUNDS else []
-                                studio_choices = list(STUDIO_BACKGROUNDS.keys()) if STUDIO_BACKGROUNDS else []
-                                nature_choices = list(NATURE_BACKGROUNDS.keys()) if NATURE_BACKGROUNDS else []
-                                indoor_choices = list(INDOOR_BACKGROUNDS.keys()) if INDOOR_BACKGROUNDS else []
-                                special_choices = list(SPECIAL_BACKGROUNDS.keys()) if SPECIAL_BACKGROUNDS else []
-                                jewelry_choices = list(JEWELRY_BACKGROUNDS.keys()) if JEWELRY_BACKGROUNDS else []
-                                special_effects_choices = list(SPECIAL_EFFECTS_BACKGROUNDS.keys()) if SPECIAL_EFFECTS_BACKGROUNDS else []
-                                
-                                # 디버깅 정보 로깅
-                                logger.info(f"Available choices - Simple: {len(simple_choices)}, Studio: {len(studio_choices)}, Nature: {len(nature_choices)}")
-                                logger.info(f"Indoor: {len(indoor_choices)}, Special: {len(special_choices)}, Jewelry: {len(jewelry_choices)}, Effects: {len(special_effects_choices)}")
-                                
                                 return {
-                                    simple_dropdown: gr.update(
-                                        visible=(bg_type == "심플 배경"), 
-                                        choices=simple_choices,
-                                        value=simple_choices[0] if simple_choices else None
-                                    ),
-                                    studio_dropdown: gr.update(
-                                        visible=(bg_type == "스튜디오 배경"),
-                                        choices=studio_choices,
-                                        value=studio_choices[0] if studio_choices else None
-                                    ),
-                                    nature_dropdown: gr.update(
-                                        visible=(bg_type == "자연 환경"),
-                                        choices=nature_choices,
-                                        value=nature_choices[0] if nature_choices else None
-                                    ),
-                                    indoor_dropdown: gr.update(
-                                        visible=(bg_type == "실내 환경"),
-                                        choices=indoor_choices,
-                                        value=indoor_choices[0] if indoor_choices else None
-                                    ),
-                                    special_dropdown: gr.update(
-                                        visible=(bg_type == "특수배경"),
-                                        choices=special_choices,
-                                        value=special_choices[0] if special_choices else None
-                                    ),
-                                    jewelry_dropdown: gr.update(
-                                        visible=(bg_type == "주얼리"),
-                                        choices=jewelry_choices,
-                                        value=jewelry_choices[0] if jewelry_choices else None
-                                    ),
-                                    special_effects_dropdown: gr.update(
-                                        visible=(bg_type == "특수효과"),
-                                        choices=special_effects_choices,
-                                        value=special_effects_choices[0] if special_effects_choices else None
-                                    )
+                                    simple_dropdown: gr.update(visible=(bg_type == "심플 배경")),
+                                    studio_dropdown: gr.update(visible=(bg_type == "스튜디오 배경")),
+                                    nature_dropdown: gr.update(visible=(bg_type == "자연 환경")),
+                                    indoor_dropdown: gr.update(visible=(bg_type == "실내 환경")),
+                                    special_dropdown: gr.update(visible=(bg_type == "특수배경")),
+                                    jewelry_dropdown: gr.update(visible=(bg_type == "주얼리")),
+                                    special_effects_dropdown: gr.update(visible=(bg_type == "특수효과"))
                                 }
                             
                             background_type.change(
@@ -843,7 +666,7 @@ def create_gradio_interface():
             
             # 프롬프트만 생성하는 함수 (비밀번호 체크 포함)
             def generate_prompt_with_password_check(password, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, request_text, aspect_ratio):
-                # 비밀번호 확인 - 환경변수에서 가져온 비밀번호와 비교
+                # 비밀번호 확인
                 if password != APP_PASSWORD:
                     return "비밀번호가 틀렸습니다. 올바른 비밀번호를 입력해주세요."
                 
@@ -994,30 +817,15 @@ def create_gradio_interface():
 # 앱 실행
 if __name__ == "__main__":
     try:
-        logger.info("=== AI 이미지 편집 및 화질 개선 애플리케이션 시작 ===")
+        logger.info("Starting application")
         
         # imgs 디렉토리 확인/생성
         os.makedirs("imgs", exist_ok=True)
         logger.info("이미지 디렉토리 준비 완료")
         
-        # 환경변수 재검증 (앱 시작 시)
-        logger.info("앱 시작 전 환경변수 최종 확인")
-        if not APP_PASSWORD:
-            logger.warning("비밀번호가 설정되지 않았습니다. 기본 비밀번호(1089)를 사용합니다.")
-        
-        if not any([
-            os.environ.get("OPENAI_API_KEY"),
-            os.environ.get("REPLICATE_API_TOKEN"),
-            os.environ.get("GEMINI_API_KEY")
-        ]):
-            logger.warning("API 키가 하나도 설정되지 않았습니다. 기능이 제한될 수 있습니다.")
-        
-        # Gradio 인터페이스 생성 및 실행
         app = create_gradio_interface()
-        logger.info("Gradio 인터페이스 생성 완료")
-        logger.info("애플리케이션을 시작합니다...")
+        logger.info("Launching Gradio app")
         app.launch(share=True)
-        
     except Exception as e:
-        logger.error(f"애플리케이션 실행 중 오류 발생: {e}")
+        logger.error(f"Error running app: {e}")
         logger.error(traceback.format_exc())