Update app.py

app.py

@@ -45,6 +45,80 @@ def load_backgrounds_from_env():
     backgrounds_code = os.environ.get("BACKGROUNDS_DATA", "")
     if not backgrounds_code:
         logger.warning("BACKGROUNDS_DATA environment variable is empty")
+        return {}, {}, {}, {}, {}, {}, {}
+    
+    try:
+        # 환경변수에서 가져온 코드 실행하여 딕셔너리들 추출
+        local_vars = {}
+        exec(backgrounds_code, {}, local_vars)
+        
+        return (
+            local_vars.get("SIMPLE_BACKGROUNDS", {}),
+            local_vars.get("STUDIO_BACKGROUNDS", {}),
+            local_vars.get("NATURE_BACKGROUNDS", {}),
+            local_vars.get("INDOOR_BACKGROUNDS", {}),
+            local_vars.get("SPECIAL_BACKGROUNDS", {}),
+            local_vars.get("JEWELRY_BACKGROUNDS", {}),
+            local_vars.get("SPECIAL_EFFECTS_BACKGROUNDS", {})
+        )
+    except Exception as e:
+        logger.error(f"Failed to parse BACKGROUNDS_DATA: {e}")
+        return {}, {}, {}, {}, {}, {}, {}
+
+# 환경변수에서 배경 딕셔너리들 로드
+(SIMPLE_BACKGROUNDS, STUDIO_BACKGROUNDS, NATURE_BACKGROUNDS, 
+ INDOOR_BACKGROUNDS, SPECIAL_BACKGROUNDS, JEWELRY_BACKGROUNDS, 
+ SPECIAL_EFFECTS_BACKGROUNDS) = load_backgrounds_from_env()
+
+# 만약 환경변수가 비어있다면 기본값 사용
+if not SIMPLE_BACKGROUNDS:
+    SIMPLE_BACKGROUNDS = {
+        "화이트 기본": "clean white background with soft lighting",
+        "회색 투톤": "light gray background with minimal shadows",
+        "라이트 그레이": "soft light gray backdrop with even illumination"
+    }
+
+if not STUDIO_BACKGROUNDS:
+    STUDIO_BACKGROUNDS = {
+        "연녹색 장미 정원": "soft green background with rose garden elements",
+        "연분홍 장미 대리석": "pink rose with marble surface backdrop"
+    }
+
+if not NATURE_BACKGROUNDS:
+    NATURE_BACKGROUNDS = {
+        "작은 파도가 있는 해변": "serene beach with gentle waves",
+        "열대해변": "tropical beach setting with palm trees"
+    }
+
+if not INDOOR_BACKGROUNDS:
+    INDOOR_BACKGROUNDS = {
+        "기본 책상": "clean modern desk setup",
+        "빛이 비치는 거실": "bright living room with natural light"
+    }
+
+if not SPECIAL_BACKGROUNDS:
+    SPECIAL_BACKGROUNDS = {
+        "네이비 빈티지 플로럴 벽지": "navy vintage floral wallpaper background"
+    }
+
+if not JEWELRY_BACKGROUNDS:
+    JEWELRY_BACKGROUNDS = {
+        "화이트 미러 스팟 라이트": "white mirror surface with spotlight"
+    }
+
+if not SPECIAL_EFFECTS_BACKGROUNDS:
+    SPECIAL_EFFECTS_BACKGROUNDS = {
+        "블루블랙 큰 물방울 효과": "blue-black background with water droplet effects"
+    }
+
+# 임시 파일 저장 함수
+def save_uploaded_file(uploaded_file, suffix='.png'):
+    try:
+        logger.info(f"Processing uploaded file: {type(uploaded_file)}")
+        
+        if uploaded_file is None:
+            logger.warning("Uploaded file is None")
+            return None
         
         with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
             temp_filename = temp_file.name
@@ -135,7 +209,7 @@ def api_check_password(password, image_file, bg_type, simple, studio, nature, in
         return result
     except Exception as e:
         logger.error(f"Error calling check_password API: {e}")
-        return None, None, None, None, "", "", f"API 호출 오류: {str(e)}"
+        return [], None, [], None, "", "", f"API 호출 오류: {str(e)}"
 
 # Gradio 인터페이스 구성 (완전히 동일한 UI)
 def create_gradio_interface():
@@ -223,8 +297,17 @@ def create_gradio_interface():
                                 interactive=True
                             )
                             
-                            # 드롭다운 변경 함수
+                            # 드롭다운 변경 함수 (원본과 동일한 엔드포인트)
                             def update_dropdowns(bg_type):
+                                # API 호출하여 원본 동작 유지
+                                try:
+                                    result = api_update_dropdowns(bg_type)
+                                    if result:
+                                        return result
+                                except:
+                                    pass
+                                
+                                # 백업용 로컬 로직
                                 return {
                                     simple_dropdown: gr.update(visible=(bg_type == "심플 배경")),
                                     studio_dropdown: gr.update(visible=(bg_type == "스튜디오 배경")),
@@ -235,10 +318,12 @@ def create_gradio_interface():
                                     special_effects_dropdown: gr.update(visible=(bg_type == "특수효과"))
                                 }
                             
+                            # 원본과 동일한 엔드포인트 생성
                             background_type.change(
                                 fn=update_dropdowns,
                                 inputs=[background_type],
-                                outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, special_dropdown, jewelry_dropdown, special_effects_dropdown]
+                                outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, special_dropdown, jewelry_dropdown, special_effects_dropdown],
+                                api_name="update_dropdowns"  # 원본과 동일한 API 이름
                             )
                     
                     # 요청사항 입력
@@ -383,77 +468,4 @@ if __name__ == "__main__":
         app.launch(share=True)
     except Exception as e:
         logger.error(f"Error running app: {e}")
-        logger.error(traceback.format_exc()), None, None, None, None, None, None
-    
-    try:
-        # 환경변수에서 가져온 코드 실행하여 딕셔너리들 추출
-        local_vars = {}
-        exec(backgrounds_code, {}, local_vars)
-        
-        return (
-            local_vars.get("SIMPLE_BACKGROUNDS", {}),
-            local_vars.get("STUDIO_BACKGROUNDS", {}),
-            local_vars.get("NATURE_BACKGROUNDS", {}),
-            local_vars.get("INDOOR_BACKGROUNDS", {}),
-            local_vars.get("SPECIAL_BACKGROUNDS", {}),
-            local_vars.get("JEWELRY_BACKGROUNDS", {}),
-            local_vars.get("SPECIAL_EFFECTS_BACKGROUNDS", {})
-        )
-    except Exception as e:
-        logger.error(f"Failed to parse BACKGROUNDS_DATA: {e}")
-        return {}, {}, {}, {}, {}, {}, {}
-
-# 환경변수에서 배경 딕셔너리들 로드
-(SIMPLE_BACKGROUNDS, STUDIO_BACKGROUNDS, NATURE_BACKGROUNDS, 
- INDOOR_BACKGROUNDS, SPECIAL_BACKGROUNDS, JEWELRY_BACKGROUNDS, 
- SPECIAL_EFFECTS_BACKGROUNDS) = load_backgrounds_from_env()
-
-# 만약 환경변수가 비어있다면 기본값 사용
-if not SIMPLE_BACKGROUNDS:
-    SIMPLE_BACKGROUNDS = {
-        "화이트 기본": "clean white background with soft lighting",
-        "회색 투톤": "light gray background with minimal shadows",
-        "라이트 그레이": "soft light gray backdrop with even illumination"
-    }
-
-if not STUDIO_BACKGROUNDS:
-    STUDIO_BACKGROUNDS = {
-        "연녹색 장미 정원": "soft green background with rose garden elements",
-        "연분홍 장미 대리석": "pink rose with marble surface backdrop"
-    }
-
-if not NATURE_BACKGROUNDS:
-    NATURE_BACKGROUNDS = {
-        "작은 파도가 있는 해변": "serene beach with gentle waves",
-        "열대해변": "tropical beach setting with palm trees"
-    }
-
-if not INDOOR_BACKGROUNDS:
-    INDOOR_BACKGROUNDS = {
-        "기본 책상": "clean modern desk setup",
-        "빛이 비치는 거실": "bright living room with natural light"
-    }
-
-if not SPECIAL_BACKGROUNDS:
-    SPECIAL_BACKGROUNDS = {
-        "네이비 빈티지 플로럴 벽지": "navy vintage floral wallpaper background"
-    }
-
-if not JEWELRY_BACKGROUNDS:
-    JEWELRY_BACKGROUNDS = {
-        "화이트 미러 스팟 라이트": "white mirror surface with spotlight"
-    }
-
-if not SPECIAL_EFFECTS_BACKGROUNDS:
-    SPECIAL_EFFECTS_BACKGROUNDS = {
-        "블루블랙 큰 물방울 효과": "blue-black background with water droplet effects"
-    }
-
-# 임시 파일 저장 함수
-def save_uploaded_file(uploaded_file, suffix='.png'):
-    try:
-        logger.info(f"Processing uploaded file: {type(uploaded_file)}")
-        
-        if uploaded_file is None:
-            logger.warning("Uploaded file is None")
-            return
+        logger.error(traceback.format_exc())