Update app.py

app.py

@@ -58,8 +58,8 @@ class GradioClientController:
                 try:
                     self.client = Client(self.api_endpoint)
                     logger.info("API 클라이언트가 성공적으로 초기화되었습니다.")
-                    # 초기화 성공 시 배경 옵션 로드
-                    self._load_background_options()
+                    # 초기화 성공 시 심플 배경 옵션 먼저 로드
+                    self._load_simple_background()
                     return
                 except Exception as e:
                     if attempt < max_retries - 1:
@@ -72,21 +72,60 @@ class GradioClientController:
             logger.error(f"클라이언트 초기화 실패: {str(e)}")
             self.client = None
     
-    def _load_background_options(self):
-        """배경 옵션을 미리 로드하여 캐시"""
+    def _load_simple_background(self):
+        """심플 배경 옵션만 먼저 로드"""
         try:
             if self.client:
-                # 각 배경 타입별로 옵션 로드
-                background_types = ["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "특수배경", "주얼리", "특수효과"]
-                for bg_type in background_types:
+                result = self.client.predict("심플 배경", api_name="/update_dropdowns")
+                if isinstance(result, (list, tuple)) and len(result) >= 7:
+                    self.background_options["심플 배경"] = result
+                    logger.info("심플 배경 옵션 로드 완료")
+                    return result[0]  # 심플 배경 선택지 반환
+        except Exception as e:
+            logger.warning(f"심플 배경 옵션 로드 실패: {str(e)}")
+        return []
+    def get_initial_dropdown_data(self, bg_type: str = "심플 배경") -> Tuple:
+        """초기 드롭다운 데이터 가져오기"""
+        try:
+            # 캐시된 데이터 먼저 확인
+            if bg_type in self.background_options:
+                result = self.background_options[bg_type]
+                logger.info(f"캐시에서 초기 드롭다운 데이터 로드: {bg_type}")
+            else:
+                # API 호출로 데이터 가져오기
+                if self.client:
                     result = self.client.predict(bg_type, api_name="/update_dropdowns")
-                    if isinstance(result, (list, tuple)) and len(result) >= 7:
-                        self.background_options[bg_type] = result
-                        logger.info(f"배경 옵션 로드 완료: {bg_type}")
-                    time.sleep(0.1)  # API 호출 간격
+                    self.background_options[bg_type] = result  # 캐시에 저장
+                    logger.info(f"API에서 초기 드롭다운 데이터 로드: {bg_type}")
+                else:
+                    logger.error("클라이언트가 초기화되지 않음")
+                    return tuple([[] for _ in range(7)])
+            
+            # 결과 반환 (실제 선택지만)
+            if isinstance(result, (list, tuple)) and len(result) >= 7:
+                return result[:7]
+            else:
+                return tuple([[] for _ in range(7)])
+                
         except Exception as e:
-            logger.warning(f"배경 옵션 미리 로드 실패: {str(e)}")
-            # 실패해도 계속 진행
+            logger.error(f"초기 드롭다운 데이터 로드 실패: {str(e)}")
+            return tuple([[] for _ in range(7)])
+
+    def get_initial_simple_choices(self) -> list:
+        """심플 배경 초기 선택지만 반환"""
+        try:
+            if "심플 배경" in self.background_options:
+                choices = self.background_options["심플 배경"][0]
+                return choices if choices else []
+            elif self.client:
+                result = self.client.predict("심플 배경", api_name="/update_dropdowns")
+                if isinstance(result, (list, tuple)) and len(result) >= 1:
+                    choices = result[0]
+                    self.background_options["심플 배경"] = result
+                    return choices if choices else []
+        except Exception as e:
+            logger.error(f"심플 배경 선택지 로드 실패: {str(e)}")
+        return []
     
     def _ensure_client(self) -> bool:
         """클라이언트 연결 상태 확인 및 재연결"""
@@ -306,9 +345,10 @@ def create_gradio_interface():
                                 value="심플 배경"
                             )
                             
-                            # 드롭다운 컴포넌트들 (API에서 동적으로 로드)
+                            # 드롭다운 컴포넌트들 - 심플 배경은 즉시 로드
                             simple_dropdown = gr.Dropdown(
-                                choices=[],
+                                choices=controller.get_initial_simple_choices(),
+                                value=None,  # 첫 번째 값은 로드 후 설정
                                 label="심플 배경 선택",
                                 visible=True,
                                 interactive=True
@@ -451,10 +491,20 @@ def create_gradio_interface():
                 ]
             )
             
-            # 앱 로드 시 초기 드롭다운 설정
+            # 앱 로드 시 모든 드롭다운 초기 설정 (더 강력한 초기화)
+            def initialize_all_dropdowns():
+                """모든 드롭다운을 초기화하는 함수"""
+                try:
+                    logger.info("드롭다운 초기화 시작")
+                    updates = controller.update_dropdowns("심플 배경")
+                    logger.info("드롭다운 초기화 완료")
+                    return updates
+                except Exception as e:
+                    logger.error(f"드롭다운 초기화 실패: {str(e)}")
+                    return tuple([gr.update() for _ in range(7)])
+            
             app.load(
-                fn=controller.update_dropdowns,
-                inputs=[background_type],
+                fn=initialize_all_dropdowns,
                 outputs=[simple_dropdown, studio_dropdown, nature_dropdown, 
                         indoor_dropdown, special_dropdown, jewelry_dropdown, special_effects_dropdown]
             )